Слияние кода завершено, страница обновится автоматически
// Copyright 2015 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 session_test
import (
"fmt"
"strconv"
"strings"
"testing"
"github.com/hanchuanchuan/goInception/config"
"github.com/hanchuanchuan/goInception/session"
"github.com/hanchuanchuan/goInception/util/testkit"
. "github.com/pingcap/check"
"golang.org/x/net/context"
)
var _ = Suite(&testSessionIncExecSuite{})
func TestExec(t *testing.T) {
TestingT(t)
}
type testSessionIncExecSuite struct {
testCommon
}
func (s *testSessionIncExecSuite) SetUpSuite(c *C) {
s.initSetUp(c)
inc := &s.defaultInc
inc.EnableFingerprint = true
inc.SqlSafeUpdates = 0
inc.EnableDropTable = true
config.GetGlobalConfig().Osc.OscMaxFlowCtl = 1
}
func (s *testSessionIncExecSuite) TearDownSuite(c *C) {
s.tearDownSuite(c)
}
func (s *testSessionIncExecSuite) TearDownTest(c *C) {
s.reset()
s.tearDownTest(c)
}
func (s *testSessionIncExecSuite) testErrorCode(c *C, sql string, errors ...*session.SQLError) [][]interface{} {
if s.isAPI {
s.sessionService.LoadOptions(session.SourceOptions{
Host: s.defaultInc.BackupHost,
Port: int(s.defaultInc.BackupPort),
User: s.defaultInc.BackupUser,
Password: s.defaultInc.BackupPassword,
RealRowCount: s.realRowCount,
IgnoreWarnings: true,
})
s.testExecuteResult(c, sql, errors...)
return nil
}
if s.tk == nil {
s.tk = testkit.NewTestKitWithInit(c, s.store)
}
session.CheckAuditSetting(config.GetGlobalConfig())
res := s.runExec(sql)
row := res.Rows()[s.getAffectedRows()-1]
errCode := 0
if len(errors) > 0 {
for _, e := range errors {
level := session.GetErrorLevel(e.Code)
if int(level) > errCode {
errCode = int(level)
}
}
}
if errCode > 0 {
errMsgs := []string{}
for _, e := range errors {
errMsgs = append(errMsgs, e.Error())
}
c.Assert(row[4], Equals, strings.Join(errMsgs, "\n"), Commentf("%v", row))
}
// 没有错误时允许有警告
// if errCode == 0 {
// c.Assert(row[2], Not(Equals), "2", Commentf("%v", res.Rows()))
// } else {
// c.Assert(row[2], Equals, strconv.Itoa(errCode), Commentf("%v", res.Rows()))
// }
// 无错误时需要校验结果是否标记为已执行
if errCode == 0 {
c.Assert(strings.Contains(row[3].(string), "Execute Successfully"), Equals, true, Commentf("%v", res.Rows()))
for _, row := range res.Rows() {
c.Assert(row[2], Not(Equals), "2", Commentf("%v", res.Rows()))
}
} else {
c.Assert(row[2], Equals, strconv.Itoa(errCode), Commentf("%v", res.Rows()))
}
return res.Rows()
}
func (s *testSessionIncExecSuite) testExecuteResult(c *C, sql string, errors ...*session.SQLError) {
result, err := s.sessionService.RunExecute(context.Background(), s.useDB+sql)
c.Assert(err, IsNil)
// for _, row := range result {
// if row.ErrLevel == 2 {
// fmt.Println(fmt.Sprintf("sql: %v, err: %v", row.Sql, row.ErrorMessage))
// } else {
// fmt.Println(fmt.Sprintf("[%v] sql: %v", session.StatusList[row.StageStatus], row.Sql))
// }
// }
s.records = result
row := result[len(result)-1]
errCode := uint8(0)
if len(errors) > 0 {
for _, e := range errors {
level := session.GetErrorLevel(e.Code)
if level > errCode {
errCode = level
}
}
}
if errCode > 0 {
errMsgs := []string{}
for _, e := range errors {
errMsgs = append(errMsgs, e.Error())
}
c.Assert(row.ErrorMessage, Equals, strings.Join(errMsgs, "\n"), Commentf("%v", result))
}
if errCode == 0 {
// Execute Successfully
c.Assert(row.StageStatus, Equals, uint8(2), Commentf("%v", s.records))
for _, row := range s.records {
c.Assert(row.ErrLevel, Not(Equals), uint8(2), Commentf("%v", s.records))
}
} else {
c.Assert(row.ErrLevel, Equals, errCode, Commentf("%#v", row))
}
}
func (s *testSessionIncExecSuite) TestCreateTable(c *C) {
sql := ""
config.GetGlobalConfig().Inc.CheckColumnComment = false
config.GetGlobalConfig().Inc.CheckTableComment = false
sql = `drop table if exists nullkeytest1;
create table nullkeytest1(c1 int, c2 int, c3 int, primary key(c1), key ix_1(c2));`
s.testErrorCode(c, sql)
// 支持innodb引擎
config.GetGlobalConfig().Inc.EnableSetEngine = true
config.GetGlobalConfig().Inc.SupportEngine = "innodb"
s.mustRunExec(c, "drop table if exists t1;create table t1(c1 varchar(10))engine = innodb;")
// 允许设置存储引擎
config.GetGlobalConfig().Inc.EnableSetEngine = true
config.GetGlobalConfig().Inc.SupportEngine = "innodb"
s.mustRunExec(c, "drop table if exists t1;create table t1(c1 varchar(10))engine = innodb;")
if s.DBVersion >= 50600 {
sql = "drop table if exists t1;create table t1(id int primary key,t1 timestamp default CURRENT_TIMESTAMP,t2 timestamp ON UPDATE CURRENT_TIMESTAMP);"
if s.explicitDefaultsForTimestamp || !(strings.Contains(s.sqlMode, "TRADITIONAL") ||
(strings.Contains(s.sqlMode, "STRICT_") && strings.Contains(s.sqlMode, "NO_ZERO_DATE"))) {
s.testErrorCode(c, sql)
} else {
s.testErrorCode(c, sql, session.NewErr(session.ER_INVALID_DEFAULT, "t2"))
}
}
indexMaxLength := 767
if s.innodbLargePrefix {
indexMaxLength = 3072
}
s.runExec("drop table if exists t1")
if s.innodbLargePrefix {
sql = "create table t1(c1 int,c2 varchar(400),c3 varchar(400),index idx_1(c1,c2))charset utf8;;"
s.testErrorCode(c, sql)
} else {
sql = "create table t1(c1 int,c2 varchar(400),c3 varchar(400),index idx_1(c1,c2))charset utf8;;"
s.testErrorCode(c, sql,
session.NewErr(session.ER_TOO_LONG_KEY, "idx_1", indexMaxLength))
}
sql = "drop table if exists t1;create table t1(c1 int,c2 varchar(200),c3 varchar(200),index idx_1(c1,c2))charset utf8;;"
s.testErrorCode(c, sql)
config.GetGlobalConfig().Inc.EnableBlobType = true
// sql = "drop table if exists t1;create table t1(c1 int,c2 text, unique uq_1(c1,c2(3068)));"
// if indexMaxLength == 3072 {
// s.testErrorCode(c, sql)
// } else {
// s.testErrorCode(c, sql,
// session.NewErr(session.ER_TOO_LONG_KEY, "", indexMaxLength))
// }
config.GetGlobalConfig().Inc.EnableBlobType = false
config.GetGlobalConfig().Inc.EnableBlobType = true
sql = "drop table if exists t1;create table t1(pt blob ,primary key (pt));"
s.testErrorCode(c, sql,
session.NewErr(session.ER_BLOB_USED_AS_KEY, "pt"))
sql = "drop table if exists t1;create table t1(`id` int, key `primary`(`id`));"
s.testErrorCode(c, sql,
session.NewErr(session.ER_WRONG_NAME_FOR_INDEX, "primary", "t1"))
sql = "drop table if exists t1;create table t1(c1.c2 varchar(10));"
s.testErrorCode(c, sql,
session.NewErr(session.ER_WRONG_TABLE_NAME, "c1"))
sql = "drop table if exists t1;create table t1(c1 int default null primary key , age int);"
s.testErrorCode(c, sql,
session.NewErr(session.ER_PRIMARY_CANT_HAVE_NULL))
sql = "drop table if exists t1;create table t1(id int null primary key , age int);"
s.testErrorCode(c, sql,
session.NewErr(session.ER_PRIMARY_CANT_HAVE_NULL))
sql = "drop table if exists t1;create table t1(id int default null, age int, primary key(id));"
s.testErrorCode(c, sql,
session.NewErr(session.ER_PRIMARY_CANT_HAVE_NULL))
sql = "drop table if exists t1;create table t1(id int null, age int, primary key(id));"
s.testErrorCode(c, sql,
session.NewErr(session.ER_PRIMARY_CANT_HAVE_NULL))
if s.DBVersion >= 50600 {
sql = `drop table if exists t1;create table t1(
id int auto_increment comment 'test',
crtTime datetime not null DEFAULT CURRENT_TIMESTAMP comment 'test',
uptTime datetime DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP comment 'test',
primary key(id)) comment 'test';`
s.testErrorCode(c, sql)
}
s.mustRunExec(c, "drop table if exists t1;create table t1(c1 int);")
// 测试表名大小写
sql = "insert into T1 values(1);"
if s.ignoreCase {
s.testErrorCode(c, sql)
} else {
s.testErrorCode(c, sql,
session.NewErr(session.ER_TABLE_NOT_EXISTED_ERROR, "test_inc.T1"))
}
}
func (s *testSessionIncExecSuite) TestDropTable(c *C) {
config.GetGlobalConfig().Inc.EnableDropTable = true
s.mustRunExec(c, "drop table if exists t1;create table t1(id int);drop table t1;")
}
func (s *testSessionIncExecSuite) TestAlterTableAddColumn(c *C) {
config.GetGlobalConfig().Inc.CheckColumnComment = false
config.GetGlobalConfig().Inc.CheckTableComment = false
config.GetGlobalConfig().Inc.EnableDropTable = true
sql := ""
sql = "drop table if exists t1;create table t1(id int);alter table t1 add column c1 int;"
s.testErrorCode(c, sql)
s.mustRunExec(c, "drop table if exists t1;create table t1(id int);alter table t1 add column c1 int first;alter table t1 add column c2 int after c1;")
// char列建议
config.GetGlobalConfig().Inc.MaxCharLength = 100
s.mustRunExec(c, `drop table if exists t1;create table t1(id int);
alter table t1 add column c1 char(200);
alter table t1 add column c2 varchar(200);`)
sql = "drop table if exists t1;create table t1(id int primary key);alter table t1 add column c1 bit default b'0';"
// pt-osc
config.GetGlobalConfig().Osc.OscOn = true
s.mustRunExec(c, sql)
// gh-ost
config.GetGlobalConfig().Osc.OscOn = false
config.GetGlobalConfig().Ghost.GhostOn = true
s.mustRunExec(c, sql)
sql = "drop table if exists t1;create table t1 (id int primary key , age int);"
s.testErrorCode(c, sql)
sql = "drop table if exists t1;create table t1 (id int primary key);alter table t1 add column (c1 int,c2 varchar(20));"
s.testErrorCode(c, sql)
if s.DBVersion >= 50600 {
// 指定特殊选项
sql = "drop table if exists t1;create table t1 (id int primary key);alter table t1 add column c1 int,ALGORITHM=INPLACE, LOCK=NONE;"
s.testErrorCode(c, sql)
}
// 特殊字符
config.GetGlobalConfig().Inc.CheckIdentifier = false
sql = "drop table if exists `t3!@#$^&*()`;create table `t3!@#$^&*()`(id int primary key);alter table `t3!@#$^&*()` add column `c3!@#$^&*()2` int comment '123';"
s.testErrorCode(c, sql)
// pt-osc
config.GetGlobalConfig().Osc.OscOn = true
s.testErrorCode(c, sql)
// gh-ost
config.GetGlobalConfig().Osc.OscOn = false
config.GetGlobalConfig().Ghost.GhostOn = true
s.testErrorCode(c, sql)
}
func (s *testSessionIncExecSuite) TestAlterTableAlterColumn(c *C) {
s.mustRunExec(c, "drop table if exists t1;create table t1(id int);alter table t1 alter column id set default '1';")
s.mustRunExec(c, "drop table if exists t1;create table t1(id int);alter table t1 alter column id drop default ;alter table t1 alter column id set default '1';")
}
func (s *testSessionIncExecSuite) TestAlterTableModifyColumn(c *C) {
config.GetGlobalConfig().Inc.CheckColumnComment = false
config.GetGlobalConfig().Inc.CheckTableComment = false
s.mustRunExec(c, "drop table if exists t1;create table t1(id int,c1 int);alter table t1 modify column c1 int first;")
s.mustRunExec(c, "drop table if exists t1;create table t1(id int,c1 int);alter table t1 modify column id int after c1;")
}
func (s *testSessionIncExecSuite) TestAlterTableDropColumn(c *C) {
s.mustRunExec(c, "drop table if exists t1;create table t1(id int,c1 int);alter table t1 drop column c1;")
}
func (s *testSessionIncExecSuite) TestInsert(c *C) {
config.GetGlobalConfig().Inc.CheckInsertField = false
config.GetGlobalConfig().IncLevel.ER_WITH_INSERT_FIELD = 0
sql := ""
// 字段警告
config.GetGlobalConfig().Inc.CheckInsertField = true
config.GetGlobalConfig().IncLevel.ER_WITH_INSERT_FIELD = 1
sql = "drop table if exists t1;create table t1(id int,c1 int);insert into t1 values(1,1);"
s.testErrorCode(c, sql,
session.NewErr(session.ER_WITH_INSERT_FIELD))
config.GetGlobalConfig().Inc.CheckInsertField = false
config.GetGlobalConfig().IncLevel.ER_WITH_INSERT_FIELD = 0
sql = "drop table if exists t1;create table t1(id int,c1 int);insert into t1(id) values();"
s.testErrorCode(c, sql,
session.NewErr(session.ER_WITH_INSERT_VALUES))
// insert select 表不存在
sql = ("drop table if exists t1,t2;create table t1(id int,c1 int );insert into t1(id,c1) select 1,null from t2;")
s.testErrorCode(c, sql,
session.NewErr(session.ER_TABLE_NOT_EXISTED_ERROR, "test_inc.t2"))
// select where
config.GetGlobalConfig().Inc.CheckDMLWhere = true
sql = "drop table if exists t1;create table t1(id int,c1 int );insert into t1(id,c1) select 1,null from t1;"
s.testErrorCode(c, sql,
session.NewErr(session.ER_NO_WHERE_CONDITION))
config.GetGlobalConfig().Inc.CheckDMLWhere = false
// limit
config.GetGlobalConfig().Inc.CheckDMLLimit = true
sql = "insert into t1(id,c1) select 1,null from t1 limit 1;"
s.testErrorCode(c, sql,
session.NewErr(session.ER_WITH_LIMIT_CONDITION))
config.GetGlobalConfig().Inc.CheckDMLLimit = false
// order by rand()
// config.GetGlobalConfig().Inc.CheckDMLOrderBy = true
sql = "insert into t1(id,c1) select 1,null from t1 order by rand();"
s.testErrorCode(c, sql,
session.NewErr(session.ER_ORDERY_BY_RAND))
// 受影响行数
s.testErrorCode(c, "insert into t1 values(1,1),(2,2);")
s.testAffectedRows(c, 2)
s.testErrorCode(c, "insert into t1(id,c1) select 1,null;")
s.testAffectedRows(c, 1)
sql = "drop table if exists t1;create table t1(c1 char(100) not null);insert into t1(c1) values(null);"
s.testErrorCode(c, sql,
session.NewErr(session.ER_BAD_NULL_ERROR, "test_inc.t1.c1", 1))
}
func (s *testSessionIncExecSuite) TestUpdate(c *C) {
config.GetGlobalConfig().Inc.CheckInsertField = false
config.GetGlobalConfig().IncLevel.ER_WITH_INSERT_FIELD = 0
sql := ""
// 表不存在
sql = "drop table if exists t1;update t1 set c1 = 1;"
s.testErrorCode(c, sql,
session.NewErr(session.ER_TABLE_NOT_EXISTED_ERROR, "test_inc.t1"))
sql = "drop table if exists t1;create table t1(id int);update t1 set c1 = 1;"
s.testErrorCode(c, sql,
session.NewErr(session.ER_COLUMN_NOT_EXISTED, "c1"))
sql = "drop table if exists t1;create table t1(id int,c1 int);update t1 set c1 = 1,c2 = 1;"
s.testErrorCode(c, sql,
session.NewErr(session.ER_COLUMN_NOT_EXISTED, "t1.c2"))
sql = (`drop table if exists t1;drop table if exists t2;create table t1(id int primary key,c1 int);
create table t2(id int primary key,c1 int,c2 int);
update t1 inner join t2 on t1.id=t2.id2 set t1.c1=t2.c1 where c11=1;`)
s.testErrorCode(c, sql,
session.NewErr(session.ER_COLUMN_NOT_EXISTED, "t2.id2"),
session.NewErr(session.ER_COLUMN_NOT_EXISTED, "c11"),
)
sql = (`drop table if exists t1;drop table if exists t2;create table t1(id int primary key,c1 int);
create table t2(id int primary key,c1 int,c2 int);
update t1,t2 t3 set t1.c1=t2.c3 where t1.id=t3.id;`)
s.testErrorCode(c, sql,
session.NewErr(session.ER_COLUMN_NOT_EXISTED, "t2.c3"),
)
sql = (`drop table if exists t1;drop table if exists t2;create table t1(id int primary key,c1 int);
create table t2(id int primary key,c1 int,c2 int);
update t1,t2 t3 set t1.c1=t2.c3 where t1.id=t3.id;`)
s.testErrorCode(c, sql,
session.NewErr(session.ER_COLUMN_NOT_EXISTED, "t2.c3"),
)
// where
config.GetGlobalConfig().Inc.CheckDMLWhere = true
sql = "drop table if exists t1;create table t1(id int,c1 int);update t1 set c1 = 1;"
s.testErrorCode(c, sql,
session.NewErr(session.ER_NO_WHERE_CONDITION))
config.GetGlobalConfig().Inc.CheckDMLWhere = false
// limit
config.GetGlobalConfig().Inc.CheckDMLLimit = true
sql = "drop table if exists t1;create table t1(id int,c1 int);update t1 set c1 = 1 limit 1;"
s.testErrorCode(c, sql,
session.NewErr(session.ER_WITH_LIMIT_CONDITION),
)
config.GetGlobalConfig().Inc.CheckDMLLimit = false
// order by rand()
config.GetGlobalConfig().Inc.CheckDMLOrderBy = true
sql = ("drop table if exists t1;create table t1(id int,c1 int);update t1 set c1 = 1 order by rand();")
s.testErrorCode(c, sql,
session.NewErr(session.ER_WITH_ORDERBY_CONDITION),
)
config.GetGlobalConfig().Inc.CheckDMLOrderBy = false
// 受影响行数
s.mustRunExec(c, "drop table if exists t1;create table t1(id int,c1 int);update t1 set c1 = 1;")
s.testAffectedRows(c, 0)
s.mustRunExec(c, "drop table if exists t1;create table t1(id int primary key,c1 int);insert into t1 values(1,1),(2,2);update t1 set c1 = 1 where id = 1;")
s.testAffectedRows(c, 0)
s.mustRunExec(c, "drop table if exists t1;create table t1(id int primary key,c1 int);insert into t1 values(1,1),(2,2);update t1 set c1 = 10 where id = 1;")
s.testAffectedRows(c, 1)
// -------------------- 多表update -------------------
sql = `drop table if exists table1;drop table if exists table2;
create table table1(id1 int primary key,c1 int,c2 int);
create table table2(id2 int primary key,c1 int,c2 int,c22 int);
insert into table1 values(1,1,1),(2,1,1);
insert into table2 values(1,1,1,null),(2,null,null,null);
update table1 t1,table2 t2 set t1.c1=10,t2.c22=20 where t1.id1=t2.id2 and t2.c1=1;`
s.mustRunExec(c, sql)
s.testAffectedRows(c, 2)
}
func (s *testSessionIncExecSuite) TestDelete(c *C) {
saved := config.GetGlobalConfig().Inc
defer func() {
config.GetGlobalConfig().Inc = saved
config.GetGlobalConfig().IncLevel.ER_WITH_INSERT_FIELD = 1
}()
config.GetGlobalConfig().Inc.CheckInsertField = false
config.GetGlobalConfig().IncLevel.ER_WITH_INSERT_FIELD = 0
sql := ""
sql = "drop table if exists t1"
s.testErrorCode(c, sql)
s.mustRunExec(c, `drop table if exists t1;drop table if exists t2;create table t1(id int primary key,c1 int);
create table t2(id int primary key,c1 int,c2 int);
delete t2 from t1 inner join t2 on t1.id=t2.id where t1.c1=1;`)
// 受影响行数
s.mustRunExec(c, "drop table if exists t1;create table t1(id int,c1 int);delete from t1 where id = 1;")
s.testAffectedRows(c, 0)
s.mustRunExec(c, "drop table if exists t1;create table t1(id int,c1 int);insert into t1(id) values(1);")
sql = `delete from t1 where id =1;`
s.testErrorCode(c, sql)
s.testAffectedRows(c, 1)
sql = `drop table if exists t1;create table t1(id int primary key,c1 int);
delete tmp from t1 as tmp where tmp.id=1;`
s.testErrorCode(c, sql)
sql = `drop table if exists t1;create table t1(id int primary key,c1 int);
delete s1 from t1 as s1 inner join t1 as s2 on s1.c1 = s2.c1 where s1.id > s2.id;`
s.testErrorCode(c, sql)
s.mustRunExec(c, "drop table if exists t1,t2;create table t1(id int primary key,c1 int);")
sql = `create table t2(id int primary key,c1 int);
delete t1 from t1 inner join t2 where t2.c1 = 1;`
s.testErrorCode(c, sql)
}
func (s *testSessionIncExecSuite) TestCreateDataBase(c *C) {
inc := &config.GetGlobalConfig().Inc
inc.EnableDropDatabase = true
sql := ""
sql = "drop database if exists test1111111111111111111;create database if not exists test1111111111111111111;"
s.testErrorCode(c, sql)
dbname := fmt.Sprintf("%s_%d_%s", strings.ReplaceAll(inc.BackupHost, ".", "_"), inc.BackupPort, "test_inc")
s.testErrorCode(c, fmt.Sprintf("drop database if exists %s;", dbname))
// 存在
sql = "create database test1111111111111111111;create database test1111111111111111111;"
s.testErrorCode(c, sql,
session.NewErrf("数据库'test1111111111111111111'已存在."))
config.GetGlobalConfig().Inc.EnableDropDatabase = false
// 不存在
sql = "drop database if exists test1111111111111111111;"
s.testErrorCode(c, sql,
session.NewErr(session.ER_CANT_DROP_DATABASE, "test1111111111111111111"))
sql = "drop database test1111111111111111111;"
s.testErrorCode(c, sql,
session.NewErr(session.ER_CANT_DROP_DATABASE, "test1111111111111111111"))
config.GetGlobalConfig().Inc.EnableDropDatabase = true
// if not exists 创建
sql = "create database if not exists test1111111111111111111;create database if not exists test1111111111111111111;"
s.testErrorCode(c, sql)
// create database
sql = "create database aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
s.testErrorCode(c, sql,
session.NewErr(session.ER_TOO_LONG_IDENT, "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"))
sql = "create database mysql"
s.testErrorCode(c, sql,
session.NewErrf("数据库'%s'已存在.", "mysql"))
sql = "drop database if exists test1111111111111111111;"
s.mustRunExec(c, sql)
// 字符集
config.GetGlobalConfig().Inc.EnableSetCharset = false
config.GetGlobalConfig().Inc.SupportCharset = ""
sql = "drop database if exists test123456;create database test123456 character set utf8;"
s.testErrorCode(c, sql,
session.NewErr(session.ER_CANT_SET_CHARSET, "utf8"))
config.GetGlobalConfig().Inc.SupportCharset = "utf8mb4"
sql = "drop database if exists test123456;create database test123456 character set utf8;"
s.testErrorCode(c, sql,
session.NewErr(session.ER_CANT_SET_CHARSET, "utf8"))
config.GetGlobalConfig().Inc.EnableSetCharset = true
config.GetGlobalConfig().Inc.SupportCharset = "utf8,utf8mb4"
sql = "drop database if exists test123456;create database test123456 character set utf8;"
s.testErrorCode(c, sql)
config.GetGlobalConfig().Inc.EnableSetCharset = true
config.GetGlobalConfig().Inc.SupportCharset = "utf8,utf8mb4"
sql = "drop database if exists test123456;create database test123456 character set latin1;"
s.testErrorCode(c, sql,
session.NewErr(session.ErrCharsetNotSupport, "utf8,utf8mb4"))
}
func (s *testSessionIncExecSuite) TestRenameTable(c *C) {
sql := ""
// 不存在
sql = "drop table if exists t1;drop table if exists t2;create table t1(id int primary key);alter table t1 rename t2;"
s.testErrorCode(c, sql)
sql = "drop table if exists t1;drop table if exists t2;create table t1(id int primary key);rename table t1 to t2;"
s.testErrorCode(c, sql)
// 存在
sql = "drop table if exists t1;create table t1(id int primary key);rename table t1 to t1;"
s.testErrorCode(c, sql,
session.NewErr(session.ER_TABLE_EXISTS_ERROR, "t1"))
}
func (s *testSessionIncExecSuite) TestCreateView(c *C) {
sql := ""
sql = "drop table if exists t1;create table t1(id int primary key);create view v1 as select * from t1;"
s.testErrorCode(c, sql,
session.NewErrf("命令禁止! 无法创建视图'v1'."))
}
func (s *testSessionIncExecSuite) TestAlterTableAddIndex(c *C) {
config.GetGlobalConfig().Inc.CheckColumnComment = false
config.GetGlobalConfig().Inc.CheckTableComment = false
sql := ""
// add index
sql = "drop table if exists t1;create table t1(id int);alter table t1 add index idx (c1)"
s.testErrorCode(c, sql,
session.NewErr(session.ER_COLUMN_NOT_EXISTED, "t1.c1"))
sql = "drop table if exists t1;create table t1(id int,c1 int);alter table t1 add index idx (c1);"
s.testErrorCode(c, sql)
sql = "drop table if exists t1;create table t1(id int,c1 int);alter table t1 add index idx (c1);alter table t1 add index idx (c1);"
s.testErrorCode(c, sql,
session.NewErr(session.ER_DUP_INDEX, "idx", "test_inc", "t1"))
}
func (s *testSessionIncExecSuite) TestAlterTableDropIndex(c *C) {
config.GetGlobalConfig().Inc.CheckColumnComment = false
config.GetGlobalConfig().Inc.CheckTableComment = false
sql := ""
// drop index
sql = "drop table if exists t1;create table t1(id int);alter table t1 drop index idx"
s.testErrorCode(c, sql,
session.NewErr(session.ER_CANT_DROP_FIELD_OR_KEY, "t1.idx"))
sql = "drop table if exists t1;create table t1(c1 int);alter table t1 add index idx (c1);alter table t1 drop index idx;"
s.testErrorCode(c, sql)
}
func (s *testSessionIncExecSuite) TestShowVariables(c *C) {
sql := ""
sql = "inception show variables;"
s.tk.MustQueryInc(sql)
c.Assert(s.tk.Se.AffectedRows(), GreaterEqual, uint64(102))
sql = "inception get variables;"
s.tk.MustQueryInc(sql)
c.Assert(s.tk.Se.AffectedRows(), GreaterEqual, uint64(102))
sql = "inception show variables like 'backup_password';"
res := s.tk.MustQueryInc(sql)
row := res.Rows()[s.getAffectedRows()-1]
if row[1].(string) != "" {
c.Assert(row[1].(string)[:1], Equals, "*")
}
}
// 无法审核，原因是需要自己做SetSessionManager
// func (s *testSessionIncExecSuite) TestShowProcesslist(c *C) {
// sql := ""
// sql = "inception show processlist;"
// tk.MustQueryInc(sql)
// c.Assert(tk.Se.AffectedRows(), GreaterEqual, 1)
// sql = "inception get processlist;"
// tk.MustQueryInc(sql)
// c.Assert(tk.Se.AffectedRows(), GreaterEqual, 1)
// sql = "inception show variables like 'backup_password';"
// res := s.tk.MustQueryInc(sql)
// row := res.Rows()[s.getAffectedRows()-1]
// if row[1].(string) != "" {
// c.Assert(row[1].(string)[:1], Equals, "*")
// }
// }
func (s *testSessionIncExecSuite) TestSetVariables(c *C) {
sql := ""
sql = "inception show variables;"
s.tk.MustQueryInc(sql)
c.Assert(s.tk.Se.AffectedRows(), GreaterEqual, uint64(102))
// 不区分session和global.所有会话全都global级别
s.tk.MustExecInc("inception set global max_keys = 20;")
s.tk.MustExecInc("inception set session max_keys = 10;")
result := s.tk.MustQueryInc("inception show variables like 'max_keys';")
result.Check(testkit.Rows("max_keys 10"))
s.tk.MustExecInc("inception set ghost_default_retries = 70;")
result = s.tk.MustQueryInc("inception show variables like 'ghost_default_retries';")
result.Check(testkit.Rows("ghost_default_retries 70"))
s.tk.MustExecInc("inception set osc_max_thread_running = 100;")
result = s.tk.MustQueryInc("inception show variables like 'osc_max_thread_running';")
result.Check(testkit.Rows("osc_max_thread_running 100"))
// 无效参数
res, err := s.tk.ExecInc("inception set osc_max_thread_running1 = 100;")
c.Assert(err, NotNil)
c.Assert(err.Error(), Equals, "无效参数")
if res != nil {
c.Assert(res.Close(), IsNil)
}
// 无效参数
res, err = s.tk.ExecInc("inception set osc_max_thread_running = 'abc';")
c.Assert(err, NotNil)
c.Assert(err.Error(), Equals, "[variable:1232]Incorrect argument type to variable 'osc_max_thread_running'")
if res != nil {
c.Assert(res.Close(), IsNil)
}
}
func (s *testSessionIncExecSuite) TestAlterTable(c *C) {
config.GetGlobalConfig().Inc.CheckColumnComment = false
config.GetGlobalConfig().Inc.CheckTableComment = false
config.GetGlobalConfig().Inc.EnableDropTable = true
sql := ""
sql = "drop table if exists t1;create table t1(id int auto_increment primary key,c1 int);"
s.mustRunExec(c, sql)
// 删除后添加列
sql = "alter table t1 drop column c1;alter table t1 add column c1 varchar(20);"
s.testErrorCode(c, sql)
sql = "alter table t1 drop column c1,add column c1 varchar(20);"
s.testErrorCode(c, sql)
// 删除后添加索引
sql = "drop table if exists t1;create table t1(id int primary key,c1 int,key ix(c1));"
s.mustRunExec(c, sql)
sql = "alter table t1 drop index ix;alter table t1 add index ix(c1);"
s.testErrorCode(c, sql)
sql = "alter table t1 drop index ix,add index ix(c1);"
s.testErrorCode(c, sql)
sql = "alter table t1 add column c2 varchar(20) comment '!@#$%^&*()_+[]{}\\|;:\",.<>/?';"
s.testErrorCode(c, sql)
sql = "alter table t1 add column c3 varchar(20) comment \"!@#$%^&*()_+[]{}\\|;:',.<>/?\";"
s.testErrorCode(c, sql)
sql = "alter table t1 add column `c4` varchar(20) comment \"!@#$%^&*()_+[]{}\\|;:',.<>/?\";"
s.testErrorCode(c, sql)
}
func (s *testSessionIncExecSuite) TestAlterTablePtOSC(c *C) {
saved := config.GetGlobalConfig().Inc
savedOsc := config.GetGlobalConfig().Osc
defer func() {
config.GetGlobalConfig().Inc = saved
config.GetGlobalConfig().Osc = savedOsc
}()
config.GetGlobalConfig().Inc.CheckColumnComment = false
config.GetGlobalConfig().Inc.CheckTableComment = false
config.GetGlobalConfig().Inc.EnableDropTable = true
config.GetGlobalConfig().Osc.OscOn = true
config.GetGlobalConfig().Ghost.GhostOn = false
config.GetGlobalConfig().Osc.OscMinTableSize = 0
sql := "drop table if exists t1;create table t1(id int auto_increment primary key,c1 int);"
s.mustRunExec(c, sql)
// 删除后添加列
sql = `# 这是一条注释
alter table t1 drop column c1;alter table t1 add column c1 varchar(20);`
s.testErrorCode(c, sql)
sql = `/* 这是一条注释 */
alter table t1 drop column c1,add column c1 varchar(20);`
s.testErrorCode(c, sql)
sql = `-- 这是一条注释
alter table t1 add column c2 varchar(20) comment '!@#$%^&*()_+[]{}\\|;:",.<>/?';`
s.testErrorCode(c, sql)
sql = "alter table t1 add column c3 varchar(20) comment \"!@#$%^&*()_+[]{}\\|;:',.<>/?\";"
s.testErrorCode(c, sql)
sql = "alter table t1 add column `c4` varchar(20) comment \"!@#$%^&*()_+[]{}\\|;:',.<>/?\";"
s.testErrorCode(c, sql)
sql = "alter table t1 add column `c5` varchar(20) comment \"!@#$%^&*()_+[]{}\\|;:',.<>/?\"; -- 测试注释"
s.testErrorCode(c, sql)
}
func (s *testSessionIncExecSuite) TestAlterTableGhost(c *C) {
saved := config.GetGlobalConfig().Inc
savedOsc := config.GetGlobalConfig().Osc
defer func() {
config.GetGlobalConfig().Inc = saved
config.GetGlobalConfig().Osc = savedOsc
}()
config.GetGlobalConfig().Inc.CheckColumnComment = false
config.GetGlobalConfig().Inc.CheckTableComment = false
config.GetGlobalConfig().Inc.EnableDropTable = true
config.GetGlobalConfig().Osc.OscOn = false
config.GetGlobalConfig().Ghost.GhostOn = true
config.GetGlobalConfig().Osc.OscMinTableSize = 0
sql := "drop table if exists t1;create table t1(id int auto_increment primary key,c1 int);"
s.mustRunExec(c, sql)
// 删除后添加列
sql = `# 这是一条注释
alter table t1 drop column c1;alter table t1 add column c1 varchar(20);`
s.testErrorCode(c, sql)
sql = `/* 这是一条注释 */
alter table t1 drop column c1,add column c1 varchar(20);`
s.testErrorCode(c, sql)
sql = `-- 这是一条注释
alter table t1 add column c2 varchar(20) comment '!@#$%^&*()_+[]{}\\|;:",.<>/?';`
s.testErrorCode(c, sql)
sql = "alter table t1 add column c3 varchar(20) comment \"!@#$%^&*()_+[]{}\\|;:',.<>/?\";"
s.testErrorCode(c, sql)
sql = "alter table t1 add column `c4` varchar(20) comment \"!@#$%^&*()_+[]{}\\|;:',.<>/?\";"
s.testErrorCode(c, sql)
sql = "alter table t1 add column `c5` varchar(20) comment \"!@#$%^&*()_+[]{}\\|;:',.<>/?\"; -- 测试注释"
s.testErrorCode(c, sql)
}
// TestDisplayWidth 测试列指定长度参数
func (s *testSessionIncExecSuite) TestDisplayWidth(c *C) {
sql := ""
config.GetGlobalConfig().Inc.CheckColumnComment = false
config.GetGlobalConfig().Inc.CheckTableComment = false
config.GetGlobalConfig().Inc.EnableEnumSetBit = true
s.mustRunExec(c, "drop table if exists t1;")
// 数据类型 警告
sql = `create table t1(c1 bit(64),
c2 tinyint(255),
c3 smallint(255),
c4 mediumint(255),
c5 int(255),
c6 bigint(255) );`
s.testErrorCode(c, sql)
// 数据类型 警告
sql = `alter table t1 add column c11 tinyint(255);`
s.testErrorCode(c, sql)
}
func (s *testSessionIncExecSuite) TestWhereCondition(c *C) {
sql := ""
s.mustRunExec(c, "drop table if exists t1,t2;create table t1(id int);")
sql = "update t1 set id = 1 where 123;"
s.mustRunExec(c, sql)
sql = "update t1 set id = 1 where null;"
s.mustRunExec(c, sql)
sql = `
delete from t1 where 1+2;
delete from t1 where 1-2;
delete from t1 where 1*2;
delete from t1 where 1/2;
update t1 set id = 1 where 1&2;
update t1 set id = 1 where 1|2;
update t1 set id = 1 where 1^2;
update t1 set id = 1 where 1 div 2;
`
s.mustRunExec(c, sql)
sql = `
update t1 set id = 1 where 1+2=3;
update t1 set id = 1 where id is null;
update t1 set id = 1 where id is not null;
update t1 set id = 1 where 1=1;
update t1 set id = 1 where id in (1,2);
update t1 set id = 1 where id is true;
`
s.mustRunExec(c, sql)
}
Вы можете оставить комментарий после Вход в систему
Неприемлемый контент может быть отображен здесь и не будет показан на странице. Вы можете проверить и изменить его с помощью соответствующей функции редактирования.
Если вы подтверждаете, что содержание не содержит непристойной лексики/перенаправления на рекламу/насилия/вульгарной порнографии/нарушений/пиратства/ложного/незначительного или незаконного контента, связанного с национальными законами и предписаниями, вы можете нажать «Отправить» для подачи апелляции, и мы обработаем ее как можно скорее.
Опубликовать ( 0 )