181 lines
8.4 KiB
Go
181 lines
8.4 KiB
Go
//go:build integration
|
||
|
||
// 集成测试：实体键上的部分唯一索引必须阻止「同一实体两行活着」。
|
||
//
|
||
// 注意：部分索引（WHERE is_deleted = 0）在 DB 层仍允许「软删后再插入同键」；
|
||
// 但入库服务层（processRunway / processStreet）现已在查重阶段跳过重爬已软删的实体，
|
||
// 从而达成「删除即永久」。本测试只验证索引这一层的行为。
|
||
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("同一实体键的第二行应被唯一索引拒绝，实际插入成功")
|
||
}
|
||
|
||
// 软删首行后，部分索引（WHERE is_deleted = 0）不再占用该键，DB 层仍允许再插入。
|
||
// 「删除即永久」由入库服务层（RunwayEntityState 命中 isDeleted 跳过重爬）保证，而非本索引。
|
||
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 + title（2026-09-22-04 细化后）。
|
||
//
|
||
// 为什么键里要有 title：同城同年可以有多个专题（开发库实例 London 2027 Day 2 / Day 3），
|
||
// title 是数据里唯一能区分它们的字段。因此：
|
||
// ① 同城同年**同标题**的第二行必须被拒（实体键仍具备唯一性）；
|
||
// ② 同城同年**不同标题**的第二行必须允许（对应「Day 2 / Day 3 分成两个文章」）。
|
||
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")
|
||
// 04 是替换索引：先 DROP 掉 03 建的 (city, year)，再按 (city, year, title) 重建同名索引。
|
||
// 两个都执行才真实还原「迁移链走完」的库结构，也才能验证旧键确实不再生效。
|
||
applyMigration(t, db, "2026-09-22-04-street-entity-key-title.sql")
|
||
repo := NewIngestRepository(db)
|
||
ctx := context.Background()
|
||
|
||
const city = "UniqTestCity"
|
||
const year = 1994
|
||
const titleDay2 = "UniqTest Spring 1994 Day 2"
|
||
const titleDay3 = "UniqTest Spring 1994 Day 3"
|
||
|
||
snap := &model.StreetSnap{JobID: 1, Title: titleDay2, 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: titleDay2, 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("同城同年同标题的第二行应被唯一索引拒绝，实际插入成功")
|
||
}
|
||
|
||
// ② 同城同年不同标题：不同专题，必须允许共存。
|
||
// 这一条是本任务的核心：若还原成 (city, year) 的旧键，这里会插入失败。
|
||
other := &model.StreetSnap{JobID: 1, Title: titleDay3, Year: year, City: city, Status: model.StatusPending}
|
||
otherID, err := repo.CreateStreetSnapWithImages(ctx, other, []model.StreetSnapImage{{Image: "uniq-c.jpg", SortOrder: 1}})
|
||
if err != nil {
|
||
t.Fatalf("同城同年不同标题应可插入，实际失败: %v", err)
|
||
}
|
||
t.Cleanup(func() {
|
||
db.Exec("DELETE FROM street_snap_images WHERE snap_id = ?", otherID)
|
||
db.Exec("DELETE FROM street_snaps WHERE id = ?", otherID)
|
||
})
|
||
|
||
// 软删首行后，实体键不再占用（与 is_deleted = 0 的部分索引口径一致）。
|
||
if err := db.Model(&model.StreetSnap{}).Where("id = ?", id).Update("is_deleted", 1).Error; err != nil {
|
||
t.Fatalf("软删失败: %v", err)
|
||
}
|
||
again := &model.StreetSnap{JobID: 1, Title: titleDay2, Year: year, City: city, Status: model.StatusPending}
|
||
newID, err := repo.CreateStreetSnapWithImages(ctx, again, []model.StreetSnapImage{{Image: "uniq-d.jpg", SortOrder: 1}})
|
||
if err != nil {
|
||
t.Fatalf("软删后应可再插入同实体键: %v", err)
|
||
}
|
||
t.Cleanup(func() {
|
||
db.Exec("DELETE FROM street_snap_images WHERE snap_id = ?", newID)
|
||
db.Exec("DELETE FROM street_snaps WHERE id = ?", newID)
|
||
})
|
||
}
|
||
|
||
// TestStreetEntityStateMatchesIndexKey 覆盖 StreetSnapEntityState 的查重口径：
|
||
// 它必须与唯一索引 uq_ss_entity 的表达式 COALESCE(title, '') 完全一致，
|
||
// 否则查重会比索引更严（漏判 → 写入注定被索引拒绝的行）或更松（重复内容被当成新实体）。
|
||
func TestStreetEntityStateMatchesIndexKey(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")
|
||
applyMigration(t, db, "2026-09-22-04-street-entity-key-title.sql")
|
||
repo := NewIngestRepository(db)
|
||
ctx := context.Background()
|
||
|
||
const city = "EntityStateTestCity"
|
||
const year = 1996
|
||
|
||
// 种子行：title = Day 2（经 ORM 写入，title 为普通空串语义）。
|
||
snap := &model.StreetSnap{JobID: 1, Title: "EntityState Day 2", Year: year, City: city, Status: model.StatusPending}
|
||
id, err := repo.CreateStreetSnapWithImages(ctx, snap, nil)
|
||
if err != nil {
|
||
t.Fatalf("种子行应能插入: %v", err)
|
||
}
|
||
t.Cleanup(func() { db.Exec("DELETE FROM street_snaps WHERE id = ?", id) })
|
||
|
||
// ① 同城同年**不同 title** → 不命中（同城同年不同专题是两个实体）。
|
||
if _, _, found, _, err := repo.StreetSnapEntityState(ctx, city, year, "EntityState Day 3"); err != nil {
|
||
t.Fatalf("查重出错: %v", err)
|
||
} else if found {
|
||
t.Fatalf("不同 title 不应命中既有行，实际命中")
|
||
}
|
||
|
||
// ② 同城同年**同 title** → 命中，且带出正确 id / status。
|
||
hitID, status, found, _, err := repo.StreetSnapEntityState(ctx, city, year, "EntityState Day 2")
|
||
if err != nil {
|
||
t.Fatalf("查重出错: %v", err)
|
||
}
|
||
if !found || hitID != id || status != model.StatusPending {
|
||
t.Fatalf("同 title 应命中 id=%d status=pending，实际 found=%v id=%d status=%q", id, found, hitID, status)
|
||
}
|
||
|
||
// ③ title 为 NULL 的历史行 + 传入空串 title → 必须命中。
|
||
// 这是 COALESCE(title, '') 与裸 title = ? 唯一的分歧点：裸写法下 NULL 永不等于空串，
|
||
// 这里会误判「未命中」并插入一行与 NULL 行同键（COALESCE 后同为 ''）的记录，被索引拒绝。
|
||
// 用原生 SQL 插入，因为 GORM 的 string 字段会写空串而非 NULL。
|
||
var nullID uint32
|
||
if err := db.Raw(
|
||
"INSERT INTO street_snaps (title, year, city, is_deleted, status, created_at, updated_at) "+
|
||
"VALUES (NULL, ?, ?, 0, ?, 0, 0) RETURNING id",
|
||
year, city, model.StatusPending,
|
||
).Scan(&nullID).Error; err != nil {
|
||
t.Fatalf("插入 NULL title 行失败: %v", err)
|
||
}
|
||
t.Cleanup(func() { db.Exec("DELETE FROM street_snaps WHERE id = ?", nullID) })
|
||
|
||
nid, _, nfound, _, err := repo.StreetSnapEntityState(ctx, city, year, "")
|
||
if err != nil {
|
||
t.Fatalf("查重出错: %v", err)
|
||
}
|
||
if !nfound || nid != nullID {
|
||
t.Fatalf("NULL title 行 + 空串应命中 id=%d，实际 found=%v id=%d", nullID, nfound, nid)
|
||
}
|
||
}
|