dump_test.go

package main

import "testing"

func TestParseTableNameFromDDL(t *testing.T) {
	tests := []struct {
		name string
		ddl  string
		want string
	}{
		{
			name: "create table",
			ddl: `CREATE TABLE table_name_1 (
  column1 STRING(32) NOT NULL,
  column2 TIMESTAMP NOT NULL OPTIONS (
    allow_commit_timestamp = true
  ),
) PRIMARY KEY(column1);`,
			want: "table_name_1",
		},
		{
			name: "create table, table enclosed by backtick (`)",
			ddl: `CREATE TABLE ` + "`table_name_1`" + ` (
  column1 STRING(32) NOT NULL,
  column2 TIMESTAMP NOT NULL OPTIONS (
    allow_commit_timestamp = true
  ),
) PRIMARY KEY(column1);`,
			want: "table_name_1",
		},
		{
			name: "create table, include multiple spaces",
			ddl: `   CREATE   TABLE    table_name_1     (
  column1 STRING(32) NOT NULL,
  column2 TIMESTAMP NOT NULL OPTIONS (
    allow_commit_timestamp = true
  ),
) PRIMARY KEY(column1);`,
			want: "table_name_1",
		},
		{
			name: "create unique index",
			ddl:  `CREATE UNIQUE INDEX table_name_1_column2_a ON table_name_1(column2);`,
			want: "table_name_1",
		},
		{
			name: "create index",
			ddl:  `CREATE INDEX table_name_1_column2_a ON table_name_1(column2);`,
			want: "table_name_1",
		},
		{
			name: "create index, index name enclosed by backtick (`)",
			ddl:  "CREATE INDEX `order` ON TABLE(`by`)",
			want: "TABLE",
		},
		{
			name: "create index, table name enclosed by backtick (`)",
			ddl:  "CREATE INDEX `order` ON `TABLE`(`by`)",
			want: "TABLE",
		},
		{
			name: "create index, include multiple spaces",
			ddl:  "  CREATE   INDEX    `order`   ON    TABLE(`by`)",
			want: "TABLE",
		},
		{
			name: "alter table",
			ddl:  "ALTER TABLE t5 ADD FOREIGN KEY(T6Id) REFERENCES t6(Id);",
			want: "t5",
		},
		{
			name: "alter table, table name enclosed by backtick (`)",
			ddl:  "ALTER TABLE `t5` ADD FOREIGN KEY(T6Id) REFERENCES t6(Id);",
			want: "t5",
		},
		{
			name: "alter table, include multiple spaces",
			ddl:  "  ALTER  TABLE \r\n `t5`   ADD   FOREIGN   KEY(T6Id) REFERENCES t6(Id);",
			want: "t5",
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if got := parseTableNameFromDDL(tt.ddl); got != tt.want {
				t.Errorf("parseTableNameFromDDL() = %v, want %v", got, tt.want)
			}
		})
	}
}