You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 

360 lines
11 KiB

// Copyright 2017 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 expression
import (
"math"
"strings"
"time"
. "github.com/pingcap/check"
"github.com/pingcap/parser/ast"
"github.com/pingcap/parser/mysql"
"github.com/pingcap/tidb/types"
"github.com/pingcap/tidb/util/chunk"
"github.com/pingcap/tidb/util/testutil"
)
func (s *testEvaluatorSuite) TestInetAton(c *C) {
tbl := []struct {
Input interface{}
Expected interface{}
}{
{"", nil},
{nil, nil},
{"255.255.255.255", 4294967295},
{"0.0.0.0", 0},
{"127.0.0.1", 2130706433},
{"0.0.0.256", nil},
{"113.14.22.3", 1896748547},
{"127", 127},
{"127.255", 2130706687},
{"127,256", nil},
{"127.2.1", 2130837505},
{"123.2.1.", nil},
{"127.0.0.1.1", nil},
}
dtbl := tblToDtbl(tbl)
fc := funcs[ast.InetAton]
for _, t := range dtbl {
f, err := fc.getFunction(s.ctx, s.datumsToConstants(t["Input"]))
c.Assert(err, IsNil)
d, err := evalBuiltinFunc(f, chunk.Row{})
c.Assert(err, IsNil)
c.Assert(d, testutil.DatumEquals, t["Expected"][0])
}
}
func (s *testEvaluatorSuite) TestIsIPv4(c *C) {
tests := []struct {
ip string
expect interface{}
}{
{"192.168.1.1", 1},
{"255.255.255.255", 1},
{"10.t.255.255", 0},
{"10.1.2.3.4", 0},
{"2001:250:207:0:0:eef2::1", 0},
{"::ffff:1.2.3.4", 0},
{"1...1", 0},
{"192.168.1.", 0},
{".168.1.2", 0},
{"168.1.2", 0},
{"1.2.3.4.5", 0},
}
fc := funcs[ast.IsIPv4]
for _, test := range tests {
ip := types.NewStringDatum(test.ip)
f, err := fc.getFunction(s.ctx, s.datumsToConstants([]types.Datum{ip}))
c.Assert(err, IsNil)
result, err := evalBuiltinFunc(f, chunk.Row{})
c.Assert(err, IsNil)
c.Assert(result, testutil.DatumEquals, types.NewDatum(test.expect))
}
// test NULL input for is_ipv4
var argNull types.Datum
f, _ := fc.getFunction(s.ctx, s.datumsToConstants([]types.Datum{argNull}))
r, err := evalBuiltinFunc(f, chunk.Row{})
c.Assert(err, IsNil)
c.Assert(r, testutil.DatumEquals, types.NewDatum(0))
}
func (s *testEvaluatorSuite) TestUUID(c *C) {
f, err := newFunctionForTest(s.ctx, ast.UUID)
c.Assert(err, IsNil)
d, err := f.Eval(chunk.Row{})
c.Assert(err, IsNil)
parts := strings.Split(d.GetString(), "-")
c.Assert(len(parts), Equals, 5)
for i, p := range parts {
switch i {
case 0:
c.Assert(len(p), Equals, 8)
case 1:
c.Assert(len(p), Equals, 4)
case 2:
c.Assert(len(p), Equals, 4)
case 3:
c.Assert(len(p), Equals, 4)
case 4:
c.Assert(len(p), Equals, 12)
}
}
_, err = funcs[ast.UUID].getFunction(s.ctx, s.datumsToConstants(nil))
c.Assert(err, IsNil)
}
func (s *testEvaluatorSuite) TestAnyValue(c *C) {
tbl := []struct {
arg interface{}
ret interface{}
}{
{nil, nil},
{1234, 1234},
{-0x99, -0x99},
{3.1415926, 3.1415926},
{"Hello, World", "Hello, World"},
}
for _, t := range tbl {
fc := funcs[ast.AnyValue]
f, err := fc.getFunction(s.ctx, s.datumsToConstants(types.MakeDatums(t.arg)))
c.Assert(err, IsNil)
r, err := evalBuiltinFunc(f, chunk.Row{})
c.Assert(err, IsNil)
c.Assert(r, testutil.DatumEquals, types.NewDatum(t.ret))
}
}
func (s *testEvaluatorSuite) TestIsIPv6(c *C) {
tests := []struct {
ip string
expect interface{}
}{
{"2001:250:207:0:0:eef2::1", 1},
{"2001:0250:0207:0001:0000:0000:0000:ff02", 1},
{"2001:250:207::eff2::1,", 0},
{"192.168.1.1", 0},
{"::ffff:1.2.3.4", 1},
}
fc := funcs[ast.IsIPv6]
for _, test := range tests {
ip := types.NewStringDatum(test.ip)
f, err := fc.getFunction(s.ctx, s.datumsToConstants([]types.Datum{ip}))
c.Assert(err, IsNil)
result, err := evalBuiltinFunc(f, chunk.Row{})
c.Assert(err, IsNil)
c.Assert(result, testutil.DatumEquals, types.NewDatum(test.expect))
}
// test NULL input for is_ipv6
var argNull types.Datum
f, _ := fc.getFunction(s.ctx, s.datumsToConstants([]types.Datum{argNull}))
r, err := evalBuiltinFunc(f, chunk.Row{})
c.Assert(err, IsNil)
c.Assert(r, testutil.DatumEquals, types.NewDatum(0))
}
func (s *testEvaluatorSuite) TestInetNtoa(c *C) {
tests := []struct {
ip int
expect interface{}
}{
{167773449, "10.0.5.9"},
{2063728641, "123.2.0.1"},
{0, "0.0.0.0"},
{545460846593, nil},
{-1, nil},
{math.MaxUint32, "255.255.255.255"},
}
fc := funcs[ast.InetNtoa]
for _, test := range tests {
ip := types.NewDatum(test.ip)
f, err := fc.getFunction(s.ctx, s.datumsToConstants([]types.Datum{ip}))
c.Assert(err, IsNil)
result, err := evalBuiltinFunc(f, chunk.Row{})
c.Assert(err, IsNil)
c.Assert(result, testutil.DatumEquals, types.NewDatum(test.expect))
}
var argNull types.Datum
f, _ := fc.getFunction(s.ctx, s.datumsToConstants([]types.Datum{argNull}))
r, err := evalBuiltinFunc(f, chunk.Row{})
c.Assert(err, IsNil)
c.Assert(r.IsNull(), IsTrue)
}
func (s *testEvaluatorSuite) TestInet6NtoA(c *C) {
tests := []struct {
ip []byte
expect interface{}
}{
// Success cases
{[]byte{0x00, 0x00, 0x00, 0x00}, "0.0.0.0"},
{[]byte{0x0A, 0x00, 0x05, 0x09}, "10.0.5.9"},
{[]byte{0xFD, 0xFE, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x5A, 0x55, 0xCA, 0xFF, 0xFE,
0xFA, 0x90, 0x89}, "fdfe::5a55:caff:fefa:9089"},
{[]byte{0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xFF, 0xFF, 0x01,
0x02, 0x03, 0x04}, "::ffff:1.2.3.4"},
{[]byte{0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xFF, 0xFF, 0xFF,
0xFF, 0xFF, 0xFF}, "::ffff:255.255.255.255"},
// Fail cases
{[]byte{}, nil}, // missing bytes
{[]byte{0x0A, 0x00, 0x05}, nil}, // missing a byte ipv4
{[]byte{0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xFF, 0xFF, 0xFF,
0xFF, 0xFF}, nil}, // missing a byte ipv6
}
fc := funcs[ast.Inet6Ntoa]
for _, test := range tests {
ip := types.NewDatum(test.ip)
f, err := fc.getFunction(s.ctx, s.datumsToConstants([]types.Datum{ip}))
c.Assert(err, IsNil)
result, err := evalBuiltinFunc(f, chunk.Row{})
c.Assert(err, IsNil)
c.Assert(result, testutil.DatumEquals, types.NewDatum(test.expect))
}
var argNull types.Datum
f, _ := fc.getFunction(s.ctx, s.datumsToConstants([]types.Datum{argNull}))
r, err := evalBuiltinFunc(f, chunk.Row{})
c.Assert(err, IsNil)
c.Assert(r.IsNull(), IsTrue)
}
func (s *testEvaluatorSuite) TestInet6AtoN(c *C) {
tests := []struct {
ip string
expect interface{}
}{
{"0.0.0.0", []byte{0x00, 0x00, 0x00, 0x00}},
{"10.0.5.9", []byte{0x0A, 0x00, 0x05, 0x09}},
{"fdfe::5a55:caff:fefa:9089", []byte{0xFD, 0xFE, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x5A, 0x55, 0xCA, 0xFF, 0xFE, 0xFA, 0x90, 0x89}},
{"::ffff:1.2.3.4", []byte{0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xFF, 0xFF, 0x01, 0x02, 0x03, 0x04}},
{"", nil},
{"Not IP address", nil},
{"::ffff:255.255.255.255", []byte{0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF}},
}
fc := funcs[ast.Inet6Aton]
for _, test := range tests {
ip := types.NewDatum(test.ip)
f, err := fc.getFunction(s.ctx, s.datumsToConstants([]types.Datum{ip}))
c.Assert(err, IsNil)
result, err := evalBuiltinFunc(f, chunk.Row{})
c.Assert(err, IsNil)
c.Assert(result, testutil.DatumEquals, types.NewDatum(test.expect))
}
var argNull types.Datum
f, _ := fc.getFunction(s.ctx, s.datumsToConstants([]types.Datum{argNull}))
r, err := evalBuiltinFunc(f, chunk.Row{})
c.Assert(err, IsNil)
c.Assert(r.IsNull(), IsTrue)
}
func (s *testEvaluatorSuite) TestIsIPv4Mapped(c *C) {
tests := []struct {
ip []byte
expect interface{}
}{
{[]byte{}, 0},
{[]byte{0x10, 0x10, 0x10, 0x10}, 0},
{[]byte{0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0xff, 0xff, 0x1, 0x2, 0x3, 0x4}, 1},
{[]byte{0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x1, 0xff, 0xff, 0x1, 0x2, 0x3, 0x4}, 0},
{[]byte{0x0, 0x1, 0x2, 0x3, 0x4, 0x5, 0x6}, 0},
}
fc := funcs[ast.IsIPv4Mapped]
for _, test := range tests {
ip := types.NewDatum(test.ip)
f, err := fc.getFunction(s.ctx, s.datumsToConstants([]types.Datum{ip}))
c.Assert(err, IsNil)
result, err := evalBuiltinFunc(f, chunk.Row{})
c.Assert(err, IsNil)
c.Assert(result, testutil.DatumEquals, types.NewDatum(test.expect))
}
var argNull types.Datum
f, _ := fc.getFunction(s.ctx, s.datumsToConstants([]types.Datum{argNull}))
r, err := evalBuiltinFunc(f, chunk.Row{})
c.Assert(err, IsNil)
c.Assert(r, testutil.DatumEquals, types.NewDatum(int64(0)))
}
func (s *testEvaluatorSuite) TestIsIPv4Compat(c *C) {
tests := []struct {
ip []byte
expect interface{}
}{
{[]byte{}, 0},
{[]byte{0x10, 0x10, 0x10, 0x10}, 0},
{[]byte{0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x1, 0x2, 0x3, 0x4}, 1},
{[]byte{0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x1, 0x0, 0x0, 0x1, 0x2, 0x3, 0x4}, 0},
{[]byte{0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x1, 0xff, 0xff, 0x1, 0x2, 0x3, 0x4}, 0},
{[]byte{0x0, 0x1, 0x2, 0x3, 0x4, 0x5, 0x6}, 0},
}
fc := funcs[ast.IsIPv4Compat]
for _, test := range tests {
ip := types.NewDatum(test.ip)
f, err := fc.getFunction(s.ctx, s.datumsToConstants([]types.Datum{ip}))
c.Assert(err, IsNil)
result, err := evalBuiltinFunc(f, chunk.Row{})
c.Assert(err, IsNil)
c.Assert(result, testutil.DatumEquals, types.NewDatum(test.expect))
}
var argNull types.Datum
f, _ := fc.getFunction(s.ctx, s.datumsToConstants([]types.Datum{argNull}))
r, err := evalBuiltinFunc(f, chunk.Row{})
c.Assert(err, IsNil)
c.Assert(r, testutil.DatumEquals, types.NewDatum(0))
}
func (s *testEvaluatorSuite) TestNameConst(c *C) {
dec := types.NewDecFromFloatForTest(123.123)
tm := types.NewTime(types.FromGoTime(time.Now()), mysql.TypeDatetime, 6)
du := types.Duration{Duration: time.Duration(12*time.Hour + 1*time.Minute + 1*time.Second), Fsp: types.DefaultFsp}
cases := []struct {
colName string
arg interface{}
isNil bool
asserts func(d types.Datum)
}{
{"test_int", 3, false, func(d types.Datum) {
c.Assert(d.GetInt64(), Equals, int64(3))
}},
{"test_float", 3.14159, false, func(d types.Datum) {
c.Assert(d.GetFloat64(), Equals, 3.14159)
}},
{"test_string", "TiDB", false, func(d types.Datum) {
c.Assert(d.GetString(), Equals, "TiDB")
}},
{"test_null", nil, true, func(d types.Datum) {
c.Assert(d.Kind(), Equals, types.KindNull)
}},
{"test_decimal", dec, false, func(d types.Datum) {
c.Assert(d.GetMysqlDecimal().String(), Equals, dec.String())
}},
{"test_time", tm, false, func(d types.Datum) {
c.Assert(d.GetMysqlTime().String(), Equals, tm.String())
}},
{"test_duration", du, false, func(d types.Datum) {
c.Assert(d.GetMysqlDuration().String(), Equals, du.String())
}},
}
for _, t := range cases {
f, err := newFunctionForTest(s.ctx, ast.NameConst, s.primitiveValsToConstants([]interface{}{t.colName, t.arg})...)
c.Assert(err, IsNil)
d, err := f.Eval(chunk.Row{})
c.Assert(err, IsNil)
t.asserts(d)
}
}