rogee
aeddedf2a3
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.
108 lines
2.7 KiB
Go
108 lines
2.7 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/gochat/gochat/internal/model"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func setupPreferenceTestDB(t *testing.T) *gorm.DB {
|
|
t.Helper()
|
|
db, err := gorm.Open(sqlite.Open("file::memory:"), &gorm.Config{})
|
|
require.NoError(t, err)
|
|
require.NoError(t, db.AutoMigrate(&model.CaptainPreference{}))
|
|
return db
|
|
}
|
|
|
|
func TestCaptainPreferenceRepo_Create(t *testing.T) {
|
|
db := setupPreferenceTestDB(t)
|
|
repo := NewCaptainPreferenceRepo(db)
|
|
ctx := context.Background()
|
|
|
|
pref := &model.CaptainPreference{
|
|
AccountID: 1,
|
|
Tone: "professional",
|
|
Language: "en",
|
|
AutoLabelEnabled: true,
|
|
AutoFollowUpEnabled: false,
|
|
AutoReplyEnabled: false,
|
|
MaxResponseLength: 500,
|
|
CustomPromptSuffix: "Be concise.",
|
|
}
|
|
|
|
err := repo.Create(ctx, pref)
|
|
require.NoError(t, err)
|
|
assert.NotZero(t, pref.ID)
|
|
assert.Equal(t, "professional", pref.Tone)
|
|
}
|
|
|
|
func TestCaptainPreferenceRepo_GetByAccountID(t *testing.T) {
|
|
db := setupPreferenceTestDB(t)
|
|
repo := NewCaptainPreferenceRepo(db)
|
|
ctx := context.Background()
|
|
|
|
pref := &model.CaptainPreference{
|
|
AccountID: 42,
|
|
Tone: "friendly",
|
|
Language: "es",
|
|
MaxResponseLength: 300,
|
|
}
|
|
require.NoError(t, repo.Create(ctx, pref))
|
|
|
|
found, err := repo.GetByAccountID(ctx, 42)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, uint(42), found.AccountID)
|
|
assert.Equal(t, "friendly", found.Tone)
|
|
assert.Equal(t, "es", found.Language)
|
|
}
|
|
|
|
func TestCaptainPreferenceRepo_GetByAccountID_NotFound(t *testing.T) {
|
|
db := setupPreferenceTestDB(t)
|
|
repo := NewCaptainPreferenceRepo(db)
|
|
ctx := context.Background()
|
|
|
|
_, err := repo.GetByAccountID(ctx, 999)
|
|
assert.Error(t, err)
|
|
}
|
|
|
|
func TestCaptainPreferenceRepo_Update(t *testing.T) {
|
|
db := setupPreferenceTestDB(t)
|
|
repo := NewCaptainPreferenceRepo(db)
|
|
ctx := context.Background()
|
|
|
|
pref := &model.CaptainPreference{
|
|
AccountID: 1,
|
|
Tone: "professional",
|
|
MaxResponseLength: 500,
|
|
}
|
|
require.NoError(t, repo.Create(ctx, pref))
|
|
|
|
pref.Tone = "casual"
|
|
pref.MaxResponseLength = 200
|
|
require.NoError(t, repo.Update(ctx, pref))
|
|
|
|
found, err := repo.GetByAccountID(ctx, 1)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "casual", found.Tone)
|
|
assert.Equal(t, 200, found.MaxResponseLength)
|
|
}
|
|
|
|
func TestCaptainPreferenceRepo_Delete(t *testing.T) {
|
|
db := setupPreferenceTestDB(t)
|
|
repo := NewCaptainPreferenceRepo(db)
|
|
ctx := context.Background()
|
|
|
|
pref := &model.CaptainPreference{AccountID: 1, Tone: "professional"}
|
|
require.NoError(t, repo.Create(ctx, pref))
|
|
|
|
require.NoError(t, repo.Delete(ctx, pref.ID))
|
|
|
|
_, err := repo.GetByAccountID(ctx, 1)
|
|
assert.Error(t, err) // soft-deleted should not be found
|
|
}
|