cmsp/internal/store/video_test.go

package store

import (
	"os"
	"path/filepath"
	"testing"
)

func TestReplaceVideos按商品全量替换(t *testing.T) {
	s := newTestStore(t)
	first := []Video{
		{SourceItem: "a", SourceURL: "https://example.invalid/a.mp4", Status: VideoStatusPending},
		{SourceItem: "b", SourceURL: "https://example.invalid/b.mp4", Status: VideoStatusDownloaded},
		{SourceItem: "c", SourceURL: "https://example.invalid/c.mp4", Status: VideoStatusFailed},
	}
	if err := s.ReplaceVideos("product-1", first, "2026-09-03 10:00:00"); err != nil {
		t.Fatalf("首次写入 3 条视频失败：%v", err)
	}
	second := []Video{{SourceItem: "d", SourceURL: "https://example.invalid/d.mp4", Status: VideoStatusDownloaded}}
	if err := s.ReplaceVideos("product-1", second, "2026-09-03 11:00:00"); err != nil {
		t.Fatalf("用 1 条视频替换失败：%v", err)
	}
	got, err := s.ListVideos("product-1")
	if err != nil {
		t.Fatalf("读取替换后视频失败：%v", err)
	}
	if len(got) != 1 || got[0].SourceItem != "d" {
		t.Fatalf("全量替换后应只剩 d，实际：%+v", got)
	}
	if got[0].ProductID != "product-1" || got[0].CreatedAt != "2026-09-03 11:00:00" {
		t.Fatalf("商品 ID 或创建时间没有按参数写入：%+v", got[0])
	}
}

func Test取第一个存在的已下载视频并写回上传状态(t *testing.T) {
	s := newTestStore(t)
	dir := t.TempDir()
	missing := filepath.Join(dir, "missing.mp4")
	available := filepath.Join(dir, "available.mp4")
	if err := os.WriteFile(available, []byte("fake-mp4"), 0o600); err != nil {
		t.Fatalf("准备本地视频失败：%v", err)
	}
	if err := s.ReplaceVideos("商品-1", []Video{
		{LocalPath: missing, Status: VideoStatusDownloaded},
		{LocalPath: available, Status: VideoStatusDownloaded},
	}, "2026-09-03 10:00:00"); err != nil {
		t.Fatalf("写入视频失败：%v", err)
	}
	video, found, err := s.FirstUploadableVideo("商品-1")
	if err != nil || !found || video.LocalPath != available {
		t.Fatalf("应跳过不存在文件并取第一个可上传视频，video=%+v found=%v err=%v", video, found, err)
	}
	if err := s.MarkVideoUploaded(video.ID, "https://cos.example.invalid/video.mp4"); err != nil {
		t.Fatalf("写回上传状态失败：%v", err)
	}
	items, err := s.ListVideos("商品-1")
	if err != nil {
		t.Fatalf("读取视频失败：%v", err)
	}
	if items[1].Status != VideoStatusUploaded || items[1].RemoteURL != "https://cos.example.invalid/video.mp4" {
		t.Fatalf("上传状态或远端地址没有写回：%+v", items[1])
	}
}

func Test按本地路径补写上传记录不删除其它视频(t *testing.T) {
	s := newTestStore(t)
	if err := s.ReplaceVideos("商品-1", []Video{
		{LocalPath: "D:/videos/其它.mp4", SourceItem: "淘宝-1", Status: VideoStatusDownloaded},
		{LocalPath: "D:/videos/目标.mp4", SourceItem: "淘宝-2", Status: VideoStatusDownloaded},
	}, "2026-09-03 10:00:00"); err != nil {
		t.Fatalf("准备已有视频记录失败：%v", err)
	}
	if err := s.UpsertUploadedVideo("商品-1", "D:/videos/目标.mp4", 123, "https://cos.example.invalid/target.mp4", "2026-09-03 11:00:00"); err != nil {
		t.Fatalf("更新同路径上传记录失败：%v", err)
	}
	if err := s.UpsertUploadedVideo("商品-1", "D:/videos/新增.mp4", 456, "https://cos.example.invalid/new.mp4", "2026-09-03 11:00:00"); err != nil {
		t.Fatalf("新增上传记录失败：%v", err)
	}
	items, err := s.ListVideos("商品-1")
	if err != nil {
		t.Fatalf("读取视频记录失败：%v", err)
	}
	if len(items) != 3 {
		t.Fatalf("补写不能删除其它视频或重复同路径，实际 %d 条：%+v", len(items), items)
	}
	var target, added Video
	for _, item := range items {
		switch item.LocalPath {
		case "D:/videos/目标.mp4":
			target = item
		case "D:/videos/新增.mp4":
			added = item
		}
	}
	if target.Status != VideoStatusUploaded || target.FileSize != 123 || target.RemoteURL == "" || target.SourceItem != "淘宝-2" {
		t.Fatalf("同路径记录未正确更新：%+v", target)
	}
	if added.Status != VideoStatusUploaded || added.FileSize != 456 || added.RemoteURL == "" || added.SourceItem != "" {
		t.Fatalf("新增记录未按磁盘来源写入：%+v", added)
	}
}