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.
129 lines
4.1 KiB
Plaintext
129 lines
4.1 KiB
Plaintext
package service
|
|
|
|
import (
|
|
"context"
|
|
"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/gorm"
|
|
)
|
|
|
|
// setupPushSubscriptionService 创建 PushTokenRepo + PushSubscriptionService 测试实例。
|
|
func setupPushSubscriptionService(t *testing.T) (*gorm.DB, *repository.PushTokenRepo, *PushSubscriptionService) {
|
|
t.Helper()
|
|
db := setupServiceTestDB(t)
|
|
repo := repository.NewPushTokenRepo(db)
|
|
svc := NewPushSubscriptionService(repo)
|
|
return db, repo, svc
|
|
}
|
|
|
|
// createTestPushToken 在数据库中创建一个测试 PushToken 并返回。
|
|
func createTestPushToken(t *testing.T, db *gorm.DB, userID uint, token string, platform string) *model.PushToken {
|
|
t.Helper()
|
|
pt := &model.PushToken{
|
|
UserID: userID,
|
|
Token: token,
|
|
Platform: platform,
|
|
DeviceID: "test-device-id",
|
|
}
|
|
if err := db.Create(pt).Error; err != nil {
|
|
t.Fatalf("无法创建测试推送令牌: %v", err)
|
|
}
|
|
return pt
|
|
}
|
|
|
|
// ========== ListPushTokens ==========
|
|
|
|
func TestPushSubscriptionService_ListPushTokens_成功(t *testing.T) {
|
|
db, _, svc := setupPushSubscriptionService(t)
|
|
account := createTestAccount(t, db)
|
|
user := createTestUser(t, db, account.ID)
|
|
|
|
// 创建多个推送令牌
|
|
createTestPushToken(t, db, user.ID, "token-abc-123", "ios")
|
|
createTestPushToken(t, db, user.ID, "token-def-456", "android")
|
|
|
|
tokens, err := svc.ListPushTokens(context.Background(), user.ID)
|
|
require.NoError(t, err)
|
|
assert.Len(t, tokens, 2)
|
|
assert.Equal(t, "token-abc-123", tokens[0].Token)
|
|
assert.Equal(t, "token-def-456", tokens[1].Token)
|
|
}
|
|
|
|
func TestPushSubscriptionService_ListPushTokens_无令牌(t *testing.T) {
|
|
db, _, svc := setupPushSubscriptionService(t)
|
|
account := createTestAccount(t, db)
|
|
user := createTestUser(t, db, account.ID)
|
|
|
|
tokens, err := svc.ListPushTokens(context.Background(), user.ID)
|
|
require.NoError(t, err)
|
|
assert.Len(t, tokens, 0)
|
|
}
|
|
|
|
// ========== RegisterPushToken ==========
|
|
|
|
func TestPushSubscriptionService_RegisterPushToken_成功(t *testing.T) {
|
|
db, _, svc := setupPushSubscriptionService(t)
|
|
account := createTestAccount(t, db)
|
|
user := createTestUser(t, db, account.ID)
|
|
|
|
pt, err := svc.RegisterPushToken(context.Background(), user.ID, "new-token-xyz", "web", "device-001")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, user.ID, pt.UserID)
|
|
assert.Equal(t, "new-token-xyz", pt.Token)
|
|
assert.Equal(t, "web", pt.Platform)
|
|
assert.Equal(t, "device-001", pt.DeviceID)
|
|
assert.NotZero(t, pt.ID)
|
|
}
|
|
|
|
// ========== RemovePushToken ==========
|
|
|
|
func TestPushSubscriptionService_RemovePushToken_成功(t *testing.T) {
|
|
db, _, svc := setupPushSubscriptionService(t)
|
|
account := createTestAccount(t, db)
|
|
user := createTestUser(t, db, account.ID)
|
|
pt := createTestPushToken(t, db, user.ID, "token-to-remove", "ios")
|
|
|
|
err := svc.RemovePushToken(context.Background(), pt.ID)
|
|
require.NoError(t, err)
|
|
|
|
// 验证令牌已被删除
|
|
var count int64
|
|
db.Model(&model.PushToken{}).Where("id = ?", pt.ID).Count(&count)
|
|
assert.Equal(t, int64(0), count)
|
|
}
|
|
|
|
func TestPushSubscriptionService_RemovePushToken_不存在(t *testing.T) {
|
|
_, _, svc := setupPushSubscriptionService(t)
|
|
|
|
err := svc.RemovePushToken(context.Background(), 9999)
|
|
assert.Error(t, err)
|
|
}
|
|
|
|
// ========== RemovePushTokenByValue ==========
|
|
|
|
func TestPushSubscriptionService_RemovePushTokenByValue_成功(t *testing.T) {
|
|
db, _, svc := setupPushSubscriptionService(t)
|
|
account := createTestAccount(t, db)
|
|
user := createTestUser(t, db, account.ID)
|
|
pt := createTestPushToken(t, db, user.ID, "token-by-value-remove", "android")
|
|
|
|
err := svc.RemovePushTokenByValue(context.Background(), pt.Token, user.ID)
|
|
require.NoError(t, err)
|
|
|
|
// 验证令牌已被删除
|
|
var count int64
|
|
db.Model(&model.PushToken{}).Where("id = ?", pt.ID).Count(&count)
|
|
assert.Equal(t, int64(0), count)
|
|
}
|
|
|
|
func TestPushSubscriptionService_RemovePushTokenByValue_不存在(t *testing.T) {
|
|
_, _, svc := setupPushSubscriptionService(t)
|
|
|
|
err := svc.RemovePushTokenByValue(context.Background(), "nonexistent-token", 9999)
|
|
// DeleteByTokenAndUser with no matching record should not error (GORM delete 0 rows is not an error)
|
|
assert.NoError(t, err)
|
|
} |