Function RunTests

testing/generation/command_docs/output/framework_test.go:106–138

RunTests runs the given collection of QueryParses tests.

(t *testing.T, tests []QueryParses)

// RunTests runs the given collection of QueryParses tests.

func RunTests(t *testing.T, tests []QueryParses) {
	for _, test := range tests {
		t.Run(test.String(), func(t *testing.T) {
			statements, err := parser.Parse(test.String())
			if !test.ShouldParse() {
				if err == nil && len(statements) > 0 {
					t.Fatal("Query now parses, please upgrade the type to `Parses`")
				}
				return
			}
			require.NoError(t, err, "Regression, query previously parsed")
			require.Truef(t, len(statements) > 0, "Regression, query previously produced a Postgres AST")
			for _, statement := range statements {
				vitessAST, err := func() (vitessAST sqlparser.Statement, err error) {
					defer func() {
						if recoverVal := recover(); recoverVal != nil {
							vitessAST = nil
						}
					}()
					return ast.Convert(statement)
				}()
				if !test.ShouldConvert() {
					if err == nil && vitessAST != nil {
						t.Fatalf("Query %s now converts, please upgrade the type to `Converts`", test.String())
					}
					return
				}
				assert.NoError(t, err, "Regression, query previously converted from a Postgres AST to a Vitess AST")
				assert.NotNil(t, vitessAST, "Regression, query now returns a nil Vitess AST")
			}
		})
	}
}

// Regex which checks for the line beginning with Unimplemented, Parses, or Converts
var testStatementRegex = regexp.MustCompile(`^(\s*)(Unimplemented|Parses|Converts)\(`)