openbilibili/app/service/main/account-recovery/dao/sqlbuilder/cond_test.go

// Copyright 2018 Huan Du. All rights reserved.
// Licensed under the MIT license that can be found in the LICENSE file.

package sqlbuilder

import (
	"testing"
)

func TestCond(t *testing.T) {
	cases := map[string]func() string{
		"$$a = $0":                  func() string { return newTestCond().Equal("$a", 123) },
		"$$b = $0":                  func() string { return newTestCond().E("$b", 123) },
		"$$a <> $0":                 func() string { return newTestCond().NotEqual("$a", 123) },
		"$$b <> $0":                 func() string { return newTestCond().NE("$b", 123) },
		"$$a > $0":                  func() string { return newTestCond().GreaterThan("$a", 123) },
		"$$b > $0":                  func() string { return newTestCond().G("$b", 123) },
		"$$a >= $0":                 func() string { return newTestCond().GreaterEqualThan("$a", 123) },
		"$$b >= $0":                 func() string { return newTestCond().GE("$b", 123) },
		"$$a < $0":                  func() string { return newTestCond().LessThan("$a", 123) },
		"$$b < $0":                  func() string { return newTestCond().L("$b", 123) },
		"$$a <= $0":                 func() string { return newTestCond().LessEqualThan("$a", 123) },
		"$$b <= $0":                 func() string { return newTestCond().LE("$b", 123) },
		"$$a IN ($0, $1, $2)":       func() string { return newTestCond().In("$a", 1, 2, 3) },
		"$$a NOT IN ($0, $1, $2)":   func() string { return newTestCond().NotIn("$a", 1, 2, 3) },
		"$$a LIKE $0":               func() string { return newTestCond().Like("$a", "%Huan%") },
		"$$a NOT LIKE $0":           func() string { return newTestCond().NotLike("$a", "%Huan%") },
		"$$a IS NULL":               func() string { return newTestCond().IsNull("$a") },
		"$$a IS NOT NULL":           func() string { return newTestCond().IsNotNull("$a") },
		"$$a BETWEEN $0 AND $1":     func() string { return newTestCond().Between("$a", 123, 456) },
		"$$a NOT BETWEEN $0 AND $1": func() string { return newTestCond().NotBetween("$a", 123, 456) },
		"(1 = 1 OR 2 = 2 OR 3 = 3)": func() string { return newTestCond().Or("1 = 1", "2 = 2", "3 = 3") },
		"$0": func() string { return newTestCond().Var(123) },
	}

	for expected, f := range cases {
		if actual := f(); expected != actual {
			t.Fatalf("invalid result. [expected:%v] [actual:%v]", expected, actual)
		}
	}
}

func newTestCond() *Cond {
	return &Cond{
		Args: &Args{},
	}
}