func (s *testVisitorSuite) TestBase(c *C) { val := expression.Value{Val: 1} visitor := expression.NewIdentEvalVisitor() var exp expression.Expression exp = &expression.Between{Expr: val, Left: val, Right: val} exp.Accept(visitor) exp = expression.NewBinaryOperation(opcode.And, val, val) exp.Accept(visitor) exp, _ = expression.NewCall("avg", []expression.Expression{val}, true) exp.Accept(visitor) rows := [][]interface{}{{1}} sq := newMockSubQuery(rows, []string{"a"}) exp = expression.NewCompareSubQuery(opcode.EQ, val, sq, true) exp.Accept(visitor) exp = &expression.Default{Name: "a"} exp.Accept(visitor) exp = expression.NewExistsSubQuery(sq) exp.Accept(visitor) when := &expression.WhenClause{Expr: val, Result: val} exp = &expression.FunctionCase{ WhenClauses: []*expression.WhenClause{when}, Value: val, ElseClause: val, } exp.Accept(visitor) exp = &expression.FunctionCast{Expr: val, Tp: types.NewFieldType(mysql.TypeLong), FunctionType: expression.ConvertFunction} exp.Accept(visitor) exp = &expression.FunctionConvert{Expr: val, Charset: "utf8"} exp.Accept(visitor) exp = &expression.FunctionSubstring{StrExpr: expression.Value{Val: "string"}, Pos: expression.Value{Val: 0}, Len: val} exp.Accept(visitor) exp = &expression.IsNull{Expr: val} exp.Accept(visitor) exp = &expression.IsTruth{Expr: val} exp.Accept(visitor) exp = &expression.ParamMarker{Expr: val} exp.Accept(visitor) exp = &expression.PatternIn{Expr: val, List: []expression.Expression{val}} exp.Accept(visitor) exp = &expression.PatternLike{Expr: val, Pattern: val} exp.Accept(visitor) exp = &expression.PatternRegexp{Expr: val, Pattern: val} exp.Accept(visitor) exp = &expression.PExpr{Expr: val} exp.Accept(visitor) exp = &expression.Position{Name: "a"} exp.Accept(visitor) exp = &expression.Row{Values: []expression.Expression{val}} exp.Accept(visitor) exp = &expression.UnaryOperation{V: val} exp.Accept(visitor) exp = &expression.Values{CIStr: model.NewCIStr("a")} exp.Accept(visitor) exp = &expression.Variable{Name: "a"} exp.Accept(visitor) }
func (s *testVisitorSuite) TestIdentEval(c *C) { exp := expression.NewBinaryOperation(opcode.Plus, &expression.Ident{CIStr: model.NewCIStr("a")}, expression.Value{Val: 1}) iev := expression.NewIdentEvalVisitor() iev.Set("a", 3) exp.Accept(iev) v, err := exp.Eval(nil, nil) c.Assert(err, IsNil) c.Assert(v, Equals, int64(4)) }