diff options
Diffstat (limited to 'pkg/database/sql/main_test.go')
-rw-r--r-- | pkg/database/sql/main_test.go | 45 |
1 files changed, 45 insertions, 0 deletions
diff --git a/pkg/database/sql/main_test.go b/pkg/database/sql/main_test.go new file mode 100644 index 0000000..c373740 --- /dev/null +++ b/pkg/database/sql/main_test.go @@ -0,0 +1,45 @@ +//go:build integration + +package sql + +import ( + "context" + "os" + "testing" + + "gorm.io/driver/sqlite" + "gorm.io/gorm" + "gorm.io/gorm/logger" +) + +var ( + ctx context.Context +) + +func setup(t *testing.T) (*gorm.DB, func()) { + t.Helper() + + ctx = context.Background() + + file, err := os.CreateTemp("", "img_user_*.db") + if err != nil { + t.Fatalf("Error creating tmp error: %s", err.Error()) + } + + db, err := gorm.Open(sqlite.Open(file.Name()), &gorm.Config{ + Logger: logger.Default.LogMode(logger.Info), + }) + if err != nil { + t.Fatalf("Error openning db, error %s", err.Error()) + } + + err = Migrate(db) + if err != nil { + t.Fatalf("Error migrating db, error %s", err.Error()) + } + + return db, func() { + //nolint:errcheck + os.Remove(file.Name()) + } +} |