// 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")
}