Files
tidb/server/util_test.go

110 lines
3.3 KiB
Go

// 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 server
import (
. "github.com/pingcap/check"
"github.com/pingcap/tidb/mysql"
"github.com/pingcap/tidb/util/testleak"
"github.com/pingcap/tidb/util/types"
)
var _ = Suite(&testUtilSuite{})
type testUtilSuite struct {
}
func (s *testUtilSuite) TestDumpBinaryTime(c *C) {
defer testleak.AfterTest(c)()
t, err := types.ParseTimestamp("0000-00-00 00:00:00.0000000")
c.Assert(err, IsNil)
d, err := dumpBinaryDateTime(t, nil)
c.Assert(err, IsNil)
c.Assert(d, DeepEquals, []byte{11, 1, 0, 1, 1, 0, 0, 0, 0, 0, 0, 0})
t, err = types.ParseDatetime("0000-00-00 00:00:00.0000000")
c.Assert(err, IsNil)
d, err = dumpBinaryDateTime(t, nil)
c.Assert(err, IsNil)
c.Assert(d, DeepEquals, []byte{11, 1, 0, 1, 1, 0, 0, 0, 0, 0, 0, 0})
t, err = types.ParseDate("0000-00-00")
c.Assert(err, IsNil)
d, err = dumpBinaryDateTime(t, nil)
c.Assert(err, IsNil)
c.Assert(d, DeepEquals, []byte{4, 1, 0, 1, 1})
myDuration, err := types.ParseDuration("0000-00-00 00:00:00.0000000", 6)
c.Assert(err, IsNil)
d = dumpBinaryTime(myDuration.Duration)
c.Assert(d, DeepEquals, []byte{0})
}
func (s *testUtilSuite) TestDumpTextValue(c *C) {
defer testleak.AfterTest(c)()
bs, err := dumpTextValue(mysql.TypeLonglong, types.NewIntDatum(10))
c.Assert(err, IsNil)
c.Assert(string(bs), Equals, "10")
bs, err = dumpTextValue(mysql.TypeLonglong, types.NewUintDatum(11))
c.Assert(err, IsNil)
c.Assert(string(bs), Equals, "11")
f32 := types.NewFloat32Datum(1.2)
bs, err = dumpTextValue(mysql.TypeDouble, f32)
c.Assert(err, IsNil)
c.Assert(string(bs), Equals, "1.2")
f32.SetFrac(2)
bs, err = dumpTextValue(mysql.TypeDouble, f32)
c.Assert(err, IsNil)
c.Assert(string(bs), Equals, "1.20")
f64 := types.NewFloat64Datum(2.2)
bs, err = dumpTextValue(mysql.TypeDouble, f64)
c.Assert(err, IsNil)
c.Assert(string(bs), Equals, "2.2")
f64.SetFrac(2)
bs, err = dumpTextValue(mysql.TypeDouble, f64)
c.Assert(err, IsNil)
c.Assert(string(bs), Equals, "2.20")
bs, err = dumpTextValue(mysql.TypeBlob, types.NewBytesDatum([]byte("foo")))
c.Assert(err, IsNil)
c.Assert(string(bs), Equals, "foo")
bs, err = dumpTextValue(mysql.TypeVarchar, types.NewStringDatum("bar"))
c.Assert(err, IsNil)
c.Assert(string(bs), Equals, "bar")
var d types.Datum
time, err := types.ParseTime("2017-01-05 23:59:59.575601", mysql.TypeDatetime, 0)
c.Assert(err, IsNil)
d.SetMysqlTime(time)
bs, err = dumpTextValue(mysql.TypeDatetime, d)
c.Assert(err, IsNil)
c.Assert(string(bs), Equals, "2017-01-06 00:00:00")
duration, err := types.ParseDuration("11:30:45", 0)
c.Assert(err, IsNil)
d.SetMysqlDuration(duration)
bs, err = dumpTextValue(mysql.TypeDuration, d)
c.Assert(err, IsNil)
c.Assert(string(bs), Equals, "11:30:45")
d.SetMysqlDecimal(types.NewDecFromStringForTest("1.23"))
bs, err = dumpTextValue(mysql.TypeNewDecimal, d)
c.Assert(err, IsNil)
c.Assert(string(bs), Equals, "1.23")
}