add test case to cover quoted identifier

DataDog · Oct 10, 2023 · 38637fa · 38637fa
1 parent 046f310
commit 38637fa
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/obfuscate_and_normalize_test.go b/obfuscate_and_normalize_test.go
@@ -156,6 +156,31 @@ multiline comment */
 				Size:     12,
 			},
 		},
+		{
+			// double quoted table name
+			input:    `SELECT * FROM "public"."users" WHERE id = 1`,
+			expected: `SELECT * FROM "public"."users" WHERE id = ?`,
+			statementMetadata: StatementMetadata{
+				Tables:   []string{`"public"."users"`},
+				Comments: []string{},
+				Commands: []string{"SELECT"},
+				Size:     22,
+			},
+		},
+		{
+			// [] quoted table name
+			input:    `SELECT * FROM [public].[users] WHERE id = 1`,
+			expected: `SELECT * FROM [public].[users] WHERE id = ?`,
+			statementMetadata: StatementMetadata{
+				Tables:   []string{"[public].[users]"},
+				Comments: []string{},
+				Commands: []string{"SELECT"},
+				Size:     22,
+			},
+			lexerOpts: []lexerOption{
+				WithDBMS(DBMSSQLServer),
+			},
+		},
 	}
 
 	obfuscator := NewObfuscator(