Fix tests of SET String()

vesoft-inc · Dec 2, 2021 · 4f0326e · 4f0326e
1 parent 1629bf4
commit 4f0326e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/client_test.go b/client_test.go
@@ -259,7 +259,8 @@ func TestServiceDataIO(t *testing.T) {
 	// test base type
 	{
 		query :=
-			"FETCH PROP ON person \"Bob\" YIELD person.name, person.age, person.grade," +
+			"FETCH PROP ON person \"Bob\" YIELD vertex as VertexID," +
+				"person.name, person.age, person.grade," +
 				"person.friends, person.book_num, person.birthday, " +
 				"person.start_school, person.morning, " +
 				"person.property, person.is_girl, person.child_name, " +
@@ -276,7 +277,8 @@ func TestServiceDataIO(t *testing.T) {
 		assert.Equal(t, "test_data", resp.GetSpaceName())
 		assert.False(t, resp.IsEmpty())
 		assert.Equal(t, 1, resp.GetRowSize())
-		names := []string{"person.name",
+		names := []string{"VertexID",
+			"person.name",
 			"person.age",
 			"person.grade",
 			"person.friends",

diff --git a/result_set_test.go b/result_set_test.go
@@ -117,7 +117,7 @@ func TestAsDedupList(t *testing.T) {
 		UVal: &nebula.NSet{Values: valList},
 	}
 	valWrap := ValueWrapper{&value, testTimezone}
-	assert.Equal(t, "[\"elem1\", \"elem2\", \"elem3\"]", valWrap.String())
+	assert.Equal(t, "{\"elem1\", \"elem2\", \"elem3\"}", valWrap.String())
 	assert.Equal(t, true, valWrap.IsSet())
 
 	res, _ := valWrap.AsList()