Files
gochat/backend/internal/canned/canned_test_helper.go
T
rogee aeddedf2a3 Reorganize repo: backend/, deploy/, docs/ layout + AGENTS.md
Restructure the monorepo into clear top-level directories:
- backend/: Go module root (cmd, internal, pkg, configs, migrations,
  docs/swagger, scripts, tests, go.mod, Makefile, .air.toml)
- deploy/: Docker (Dockerfile, docker-compose*), quickstart, fluentd
- docs/: project documentation + reports/ (moved from repo root)
- AGENTS.md: new AI coding-agent guide at repo root

Update all references to the new layout:
- Dockerfile: COPY backend/go.mod, COPY backend/ (context = repo root)
- docker-compose files: context ../.., dockerfile deploy/docker/Dockerfile,
  env_file ../../.env, volume mounts ../../backend:/app
- deploy/quickstart/compose.yaml: dockerfile deploy/docker/Dockerfile
- CI: working-directory: backend for go commands, file deploy/docker/Dockerfile,
  coverage path backend/coverage.out, health_check backend/scripts/
- backend/Makefile: docker target uses -f ../deploy/docker/Dockerfile ../
- README: architecture tree, quickstart, config paths updated

Move root stray scripts (rename_models.*, run_m11_tests.sh, verify_build.sh,
gorm_bool_main.go) to backend/scripts/legacy/. All moves via git mv to
preserve history. Build, vet, SQLite tests, and docker compose config verified.
2026-07-07 14:44:12 +08:00

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
}