Add another test for CTE

pg-sharding · Apr 25, 2024 · a25f188 · a25f188
1 parent f19a1ac
commit a25f188
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/router/qrouter/proxy_routing.go b/router/qrouter/proxy_routing.go
@@ -105,7 +105,7 @@ func (meta *RoutingMetadataContext) RecordConstExpr(resolvedRelation RelationFQN
 	meta.rels[resolvedRelation] = struct{}{}
 	if _, ok := meta.exprs[resolvedRelation]; !ok {
 		meta.exprs[resolvedRelation] = map[string]string{}
-	}
+	} // TODO: else branch
 	delete(meta.unparsed_columns, colname)
 	meta.exprs[resolvedRelation][colname] = expr
 }

diff --git a/router/qrouter/proxy_routing_test.go b/router/qrouter/proxy_routing_test.go
@@ -320,6 +320,21 @@ func TestCTE(t *testing.T) {
 				TargetSessionAttrs: "any",
 			},
 		},
+		{
+			query: `
+			WITH xxxx AS (
+			SELECT * from t where i = 1
+		),
+			zzzz AS (
+				UPDATE t
+				SET a = 0
+				WHERE i = 12
+			)
+			SELECT * FROM xxxx;
+			`,
+			err: nil,
+			exp: routingstate.MultiMatchState{},
+		},
 	} {
 		parserRes, err := lyx.Parse(tt.query)