Function TestSetKindCase

common/ast/expr_test.go:27–107

(t *testing.T)

)

func TestSetKindCase(t *testing.T) {
	fac := ast.NewExprFactory()
	tests := []ast.Expr{
		fac.NewUnspecifiedExpr(1),
		fac.NewCall(1, "_==_", fac.NewLiteral(2, types.True), fac.NewLiteral(3, types.False)),
		fac.NewMemberCall(1, overloads.Size, fac.NewLiteral(2, types.String("hello"))),
		fac.NewComprehension(12,
			fac.NewList(1, []ast.Expr{}, []int32{}),
			"i",
			"__result__",
			fac.NewLiteral(5, types.False),
			fac.NewCall(8, "@not_strictly_false", fac.NewCall(7, "!_", fac.NewAccuIdent(6))),
			fac.NewCall(10, "_||_", fac.NewAccuIdent(9), fac.NewIdent(4, "i")),
			fac.NewAccuIdent(11),
		),
		fac.NewIdent(1, "a"),
		fac.NewLiteral(1, types.Bytes("hello")),
		fac.NewList(1, []ast.Expr{fac.NewIdent(2, "a"), fac.NewIdent(3, "b")}, []int32{}),
		fac.NewMap(1, []ast.EntryExpr{
			fac.NewMapEntry(2,
				fac.NewLiteral(3, types.String("string")),
				fac.NewCall(6, "_?._", fac.NewIdent(4, "a"), fac.NewLiteral(5, types.String("b"))),
				true),
		}),
		fac.NewSelect(2, fac.NewIdent(1, "a"), "b"),
		fac.NewStruct(1,
			"custom.StructType",
			[]ast.EntryExpr{
				fac.NewStructField(2,
					"uint_field",
					fac.NewLiteral(3, types.Uint(42)),
					false),
			}),
	}
	expr := nilTestExpr(t)
	for i, tst := range tests {
		t.Run(fmt.Sprintf("%d", i), func(t *testing.T) {
			expr.SetKindCase(tst)
			switch expr.Kind() {
			case ast.CallKind:
				if !reflect.DeepEqual(expr.AsCall(), tst.AsCall()) {
					t.Errorf("got %v, wanted %v", expr.AsCall(), tst.AsCall())
				}
			case ast.ComprehensionKind:
				if !reflect.DeepEqual(expr.AsComprehension(), tst.AsComprehension()) {
					t.Errorf("got %v, wanted %v", expr.AsComprehension(), tst.AsComprehension())
				}
			case ast.IdentKind:
				if !reflect.DeepEqual(expr.AsIdent(), tst.AsIdent()) {
					t.Errorf("got %v, wanted %v", expr.AsIdent(), tst.AsIdent())
				}
			case ast.LiteralKind:
				if !reflect.DeepEqual(expr.AsLiteral(), tst.AsLiteral()) {
					t.Errorf("got %v, wanted %v", expr.AsLiteral(), tst.AsLiteral())
				}
			case ast.ListKind:
				if !reflect.DeepEqual(expr.AsList(), tst.AsList()) {
					t.Errorf("got %v, wanted %v", expr.AsList(), tst.AsList())