88 lines
1.9 KiB
Go
88 lines
1.9 KiB
Go
package canned
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/logger"
|
|
|
|
"github.com/gochat/gochat/internal/model"
|
|
)
|
|
|
|
// testDBProvider implements DBProvider for tests.
|
|
type testDBProvider struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func (p *testDBProvider) DB() *gorm.DB {
|
|
return p.db
|
|
}
|
|
|
|
// setupCannedTestDB creates an SQLite in-memory database with all required migrations.
|
|
func setupCannedTestDB(t *testing.T) *gorm.DB {
|
|
t.Helper()
|
|
|
|
db, err := gorm.Open(sqlite.Open("file::memory:"), &gorm.Config{
|
|
Logger: logger.Default.LogMode(logger.Silent),
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("failed to open SQLite test database: %v", err)
|
|
}
|
|
|
|
if err := db.AutoMigrate(
|
|
&model.Account{},
|
|
&model.User{},
|
|
&model.AccountUser{},
|
|
&CannedResponse{},
|
|
); err != nil {
|
|
t.Fatalf("failed to auto-migrate models: %v", err)
|
|
}
|
|
|
|
t.Cleanup(func() {
|
|
sqlDB, _ := db.DB()
|
|
sqlDB.Close()
|
|
})
|
|
|
|
return db
|
|
}
|
|
|
|
// setupCannedTestDBProvider creates a DBProvider-backed test database.
|
|
func setupCannedTestDBProvider(t *testing.T) DBProvider {
|
|
t.Helper()
|
|
db := setupCannedTestDB(t)
|
|
return &testDBProvider{db: db}
|
|
}
|
|
|
|
// seedTestAccount creates a minimal account + user for canned response tests.
|
|
func seedTestAccount(db *gorm.DB, t *testing.T) (accountID uint, userID uint) {
|
|
t.Helper()
|
|
|
|
account := &model.Account{Name: "Test Account"}
|
|
if err := db.Create(account).Error; err != nil {
|
|
t.Fatalf("failed to create test account: %v", err)
|
|
}
|
|
accountID = account.ID
|
|
|
|
user := &model.User{
|
|
AccountID: accountID,
|
|
Name: "Test User",
|
|
Email: "test@example.com",
|
|
Password: "testpassword",
|
|
}
|
|
if err := db.Create(user).Error; err != nil {
|
|
t.Fatalf("failed to create test user: %v", err)
|
|
}
|
|
userID = user.ID
|
|
|
|
au := &model.AccountUser{
|
|
AccountID: accountID,
|
|
UserID: userID,
|
|
Role: "administrator",
|
|
}
|
|
if err := db.Create(au).Error; err != nil {
|
|
t.Fatalf("failed to create account_user: %v", err)
|
|
}
|
|
|
|
return accountID, userID
|
|
} |