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

212 lines
6.4 KiB
Plaintext

package service
import (
"context"
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"gorm.io/gorm"
"github.com/gochat/gochat/internal/model"
)
// ========== Team Service test utilities ==========
func createTeamTestAccount(t *testing.T, db *gorm.DB) *model.Account {
t.Helper()
account := &model.Account{Name: "TeamSvcOrg", Locale: "en", Active: true}
require.NoError(t, db.Create(account).Error)
return account
}
func createTeamTestUser(t *testing.T, db *gorm.DB, name, email string) *model.User {
t.Helper()
user := &model.User{Name: name, Email: email, Provider: "email"}
require.NoError(t, db.Create(user).Error)
return user
}
// ========== Team Service tests ==========
func TestTeamService_Get(t *testing.T) {
db, svc := setupTeamService(t)
account := createTeamTestAccount(t, db)
team := &model.Team{AccountID: account.ID, Name: "SvcGetTeam", Description: "Desc"}
require.NoError(t, db.Create(team).Error)
result, err := svc.Get(context.Background(), team.ID, account.ID)
require.NoError(t, err)
assert.Equal(t, "SvcGetTeam", result.Name)
}
func TestTeamService_Get_NotFound(t *testing.T) {
_, svc := setupTeamService(t)
result, err := svc.Get(context.Background(), 9999, 9999)
assert.Error(t, err)
assert.Nil(t, result)
}
func TestTeamService_List(t *testing.T) {
db, svc := setupTeamService(t)
account := createTeamTestAccount(t, db)
for i := 0; i < 3; i++ {
require.NoError(t, db.Create(&model.Team{AccountID: account.ID, Name: "TeamSvc" + string(rune('A'+i))}).Error)
}
teams, total, err := svc.List(context.Background(), account.ID, 0, 10)
require.NoError(t, err)
assert.Equal(t, int64(3), total)
assert.Len(t, teams, 3)
}
func TestTeamService_Create(t *testing.T) {
db, svc := setupTeamService(t)
account := createTeamTestAccount(t, db)
autoAssign := true
team, err := svc.Create(context.Background(), account.ID, CreateTeamRequest{
Name: "NewSvcTeam",
Description: "Service created team",
AllowAutoAssignment: &autoAssign,
})
require.NoError(t, err)
assert.NotZero(t, team.ID)
assert.Equal(t, account.ID, team.AccountID)
assert.Equal(t, "NewSvcTeam", team.Name)
assert.True(t, team.AllowAutoAssignment)
}
func TestTeamService_Update(t *testing.T) {
db, svc := setupTeamService(t)
account := createTeamTestAccount(t, db)
team := &model.Team{AccountID: account.ID, Name: "SvcBeforeUpdate"}
require.NoError(t, db.Create(team).Error)
updated, err := svc.Update(context.Background(), team.ID, account.ID, UpdateTeamRequest{
Name: "SvcAfterUpdate",
Description: "Updated via service",
})
require.NoError(t, err)
assert.Equal(t, "SvcAfterUpdate", updated.Name)
assert.Equal(t, "Updated via service", updated.Description)
}
func TestTeamService_Delete(t *testing.T) {
db, svc := setupTeamService(t)
account := createTeamTestAccount(t, db)
team := &model.Team{AccountID: account.ID, Name: "SvcDeleteTeam"}
require.NoError(t, db.Create(team).Error)
err := svc.Delete(context.Background(), team.ID, account.ID)
assert.NoError(t, err)
found, err := svc.Get(context.Background(), team.ID, account.ID)
assert.Error(t, err)
assert.Nil(t, found)
}
func TestTeamService_AddMembers(t *testing.T) {
db, svc := setupTeamService(t)
account := createTeamTestAccount(t, db)
team := &model.Team{AccountID: account.ID, Name: "SvcAddMemberTeam"}
require.NoError(t, db.Create(team).Error)
user1 := createTeamTestUser(t, db, "SvcAgent1", "svc1@example.com")
user2 := createTeamTestUser(t, db, "SvcAgent2", "svc2@example.com")
members, err := svc.AddMembers(context.Background(), team.ID, account.ID, []uint{user1.ID, user2.ID})
require.NoError(t, err)
assert.Len(t, members, 2)
}
func TestTeamService_RemoveMember(t *testing.T) {
db, svc := setupTeamService(t)
account := createTeamTestAccount(t, db)
team := &model.Team{AccountID: account.ID, Name: "SvcRmMemberTeam"}
require.NoError(t, db.Create(team).Error)
user := createTeamTestUser(t, db, "SvcAgent3", "svc3@example.com")
_, err := svc.AddMembers(context.Background(), team.ID, account.ID, []uint{user.ID})
require.NoError(t, err)
err = svc.RemoveMember(context.Background(), team.ID, user.ID, account.ID)
assert.NoError(t, err)
}
func TestTeamService_ListMembers(t *testing.T) {
db, svc := setupTeamService(t)
account := createTeamTestAccount(t, db)
team := &model.Team{AccountID: account.ID, Name: "SvcListMembersTeam"}
require.NoError(t, db.Create(team).Error)
for i := 0; i < 3; i++ {
user := createTeamTestUser(t, db, "SvcListAgent"+string(rune('A'+i)), "svclist"+string(rune('A'+i))+"@example.com")
_, err := svc.AddMembers(context.Background(), team.ID, account.ID, []uint{user.ID})
require.NoError(t, err)
}
members, err := svc.ListMembers(context.Background(), team.ID, account.ID)
require.NoError(t, err)
assert.Len(t, members, 3)
}
// ========== Profile Service tests ==========
func TestProfileService_Get(t *testing.T) {
db, svc := setupProfileService(t)
user := &model.User{Name: "ProfileUser", Email: "profile@example.com", Provider: "email"}
require.NoError(t, db.Create(user).Error)
result, err := svc.Get(context.Background(), user.ID)
require.NoError(t, err)
assert.Equal(t, "ProfileUser", result.Name)
assert.Equal(t, "profile@example.com", result.Email)
}
func TestProfileService_Get_NotFound(t *testing.T) {
_, svc := setupProfileService(t)
result, err := svc.Get(context.Background(), 9999)
assert.Error(t, err)
assert.Nil(t, result)
}
func TestProfileService_Update(t *testing.T) {
db, svc := setupProfileService(t)
user := &model.User{Name: "BeforeProfile", Email: "before@example.com", Provider: "email"}
require.NoError(t, db.Create(user).Error)
updated, err := svc.Update(context.Background(), user.ID, UpdateProfileRequest{
Name: "AfterProfile",
Email: "after@example.com",
Availability: "online",
})
require.NoError(t, err)
assert.Equal(t, "AfterProfile", updated.Name)
assert.Equal(t, "after@example.com", updated.Email)
}
func TestProfileService_UpdateAvatar(t *testing.T) {
db, svc := setupProfileService(t)
user := &model.User{Name: "AvatarUser", Email: "avatar@example.com", Provider: "email"}
require.NoError(t, db.Create(user).Error)
updated, err := svc.UpdateAvatar(context.Background(), user.ID, UpdateAvatarRequest{
AvatarURL: "https://example.com/avatar.png",
})
require.NoError(t, err)
assert.Equal(t, "https://example.com/avatar.png", updated.AvatarURL)
}