240 lines
9.6 KiB
Go
240 lines
9.6 KiB
Go
// Copyright 2016 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 executor_test
|
|
|
|
import (
|
|
. "github.com/pingcap/check"
|
|
"github.com/pingcap/tidb/sessionctx"
|
|
"github.com/pingcap/tidb/sessionctx/variable"
|
|
"github.com/pingcap/tidb/util/testkit"
|
|
"golang.org/x/net/context"
|
|
)
|
|
|
|
func (s *testSuite) TestSetVar(c *C) {
|
|
tk := testkit.NewTestKit(c, s.store)
|
|
testSQL := "SET @a = 1;"
|
|
tk.MustExec(testSQL)
|
|
|
|
testSQL = `SET @a = "1";`
|
|
tk.MustExec(testSQL)
|
|
|
|
testSQL = "SET @a = null;"
|
|
tk.MustExec(testSQL)
|
|
|
|
testSQL = "SET @@global.autocommit = 1;"
|
|
tk.MustExec(testSQL)
|
|
|
|
// TODO: this test case should returns error.
|
|
// testSQL = "SET @@global.autocommit = null;"
|
|
// _, err := tk.Exec(testSQL)
|
|
// c.Assert(err, NotNil)
|
|
|
|
testSQL = "SET @@autocommit = 1;"
|
|
tk.MustExec(testSQL)
|
|
|
|
testSQL = "SET @@autocommit = null;"
|
|
_, err := tk.Exec(testSQL)
|
|
c.Assert(err, NotNil)
|
|
|
|
errTestSql := "SET @@date_format = 1;"
|
|
_, err = tk.Exec(errTestSql)
|
|
c.Assert(err, NotNil)
|
|
|
|
errTestSql = "SET @@rewriter_enabled = 1;"
|
|
_, err = tk.Exec(errTestSql)
|
|
c.Assert(err, NotNil)
|
|
|
|
errTestSql = "SET xxx = abcd;"
|
|
_, err = tk.Exec(errTestSql)
|
|
c.Assert(err, NotNil)
|
|
|
|
errTestSql = "SET @@global.a = 1;"
|
|
_, err = tk.Exec(errTestSql)
|
|
c.Assert(err, NotNil)
|
|
|
|
errTestSql = "SET @@global.timestamp = 1;"
|
|
_, err = tk.Exec(errTestSql)
|
|
c.Assert(err, NotNil)
|
|
|
|
// For issue 998
|
|
testSQL = "SET @issue998a=1, @issue998b=5;"
|
|
tk.MustExec(testSQL)
|
|
tk.MustQuery(`select @issue998a, @issue998b;`).Check(testkit.Rows("1 5"))
|
|
testSQL = "SET @@autocommit=0, @issue998a=2;"
|
|
tk.MustExec(testSQL)
|
|
tk.MustQuery(`select @issue998a, @@autocommit;`).Check(testkit.Rows("2 0"))
|
|
testSQL = "SET @@global.autocommit=1, @issue998b=6;"
|
|
tk.MustExec(testSQL)
|
|
tk.MustQuery(`select @issue998b, @@global.autocommit;`).Check(testkit.Rows("6 1"))
|
|
|
|
// For issue 4302
|
|
testSQL = "use test;drop table if exists x;create table x(a int);insert into x value(1);"
|
|
tk.MustExec(testSQL)
|
|
testSQL = "SET @issue4302=(select a from x limit 1);"
|
|
tk.MustExec(testSQL)
|
|
tk.MustQuery(`select @issue4302;`).Check(testkit.Rows("1"))
|
|
|
|
// Set default
|
|
// {ScopeGlobal | ScopeSession, "low_priority_updates", "OFF"},
|
|
// For global var
|
|
tk.MustQuery(`select @@global.low_priority_updates;`).Check(testkit.Rows("OFF"))
|
|
tk.MustExec(`set @@global.low_priority_updates="ON";`)
|
|
tk.MustQuery(`select @@global.low_priority_updates;`).Check(testkit.Rows("ON"))
|
|
tk.MustExec(`set @@global.low_priority_updates=DEFAULT;`) // It will be set to compiled-in default value.
|
|
tk.MustQuery(`select @@global.low_priority_updates;`).Check(testkit.Rows("OFF"))
|
|
// For session
|
|
tk.MustQuery(`select @@session.low_priority_updates;`).Check(testkit.Rows("OFF"))
|
|
tk.MustExec(`set @@global.low_priority_updates="ON";`)
|
|
tk.MustExec(`set @@session.low_priority_updates=DEFAULT;`) // It will be set to global var value.
|
|
tk.MustQuery(`select @@session.low_priority_updates;`).Check(testkit.Rows("ON"))
|
|
|
|
// For mysql jdbc driver issue.
|
|
tk.MustQuery(`select @@session.tx_read_only;`).Check(testkit.Rows("0"))
|
|
|
|
// Test session variable states.
|
|
vars := tk.Se.(sessionctx.Context).GetSessionVars()
|
|
tk.Se.CommitTxn(context.TODO())
|
|
tk.MustExec("set @@autocommit = 1")
|
|
c.Assert(vars.InTxn(), IsFalse)
|
|
c.Assert(vars.IsAutocommit(), IsTrue)
|
|
tk.MustExec("set @@autocommit = 0")
|
|
c.Assert(vars.IsAutocommit(), IsFalse)
|
|
|
|
tk.MustExec("set @@sql_mode = 'strict_trans_tables'")
|
|
c.Assert(vars.StrictSQLMode, IsTrue)
|
|
tk.MustExec("set @@sql_mode = ''")
|
|
c.Assert(vars.StrictSQLMode, IsFalse)
|
|
|
|
tk.MustExec("set names utf8")
|
|
charset, collation := vars.GetCharsetInfo()
|
|
c.Assert(charset, Equals, "utf8")
|
|
c.Assert(collation, Equals, "utf8_bin")
|
|
|
|
tk.MustExec("set @@character_set_results = NULL")
|
|
|
|
c.Assert(vars.ImportingData, IsFalse)
|
|
tk.MustExec("set @@tidb_import_data = '1'")
|
|
c.Assert(vars.ImportingData, IsTrue)
|
|
tk.MustExec("set @@tidb_import_data = '0'")
|
|
c.Assert(vars.ImportingData, IsFalse)
|
|
|
|
// Test set transaction isolation level, which is equivalent to setting variable "tx_isolation".
|
|
tk.MustExec("SET SESSION TRANSACTION ISOLATION LEVEL READ COMMITTED")
|
|
tk.MustQuery("select @@session.tx_isolation").Check(testkit.Rows("READ-COMMITTED"))
|
|
tk.MustQuery("select @@session.transaction_isolation").Check(testkit.Rows("READ-COMMITTED"))
|
|
tk.MustExec("SET SESSION TRANSACTION ISOLATION LEVEL READ UNCOMMITTED")
|
|
tk.MustQuery("select @@session.tx_isolation").Check(testkit.Rows("READ-UNCOMMITTED"))
|
|
tk.MustQuery("select @@session.transaction_isolation").Check(testkit.Rows("READ-UNCOMMITTED"))
|
|
tk.MustExec("SET GLOBAL TRANSACTION ISOLATION LEVEL SERIALIZABLE")
|
|
tk.MustQuery("select @@global.tx_isolation").Check(testkit.Rows("SERIALIZABLE"))
|
|
tk.MustQuery("select @@global.transaction_isolation").Check(testkit.Rows("SERIALIZABLE"))
|
|
|
|
// test synonyms variables
|
|
tk.MustExec("SET SESSION tx_isolation = 'READ-COMMITTED'")
|
|
tk.MustQuery("select @@session.tx_isolation").Check(testkit.Rows("READ-COMMITTED"))
|
|
tk.MustQuery("select @@session.transaction_isolation").Check(testkit.Rows("READ-COMMITTED"))
|
|
|
|
tk.MustExec("SET SESSION tx_isolation = 'READ-UNCOMMITTED'")
|
|
tk.MustQuery("select @@session.tx_isolation").Check(testkit.Rows("READ-UNCOMMITTED"))
|
|
tk.MustQuery("select @@session.transaction_isolation").Check(testkit.Rows("READ-UNCOMMITTED"))
|
|
|
|
tk.MustExec("SET SESSION transaction_isolation = 'SERIALIZABLE'")
|
|
tk.MustQuery("select @@session.tx_isolation").Check(testkit.Rows("SERIALIZABLE"))
|
|
tk.MustQuery("select @@session.transaction_isolation").Check(testkit.Rows("SERIALIZABLE"))
|
|
|
|
tk.MustExec("SET GLOBAL transaction_isolation = 'SERIALIZABLE'")
|
|
tk.MustQuery("select @@global.tx_isolation").Check(testkit.Rows("SERIALIZABLE"))
|
|
tk.MustQuery("select @@global.transaction_isolation").Check(testkit.Rows("SERIALIZABLE"))
|
|
|
|
tk.MustExec("SET GLOBAL transaction_isolation = 'READ-UNCOMMITTED'")
|
|
tk.MustQuery("select @@global.tx_isolation").Check(testkit.Rows("READ-UNCOMMITTED"))
|
|
tk.MustQuery("select @@global.transaction_isolation").Check(testkit.Rows("READ-UNCOMMITTED"))
|
|
|
|
tk.MustExec("SET GLOBAL tx_isolation = 'SERIALIZABLE'")
|
|
tk.MustQuery("select @@global.tx_isolation").Check(testkit.Rows("SERIALIZABLE"))
|
|
tk.MustQuery("select @@global.transaction_isolation").Check(testkit.Rows("SERIALIZABLE"))
|
|
|
|
tk.MustExec("SET SESSION tx_read_only = 1")
|
|
tk.MustExec("SET SESSION tx_read_only = 0")
|
|
tk.MustQuery("select @@session.tx_read_only").Check(testkit.Rows("0"))
|
|
tk.MustQuery("select @@session.transaction_read_only").Check(testkit.Rows("0"))
|
|
|
|
tk.MustExec("SET GLOBAL tx_read_only = 1")
|
|
tk.MustExec("SET GLOBAL tx_read_only = 0")
|
|
tk.MustQuery("select @@global.tx_read_only").Check(testkit.Rows("0"))
|
|
tk.MustQuery("select @@global.transaction_read_only").Check(testkit.Rows("0"))
|
|
|
|
tk.MustExec("SET SESSION transaction_read_only = 1")
|
|
tk.MustExec("SET SESSION transaction_read_only = 0")
|
|
tk.MustQuery("select @@session.tx_read_only").Check(testkit.Rows("0"))
|
|
tk.MustQuery("select @@session.transaction_read_only").Check(testkit.Rows("0"))
|
|
|
|
tk.MustExec("SET SESSION transaction_read_only = 1")
|
|
tk.MustQuery("select @@session.tx_read_only").Check(testkit.Rows("1"))
|
|
tk.MustQuery("select @@session.transaction_read_only").Check(testkit.Rows("1"))
|
|
|
|
tk.MustExec("SET GLOBAL transaction_read_only = 1")
|
|
tk.MustExec("SET GLOBAL transaction_read_only = 0")
|
|
tk.MustQuery("select @@global.tx_read_only").Check(testkit.Rows("0"))
|
|
tk.MustQuery("select @@global.transaction_read_only").Check(testkit.Rows("0"))
|
|
|
|
tk.MustExec("SET GLOBAL transaction_read_only = 1")
|
|
tk.MustQuery("select @@global.tx_read_only").Check(testkit.Rows("1"))
|
|
tk.MustQuery("select @@global.transaction_read_only").Check(testkit.Rows("1"))
|
|
|
|
// Even the transaction fail, set session variable would success.
|
|
tk.MustExec("BEGIN")
|
|
tk.MustExec("SET SESSION TRANSACTION ISOLATION LEVEL READ COMMITTED")
|
|
_, err = tk.Exec(`INSERT INTO t VALUES ("sdfsdf")`)
|
|
c.Assert(err, NotNil)
|
|
tk.MustExec("COMMIT")
|
|
tk.MustQuery("select @@session.tx_isolation").Check(testkit.Rows("READ-COMMITTED"))
|
|
|
|
tk.MustExec("set global avoid_temporal_upgrade = on")
|
|
tk.MustQuery(`select @@global.avoid_temporal_upgrade;`).Check(testkit.Rows("ON"))
|
|
tk.MustExec("set @@global.avoid_temporal_upgrade = off")
|
|
tk.MustQuery(`select @@global.avoid_temporal_upgrade;`).Check(testkit.Rows("off"))
|
|
tk.MustExec("set session sql_log_bin = on")
|
|
tk.MustQuery(`select @@session.sql_log_bin;`).Check(testkit.Rows("ON"))
|
|
tk.MustExec("set sql_log_bin = off")
|
|
tk.MustQuery(`select @@session.sql_log_bin;`).Check(testkit.Rows("off"))
|
|
tk.MustExec("set @@sql_log_bin = on")
|
|
tk.MustQuery(`select @@session.sql_log_bin;`).Check(testkit.Rows("ON"))
|
|
}
|
|
|
|
func (s *testSuite) TestSetCharset(c *C) {
|
|
tk := testkit.NewTestKit(c, s.store)
|
|
tk.MustExec(`SET NAMES latin1`)
|
|
|
|
ctx := tk.Se.(sessionctx.Context)
|
|
sessionVars := ctx.GetSessionVars()
|
|
for _, v := range variable.SetNamesVariables {
|
|
sVar, err := variable.GetSessionSystemVar(sessionVars, v)
|
|
c.Assert(err, IsNil)
|
|
c.Assert(sVar != "utf8", IsTrue)
|
|
}
|
|
tk.MustExec(`SET NAMES utf8`)
|
|
for _, v := range variable.SetNamesVariables {
|
|
sVar, err := variable.GetSessionSystemVar(sessionVars, v)
|
|
c.Assert(err, IsNil)
|
|
c.Assert(sVar, Equals, "utf8")
|
|
}
|
|
sVar, err := variable.GetSessionSystemVar(sessionVars, variable.CollationConnection)
|
|
c.Assert(err, IsNil)
|
|
c.Assert(sVar, Equals, "utf8_bin")
|
|
|
|
// Issue 1523
|
|
tk.MustExec(`SET NAMES binary`)
|
|
}
|