github.com/whtcorpsinc/MilevaDB-Prod@v0.0.0-20211104133533-f57f4be3b597/dbs/memristed/memex/builtin_miscellaneous_test.go

// Copyright 2020 WHTCORPS INC, 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 memex

import (
	"math"
	"strings"
	"time"

	. "github.com/whtcorpsinc/check"
	"github.com/whtcorpsinc/BerolinaSQL/ast"
	"github.com/whtcorpsinc/BerolinaSQL/allegrosql"
	"github.com/whtcorpsinc/milevadb/types"
	"github.com/whtcorpsinc/milevadb/soliton/chunk"
	"github.com/whtcorpsinc/milevadb/soliton/solitonutil"
)

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.Event{})
		c.Assert(err, IsNil)
		c.Assert(d, solitonutil.CausetEquals, 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.NewStringCauset(test.ip)
		f, err := fc.getFunction(s.ctx, s.datumsToConstants([]types.Causet{ip}))
		c.Assert(err, IsNil)
		result, err := evalBuiltinFunc(f, chunk.Event{})
		c.Assert(err, IsNil)
		c.Assert(result, solitonutil.CausetEquals, types.NewCauset(test.expect))
	}
	// test NULL input for is_ipv4
	var argNull types.Causet
	f, _ := fc.getFunction(s.ctx, s.datumsToConstants([]types.Causet{argNull}))
	r, err := evalBuiltinFunc(f, chunk.Event{})
	c.Assert(err, IsNil)
	c.Assert(r, solitonutil.CausetEquals, types.NewCauset(0))
}

func (s *testEvaluatorSuite) TestUUID(c *C) {
	f, err := newFunctionForTest(s.ctx, ast.UUID)
	c.Assert(err, IsNil)
	d, err := f.Eval(chunk.Event{})
	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.MakeCausets(t.arg)))
		c.Assert(err, IsNil)
		r, err := evalBuiltinFunc(f, chunk.Event{})
		c.Assert(err, IsNil)
		c.Assert(r, solitonutil.CausetEquals, types.NewCauset(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.NewStringCauset(test.ip)
		f, err := fc.getFunction(s.ctx, s.datumsToConstants([]types.Causet{ip}))
		c.Assert(err, IsNil)
		result, err := evalBuiltinFunc(f, chunk.Event{})
		c.Assert(err, IsNil)
		c.Assert(result, solitonutil.CausetEquals, types.NewCauset(test.expect))
	}
	// test NULL input for is_ipv6
	var argNull types.Causet
	f, _ := fc.getFunction(s.ctx, s.datumsToConstants([]types.Causet{argNull}))
	r, err := evalBuiltinFunc(f, chunk.Event{})
	c.Assert(err, IsNil)
	c.Assert(r, solitonutil.CausetEquals, types.NewCauset(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.NewCauset(test.ip)
		f, err := fc.getFunction(s.ctx, s.datumsToConstants([]types.Causet{ip}))
		c.Assert(err, IsNil)
		result, err := evalBuiltinFunc(f, chunk.Event{})
		c.Assert(err, IsNil)
		c.Assert(result, solitonutil.CausetEquals, types.NewCauset(test.expect))
	}

	var argNull types.Causet
	f, _ := fc.getFunction(s.ctx, s.datumsToConstants([]types.Causet{argNull}))
	r, err := evalBuiltinFunc(f, chunk.Event{})
	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.NewCauset(test.ip)
		f, err := fc.getFunction(s.ctx, s.datumsToConstants([]types.Causet{ip}))
		c.Assert(err, IsNil)
		result, err := evalBuiltinFunc(f, chunk.Event{})
		c.Assert(err, IsNil)
		c.Assert(result, solitonutil.CausetEquals, types.NewCauset(test.expect))
	}

	var argNull types.Causet
	f, _ := fc.getFunction(s.ctx, s.datumsToConstants([]types.Causet{argNull}))
	r, err := evalBuiltinFunc(f, chunk.Event{})
	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.NewCauset(test.ip)
		f, err := fc.getFunction(s.ctx, s.datumsToConstants([]types.Causet{ip}))
		c.Assert(err, IsNil)
		result, err := evalBuiltinFunc(f, chunk.Event{})
		c.Assert(err, IsNil)
		c.Assert(result, solitonutil.CausetEquals, types.NewCauset(test.expect))
	}

	var argNull types.Causet
	f, _ := fc.getFunction(s.ctx, s.datumsToConstants([]types.Causet{argNull}))
	r, err := evalBuiltinFunc(f, chunk.Event{})
	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.NewCauset(test.ip)
		f, err := fc.getFunction(s.ctx, s.datumsToConstants([]types.Causet{ip}))
		c.Assert(err, IsNil)
		result, err := evalBuiltinFunc(f, chunk.Event{})
		c.Assert(err, IsNil)
		c.Assert(result, solitonutil.CausetEquals, types.NewCauset(test.expect))
	}

	var argNull types.Causet
	f, _ := fc.getFunction(s.ctx, s.datumsToConstants([]types.Causet{argNull}))
	r, err := evalBuiltinFunc(f, chunk.Event{})
	c.Assert(err, IsNil)
	c.Assert(r, solitonutil.CausetEquals, types.NewCauset(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.NewCauset(test.ip)
		f, err := fc.getFunction(s.ctx, s.datumsToConstants([]types.Causet{ip}))
		c.Assert(err, IsNil)
		result, err := evalBuiltinFunc(f, chunk.Event{})
		c.Assert(err, IsNil)
		c.Assert(result, solitonutil.CausetEquals, types.NewCauset(test.expect))
	}

	var argNull types.Causet
	f, _ := fc.getFunction(s.ctx, s.datumsToConstants([]types.Causet{argNull}))
	r, err := evalBuiltinFunc(f, chunk.Event{})
	c.Assert(err, IsNil)
	c.Assert(r, solitonutil.CausetEquals, types.NewCauset(0))
}

func (s *testEvaluatorSuite) TestNameConst(c *C) {
	dec := types.NewDecFromFloatForTest(123.123)
	tm := types.NewTime(types.FromGoTime(time.Now()), allegrosql.TypeDatetime, 6)
	du := types.Duration{Duration: 12*time.Hour + 1*time.Minute + 1*time.Second, Fsp: types.DefaultFsp}
	cases := []struct {
		defCausName string
		arg     interface{}
		isNil   bool
		asserts func(d types.Causet)
	}{
		{"test_int", 3, false, func(d types.Causet) {
			c.Assert(d.GetInt64(), Equals, int64(3))
		}},
		{"test_float", 3.14159, false, func(d types.Causet) {
			c.Assert(d.GetFloat64(), Equals, 3.14159)
		}},
		{"test_string", "MilevaDB", false, func(d types.Causet) {
			c.Assert(d.GetString(), Equals, "MilevaDB")
		}},
		{"test_null", nil, true, func(d types.Causet) {
			c.Assert(d.HoTT(), Equals, types.HoTTNull)
		}},
		{"test_decimal", dec, false, func(d types.Causet) {
			c.Assert(d.GetMysqlDecimal().String(), Equals, dec.String())
		}},
		{"test_time", tm, false, func(d types.Causet) {
			c.Assert(d.GetMysqlTime().String(), Equals, tm.String())
		}},
		{"test_duration", du, false, func(d types.Causet) {
			c.Assert(d.GetMysqlDuration().String(), Equals, du.String())
		}},
	}

	for _, t := range cases {
		f, err := newFunctionForTest(s.ctx, ast.NameConst, s.primitiveValsToConstants([]interface{}{t.defCausName, t.arg})...)
		c.Assert(err, IsNil)
		d, err := f.Eval(chunk.Event{})
		c.Assert(err, IsNil)
		t.asserts(d)
	}
}