test: fix tests that check error

Signed-off-by: Andres Taylor <[email protected]>
vitessio · Dec 13, 2023 · da6efc9 · da6efc9
1 parent f8dcfa7
commit da6efc9
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/go/vt/vtgate/planbuilder/operators/queryprojection_test.go b/go/vt/vtgate/planbuilder/operators/queryprojection_test.go
@@ -86,7 +86,7 @@ func TestQP(t *testing.T) {
 			_, err = semantics.Analyze(sel, "", &semantics.FakeSI{})
 			require.NoError(t, err)
 
-			qp := createQPFromSelect(ctx, sel)
+			qp, err := getQPAndError(ctx, sel)
 			if tcase.expErr != "" {
 				require.Error(t, err)
 				require.Contains(t, err.Error(), tcase.expErr)
@@ -103,6 +103,12 @@ func TestQP(t *testing.T) {
 	}
 }
 
+func getQPAndError(ctx *plancontext.PlanningContext, sel *sqlparser.Select) (qp *QueryProjection, err error) {
+	defer PanicHandler(&err)
+	qp = createQPFromSelect(ctx, sel)
+	return
+}
+
 func TestQPSimplifiedExpr(t *testing.T) {
 	testCases := []struct {
 		query, expected string