69 lines
2.3 KiB
Go
69 lines
2.3 KiB
Go
package postgres
|
||
|
||
import (
|
||
"context"
|
||
"testing"
|
||
|
||
"telesrv/internal/domain"
|
||
)
|
||
|
||
// TestWebPageStoreRoundTrip 验证 web_pages 的 upsert/读回（snapshot JSONB 字节级往返）与
|
||
// 按 url_hash 去重覆盖。需 TELESRV_TEST_POSTGRES_DSN。
|
||
func TestWebPageStoreRoundTrip(t *testing.T) {
|
||
pool := testPool(t)
|
||
ctx := context.Background()
|
||
s := NewMediaStore(pool)
|
||
|
||
const urlHash = int64(9100000000000000777)
|
||
_, _ = pool.Exec(ctx, `DELETE FROM web_pages WHERE url_hash = $1`, urlHash)
|
||
t.Cleanup(func() {
|
||
_, _ = pool.Exec(context.Background(), `DELETE FROM web_pages WHERE url_hash = $1`, urlHash)
|
||
})
|
||
|
||
// miss
|
||
if _, _, found, err := s.GetWebPageByURLHash(ctx, urlHash); err != nil || found {
|
||
t.Fatalf("expected miss, got found=%v err=%v", found, err)
|
||
}
|
||
|
||
page := domain.MessageWebPage{
|
||
State: domain.MessageWebPageStateDone,
|
||
ID: urlHash,
|
||
URL: "https://example.com/a",
|
||
DisplayURL: "example.com",
|
||
Title: "Title",
|
||
Description: "Desc",
|
||
SiteName: "Site",
|
||
Type: "article",
|
||
Hash: 123,
|
||
Photo: &domain.Photo{ID: 5, AccessHash: 6, DCID: 2, Sizes: []domain.PhotoSize{{Kind: domain.PhotoSizeKindDefault, Type: "x", W: 320, H: 200, Size: 1024}}},
|
||
}
|
||
if err := s.PutWebPage(ctx, urlHash, page, 1700000000); err != nil {
|
||
t.Fatalf("PutWebPage: %v", err)
|
||
}
|
||
got, refreshedAt, found, err := s.GetWebPageByURLHash(ctx, urlHash)
|
||
if err != nil || !found {
|
||
t.Fatalf("GetWebPageByURLHash: found=%v err=%v", found, err)
|
||
}
|
||
if got.State != domain.MessageWebPageStateDone || got.Title != "Title" || got.Photo == nil || got.Photo.ID != 5 {
|
||
t.Fatalf("round-trip mismatch: %+v", got)
|
||
}
|
||
if refreshedAt != 1700000000 {
|
||
t.Fatalf("refreshedAt = %d, want 1700000000", refreshedAt)
|
||
}
|
||
|
||
// upsert 覆盖为 empty 态。
|
||
empty := domain.MessageWebPage{State: domain.MessageWebPageStateEmpty, ID: urlHash, URL: "https://example.com/a"}
|
||
if err := s.PutWebPage(ctx, urlHash, empty, 1700000500); err != nil {
|
||
t.Fatalf("PutWebPage(empty): %v", err)
|
||
}
|
||
got2, refreshed2, _, err := s.GetWebPageByURLHash(ctx, urlHash)
|
||
if err != nil {
|
||
t.Fatalf("GetWebPageByURLHash after upsert: %v", err)
|
||
}
|
||
if got2.State != domain.MessageWebPageStateEmpty || got2.Photo != nil {
|
||
t.Fatalf("upsert did not overwrite: %+v", got2)
|
||
}
|
||
if refreshed2 != 1700000500 {
|
||
t.Fatalf("refreshed_at not updated on upsert: %d", refreshed2)
|
||
}
|
||
}
|