mirror of
https://github.com/kevin-DL/services.git
synced 2026-01-12 19:25:16 +00:00
31 lines
748 B
Go
31 lines
748 B
Go
package handler_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/micro/services/codes/handler"
|
|
"gorm.io/driver/postgres"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func testHandler(t *testing.T) *handler.Codes {
|
|
// connect to the database
|
|
db, err := gorm.Open(postgres.Open("postgresql://postgres@localhost:5432/codes?sslmode=disable"), &gorm.Config{})
|
|
if err != nil {
|
|
t.Fatalf("Error connecting to database: %v", err)
|
|
}
|
|
|
|
// migrate the database
|
|
if err := db.AutoMigrate(&handler.Code{}); err != nil {
|
|
t.Fatalf("Error migrating database: %v", err)
|
|
}
|
|
|
|
// clean any data from a previous run
|
|
if err := db.Exec("TRUNCATE TABLE codes CASCADE").Error; err != nil {
|
|
t.Fatalf("Error cleaning database: %v", err)
|
|
}
|
|
|
|
return &handler.Codes{DB: db, Time: time.Now}
|
|
}
|