mirror of
https://github.com/truewhile/MeBox.git
synced 2026-09-28 11:16:37 +08:00
truewhile
106 lines
3.2 KiB
Go
106 lines
3.2 KiB
Go
package repository
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/truewhile/MeBox/internal/model"
|
|
)
|
|
|
|
func TestMediaProbeUpsertReplacesAndGetReturnsLatest(t *testing.T) {
|
|
repos := newSegmentTestRepos(t)
|
|
ctx := t.Context()
|
|
|
|
first := &model.MediaProbe{
|
|
MediaID: "m-1", Signature: "sig-1", Source: "local",
|
|
Container: "matroska,webm", DurationSec: 1451, ChapterCount: 2,
|
|
Payload: `{"container":"matroska,webm"}`, ProbedAt: time.Now(),
|
|
}
|
|
if err := repos.MediaProbe.Upsert(ctx, first); err != nil {
|
|
t.Fatalf("upsert #1: %v", err)
|
|
}
|
|
second := &model.MediaProbe{
|
|
MediaID: "m-1", Signature: "sig-2", Source: "strm",
|
|
Container: "mp4", DurationSec: 900, ChapterCount: 0,
|
|
Payload: `{"container":"mp4"}`, ProbedAt: time.Now(),
|
|
}
|
|
if err := repos.MediaProbe.Upsert(ctx, second); err != nil {
|
|
t.Fatalf("upsert #2: %v", err)
|
|
}
|
|
|
|
got, err := repos.MediaProbe.Get(ctx, "m-1")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if got == nil {
|
|
t.Fatal("probe row missing")
|
|
}
|
|
if got.Container != "mp4" || got.DurationSec != 900 || got.Signature != "sig-2" || got.Payload != `{"container":"mp4"}` {
|
|
t.Fatalf("probe = %#v, want the second upsert to win", got)
|
|
}
|
|
// 重复探测不能累积重复行：media_id 是唯一索引。
|
|
var count int64
|
|
if err := repos.DB.Model(&model.MediaProbe{}).Where("media_id = ?", "m-1").Count(&count).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if count != 1 {
|
|
t.Fatalf("rows = %d, want 1", count)
|
|
}
|
|
}
|
|
|
|
func TestMediaProbeGetReturnsNilWhenMissing(t *testing.T) {
|
|
repos := newSegmentTestRepos(t)
|
|
got, err := repos.MediaProbe.Get(t.Context(), "nope")
|
|
if err != nil {
|
|
t.Fatalf("Get: %v", err)
|
|
}
|
|
if got != nil {
|
|
t.Fatalf("probe = %#v, want nil", got)
|
|
}
|
|
}
|
|
|
|
func TestMediaProbeMarkFailureKeepsPreviousPayload(t *testing.T) {
|
|
repos := newSegmentTestRepos(t)
|
|
ctx := t.Context()
|
|
|
|
if err := repos.MediaProbe.Upsert(ctx, &model.MediaProbe{
|
|
MediaID: "m-1", Container: "matroska,webm", DurationSec: 1451,
|
|
ChapterCount: 2, Payload: `{"container":"matroska,webm"}`, ProbedAt: time.Now(),
|
|
}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := repos.MediaProbe.MarkFailure(ctx, "m-1", "ffprobe full: exit status 1", time.Now()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
got, err := repos.MediaProbe.Get(ctx, "m-1")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if got == nil || got.LastError != "ffprobe full: exit status 1" {
|
|
t.Fatalf("probe = %#v, want the recorded failure", got)
|
|
}
|
|
// 一次失败的重探不该把上一次成功拿到的媒体信息抹掉。
|
|
if got.Container != "matroska,webm" || got.DurationSec != 1451 || got.Payload == "" || got.ChapterCount != 2 {
|
|
t.Fatalf("a failed re-probe wiped the previous summary: %#v", got)
|
|
}
|
|
}
|
|
|
|
func TestMediaProbeMarkFailureInsertsRowWhenAbsent(t *testing.T) {
|
|
repos := newSegmentTestRepos(t)
|
|
ctx := t.Context()
|
|
|
|
// 没有历史成功记录时，失败也必须落一行：否则冷却期没有时间戳，
|
|
// 每次播放都会重跑一次注定失败的探测。
|
|
if err := repos.MediaProbe.MarkFailure(ctx, "m-2", "boom", time.Now()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
got, err := repos.MediaProbe.Get(ctx, "m-2")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if got == nil || got.LastError != "boom" || got.ProbedAt.IsZero() {
|
|
t.Fatalf("probe = %#v, want a failure row with a timestamp", got)
|
|
}
|
|
}
|