From 237cdfc544f8674c238a76fe38ebc90a412f80a3 Mon Sep 17 00:00:00 2001
From: Rohit Nayak <rohit@planetscale.com>
Date: Thu, 25 Apr 2024 12:19:05 +0200
Subject: [PATCH] Add test cases for insert/update/delete rerouting for
 reference tables

Signed-off-by: Rohit Nayak <rohit@planetscale.com>
---
 .../vtgate/planbuilder/testdata/onecase.json  | 67 +++++++++++++++++--
 1 file changed, 63 insertions(+), 4 deletions(-)

diff --git a/go/vt/vtgate/planbuilder/testdata/onecase.json b/go/vt/vtgate/planbuilder/testdata/onecase.json
index da7543f706a..59e8391678d 100644
--- a/go/vt/vtgate/planbuilder/testdata/onecase.json
+++ b/go/vt/vtgate/planbuilder/testdata/onecase.json
@@ -1,9 +1,68 @@
 [
   {
-    "comment": "Add your test case here for debugging and run go test -run=One.",
-    "query": "",
+    "comment": "delete from reference table - query send to source table",
+    "query": "delete from user.ref_with_source where col = 1",
     "plan": {
-
+      "QueryType": "DELETE",
+      "Original": "delete from user.ref_with_source where col = 1",
+      "Instructions": {
+        "OperatorType": "Delete",
+        "Variant": "Unsharded",
+        "Keyspace": {
+          "Name": "main",
+          "Sharded": false
+        },
+        "TargetTabletType": "PRIMARY",
+        "Query": "delete from source_of_ref where col = 1",
+        "Table": "source_of_ref"
+      },
+      "TablesUsed": [
+        "main.source_of_ref"
+      ]
+    }
+  },
+  {
+    "comment": "update from reference table - query send to source table",
+    "query": "update user.ref_with_source set x = 4 where col = 1",
+    "plan": {
+      "QueryType": "UPDATE",
+      "Original": "update user.ref_with_source set x = 4 where col = 1",
+      "Instructions": {
+        "OperatorType": "Update",
+        "Variant": "Unsharded",
+        "Keyspace": {
+          "Name": "main",
+          "Sharded": false
+        },
+        "TargetTabletType": "PRIMARY",
+        "Query": "update source_of_ref set x = 4 where col = 1",
+        "Table": "source_of_ref"
+      },
+      "TablesUsed": [
+        "main.source_of_ref"
+      ]
+    }
+  },
+  {
+    "comment": "insert from reference table - query send to source table",
+    "query": "insert into user.ref_with_source(x) values(4)",
+    "plan": {
+      "QueryType": "INSERT",
+      "Original": "insert into user.ref_with_source(x) values(4)",
+      "Instructions": {
+        "OperatorType": "Insert",
+        "Variant": "Unsharded",
+        "Keyspace": {
+          "Name": "main",
+          "Sharded": false
+        },
+        "TargetTabletType": "PRIMARY",
+        "Query": "insert into source_of_ref(x) values (4)",
+        "TableName": "source_of_ref"
+      },
+      "TablesUsed": [
+        "main.source_of_ref"
+      ]
     }
   }
-]
\ No newline at end of file
+]