fix tests

go-xorm · Aug 20, 2019 · 3e09cc8 · 3e09cc8
1 parent ff5a5fd
commit 3e09cc8
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/session_find_test.go b/session_find_test.go
@@ -812,11 +812,15 @@ func TestFindMapCols(t *testing.T) {
 	assert.NoError(t, prepareEngine())
 	assertSync(t, new(FindMapCols))
 
+	id := testEngine.GetColumnMapper().Obj2Table("Id")
+	colA := testEngine.GetColumnMapper().Obj2Table("ColA")
+	colB := testEngine.GetColumnMapper().Obj2Table("ColB")
+
 	var objs = make(map[int64]*FindMapCols)
-	err := testEngine.Cols("col_a, col_b").Find(&objs)
+	err := testEngine.Cols(colA, colB).Find(&objs)
 	assert.Error(t, err)
 	assert.True(t, IsErrPrimaryKeyNoSelected(err))
 
-	err = testEngine.Cols("id, col_a, col_b").Find(&objs)
+	err = testEngine.Cols(id, colA, colB).Find(&objs)
 	assert.NoError(t, err)
 }