session_test: remove session_test.go (#5862)
This commit is contained in:
Han Fei
@ -1868,3 +1868,37 @@ func (s *testSessionSuite) TestCastTimeToDate(c *C) {
|
||||
date = time.Now().In(time.FixedZone("UTC", 8*int(time.Hour/time.Second)))
|
||||
tk.MustQuery("select cast(time('12:23:34') as date)").Check(testkit.Rows(date.Format("2006-01-02")))
|
||||
}
|
||||
|
||||
func (s *testSessionSuite) TestSetGlobalTZ(c *C) {
|
||||
defer testleak.AfterTest(c)()
|
||||
goCtx := goctx.Background()
|
||||
|
||||
tk := testkit.NewTestKitWithInit(c, s.store)
|
||||
tk.MustExec("set time_zone = '+08:00'")
|
||||
rs, err := tk.Exec("show variables like 'time_zone'")
|
||||
c.Assert(err, IsNil)
|
||||
row0, err := rs.Next(goCtx)
|
||||
c.Assert(err, IsNil)
|
||||
c.Assert(row0, NotNil)
|
||||
c.Assert(row0.Len(), Equals, 2)
|
||||
c.Assert(row0.GetBytes(1), BytesEquals, []byte("+08:00"))
|
||||
|
||||
tk.MustExec("set global time_zone = '+00:00'")
|
||||
|
||||
rs, err = tk.Exec("show variables like 'time_zone'")
|
||||
c.Assert(err, IsNil)
|
||||
row0, err = rs.Next(goCtx)
|
||||
c.Assert(err, IsNil)
|
||||
c.Assert(row0, NotNil)
|
||||
c.Assert(row0.Len(), Equals, 2)
|
||||
c.Assert(row0.GetBytes(1), BytesEquals, []byte("+08:00"))
|
||||
|
||||
tk1 := testkit.NewTestKitWithInit(c, s.store)
|
||||
rs1, err := tk1.Exec("show variables like 'time_zone'")
|
||||
c.Assert(err, IsNil)
|
||||
row1, err := rs1.Next(goCtx)
|
||||
c.Assert(err, IsNil)
|
||||
c.Assert(row1, NotNil)
|
||||
c.Assert(row1.Len(), Equals, 2)
|
||||
c.Assert(row1.GetBytes(1), BytesEquals, []byte("+00:00"))
|
||||
}
|
||||
|
||||
140
session_test.go
140
session_test.go
@ -1,140 +0,0 @@
|
||||
// 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 tidb
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
. "github.com/pingcap/check"
|
||||
"github.com/pingcap/tidb/domain"
|
||||
"github.com/pingcap/tidb/kv"
|
||||
"github.com/pingcap/tidb/terror"
|
||||
"github.com/pingcap/tidb/util/testleak"
|
||||
goctx "golang.org/x/net/context"
|
||||
)
|
||||
|
||||
var (
|
||||
_ = Suite(&testSessionSuite{})
|
||||
)
|
||||
|
||||
type testSessionSuite struct {
|
||||
dbName string
|
||||
createDBSQL string
|
||||
dropDBSQL string
|
||||
useDBSQL string
|
||||
createTableSQL string
|
||||
dropTableSQL string
|
||||
selectSQL string
|
||||
|
||||
store kv.Storage
|
||||
dom *domain.Domain
|
||||
}
|
||||
|
||||
func (s *testSessionSuite) SetUpSuite(c *C) {
|
||||
s.dbName = "test_session_db"
|
||||
s.dropTableSQL = `Drop TABLE if exists t;`
|
||||
s.createTableSQL = `CREATE TABLE t(id TEXT);`
|
||||
s.selectSQL = `SELECT * from t;`
|
||||
|
||||
s.store = newStore(c, s.dbName)
|
||||
dom, err := BootstrapSession(s.store)
|
||||
c.Assert(err, IsNil)
|
||||
s.dom = dom
|
||||
}
|
||||
|
||||
func (s *testSessionSuite) TearDownSuite(c *C) {
|
||||
removeStore(c, s.dbName)
|
||||
s.dom.Close()
|
||||
err := s.store.Close()
|
||||
c.Assert(err, IsNil)
|
||||
}
|
||||
|
||||
func (s *testSessionSuite) TestSchemaCheckerSimple(c *C) {
|
||||
defer testleak.AfterTest(c)()
|
||||
lease := 5 * time.Millisecond
|
||||
validator := domain.NewSchemaValidator(lease)
|
||||
checker := &schemaLeaseChecker{SchemaValidator: validator}
|
||||
|
||||
// Add some schema versions and delta table IDs.
|
||||
ts := uint64(time.Now().UnixNano())
|
||||
validator.Update(ts, 0, 2, []int64{1})
|
||||
validator.Update(ts, 2, 4, []int64{2})
|
||||
|
||||
// checker's schema version is the same as the current schema version.
|
||||
checker.schemaVer = 4
|
||||
err := checker.Check(ts)
|
||||
c.Assert(err, IsNil)
|
||||
|
||||
// checker's schema version is less than the current schema version, and it doesn't exist in validator's items.
|
||||
// checker's related table ID isn't in validator's changed table IDs.
|
||||
checker.schemaVer = 2
|
||||
checker.relatedTableIDs = []int64{3}
|
||||
err = checker.Check(ts)
|
||||
c.Assert(err, IsNil)
|
||||
// The checker's schema version isn't in validator's items.
|
||||
checker.schemaVer = 1
|
||||
checker.relatedTableIDs = []int64{3}
|
||||
err = checker.Check(ts)
|
||||
c.Assert(terror.ErrorEqual(err, domain.ErrInfoSchemaChanged), IsTrue)
|
||||
// checker's related table ID is in validator's changed table IDs.
|
||||
checker.relatedTableIDs = []int64{2}
|
||||
err = checker.Check(ts)
|
||||
c.Assert(terror.ErrorEqual(err, domain.ErrInfoSchemaChanged), IsTrue)
|
||||
|
||||
// validator's latest schema version is expired.
|
||||
time.Sleep(lease + time.Microsecond)
|
||||
checker.schemaVer = 4
|
||||
checker.relatedTableIDs = []int64{3}
|
||||
err = checker.Check(ts)
|
||||
c.Assert(err, IsNil)
|
||||
nowTS := uint64(time.Now().UnixNano())
|
||||
// Use checker.SchemaValidator.Check instead of checker.Check here because backoff make CI slow.
|
||||
result := checker.SchemaValidator.Check(nowTS, checker.schemaVer, checker.relatedTableIDs)
|
||||
c.Assert(result, Equals, domain.ResultUnknown)
|
||||
}
|
||||
|
||||
func (s *testSessionSuite) TestSetGlobalTZ(c *C) {
|
||||
defer testleak.AfterTest(c)()
|
||||
dbName := "testTZ"
|
||||
dropDBSQL := fmt.Sprintf("drop database %s;", dbName)
|
||||
goCtx := goctx.Background()
|
||||
|
||||
se0 := newSession(c, s.store, dbName)
|
||||
mustExecSQL(c, se0, "set time_zone = '+08:00'")
|
||||
rs0 := mustExecSQL(c, se0, "show variables like 'time_zone'")
|
||||
row0, err := rs0.Next(goCtx)
|
||||
c.Assert(err, IsNil)
|
||||
c.Assert(row0, NotNil)
|
||||
c.Assert(row0.Len(), Equals, 2)
|
||||
c.Assert(row0.GetBytes(1), BytesEquals, []byte("+08:00"))
|
||||
|
||||
mustExecSQL(c, se0, "set global time_zone = '+00:00'")
|
||||
|
||||
rs0 = mustExecSQL(c, se0, "show variables like 'time_zone'")
|
||||
c.Assert(err, IsNil)
|
||||
c.Assert(row0, NotNil)
|
||||
c.Assert(row0.Len(), Equals, 2)
|
||||
c.Assert(row0.GetBytes(1), BytesEquals, []byte("+08:00"))
|
||||
|
||||
se1 := newSession(c, s.store, dbName)
|
||||
rs1 := mustExecSQL(c, se1, "show variables like 'time_zone'")
|
||||
row1, err := rs1.Next(goCtx)
|
||||
c.Assert(err, IsNil)
|
||||
c.Assert(row1, NotNil)
|
||||
c.Assert(row1.Len(), Equals, 2)
|
||||
c.Assert(row1.GetBytes(1), BytesEquals, []byte("+00:00"))
|
||||
|
||||
mustExecSQL(c, se0, dropDBSQL)
|
||||
}
|
||||
54
tidb_test.go
54
tidb_test.go
@ -28,6 +28,7 @@ import (
|
||||
"github.com/pingcap/tidb/domain"
|
||||
"github.com/pingcap/tidb/kv"
|
||||
"github.com/pingcap/tidb/store/mockstore"
|
||||
"github.com/pingcap/tidb/terror"
|
||||
"github.com/pingcap/tidb/types"
|
||||
"github.com/pingcap/tidb/util"
|
||||
"github.com/pingcap/tidb/util/auth"
|
||||
@ -176,6 +177,50 @@ func (s *testMainSuite) TestSysSessionPoolGoroutineLeak(c *C) {
|
||||
c.Assert(after-before, Less, 3)
|
||||
}
|
||||
|
||||
func (s *testMainSuite) TestSchemaCheckerSimple(c *C) {
|
||||
defer testleak.AfterTest(c)()
|
||||
lease := 5 * time.Millisecond
|
||||
validator := domain.NewSchemaValidator(lease)
|
||||
checker := &schemaLeaseChecker{SchemaValidator: validator}
|
||||
|
||||
// Add some schema versions and delta table IDs.
|
||||
ts := uint64(time.Now().UnixNano())
|
||||
validator.Update(ts, 0, 2, []int64{1})
|
||||
validator.Update(ts, 2, 4, []int64{2})
|
||||
|
||||
// checker's schema version is the same as the current schema version.
|
||||
checker.schemaVer = 4
|
||||
err := checker.Check(ts)
|
||||
c.Assert(err, IsNil)
|
||||
|
||||
// checker's schema version is less than the current schema version, and it doesn't exist in validator's items.
|
||||
// checker's related table ID isn't in validator's changed table IDs.
|
||||
checker.schemaVer = 2
|
||||
checker.relatedTableIDs = []int64{3}
|
||||
err = checker.Check(ts)
|
||||
c.Assert(err, IsNil)
|
||||
// The checker's schema version isn't in validator's items.
|
||||
checker.schemaVer = 1
|
||||
checker.relatedTableIDs = []int64{3}
|
||||
err = checker.Check(ts)
|
||||
c.Assert(terror.ErrorEqual(err, domain.ErrInfoSchemaChanged), IsTrue)
|
||||
// checker's related table ID is in validator's changed table IDs.
|
||||
checker.relatedTableIDs = []int64{2}
|
||||
err = checker.Check(ts)
|
||||
c.Assert(terror.ErrorEqual(err, domain.ErrInfoSchemaChanged), IsTrue)
|
||||
|
||||
// validator's latest schema version is expired.
|
||||
time.Sleep(lease + time.Microsecond)
|
||||
checker.schemaVer = 4
|
||||
checker.relatedTableIDs = []int64{3}
|
||||
err = checker.Check(ts)
|
||||
c.Assert(err, IsNil)
|
||||
nowTS := uint64(time.Now().UnixNano())
|
||||
// Use checker.SchemaValidator.Check instead of checker.Check here because backoff make CI slow.
|
||||
result := checker.SchemaValidator.Check(nowTS, checker.schemaVer, checker.relatedTableIDs)
|
||||
c.Assert(result, Equals, domain.ResultUnknown)
|
||||
}
|
||||
|
||||
func newStore(c *C, dbPath string) kv.Storage {
|
||||
store, err := mockstore.NewMockTikvStore()
|
||||
c.Assert(err, IsNil)
|
||||
@ -241,12 +286,3 @@ func match(c *C, row []types.Datum, expected ...interface{}) {
|
||||
c.Assert(got, Equals, need)
|
||||
}
|
||||
}
|
||||
|
||||
func mustExecFailed(c *C, se Session, sql string, args ...interface{}) {
|
||||
r, err := exec(se, sql, args...)
|
||||
if err == nil && r != nil {
|
||||
// sometimes we may meet error after executing first row.
|
||||
_, err = r.Next(goctx.Background())
|
||||
}
|
||||
c.Assert(err, NotNil)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user