Files
gochat/backend/internal/service/captain_preference_service_test.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

123 lines
3.3 KiB
Go

package service
import (
"context"
"fmt"
"testing"
"github.com/gochat/gochat/internal/model"
"github.com/gochat/gochat/internal/repository"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"gorm.io/driver/sqlite"
"gorm.io/gorm"
)
func setupPreferenceServiceTestDB(t *testing.T) *gorm.DB {
t.Helper()
// Use unique DB per test to avoid cross-test data leakage
dbName := fmt.Sprintf("file:%s?mode=memory&cache=private", t.Name())
db, err := gorm.Open(sqlite.Open(dbName), &gorm.Config{})
require.NoError(t, err)
require.NoError(t, db.AutoMigrate(&model.CaptainPreference{}))
return db
}
func newTestPreferenceService(t *testing.T) (*CaptainPreferenceService, *repository.CaptainPreferenceRepo) {
t.Helper()
db := setupPreferenceServiceTestDB(t)
repo := repository.NewCaptainPreferenceRepo(db)
svc := NewCaptainPreferenceService(repo)
return svc, repo
}
func intPtr(v int) *int { return &v }
func TestCaptainPreferenceService_Create(t *testing.T) {
svc, _ := newTestPreferenceService(t)
ctx := context.Background()
req := &CreatePreferenceRequest{
Tone: "professional",
Language: "en",
AutoLabelEnabled: true,
MaxResponseLength: 500,
}
pref, err := svc.Create(ctx, 1, req)
require.NoError(t, err)
assert.NotZero(t, pref.ID)
assert.Equal(t, uint(1), pref.AccountID)
assert.Equal(t, "professional", pref.Tone)
assert.Equal(t, "en", pref.Language)
assert.True(t, pref.AutoLabelEnabled)
}
func TestCaptainPreferenceService_Create_DuplicateAccount(t *testing.T) {
svc, _ := newTestPreferenceService(t)
ctx := context.Background()
req := &CreatePreferenceRequest{Tone: "professional"}
_, err := svc.Create(ctx, 1, req)
require.NoError(t, err)
req2 := &CreatePreferenceRequest{Tone: "casual"}
_, err = svc.Create(ctx, 1, req2)
assert.Error(t, err) // duplicate account_id should fail
}
func TestCaptainPreferenceService_Get(t *testing.T) {
svc, _ := newTestPreferenceService(t)
ctx := context.Background()
req := &CreatePreferenceRequest{Tone: "friendly", Language: "es"}
pref, err := svc.Create(ctx, 42, req)
require.NoError(t, err)
found, err := svc.Get(ctx, 42)
require.NoError(t, err)
assert.Equal(t, pref.ID, found.ID)
assert.Equal(t, "friendly", found.Tone)
}
func TestCaptainPreferenceService_Get_NotFound(t *testing.T) {
svc, _ := newTestPreferenceService(t)
ctx := context.Background()
_, err := svc.Get(ctx, 999)
assert.Error(t, err)
}
func TestCaptainPreferenceService_Update(t *testing.T) {
svc, _ := newTestPreferenceService(t)
ctx := context.Background()
req := &CreatePreferenceRequest{Tone: "professional", MaxResponseLength: 500}
_, err := svc.Create(ctx, 1, req)
require.NoError(t, err)
updateReq := &UpdatePreferenceRequest{
Tone: "casual",
MaxResponseLength: intPtr(200),
}
updated, err := svc.Update(ctx, 1, updateReq)
require.NoError(t, err)
assert.Equal(t, "casual", updated.Tone)
assert.Equal(t, 200, updated.MaxResponseLength)
}
func TestCaptainPreferenceService_Delete(t *testing.T) {
svc, _ := newTestPreferenceService(t)
ctx := context.Background()
req := &CreatePreferenceRequest{Tone: "formal"}
_, err := svc.Create(ctx, 1, req)
require.NoError(t, err)
// Delete and verify soft delete
err = svc.Delete(ctx, 1)
require.NoError(t, err)
_, err = svc.Get(ctx, 1)
assert.Error(t, err) // soft deleted, should not be found
}