types: move file const_test.go from the parser repository (pingcap#…

…8211)

types/const_test.go

@@ -0,0 +1,321 @@
+// Copyright 2017 PingCAP, Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package types_test
+
+import (
+	"flag"
+	"testing"
+
+	. "github.com/pingcap/check"
+	"github.com/pingcap/parser"
+	"github.com/pingcap/parser/mysql"
+	"github.com/pingcap/tidb/domain"
+	"github.com/pingcap/tidb/kv"
+	"github.com/pingcap/tidb/session"
+	"github.com/pingcap/tidb/store/mockstore"
+	"github.com/pingcap/tidb/store/mockstore/mocktikv"
+	"github.com/pingcap/tidb/util/testkit"
+	"github.com/pingcap/tidb/util/testleak"
+	"golang.org/x/net/context"
+)
+
+func TestT(t *testing.T) {
+	CustomVerboseFlag = true
+	TestingT(t)
+}
+
+var _ = Suite(&testMySQLConstSuite{})
+
+type testMySQLConstSuite struct {
+	cluster   *mocktikv.Cluster
+	mvccStore mocktikv.MVCCStore
+	store     kv.Storage
+	dom       *domain.Domain
+	*parser.Parser
+}
+
+var mockTikv = flag.Bool("mockTikv", true, "use mock tikv store in executor test")
+
+func (s *testMySQLConstSuite) SetUpSuite(c *C) {
+	s.Parser = parser.New()
+	flag.Lookup("mockTikv")
+	useMockTikv := *mockTikv
+	if useMockTikv {
+		s.cluster = mocktikv.NewCluster()
+		mocktikv.BootstrapWithSingleStore(s.cluster)
+		s.mvccStore = mocktikv.MustNewMVCCStore()
+		store, err := mockstore.NewMockTikvStore(
+			mockstore.WithCluster(s.cluster),
+			mockstore.WithMVCCStore(s.mvccStore),
+		)
+		c.Assert(err, IsNil)
+		s.store = store
+		session.SetSchemaLease(0)
+		session.SetStatsLease(0)
+	}
+	var err error
+	s.dom, err = session.BootstrapSession(s.store)
+	c.Assert(err, IsNil)
+}
+
+func (s *testMySQLConstSuite) TearDownSuite(c *C) {
+	s.dom.Close()
+	s.store.Close()
+	testleak.AfterTest(c)()
+}
+
+func (s *testMySQLConstSuite) TestGetSQLMode(c *C) {
+	positiveCases := []struct {
+		arg string
+	}{
+		{"NO_ZERO_DATE"},
+		{",,NO_ZERO_DATE"},
+		{"NO_ZERO_DATE,NO_ZERO_IN_DATE"},
+		{""},
+		{", "},
+		{","},
+	}
+
+	for _, t := range positiveCases {
+		_, err := mysql.GetSQLMode(mysql.FormatSQLModeStr(t.arg))
+		c.Assert(err, IsNil)
+	}
+
+	negativeCases := []struct {
+		arg string
+	}{
+		{"NO_ZERO_DATE, NO_ZERO_IN_DATE"},
+		{"NO_ZERO_DATE,adfadsdfasdfads"},
+		{", ,NO_ZERO_DATE"},
+		{" ,"},
+	}
+
+	for _, t := range negativeCases {
+		_, err := mysql.GetSQLMode(mysql.FormatSQLModeStr(t.arg))
+		c.Assert(err, NotNil)
+	}
+}
+
+func (s *testMySQLConstSuite) TestSQLMode(c *C) {
+	tests := []struct {
+		arg                           string
+		hasNoZeroDateMode             bool
+		hasNoZeroInDateMode           bool
+		hasErrorForDivisionByZeroMode bool
+	}{
+		{"NO_ZERO_DATE", true, false, false},
+		{"NO_ZERO_IN_DATE", false, true, false},
+		{"ERROR_FOR_DIVISION_BY_ZERO", false, false, true},
+		{"NO_ZERO_IN_DATE,NO_ZERO_DATE", true, true, false},
+		{"NO_ZERO_DATE,NO_ZERO_IN_DATE", true, true, false},
+		{"NO_ZERO_DATE,NO_ZERO_IN_DATE", true, true, false},
+		{"NO_ZERO_DATE,NO_ZERO_IN_DATE,ERROR_FOR_DIVISION_BY_ZERO", true, true, true},
+		{"NO_ZERO_IN_DATE,ERROR_FOR_DIVISION_BY_ZERO", false, true, true},
+		{"", false, false, false},
+	}
+
+	for _, t := range tests {
+		sqlMode, _ := mysql.GetSQLMode(t.arg)
+		c.Assert(sqlMode.HasNoZeroDateMode(), Equals, t.hasNoZeroDateMode)
+		c.Assert(sqlMode.HasNoZeroInDateMode(), Equals, t.hasNoZeroInDateMode)
+		c.Assert(sqlMode.HasErrorForDivisionByZeroMode(), Equals, t.hasErrorForDivisionByZeroMode)
+	}
+}
+
+func (s *testMySQLConstSuite) TestRealAsFloatMode(c *C) {
+	tk := testkit.NewTestKit(c, s.store)
+	tk.MustExec("use test")
+	tk.MustExec("drop table if exists t;")
+	tk.MustExec("create table t (a real);")
+	result := tk.MustQuery("desc t")
+	c.Check(result.Rows(), HasLen, 1)
+	row := result.Rows()[0]
+	c.Assert(row[1], Equals, "double")
+
+	tk.MustExec("drop table if exists t;")
+	tk.MustExec("set sql_mode='REAL_AS_FLOAT'")
+	tk.MustExec("create table t (a real)")
+	result = tk.MustQuery("desc t")
+	c.Check(result.Rows(), HasLen, 1)
+	row = result.Rows()[0]
+	c.Assert(row[1], Equals, "float")
+}
+
+func (s *testMySQLConstSuite) TestPipesAsConcatMode(c *C) {
+	tk := testkit.NewTestKit(c, s.store)
+	tk.MustExec("SET sql_mode='PIPES_AS_CONCAT';")
+	r := tk.MustQuery(`SELECT 'hello' || 'world';`)
+	r.Check(testkit.Rows("helloworld"))
+}
+
+func (s *testMySQLConstSuite) TestNoUnsignedSubtractionMode(c *C) {
+	tk := testkit.NewTestKit(c, s.store)
+	ctx := context.Background()
+	tk.MustExec("set sql_mode='NO_UNSIGNED_SUBTRACTION'")
+	r := tk.MustQuery("SELECT CAST(0 as UNSIGNED) - 1;")
+	r.Check(testkit.Rows("-1"))
+	rs, _ := tk.Exec("SELECT CAST(18446744073709551615 as UNSIGNED) - 1;")
+	_, err := session.GetRows4Test(ctx, tk.Se, rs)
+	c.Assert(err, NotNil)
+	c.Assert(rs.Close(), IsNil)
+	rs, _ = tk.Exec("SELECT 1 - CAST(18446744073709551615 as UNSIGNED);")
+	_, err = session.GetRows4Test(ctx, tk.Se, rs)
+	c.Assert(err, NotNil)
+	c.Assert(rs.Close(), IsNil)
+	rs, _ = tk.Exec("SELECT CAST(-1 as UNSIGNED) - 1")
+	_, err = session.GetRows4Test(ctx, tk.Se, rs)
+	c.Assert(err, NotNil)
+	c.Assert(rs.Close(), IsNil)
+	rs, _ = tk.Exec("SELECT CAST(9223372036854775808 as UNSIGNED) - 1")
+	_, err = session.GetRows4Test(ctx, tk.Se, rs)
+	c.Assert(err, NotNil)
+	c.Assert(rs.Close(), IsNil)
+}
+
+func (s *testMySQLConstSuite) TestHighNotPrecedenceMode(c *C) {
+	tk := testkit.NewTestKit(c, s.store)
+	tk.MustExec("use test")
+	tk.MustExec("drop table if exists t1")
+	tk.MustExec("create table t1 (a int);")
+	tk.MustExec("insert into t1 values (0),(1),(NULL);")
+	r := tk.MustQuery(`SELECT * FROM t1 WHERE NOT a BETWEEN 2 AND 3;`)
+	r.Check(testkit.Rows("0", "1"))
+	r = tk.MustQuery(`SELECT NOT 1 BETWEEN -5 AND 5;`)
+	r.Check(testkit.Rows("0"))
+	tk.MustExec("set sql_mode='high_not_precedence';")
+	r = tk.MustQuery(`SELECT * FROM t1 WHERE NOT a BETWEEN 2 AND 3;`)
+	r.Check(testkit.Rows())
+	r = tk.MustQuery(`SELECT NOT 1 BETWEEN -5 AND 5;`)
+	r.Check(testkit.Rows("1"))
+}
+
+func (s *testMySQLConstSuite) TestIgnoreSpaceMode(c *C) {
+	tk := testkit.NewTestKit(c, s.store)
+	tk.MustExec("use test")
+	tk.MustExec("set sql_mode=''")
+	tk.MustExec("CREATE TABLE COUNT (a bigint);")
+	tk.MustExec("DROP TABLE COUNT;")
+	tk.MustExec("CREATE TABLE `COUNT` (a bigint);")
+	tk.MustExec("DROP TABLE COUNT;")
+	_, err := tk.Exec("CREATE TABLE COUNT(a bigint);")
+	c.Assert(err, NotNil)
+	tk.MustExec("CREATE TABLE test.COUNT(a bigint);")
+	tk.MustExec("DROP TABLE COUNT;")
+
+	tk.MustExec("CREATE TABLE BIT_AND (a bigint);")
+	tk.MustExec("DROP TABLE BIT_AND;")
+	tk.MustExec("CREATE TABLE `BIT_AND` (a bigint);")
+	tk.MustExec("DROP TABLE BIT_AND;")
+	_, err = tk.Exec("CREATE TABLE BIT_AND(a bigint);")
+	c.Assert(err, NotNil)
+	tk.MustExec("CREATE TABLE test.BIT_AND(a bigint);")
+	tk.MustExec("DROP TABLE BIT_AND;")
+
+	tk.MustExec("CREATE TABLE NOW (a bigint);")
+	tk.MustExec("DROP TABLE NOW;")
+	tk.MustExec("CREATE TABLE `NOW` (a bigint);")
+	tk.MustExec("DROP TABLE NOW;")
+	_, err = tk.Exec("CREATE TABLE NOW(a bigint);")
+	c.Assert(err, NotNil)
+	tk.MustExec("CREATE TABLE test.NOW(a bigint);")
+	tk.MustExec("DROP TABLE NOW;")
+
+	tk.MustExec("set sql_mode='IGNORE_SPACE'")
+	_, err = tk.Exec("CREATE TABLE COUNT (a bigint);")
+	c.Assert(err, NotNil)
+	tk.MustExec("CREATE TABLE `COUNT` (a bigint);")
+	tk.MustExec("DROP TABLE COUNT;")
+	_, err = tk.Exec("CREATE TABLE COUNT(a bigint);")
+	c.Assert(err, NotNil)
+	tk.MustExec("CREATE TABLE test.COUNT(a bigint);")
+	tk.MustExec("DROP TABLE COUNT;")
+
+	_, err = tk.Exec("CREATE TABLE BIT_AND (a bigint);")
+	c.Assert(err, NotNil)
+	tk.MustExec("CREATE TABLE `BIT_AND` (a bigint);")
+	tk.MustExec("DROP TABLE BIT_AND;")
+	_, err = tk.Exec("CREATE TABLE BIT_AND(a bigint);")
+	c.Assert(err, NotNil)
+	tk.MustExec("CREATE TABLE test.BIT_AND(a bigint);")
+	tk.MustExec("DROP TABLE BIT_AND;")
+
+	_, err = tk.Exec("CREATE TABLE NOW (a bigint);")
+	c.Assert(err, NotNil)
+	tk.MustExec("CREATE TABLE `NOW` (a bigint);")
+	tk.MustExec("DROP TABLE NOW;")
+	_, err = tk.Exec("CREATE TABLE NOW(a bigint);")
+	c.Assert(err, NotNil)
+	tk.MustExec("CREATE TABLE test.NOW(a bigint);")
+	tk.MustExec("DROP TABLE NOW;")
+
+}
+
+func (s *testMySQLConstSuite) TestPadCharToFullLengthMode(c *C) {
+	tk := testkit.NewTestKit(c, s.store)
+	tk.MustExec("use test")
+	// test type `CHAR(n)`
+	tk.MustExec("drop table if exists t1")
+	tk.MustExec("create table t1 (a char(10));")
+	tk.MustExec("insert into t1 values ('xy');")
+	tk.MustExec("set sql_mode='';")
+	r := tk.MustQuery(`SELECT a='xy        ', char_length(a) FROM t1;`)
+	r.Check(testkit.Rows("0 2"))
+	r = tk.MustQuery(`SELECT count(*) FROM t1 WHERE a='xy        ';`)
+	r.Check(testkit.Rows("0"))
+	tk.MustExec("set sql_mode='PAD_CHAR_TO_FULL_LENGTH';")
+	r = tk.MustQuery(`SELECT a='xy        ', char_length(a) FROM t1;`)
+	r.Check(testkit.Rows("1 10"))
+	r = tk.MustQuery(`SELECT count(*) FROM t1 WHERE a='xy        ';`)
+	r.Check(testkit.Rows("1"))
+
+	// test type `VARCHAR(n)`
+	tk.MustExec("drop table if exists t1")
+	tk.MustExec("create table t1 (a varchar(10));")
+	tk.MustExec("insert into t1 values ('xy');")
+	tk.MustExec("set sql_mode='';")
+	r = tk.MustQuery(`SELECT a='xy        ', char_length(a) FROM t1;`)
+	r.Check(testkit.Rows("0 2"))
+	r = tk.MustQuery(`SELECT count(*) FROM t1 WHERE a='xy        ';`)
+	r.Check(testkit.Rows("0"))
+	tk.MustExec("set sql_mode='PAD_CHAR_TO_FULL_LENGTH';")
+	r = tk.MustQuery(`SELECT a='xy        ', char_length(a) FROM t1;`)
+	r.Check(testkit.Rows("0 2"))
+}
+
+func (s *testMySQLConstSuite) TestNoBackslashEscapesMode(c *C) {
+	tk := testkit.NewTestKit(c, s.store)
+	tk.MustExec("set sql_mode=''")
+	r := tk.MustQuery("SELECT '\\\\'")
+	r.Check(testkit.Rows("\\"))
+	tk.MustExec("set sql_mode='NO_BACKSLASH_ESCAPES'")
+	r = tk.MustQuery("SELECT '\\\\'")
+	r.Check(testkit.Rows("\\\\"))
+}
+
+func (s *testMySQLConstSuite) TestServerStatus(c *C) {
+	tests := []struct {
+		arg            uint16
+		IsCursorExists bool
+	}{
+		{0, false},
+		{mysql.ServerStatusInTrans | mysql.ServerStatusNoBackslashEscaped, false},
+		{mysql.ServerStatusCursorExists, true},
+		{mysql.ServerStatusCursorExists | mysql.ServerStatusLastRowSend, true},
+	}
+
+	for _, t := range tests {
+		ret := mysql.HasCursorExistsFlag(t.arg)
+		c.Assert(ret, Equals, t.IsCursorExists)
+	}
+}