test: add test asserting I_S queries are correct

Signed-off-by: Andres Taylor <andres@planetscale.com>
vitessio · Jun 4, 2024 · 69bb0f1 · 69bb0f1
1 parent b391a63
commit 69bb0f1
Show file tree

Hide file tree

Showing 2 changed files with 105 additions and 0 deletions.
diff --git a/go/vt/vtgate/planbuilder/testdata/expected/onecase.json b/go/vt/vtgate/planbuilder/testdata/expected/onecase.json
@@ -0,0 +1,85 @@
+[
+  {
+    "comment": "Add your test case here for debugging and run go test -run=One.",
+    "query": "select (select count(*) from user_extra) x from user order by x",
+    "plan": {
+      "QueryType": "SELECT",
+      "Original": "select (select count(*) from user_extra) x from user order by x",
+      "Instructions": {
+        "OperatorType": "UncorrelatedSubquery",
+        "Variant": "PulloutValue",
+        "PulloutVars": [
+          "__sq1"
+        ],
+        "Inputs": [
+          {
+            "InputName": "SubQuery",
+            "OperatorType": "Aggregate",
+            "Variant": "Scalar",
+            "Aggregates": "sum_count_star(0) AS count(*)",
+            "Inputs": [
+              {
+                "OperatorType": "Route",
+                "Variant": "Scatter",
+                "Keyspace": {
+                  "Name": "user",
+                  "Sharded": true
+                },
+                "FieldQuery": "select count(*) from user_extra where 1 != 1",
+                "Query": "select count(*) from user_extra",
+                "Table": "user_extra"
+              }
+            ]
+          },
+          {
+            "InputName": "Outer",
+            "OperatorType": "UncorrelatedSubquery",
+            "Variant": "PulloutValue",
+            "PulloutVars": [
+              "__sq1"
+            ],
+            "Inputs": [
+              {
+                "InputName": "SubQuery",
+                "OperatorType": "Aggregate",
+                "Variant": "Scalar",
+                "Aggregates": "sum_count_star(0) AS count(*)",
+                "Inputs": [
+                  {
+                    "OperatorType": "Route",
+                    "Variant": "Scatter",
+                    "Keyspace": {
+                      "Name": "user",
+                      "Sharded": true
+                    },
+                    "FieldQuery": "select count(*) from user_extra where 1 != 1",
+                    "Query": "select count(*) from user_extra",
+                    "Table": "user_extra"
+                  }
+                ]
+              },
+              {
+                "InputName": "Outer",
+                "OperatorType": "Route",
+                "Variant": "Scatter",
+                "Keyspace": {
+                  "Name": "user",
+                  "Sharded": true
+                },
+                "FieldQuery": "select :__sq1 as x, __sq1, weight_string(__sq1) from `user` where 1 != 1",
+                "OrderBy": "(1|2) ASC",
+                "Query": "select :__sq1 as x, __sq1, weight_string(__sq1) from `user` order by __sq1 asc",
+                "ResultColumns": 1,
+                "Table": "`user`"
+              }
+            ]
+          }
+        ]
+      },
+      "TablesUsed": [
+        "user.user",
+        "user.user_extra"
+      ]
+    }
+  }
+]
diff --git a/go/vt/vtgate/planbuilder/testdata/filter_cases.json b/go/vt/vtgate/planbuilder/testdata/filter_cases.json
@@ -884,6 +884,26 @@
       ]
     }
   },
+  {
+    "comment": "Check that we don't remove information_schema qualifier",
+    "query": "select 1 from information_schema.tables where exists(select 1 from information_schema.tables where table_name = 'tables')",
+    "plan": {
+      "QueryType": "SELECT",
+      "Original": "select 1 from information_schema.tables where exists(select 1 from information_schema.tables where table_name = 'tables')",
+      "Instructions": {
+        "OperatorType": "Route",
+        "Variant": "DBA",
+        "Keyspace": {
+          "Name": "main",
+          "Sharded": false
+        },
+        "FieldQuery": "select 1 from information_schema.`tables` where 1 != 1",
+        "Query": "select 1 from information_schema.`tables` where exists (select 1 from information_schema.`tables` where table_name = :table_name /* VARCHAR */ limit 1)",
+        "SysTableTableName": "[table_name:VARCHAR(\"tables\")]",
+        "Table": "information_schema.`tables`"
+      }
+    }},
+
   {
     "comment": "Composite IN: tuple inside tuple",
     "query": "select id from user where ((col1, name), col2) in ((('aa', 'bb'), 'cc'), (('dd', 'ee'), 'ff'))",