191 lines
6.7 KiB
Go
191 lines
6.7 KiB
Go
package repository
|
||
|
||
import (
|
||
"context"
|
||
"database/sql"
|
||
"errors"
|
||
"path/filepath"
|
||
"strings"
|
||
"testing"
|
||
"time"
|
||
|
||
"git.pcux.cn/xuans/Scert/backend/internal/config"
|
||
"git.pcux.cn/xuans/Scert/backend/internal/database"
|
||
"git.pcux.cn/xuans/Scert/backend/internal/domain"
|
||
"git.pcux.cn/xuans/Scert/backend/internal/testsupport/mysqltest"
|
||
)
|
||
|
||
func TestSQLProjectRepositorySQLite(t *testing.T) {
|
||
cfg := config.Default(t.TempDir()).Database
|
||
cfg.DSN = filepath.Join(t.TempDir(), "projects.db")
|
||
db, err := database.Open(t.Context(), cfg)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
t.Cleanup(func() { _ = db.Close() })
|
||
if err := database.Migrate(t.Context(), db.DB, cfg.Driver); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
checkProjectRepository(t, NewSQLProjectRepository(db.DB))
|
||
if err := db.Close(); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
db, err = database.Open(t.Context(), cfg)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if err := database.Migrate(t.Context(), db.DB, cfg.Driver); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
items, err := NewSQLProjectRepository(db.DB).ListProjects(t.Context())
|
||
if err != nil || len(items) != 2 || items[0].Name != "调整后的默认项目" || items[1].ID != "retained" {
|
||
t.Fatalf("关闭重开后项目未保留：%+v，%v", items, err)
|
||
}
|
||
}
|
||
|
||
func TestSQLProjectRepositoryMySQLReal(t *testing.T) {
|
||
server := mysqltest.New(t)
|
||
cfg := server.Database(t)
|
||
db, err := database.Open(t.Context(), cfg)
|
||
if err != nil {
|
||
t.Fatal("打开独立 MySQL 项目测试库失败")
|
||
}
|
||
t.Cleanup(func() { _ = db.Close() })
|
||
if err := database.Migrate(t.Context(), db.DB, cfg.Driver); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
checkProjectRepository(t, NewSQLProjectRepository(db.DB, cfg.Driver))
|
||
}
|
||
|
||
func checkProjectRepository(t *testing.T, repo *SQLProjectRepository) {
|
||
t.Helper()
|
||
ctx := t.Context()
|
||
items, err := repo.ListProjects(ctx)
|
||
if err != nil || len(items) != 1 || items[0].ID != domain.DefaultProjectID || items[0].Name != "默认项目" {
|
||
t.Fatalf("默认项目缺失：%+v，%v", items, err)
|
||
}
|
||
initial := items[0]
|
||
t.Run("reserved_default", func(t *testing.T) {
|
||
if err := repo.CreateProject(ctx, initial); !errors.Is(err, ErrDefaultProject) {
|
||
t.Fatalf("未拒绝重复创建默认项目：%v", err)
|
||
}
|
||
if err := repo.DeleteProject(ctx, " default "); !errors.Is(err, ErrDefaultProject) {
|
||
t.Fatalf("未阻止删除默认项目：%v", err)
|
||
}
|
||
})
|
||
item := domain.Project{ID: "project-a", Name: " 证书项目🔒 ", Description: " 项目说明\n第二行 "}
|
||
item.CreatedAt = time.Date(2000, 1, 1, 0, 0, 0, 0, time.UTC)
|
||
t.Run("create_and_read", func(t *testing.T) {
|
||
if err := repo.CreateProject(ctx, item); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
loaded, err := repo.GetProject(ctx, " project-a ")
|
||
if err != nil || loaded.Name != strings.TrimSpace(item.Name) || loaded.Description != strings.TrimSpace(item.Description) {
|
||
t.Fatalf("项目数据读取不符：%+v，%v", loaded, err)
|
||
}
|
||
if loaded.CreatedAt.Equal(item.CreatedAt) || loaded.CreatedAt.IsZero() || !loaded.CreatedAt.Equal(loaded.UpdatedAt) {
|
||
t.Fatal("仓储必须生成自己的创建和更新时间")
|
||
}
|
||
item = loaded
|
||
})
|
||
t.Run("duplicate_and_invalid", func(t *testing.T) {
|
||
if err := repo.CreateProject(ctx, item); err == nil {
|
||
t.Fatal("重复项目主键未拒绝")
|
||
}
|
||
bad := item
|
||
bad.Name = " "
|
||
if err := repo.CreateProject(ctx, bad); err == nil {
|
||
t.Fatal("创建未校验项目")
|
||
}
|
||
if err := repo.UpdateProject(ctx, bad); err == nil {
|
||
t.Fatal("更新未校验项目")
|
||
}
|
||
})
|
||
t.Run("update_and_order", func(t *testing.T) {
|
||
before, err := repo.GetProject(ctx, item.ID)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
item.Name = "更新后的项目"
|
||
item.CreatedAt = time.Time{}
|
||
for round := 0; round < 2; round++ {
|
||
if err := repo.UpdateProject(ctx, item); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
}
|
||
loaded, err := repo.GetProject(ctx, item.ID)
|
||
if err != nil || loaded.Name != item.Name || !loaded.CreatedAt.Equal(before.CreatedAt) || loaded.UpdatedAt.Before(before.UpdatedAt) {
|
||
t.Fatalf("更新改变了创建时间或丢失内容：%+v，%v", loaded, err)
|
||
}
|
||
if err := repo.CreateProject(ctx, domain.Project{ID: "retained", Name: item.Name}); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
items, err := repo.ListProjects(ctx)
|
||
if err != nil || len(items) != 3 || items[0].ID != domain.DefaultProjectID {
|
||
t.Fatalf("项目列表必须默认项目在前：%+v，%v", items, err)
|
||
}
|
||
initial.Name = "调整后的默认项目"
|
||
if err := repo.UpdateProject(ctx, initial); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
})
|
||
t.Run("cancelled_context", func(t *testing.T) {
|
||
cancelled, cancel := context.WithCancel(ctx)
|
||
cancel()
|
||
if err := repo.CreateProject(cancelled, domain.Project{ID: "cancelled", Name: "不应写入"}); !errors.Is(err, context.Canceled) {
|
||
t.Fatalf("取消后创建未被拒绝：%v", err)
|
||
}
|
||
if _, err := repo.ListProjects(cancelled); !errors.Is(err, context.Canceled) {
|
||
t.Fatalf("取消后查询未被拒绝：%v", err)
|
||
}
|
||
if err := repo.DeleteProject(cancelled, item.ID); !errors.Is(err, context.Canceled) {
|
||
t.Fatalf("取消后删除未被拒绝：%v", err)
|
||
}
|
||
})
|
||
t.Run("missing_and_injection", func(t *testing.T) {
|
||
for _, id := range []string{"missing", "' OR 1=1 --"} {
|
||
if _, err := repo.GetProject(ctx, id); !errors.Is(err, sql.ErrNoRows) {
|
||
t.Fatalf("查询不存在项目应保留未找到语义：%v", err)
|
||
}
|
||
if err := repo.DeleteProject(ctx, id); !errors.Is(err, sql.ErrNoRows) {
|
||
t.Fatalf("删除不存在项目应保留未找到语义：%v", err)
|
||
}
|
||
}
|
||
if err := repo.UpdateProject(ctx, domain.Project{ID: "missing", Name: "不存在"}); !errors.Is(err, sql.ErrNoRows) {
|
||
t.Fatalf("更新不存在项目应保留未找到语义：%v", err)
|
||
}
|
||
})
|
||
t.Run("delete", func(t *testing.T) {
|
||
if err := repo.DeleteProject(ctx, item.ID); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if _, err := repo.GetProject(ctx, item.ID); !errors.Is(err, sql.ErrNoRows) {
|
||
t.Fatalf("删除后仍能读取项目：%v", err)
|
||
}
|
||
items, err := repo.ListProjects(ctx)
|
||
if err != nil || len(items) != 2 || items[1].ID != "retained" {
|
||
t.Fatalf("删除项目影响了其他项目：%+v，%v", items, err)
|
||
}
|
||
})
|
||
}
|
||
|
||
func TestSQLProjectRepositoryNilDatabase(t *testing.T) {
|
||
for _, repo := range []*SQLProjectRepository{nil, NewSQLProjectRepository(nil)} {
|
||
if err := repo.CreateProject(t.Context(), domain.Project{}); err == nil {
|
||
t.Fatal("创建没有拒绝空数据库")
|
||
}
|
||
if _, err := repo.GetProject(t.Context(), "a"); err == nil {
|
||
t.Fatal("查询没有拒绝空数据库")
|
||
}
|
||
if _, err := repo.ListProjects(t.Context()); err == nil {
|
||
t.Fatal("列表没有拒绝空数据库")
|
||
}
|
||
if err := repo.UpdateProject(t.Context(), domain.Project{}); err == nil {
|
||
t.Fatal("更新没有拒绝空数据库")
|
||
}
|
||
if err := repo.DeleteProject(t.Context(), "a"); err == nil {
|
||
t.Fatal("删除没有拒绝空数据库")
|
||
}
|
||
}
|
||
}
|