修改测试

test/sdk_test.go

@@ -24,15 +24,16 @@ type Class struct {
 }
 
 const (
-	token          = "IpTTwAQweh/BP51fz5CzWKQFaXHvZe6ewvk6yOcAOkU="
-	address        = "localhost"
-	httpPort       = "10000"
-	grpcPort       = "10001"
-	namespace      = "ns-sdk-demo"
-	dataSource     = "ds-sdk-demo"
-	deleteSql      = "delete-sdk-demo"
-	goRoutineCount = 100
-	tableName      = "test.classes"
+	token               = "IpTTwAQweh/BP51fz5CzWKQFaXHvZe6ewvk6yOcAOkU="
+	address             = "localhost"
+	httpPort            = "10000"
+	grpcPort            = "10001"
+	namespace           = "ns-sdk-demo"
+	dataSource          = "ds-sdk-demo"
+	deleteSql           = "delete-sdk-demo"
+	goRoutineCount      = 100
+	tableName           = "test.classes"
+	sqlResultTimeFormat = "2006-01-02T15:04:05.000000+08:00"
 )
 
 var (
@@ -142,7 +143,7 @@ func TestRawSqlTemplate(t *testing.T) {
 	newStudentNum := rand.Int31n(100)
 
 	now := time.Now()
-	exceptedNowStr := now.Format("2006-01-02T15:04:05.000000+08:00")
+	exceptedNowStr := now.Format(sqlResultTimeFormat)
 
 	insertExecuteParams, err := raw_sql_tpl.InsertExecuteParams{
 		TableName: tableName,
@@ -252,16 +253,22 @@ func TestRawSqlTemplate(t *testing.T) {
 		t.Fatal(err)
 	}
 
-	if float64(len(queryResults)) != countResults[0]["count"].(float64) {
+	classes := make([]Class, 0)
+	err = sdk.ParseSqlResults(queryResults, &classes)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if float64(len(classes)) != countResults[0]["count"].(float64) {
 		t.Fatal("总数不正确")
 	}
 
-	if queryResults[0]["id"].(string) != classID ||
-		queryResults[0]["name"].(string) != className ||
-		queryResults[0]["student_num_alias"].(float64) != float64(studentNum) ||
-		queryResults[0]["graduated_time"].(string) != exceptedNowStr ||
-		queryResults[0]["created_time"].(string) != exceptedNowStr ||
-		queryResults[0]["last_updated_time"].(string) != exceptedNowStr {
+	if classes[0].ID != classID ||
+		classes[0].Name != className ||
+		classes[0].StudentNum != int(studentNum) ||
+		classes[0].GraduatedTime.Format(sqlResultTimeFormat) != exceptedNowStr ||
+		classes[0].CreatedTime.Format(sqlResultTimeFormat) != exceptedNowStr ||
+		classes[0].LastUpdatedTime.Format(sqlResultTimeFormat) != exceptedNowStr {
 		t.Fatal("查询数据不正确")
 	}
 
@@ -280,16 +287,22 @@ func TestRawSqlTemplate(t *testing.T) {
 		t.Fatal(err)
 	}
 
-	if float64(len(queryResults)) != countResults[0]["count"].(float64) {
+	classes = make([]Class, 0)
+	err = sdk.ParseSqlResults(queryResults, &classes)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if float64(len(classes)) != countResults[0]["count"].(float64) {
 		t.Fatal("总数不正确")
 	}
 
-	if queryResults[0]["id"].(string) != classID ||
-		queryResults[0]["name"].(string) != newClassName ||
-		queryResults[0]["student_num_alias"].(float64) != float64(newStudentNum) ||
-		queryResults[0]["graduated_time"].(string) != exceptedNowStr ||
-		queryResults[0]["created_time"].(string) != exceptedNowStr ||
-		queryResults[0]["last_updated_time"].(string) != exceptedNowStr {
+	if classes[0].ID != classID ||
+		classes[0].Name != newClassName ||
+		classes[0].StudentNum != int(newStudentNum) ||
+		classes[0].GraduatedTime.Format(sqlResultTimeFormat) != exceptedNowStr ||
+		classes[0].CreatedTime.Format(sqlResultTimeFormat) != exceptedNowStr ||
+		classes[0].LastUpdatedTime.Format(sqlResultTimeFormat) != exceptedNowStr {
 		t.Fatal("查询数据不正确")
 	}