Additional unit tests to address open issues in github

golang-migrate · Feb 5, 2025 · 5973c68 · 5973c68
1 parent e53b7ac
commit 5973c68
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/database/spanner/spanner_test.go b/database/spanner/spanner_test.go
@@ -313,6 +313,34 @@ func Test_statementGroups(t *testing.T) {
 				},
 			},
 		},
+		{
+			name: "ALTER with DEFAULT",
+			// From https://github.com/golang-migrate/migrate/issues/918
+			multiStatement: "ALTER TABLE t1 ADD COLUMN c1 STRING(MAX) DEFAULT ('');ALTER TABLE t1 ADD COLUMN c1 STRING(MAX) NOT NULL DEFAULT ('');",
+			expected: []*statementGroup{
+				{
+					typ: statementTypeDDL,
+					stmts: []string{
+						"ALTER TABLE t1 ADD COLUMN c1 STRING(MAX) DEFAULT ('')",
+						"ALTER TABLE t1 ADD COLUMN c1 STRING(MAX) NOT NULL DEFAULT ('')",
+					},
+				},
+			},
+		},
+		{
+			name: "Change Streams",
+			multiStatement: `CREATE CHANGE STREAM NamesAndAlbums
+FOR Singers(FirstName, LastName), Albums
+OPTIONS ( retention_period = '36h' );`,
+			expected: []*statementGroup{
+				{
+					typ: statementTypeDDL,
+					stmts: []string{
+						"CREATE CHANGE STREAM NamesAndAlbums\nFOR Singers(FirstName, LastName), Albums\nOPTIONS ( retention_period = '36h' )",
+					},
+				},
+			},
+		},
 	}
 
 	for _, tc := range testCases {