cond_test.go

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

package sphinxql

import (
	"testing"

	"github.com/huandu/go-assert"
)

func TestCond(t *testing.T) {
	a := assert.New(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") },
		"(1 = 1 AND 2 = 2 AND 3 = 3)": func() string { return newTestCond().And("1 = 1", "2 = 2", "3 = 3") },
		"$0":                          func() string { return newTestCond().Var(123) },
	}

	for expected, f := range cases {
		actual := f()
		a.Equal(actual, expected)
	}
}

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