refactor: unify code style, use existing patterns, and highlight impl…

…emented interfaces in structs
uptrace · Dec 6, 2022 · 533383a · 533383a
1 parent 8abcf63
commit 533383a
Show file tree

Hide file tree

Showing 3 changed files with 137 additions and 84 deletions.
diff --git a/internal/dbtest/query_test.go b/internal/dbtest/query_test.go
@@ -958,7 +958,8 @@ func TestAlterTable(t *testing.T) {
 		{
 			"rename column", needsAlterTable,
 			func(db *bun.DB) schema.QueryAppender {
-				return db.NewAlterTable().Model((*Model)(nil)).RenameColumn("old", "new")
+				return db.NewAlterTable().Model((*Model)(nil)).
+					RenameColumn().Column("old").To("new")
 			},
 		},
 		{
@@ -970,29 +971,32 @@ func TestAlterTable(t *testing.T) {
 		{
 			"rename table", needsAlterTable,
 			func(db *bun.DB) schema.QueryAppender {
-				return db.NewAlterTable().Model((*Model)(nil)).Rename("new_models")
+				return db.NewAlterTable().Model((*Model)(nil)).
+					Rename().To("new_models")
 			},
 		},
 		{
 			"rename table if exists", needsAlterTable,
 			func(db *bun.DB) schema.QueryAppender {
-				return db.NewAlterTable().Model((*Model)(nil)).IfExists().Rename("new_models")
+				return db.NewAlterTable().Model((*Model)(nil)).IfExists().
+					Rename().To("new_models")
 			},
 		},
 		{
 			"change column type", needsAlterTable,
 			func(db *bun.DB) schema.QueryAppender {
 				// Change column type with common SQL data type
-				return db.NewAlterTable().Model((*Model)(nil)).AlterColumn("old").Type(sqltype.Blob)
+				return db.NewAlterTable().Model((*Model)(nil)).
+					AlterColumn().Column("old").Type(sqltype.Blob)
 			},
 		},
 		{
 			"change column type chained", needsAlterTable,
 			func(db *bun.DB) schema.QueryAppender {
 				// Alter 2 columns in a chained query
 				return db.NewAlterTable().Model((*Model)(nil)).
-					AlterColumn("old").Type(sqltype.Blob).
-					AlterColumn("active").Type(sqltype.SmallInt)
+					AlterColumn().Column("old").Type(sqltype.Blob).
+					AlterColumn().Column("active").Type(sqltype.SmallInt)
 			},
 		},
 		{