78 lines
3.0 KiB
Go
78 lines
3.0 KiB
Go
//go:build integration
|
||
|
||
// 集成测试：实体键上的部分唯一索引必须阻止「同一实体两行」，
|
||
// 同时不阻止「实体键相同但已软删」的行（与 RunwayEntityState 的 is_deleted = 0 口径一致）。
|
||
package repository
|
||
|
||
import (
|
||
"context"
|
||
"testing"
|
||
|
||
"fashionapi/internal/model"
|
||
)
|
||
|
||
// TestRunwayEntityKeyUnique 同一实体键的第二行应被拒绝；软删后可以再插入。
|
||
func TestRunwayEntityKeyUnique(t *testing.T) {
|
||
db := testDB(t)
|
||
applyMigration(t, db, "2026-09-22-01-single-table-publish.sql")
|
||
applyMigration(t, db, "2026-09-22-03-entity-key-unique.sql")
|
||
repo := NewIngestRepository(db)
|
||
ctx := context.Background()
|
||
|
||
const season = "SS95"
|
||
rw := newRunwayForIngest(1, 1, season, "rtw", "uniq-a.jpg")
|
||
id, err := repo.CreateRunwayWithImages(ctx, rw, newRunwayImages("uniq-a.jpg"))
|
||
if err != nil {
|
||
t.Fatalf("首行应能插入: %v", err)
|
||
}
|
||
t.Cleanup(func() {
|
||
db.Exec("DELETE FROM brand_runway_images WHERE runway_id = ?", id)
|
||
db.Exec("DELETE FROM brand_runways WHERE id = ?", id)
|
||
})
|
||
|
||
dup := newRunwayForIngest(1, 1, season, "rtw", "uniq-b.jpg")
|
||
if _, err := repo.CreateRunwayWithImages(ctx, dup, newRunwayImages("uniq-b.jpg")); err == nil {
|
||
t.Fatalf("同一实体键的第二行应被唯一索引拒绝，实际插入成功")
|
||
}
|
||
|
||
// 软删首行后，实体键不再占用（与 is_deleted = 0 的部分索引口径一致）。
|
||
if err := db.Model(&model.BrandRunway{}).Where("id = ?", id).Update("is_deleted", 1).Error; err != nil {
|
||
t.Fatalf("软删失败: %v", err)
|
||
}
|
||
again := newRunwayForIngest(1, 1, season, "rtw", "uniq-c.jpg")
|
||
newID, err := repo.CreateRunwayWithImages(ctx, again, newRunwayImages("uniq-c.jpg"))
|
||
if err != nil {
|
||
t.Fatalf("软删后应可再插入同实体键: %v", err)
|
||
}
|
||
t.Cleanup(func() {
|
||
db.Exec("DELETE FROM brand_runway_images WHERE runway_id = ?", newID)
|
||
db.Exec("DELETE FROM brand_runways WHERE id = ?", newID)
|
||
})
|
||
}
|
||
|
||
// TestStreetEntityKeyUnique 街拍侧同理（实体键 = city + year）。
|
||
func TestStreetEntityKeyUnique(t *testing.T) {
|
||
db := testDB(t)
|
||
applyMigration(t, db, "2026-09-22-01-single-table-publish.sql")
|
||
applyMigration(t, db, "2026-09-22-03-entity-key-unique.sql")
|
||
repo := NewIngestRepository(db)
|
||
ctx := context.Background()
|
||
|
||
const city = "UniqTestCity"
|
||
const year = 1994
|
||
snap := &model.StreetSnap{JobID: 1, Title: "uniq", Year: year, City: city, Status: model.StatusPending}
|
||
id, err := repo.CreateStreetSnapWithImages(ctx, snap, []model.StreetSnapImage{{Image: "uniq-a.jpg", SortOrder: 1}})
|
||
if err != nil {
|
||
t.Fatalf("首行应能插入: %v", err)
|
||
}
|
||
t.Cleanup(func() {
|
||
db.Exec("DELETE FROM street_snap_images WHERE snap_id = ?", id)
|
||
db.Exec("DELETE FROM street_snaps WHERE id = ?", id)
|
||
})
|
||
|
||
dup := &model.StreetSnap{JobID: 1, Title: "uniq", Year: year, City: city, Status: model.StatusPending}
|
||
if _, err := repo.CreateStreetSnapWithImages(ctx, dup, []model.StreetSnapImage{{Image: "uniq-b.jpg", SortOrder: 1}}); err == nil {
|
||
t.Fatalf("同一实体键的第二行应被唯一索引拒绝，实际插入成功")
|
||
}
|
||
}
|