Use assertion function

xataio · Nov 26, 2024 · 31ce387 · 31ce387
1 parent 3ae8de8
commit 31ce387
Show file tree

Hide file tree

Showing 9 changed files with 54 additions and 246 deletions.
diff --git a/pkg/migrations/op_change_type_test.go b/pkg/migrations/op_change_type_test.go
@@ -101,24 +101,14 @@ func TestChangeColumnType(t *testing.T) {
 				}, rows)
 			},
 			afterRollback: func(t *testing.T, db *sql.DB, schema string) {
-				// The new (temporary) `rating` column should not exist on the underlying table.
-				ColumnMustNotExist(t, db, schema, "reviews", migrations.TemporaryName("rating"))
-
-				// The up function no longer exists.
-				FunctionMustNotExist(t, db, schema, migrations.TriggerFunctionName("reviews", "rating"))
-				// The down function no longer exists.
-				FunctionMustNotExist(t, db, schema, migrations.TriggerFunctionName("reviews", migrations.TemporaryName("rating")))
-
-				// The up trigger no longer exists.
-				TriggerMustNotExist(t, db, schema, "reviews", migrations.TriggerName("reviews", "rating"))
-				// The down trigger no longer exists.
-				TriggerMustNotExist(t, db, schema, "reviews", migrations.TriggerName("reviews", migrations.TemporaryName("rating")))
+				// The table is cleaned up; temporary columns, trigger functions and triggers no longer exist.
+				TableMustBeCleanedUp(t, db, schema, "reviews", "rating")
 			},
 			afterComplete: func(t *testing.T, db *sql.DB, schema string) {
 				newVersionSchema := roll.VersionedSchemaName(schema, "02_change_type")
 
-				// The new (temporary) `rating` column should not exist on the underlying table.
-				ColumnMustNotExist(t, db, schema, "reviews", migrations.TemporaryName("rating"))
+				// The table is cleaned up; temporary columns, trigger functions and triggers no longer exist.
+				TableMustBeCleanedUp(t, db, schema, "reviews", "rating")
 
 				// The `rating` column in the new view must have the correct type.
 				ColumnMustHaveType(t, db, newVersionSchema, "reviews", "rating", "integer")

diff --git a/pkg/migrations/op_create_constraint_test.go b/pkg/migrations/op_create_constraint_test.go
@@ -84,11 +84,11 @@ func TestCreateConstraint(t *testing.T) {
 				IndexMustNotExist(t, db, schema, "users", "uniue_name")
 
 				// Functions, triggers and temporary columns are dropped.
-				tableCleanedUp(t, db, schema, "users", "name")
+				TableMustBeCleanedUp(t, db, schema, "users", "name")
 			},
 			afterComplete: func(t *testing.T, db *sql.DB, schema string) {
 				// Functions, triggers and temporary columns are dropped.
-				tableCleanedUp(t, db, schema, "users", "name")
+				TableMustBeCleanedUp(t, db, schema, "users", "name")
 
 				// Inserting values into the new schema that violate uniqueness should fail.
 				MustInsert(t, db, schema, "02_create_constraint", "users", map[string]string{
@@ -161,11 +161,11 @@ func TestCreateConstraint(t *testing.T) {
 			},
 			afterRollback: func(t *testing.T, db *sql.DB, schema string) {
 				// Functions, triggers and temporary columns are dropped.
-				tableCleanedUp(t, db, schema, "users", "name")
+				TableMustBeCleanedUp(t, db, schema, "users", "name")
 			},
 			afterComplete: func(t *testing.T, db *sql.DB, schema string) {
 				// Functions, triggers and temporary columns are dropped.
-				tableCleanedUp(t, db, schema, "users", "name")
+				TableMustBeCleanedUp(t, db, schema, "users", "name")
 
 				// Inserting values into the new schema that violate the check constraint should fail.
 				MustNotInsert(t, db, schema, "02_create_constraint", "users", map[string]string{
@@ -250,8 +250,8 @@ func TestCreateConstraint(t *testing.T) {
 				IndexMustNotExist(t, db, schema, "users", "unique_name_email")
 
 				// Functions, triggers and temporary columns are dropped.
-				tableCleanedUp(t, db, schema, "users", "name")
-				tableCleanedUp(t, db, schema, "users", "email")
+				TableMustBeCleanedUp(t, db, schema, "users", "name")
+				TableMustBeCleanedUp(t, db, schema, "users", "email")
 			},
 			afterComplete: func(t *testing.T, db *sql.DB, schema string) {
 				// Complete is a no-op.
@@ -334,13 +334,13 @@ func TestCreateConstraint(t *testing.T) {
 				// The check constraint must not exists on the table.
 				CheckConstraintMustNotExist(t, db, schema, "users", "check_name_email")
 				// Functions, triggers and temporary columns are dropped.
-				tableCleanedUp(t, db, schema, "users", "name")
-				tableCleanedUp(t, db, schema, "users", "email")
+				TableMustBeCleanedUp(t, db, schema, "users", "name")
+				TableMustBeCleanedUp(t, db, schema, "users", "email")
 			},
 			afterComplete: func(t *testing.T, db *sql.DB, schema string) {
 				// Functions, triggers and temporary columns are dropped.
-				tableCleanedUp(t, db, schema, "users", "name")
-				tableCleanedUp(t, db, schema, "users", "email")
+				TableMustBeCleanedUp(t, db, schema, "users", "name")
+				TableMustBeCleanedUp(t, db, schema, "users", "email")
 
 				// Inserting values into the new schema that the violate the check constraint must fail.
 				MustNotInsert(t, db, schema, "02_create_constraint", "users", map[string]string{
@@ -474,13 +474,13 @@ func TestCreateConstraint(t *testing.T) {
 				// The check constraint must not exists on the table.
 				CheckConstraintMustNotExist(t, db, schema, "reports", "fk_users")
 				// Functions, triggers and temporary columns are dropped.
-				tableCleanedUp(t, db, schema, "reports", "users_id")
-				tableCleanedUp(t, db, schema, "reports", "users_zip")
+				TableMustBeCleanedUp(t, db, schema, "reports", "users_id")
+				TableMustBeCleanedUp(t, db, schema, "reports", "users_zip")
 			},
 			afterComplete: func(t *testing.T, db *sql.DB, schema string) {
 				// Functions, triggers and temporary columns are dropped.
-				tableCleanedUp(t, db, schema, "reports", "users_id")
-				tableCleanedUp(t, db, schema, "reports", "users_zip")
+				TableMustBeCleanedUp(t, db, schema, "reports", "users_id")
+				TableMustBeCleanedUp(t, db, schema, "reports", "users_zip")
 
 				// Inserting values into the new schema that the violate the check constraint must fail.
 				MustNotInsert(t, db, schema, "02_create_constraint", "reports", map[string]string{

diff --git a/pkg/migrations/op_drop_constraint_test.go b/pkg/migrations/op_drop_constraint_test.go
@@ -106,18 +106,8 @@ func TestDropConstraint(t *testing.T) {
 				}, rows)
 			},
 			afterRollback: func(t *testing.T, db *sql.DB, schema string) {
-				// The new (temporary) `title` column should not exist on the underlying table.
-				ColumnMustNotExist(t, db, schema, "posts", migrations.TemporaryName("title"))
-
-				// The up function no longer exists.
-				FunctionMustNotExist(t, db, schema, migrations.TriggerFunctionName("posts", "title"))
-				// The down function no longer exists.
-				FunctionMustNotExist(t, db, schema, migrations.TriggerFunctionName("posts", migrations.TemporaryName("title")))
-
-				// The up trigger no longer exists.
-				TriggerMustNotExist(t, db, schema, "posts", migrations.TriggerName("posts", "title"))
-				// The down trigger no longer exists.
-				TriggerMustNotExist(t, db, schema, "posts", migrations.TriggerName("posts", migrations.TemporaryName("title")))
+				// The table is cleaned up; temporary columns, trigger functions and triggers no longer exist.
+				TableMustBeCleanedUp(t, db, schema, "posts", "title")
 			},
 			afterComplete: func(t *testing.T, db *sql.DB, schema string) {
 				// Inserting a row that does not meet the check constraint into the new view works.
@@ -134,15 +124,8 @@ func TestDropConstraint(t *testing.T) {
 					{"id": 5, "title": "e"},
 				}, rows)
 
-				// The up function no longer exists.
-				FunctionMustNotExist(t, db, schema, migrations.TriggerFunctionName("posts", "title"))
-				// The down function no longer exists.
-				FunctionMustNotExist(t, db, schema, migrations.TriggerFunctionName("posts", migrations.TemporaryName("title")))
-
-				// The up trigger no longer exists.
-				TriggerMustNotExist(t, db, schema, "posts", migrations.TriggerName("posts", "title"))
-				// The down trigger no longer exists.
-				TriggerMustNotExist(t, db, schema, "posts", migrations.TriggerName("posts", migrations.TemporaryName("title")))
+				// The table is cleaned up; temporary columns, trigger functions and triggers no longer exist.
+				TableMustBeCleanedUp(t, db, schema, "posts", "title")
 			},
 		},
 		{
@@ -332,18 +315,8 @@ func TestDropConstraint(t *testing.T) {
 				}, rows)
 			},
 			afterRollback: func(t *testing.T, db *sql.DB, schema string) {
-				// The new (temporary) `user_id` column should not exist on the underlying table.
-				ColumnMustNotExist(t, db, schema, "posts", migrations.TemporaryName("user_id"))
-
-				// The up function no longer exists.
-				FunctionMustNotExist(t, db, schema, migrations.TriggerFunctionName("posts", "user_id"))
-				// The down function no longer exists.
-				FunctionMustNotExist(t, db, schema, migrations.TriggerFunctionName("posts", migrations.TemporaryName("user_id")))
-
-				// The up trigger no longer exists.
-				TriggerMustNotExist(t, db, schema, "posts", migrations.TriggerName("posts", "user_id"))
-				// The down trigger no longer exists.
-				TriggerMustNotExist(t, db, schema, "posts", migrations.TriggerName("posts", migrations.TemporaryName("user_id")))
+				// The table is cleaned up; temporary columns, trigger functions and triggers no longer exist.
+				TableMustBeCleanedUp(t, db, schema, "posts", "user_id")
 			},
 			afterComplete: func(t *testing.T, db *sql.DB, schema string) {
 				// The new (temporary) `user_id` column should not exist on the underlying table.
@@ -364,15 +337,8 @@ func TestDropConstraint(t *testing.T) {
 					{"id": 5, "title": "another post by an unknown user", "user_id": 4},
 				}, rows)
 
-				// The up function no longer exists.
-				FunctionMustNotExist(t, db, schema, migrations.TriggerFunctionName("posts", "user_id"))
-				// The down function no longer exists.
-				FunctionMustNotExist(t, db, schema, migrations.TriggerFunctionName("posts", migrations.TemporaryName("user_id")))
-
-				// The up trigger no longer exists.
-				TriggerMustNotExist(t, db, schema, "posts", migrations.TriggerName("posts", "user_id"))
-				// The down trigger no longer exists.
-				TriggerMustNotExist(t, db, schema, "posts", migrations.TriggerName("posts", migrations.TemporaryName("user_id")))
+				// The table is cleaned up; temporary columns, trigger functions and triggers no longer exist.
+				TableMustBeCleanedUp(t, db, schema, "posts", "user_id")
 			},
 		},
 		{
@@ -430,22 +396,12 @@ func TestDropConstraint(t *testing.T) {
 				})
 			},
 			afterRollback: func(t *testing.T, db *sql.DB, schema string) {
-				// The new (temporary) `name` column should not exist on the underlying table.
-				ColumnMustNotExist(t, db, schema, "users", migrations.TemporaryName("name"))
-
-				// The up function no longer exists.
-				FunctionMustNotExist(t, db, schema, migrations.TriggerFunctionName("users", "name"))
-				// The down function no longer exists.
-				FunctionMustNotExist(t, db, schema, migrations.TriggerFunctionName("users", migrations.TemporaryName("name")))
-
-				// The up trigger no longer exists.
-				TriggerMustNotExist(t, db, schema, "users", migrations.TriggerName("users", "name"))
-				// The down trigger no longer exists.
-				TriggerMustNotExist(t, db, schema, "users", migrations.TriggerName("users", migrations.TemporaryName("name")))
+				// The table is cleaned up; temporary columns, trigger functions and triggers no longer exist.
+				TableMustBeCleanedUp(t, db, schema, "users", "name")
 			},
 			afterComplete: func(t *testing.T, db *sql.DB, schema string) {
-				// The new (temporary) `name` column should not exist on the underlying table.
-				ColumnMustNotExist(t, db, schema, "users", migrations.TemporaryName("name"))
+				// The table is cleaned up; temporary columns, trigger functions and triggers no longer exist.
+				TableMustBeCleanedUp(t, db, schema, "users", "name")
 
 				// Inserting a row that does not meet the unique constraint into the new view works.
 				MustInsert(t, db, schema, "02_drop_unique_constraint", "users", map[string]string{

diff --git a/pkg/migrations/op_drop_multicolumn_constraint_test.go b/pkg/migrations/op_drop_multicolumn_constraint_test.go
@@ -124,25 +124,8 @@ func TestDropMultiColumnConstraint(t *testing.T) {
 				}, rows)
 			},
 			afterRollback: func(t *testing.T, db *sql.DB, schema string) {
-				// The temporary columns no longer exist.
-				ColumnMustNotExist(t, db, schema, "products", migrations.TemporaryName("price"))
-				ColumnMustNotExist(t, db, schema, "products", migrations.TemporaryName("discount"))
-
-				// The up functions no longer exist.
-				FunctionMustNotExist(t, db, schema, migrations.TriggerFunctionName("products", "price"))
-				FunctionMustNotExist(t, db, schema, migrations.TriggerFunctionName("products", "discount"))
-
-				// The down functions no longer exist.
-				FunctionMustNotExist(t, db, schema, migrations.TriggerFunctionName("products", migrations.TemporaryName("price")))
-				FunctionMustNotExist(t, db, schema, migrations.TriggerFunctionName("products", migrations.TemporaryName("discount")))
-
-				// The up triggers no longer exist.
-				TriggerMustNotExist(t, db, schema, "products", migrations.TriggerName("products", "price"))
-				TriggerMustNotExist(t, db, schema, "products", migrations.TriggerName("products", "discount"))
-
-				// The down triggers no longer exist.
-				TriggerMustNotExist(t, db, schema, "products", migrations.TriggerName("products", migrations.TemporaryName("price")))
-				TriggerMustNotExist(t, db, schema, "products", migrations.TriggerName("products", migrations.TemporaryName("discount")))
+				// The table is cleaned up; temporary columns, trigger functions and triggers no longer exist.
+				TableMustBeCleanedUp(t, db, schema, "products", "price", "discount")
 			},
 			afterComplete: func(t *testing.T, db *sql.DB, schema string) {
 				// Inserting a row into the new schema that violates the check constraint succeeds.
@@ -257,25 +240,8 @@ func TestDropMultiColumnConstraint(t *testing.T) {
 				}, rows)
 			},
 			afterRollback: func(t *testing.T, db *sql.DB, schema string) {
-				// The temporary columns no longer exist.
-				ColumnMustNotExist(t, db, schema, "products", migrations.TemporaryName("name"))
-				ColumnMustNotExist(t, db, schema, "products", migrations.TemporaryName("description"))
-
-				// The up functions no longer exist.
-				FunctionMustNotExist(t, db, schema, migrations.TriggerFunctionName("products", "name"))
-				FunctionMustNotExist(t, db, schema, migrations.TriggerFunctionName("products", "description"))
-
-				// The down functions no longer exist.
-				FunctionMustNotExist(t, db, schema, migrations.TriggerFunctionName("products", migrations.TemporaryName("name")))
-				FunctionMustNotExist(t, db, schema, migrations.TriggerFunctionName("products", migrations.TemporaryName("description")))
-
-				// The up triggers no longer exist.
-				TriggerMustNotExist(t, db, schema, "products", migrations.TriggerName("products", "name"))
-				TriggerMustNotExist(t, db, schema, "products", migrations.TriggerName("products", "description"))
-
-				// The down triggers no longer exist.
-				TriggerMustNotExist(t, db, schema, "products", migrations.TriggerName("products", migrations.TemporaryName("name")))
-				TriggerMustNotExist(t, db, schema, "products", migrations.TriggerName("products", migrations.TemporaryName("description")))
+				// The table is cleaned up; temporary columns, trigger functions and triggers no longer exist.
+				TableMustBeCleanedUp(t, db, schema, "products", "name", "description")
 			},
 			afterComplete: func(t *testing.T, db *sql.DB, schema string) {
 				// Inserting a row into the new schema that violates the unique constraint succeeds.
@@ -423,25 +389,8 @@ func TestDropMultiColumnConstraint(t *testing.T) {
 				}, rows)
 			},
 			afterRollback: func(t *testing.T, db *sql.DB, schema string) {
-				// The temporary columns no longer exist.
-				ColumnMustNotExist(t, db, schema, "reports", migrations.TemporaryName("user_id"))
-				ColumnMustNotExist(t, db, schema, "reports", migrations.TemporaryName("user_zip"))
-
-				// The up functions no longer exist.
-				FunctionMustNotExist(t, db, schema, migrations.TriggerFunctionName("reports", "user_id"))
-				FunctionMustNotExist(t, db, schema, migrations.TriggerFunctionName("reports", "user_zip"))
-
-				// The down functions no longer exist.
-				FunctionMustNotExist(t, db, schema, migrations.TriggerFunctionName("reports", migrations.TemporaryName("user_id")))
-				FunctionMustNotExist(t, db, schema, migrations.TriggerFunctionName("reports", migrations.TemporaryName("user_zip")))
-
-				// The up triggers no longer exist.
-				TriggerMustNotExist(t, db, schema, "reports", migrations.TriggerName("reports", "user_id"))
-				TriggerMustNotExist(t, db, schema, "reports", migrations.TriggerName("reports", "user_zip"))
-
-				// The down triggers no longer exist.
-				TriggerMustNotExist(t, db, schema, "reports", migrations.TriggerName("reports", migrations.TemporaryName("user_id")))
-				TriggerMustNotExist(t, db, schema, "reports", migrations.TriggerName("reports", migrations.TemporaryName("user_zip")))
+				// The table is cleaned up; temporary columns, trigger functions and triggers no longer exist.
+				TableMustBeCleanedUp(t, db, schema, "reports", "user_id", "user_zip")
 			},
 			afterComplete: func(t *testing.T, db *sql.DB, schema string) {
 				// Inserting a row that violates the FK constraint into the new schema succeeds.

diff --git a/pkg/migrations/op_drop_not_null_test.go b/pkg/migrations/op_drop_not_null_test.go
@@ -110,22 +110,12 @@ func TestDropNotNull(t *testing.T) {
 				}, rows)
 			},
 			afterRollback: func(t *testing.T, db *sql.DB, schema string) {
-				// The new (temporary) `review` column should not exist on the underlying table.
-				ColumnMustNotExist(t, db, schema, "reviews", migrations.TemporaryName("review"))
-
-				// The up function no longer exists.
-				FunctionMustNotExist(t, db, schema, migrations.TriggerFunctionName("reviews", "review"))
-				// The down function no longer exists.
-				FunctionMustNotExist(t, db, schema, migrations.TriggerFunctionName("reviews", migrations.TemporaryName("review")))
-
-				// The up trigger no longer exists.
-				TriggerMustNotExist(t, db, schema, "reviews", migrations.TriggerName("reviews", "review"))
-				// The down trigger no longer exists.
-				TriggerMustNotExist(t, db, schema, "reviews", migrations.TriggerName("reviews", migrations.TemporaryName("review")))
+				// The table is cleaned up; temporary columns, trigger functions and triggers no longer exist.
+				TableMustBeCleanedUp(t, db, schema, "reviews", "review")
 			},
 			afterComplete: func(t *testing.T, db *sql.DB, schema string) {
-				// The new (temporary) `review` column should not exist on the underlying table.
-				ColumnMustNotExist(t, db, schema, "reviews", migrations.TemporaryName("review"))
+				// The table is cleaned up; temporary columns, trigger functions and triggers no longer exist.
+				TableMustBeCleanedUp(t, db, schema, "reviews", "review")
 
 				// Writing a NULL review into the `review` column should succeed.
 				MustInsert(t, db, schema, "02_set_nullable", "reviews", map[string]string{
@@ -141,16 +131,6 @@ func TestDropNotNull(t *testing.T) {
 					{"id": 4, "username": "dana", "product": "durian", "review": "delicious"},
 					{"id": 5, "username": "earl", "product": "eggplant", "review": nil},
 				}, rows)
-
-				// The up function no longer exists.
-				FunctionMustNotExist(t, db, schema, migrations.TriggerFunctionName("reviews", "review"))
-				// The down function no longer exists.
-				FunctionMustNotExist(t, db, schema, migrations.TriggerFunctionName("reviews", migrations.TemporaryName("review")))
-
-				// The up trigger no longer exists.
-				TriggerMustNotExist(t, db, schema, "reviews", migrations.TriggerName("reviews", "review"))
-				// The down trigger no longer exists.
-				TriggerMustNotExist(t, db, schema, "reviews", migrations.TriggerName("reviews", migrations.TemporaryName("review")))
 			},
 		},
 		{