From 1400fea144a1542314e7c1dee4ad1d7798f63ee2 Mon Sep 17 00:00:00 2001
From: c-r-dev <crangavajha1@bloomberg.net>
Date: Tue, 17 Dec 2024 15:57:41 -0500
Subject: [PATCH] added sample test data for filter_cases e2e testing

Signed-off-by: c-r-dev <crangavajha1@bloomberg.net>
---
 .../endtoend/vtgate/plan_tests/main_test.go   | 26 +++++++++++++++++++
 .../vtgate/plan_tests/plan_e2e_test.go        |  1 +
 .../planbuilder/testdata/filter_cases.json    |  3 +--
 .../planbuilder/testdata/sampledata/user.sql  | 14 ++++++++++
 4 files changed, 42 insertions(+), 2 deletions(-)
 create mode 100644 go/vt/vtgate/planbuilder/testdata/sampledata/user.sql

diff --git a/go/test/endtoend/vtgate/plan_tests/main_test.go b/go/test/endtoend/vtgate/plan_tests/main_test.go
index d3915af0c8d..32bacdf3c85 100644
--- a/go/test/endtoend/vtgate/plan_tests/main_test.go
+++ b/go/test/endtoend/vtgate/plan_tests/main_test.go
@@ -27,6 +27,7 @@ import (
 	"vitess.io/vitess/go/mysql"
 	"vitess.io/vitess/go/test/endtoend/cluster"
 	"vitess.io/vitess/go/test/endtoend/utils"
+	"vitess.io/vitess/go/vt/sqlparser"
 	"vitess.io/vitess/go/vt/vtgate/engine"
 	"vitess.io/vitess/go/vt/vtgate/planbuilder"
 )
@@ -128,6 +129,31 @@ func start(t *testing.T) (utils.MySQLCompare, func()) {
 	}
 }
 
+// parseSQL statements - querySQL may be a multi-line sql blob
+func parseSQL(querySQL ...string) ([]string, error) {
+	parser := sqlparser.NewTestParser()
+	var sqls []string
+	for _, sql := range querySQL {
+		split, err := parser.SplitStatementToPieces(sql)
+		if err != nil {
+			return nil, err
+		}
+		sqls = append(sqls, split...)
+	}
+	return sqls, nil
+}
+
+func loadSampleData(t *testing.T, mcmp utils.MySQLCompare) {
+	sampleDataSQL := readFile("sampledata/user.sql")
+	insertSQL, err := parseSQL(sampleDataSQL)
+	if err != nil {
+		require.NoError(t, err)
+	}
+	for _, sql := range insertSQL {
+		mcmp.ExecNoCompare(sql)
+	}
+}
+
 func readJSONTests(filename string) []planbuilder.PlanTest {
 	var output []planbuilder.PlanTest
 	file, err := os.Open(locateFile(filename))
diff --git a/go/test/endtoend/vtgate/plan_tests/plan_e2e_test.go b/go/test/endtoend/vtgate/plan_tests/plan_e2e_test.go
index 9c110df020a..4e029582d2f 100644
--- a/go/test/endtoend/vtgate/plan_tests/plan_e2e_test.go
+++ b/go/test/endtoend/vtgate/plan_tests/plan_e2e_test.go
@@ -44,6 +44,7 @@ func TestSelectCases(t *testing.T) {
 func TestFilterCases(t *testing.T) {
 	mcmp, closer := start(t)
 	defer closer()
+	loadSampleData(t, mcmp)
 	tests := readJSONTests("filter_cases.json")
 	for _, test := range tests {
 		mcmp.Run(test.Comment, func(mcmp *utils.MySQLCompare) {
diff --git a/go/vt/vtgate/planbuilder/testdata/filter_cases.json b/go/vt/vtgate/planbuilder/testdata/filter_cases.json
index 4bf7a969a50..72b6c4ddd46 100644
--- a/go/vt/vtgate/planbuilder/testdata/filter_cases.json
+++ b/go/vt/vtgate/planbuilder/testdata/filter_cases.json
@@ -5109,7 +5109,6 @@
       "user.sales",
       "user.sales_extra"
     ]
-  },
-  "skip_e2e":true
+  }
 }
 ]
diff --git a/go/vt/vtgate/planbuilder/testdata/sampledata/user.sql b/go/vt/vtgate/planbuilder/testdata/sampledata/user.sql
new file mode 100644
index 00000000000..044a1ee140d
--- /dev/null
+++ b/go/vt/vtgate/planbuilder/testdata/sampledata/user.sql
@@ -0,0 +1,14 @@
+INSERT INTO sales (oid, col1)
+ VALUES (1, 'a_1');
+
+INSERT INTO sales_extra(colx, cola, colb, start, end)
+VALUES (11, 'a_1', 'b_1',0, 500);
+
+INSERT INTO sales_extra(colx, cola, colb, start, end)
+VALUES (12, 'a_2', 'b_2',500, 1000);
+
+INSERT INTO sales_extra(colx, cola, colb, start, end)
+VALUES (13, 'a_3', 'b_3',1000, 1500);
+
+INSERT INTO sales_extra(colx, cola, colb, start, end)
+VALUES (14, 'a_4', 'b_4',1500, 2000);
\ No newline at end of file