openbilibili/app/service/live/zeus/expr/expr_test.go

package expr

import (
	"fmt"
	"testing"
)

func TestExpr(t *testing.T) {
	defer func() {
		switch x := recover().(type) {
		case nil:
			// no panic
		case runtimePanic:
			fmt.Println(x)
			panic(x)
		default:
			// unexpected panic: resume state of panic.
			panic(x)
		}
	}()
	tests := []struct {
		expr string
		env  Env
		want string
	}{
		{`$x == "android"`, Env{"$x": "android"}, "true"},
		{"$x == `android`", Env{"$x": "android"}, "true"},
		{"to_lower($x) == `android`", Env{"$x": "AnDroId"}, "true"},
		{"to_lower($x) == `android`", Env{"$x": "iOS"}, "false"},
		{"log($1 + 9 * $1)", Env{"$1": 100}, "3"},
		{"$1 > 80 && $2 <9", Env{"$1": 100, "$2": 2}, "true"},
		{"pow(x, false) + pow(y, false)", Env{"x": 12, "y": 1}, "2"},
		{"pow(x, 3) + pow(y, 3)", Env{"x": 9, "y": 10}, "1729"},
		{"9 * (F - 32)/5", Env{"F": -40}, "-129"},
		{"-1 + -x", Env{"x": 1}, "-2"},
		{"-1 - x", Env{"x": 1}, "-2"},
		{"a >= 10", Env{"a": 15}, "true"},
		{"b >= sin(10) && a < 1", Env{"a": 9, "b": 10}, "false"},
		{"!!!true", Env{"a": 9, "b": 10}, "false"},
	}
	var prevExpr string
	parser := NewExpressionParser()
	for _, test := range tests {
		// Print expr only when it changes.
		if test.expr != prevExpr {
			t.Logf("\n%s\n", test.expr)
			prevExpr = test.expr
		}
		if err := parser.Parse(test.expr); err != nil {
			t.Error(err) // parse error
			continue
		}
		got := fmt.Sprintf("%v", parser.GetExpr().Eval(test.env))
		t.Logf("%s\t%v => %s\n", test.expr, test.env, got)
		if got != test.want {
			t.Errorf("%s.Eval() in %v = %q, want %q\n",
				test.expr, test.env, got, test.want)
		}
	}
}