fix

Signed-off-by: wjhuang2016 <huangwenjun1997@gmail.com>
pingcap · Dec 23, 2020 · dd4f56b · dd4f56b
1 parent 836466a
commit dd4f56b
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/server/tidb_test.go b/server/tidb_test.go
@@ -937,15 +937,15 @@ func (ts *tidbTestSuite) TestNO_DEFAULT_VALUEFlag(c *C) {
 	c.Assert(err, IsNil)
 
 	ctx := context.Background()
-	_, err = Execute(ctx, qctx, "use test")
+	_, err = qctx.Execute(ctx, "use test")
 	c.Assert(err, IsNil)
-	_, err = Execute(ctx, qctx, "drop table if exists t")
+	_, err = qctx.Execute(ctx, "drop table if exists t")
 	c.Assert(err, IsNil)
-	_, err = Execute(ctx, qctx, "create table t(c1 int key, c2 int);")
+	_, err = qctx.Execute(ctx, "create table t(c1 int key, c2 int);")
 	c.Assert(err, IsNil)
-	rs, err := Execute(ctx, qctx, "select c1 from t;")
+	rs, err := qctx.Execute(ctx, "select c1 from t;")
 	c.Assert(err, IsNil)
-	cols := rs.Columns()
+	cols := rs[0].Columns()
 	c.Assert(len(cols), Equals, 1)
 	expectFlag := uint16(tmysql.NotNullFlag | tmysql.PriKeyFlag | tmysql.NoDefaultValueFlag)
 	c.Assert(dumpFlag(cols[0].Type, cols[0].Flag), Equals, expectFlag)