Update.

artie-labs · Dec 19, 2024 · d584a44 · d584a44
1 parent 97409a5
commit d584a44
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/lib/antlr/rename_table.go b/lib/antlr/rename_table.go
@@ -13,7 +13,12 @@ func processRenameTable(ctx *generated.RenameTableContext) ([]Event, error) {
 		case *generated.RenameTableClauseContext:
 			var allTableNames []string
 			for _, tableName := range castedChild.AllTableName() {
-				allTableNames = append(allTableNames, tableName.GetText())
+				parsedTableName, err := getTableNameFromNode(tableName)
+				if err != nil {
+					return nil, fmt.Errorf("failed to get table name: %w", err)
+				}
+
+				allTableNames = append(allTableNames, parsedTableName)
 			}
 
 			// Must be at least two table names

diff --git a/lib/antlr/rename_table_test.go b/lib/antlr/rename_table_test.go
@@ -20,15 +20,15 @@ func TestRenameTable(t *testing.T) {
 	}
 	{
 		// Another one table variant
-		events, err := Parse(`RENAME TABLE current_db.tbl_name TO other_db.tbl_name;`)
+		events, err := Parse(`RENAME TABLE current_db.tbl_name TO current_db.tbl_name;`)
 		assert.NoError(t, err)
 		assert.Len(t, events, 1)
 
 		renameTableEvent, isOk := events[0].(RenameTableEvent)
 		assert.True(t, isOk)
 
-		assert.Equal(t, "current_db.tbl_name", renameTableEvent.GetTable())
-		assert.Equal(t, "other_db.tbl_name", renameTableEvent.GetNewTableName())
+		assert.Equal(t, "tbl_name", renameTableEvent.GetTable())
+		assert.Equal(t, "tbl_name", renameTableEvent.GetNewTableName())
 	}
 	{
 		// Multiple tables