143 lines
4.1 KiB
Go
143 lines
4.1 KiB
Go
package preview
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"convert/internal/media"
|
|
)
|
|
|
|
func TestPreviewServer_RangeThumbAndProxy(t *testing.T) {
|
|
bins := media.FindBinaries("", "")
|
|
if bins.FFmpeg == "" || bins.FFprobe == "" {
|
|
t.Skip("ffmpeg/ffprobe not installed")
|
|
}
|
|
|
|
tmpDir := t.TempDir()
|
|
src := filepath.Join(tmpDir, "clip.mkv")
|
|
gen := exec.Command(bins.FFmpeg,
|
|
"-hide_banner", "-loglevel", "error",
|
|
"-f", "lavfi", "-i", "testsrc=duration=2:size=320x180:rate=30",
|
|
"-f", "lavfi", "-i", "sine=frequency=440:duration=2",
|
|
"-c:v", "libx264", "-preset", "ultrafast", "-c:a", "aac", "-y", src,
|
|
)
|
|
if out, err := gen.CombinedOutput(); err != nil {
|
|
t.Fatalf("failed generating sample mkv: %v (%s)", err, string(out))
|
|
}
|
|
|
|
info, err := media.Probe(bins.FFprobe, src)
|
|
if err != nil {
|
|
t.Fatalf("probe failed: %v", err)
|
|
}
|
|
|
|
srv, err := Start(bins.FFmpeg)
|
|
if err != nil {
|
|
t.Fatalf("preview server start failed: %v", err)
|
|
}
|
|
defer srv.Close()
|
|
|
|
id := srv.Register(info)
|
|
|
|
// 1. Test Range request on /video
|
|
req, _ := http.NewRequest("GET", srv.BaseURL()+"/video?id="+id, nil)
|
|
req.Header.Set("Range", "bytes=0-127")
|
|
resp, err := http.DefaultClient.Do(req)
|
|
if err != nil {
|
|
t.Fatalf("GET /video failed: %v", err)
|
|
}
|
|
resp.Body.Close()
|
|
if resp.StatusCode != http.StatusPartialContent {
|
|
t.Fatalf("expected 206 Partial Content, got %d", resp.StatusCode)
|
|
}
|
|
|
|
// 2. Test /thumb JPEG generation
|
|
tResp, err := http.Get(srv.BaseURL() + "/thumb?id=" + id + "&t=0.5")
|
|
if err != nil {
|
|
t.Fatalf("GET /thumb failed: %v", err)
|
|
}
|
|
tBody, _ := io.ReadAll(tResp.Body)
|
|
tResp.Body.Close()
|
|
if tResp.StatusCode != http.StatusOK || len(tBody) < 100 || tBody[0] != 0xFF || tBody[1] != 0xD8 {
|
|
t.Fatalf("expected valid JPEG thumbnail, status=%d len=%d", tResp.StatusCode, len(tBody))
|
|
}
|
|
|
|
// 3. Test /proxy MKV -> MP4 remux
|
|
pResp, err := http.Get(srv.BaseURL() + "/proxy?id=" + id + "&merge=0")
|
|
if err != nil {
|
|
t.Fatalf("GET /proxy failed: %v", err)
|
|
}
|
|
pBody, _ := io.ReadAll(pResp.Body)
|
|
pResp.Body.Close()
|
|
if pResp.StatusCode != http.StatusOK || len(pBody) < 500 {
|
|
t.Fatalf("expected valid MP4 proxy, status=%d len=%d", pResp.StatusCode, len(pBody))
|
|
}
|
|
|
|
// 4. Test unknown ID returns 404
|
|
nfResp, err := http.Get(srv.BaseURL() + "/video?id=nonexistent")
|
|
if err != nil {
|
|
t.Fatalf("GET /video 404 failed: %v", err)
|
|
}
|
|
nfResp.Body.Close()
|
|
if nfResp.StatusCode != http.StatusNotFound {
|
|
t.Fatalf("expected 404, got %d", nfResp.StatusCode)
|
|
}
|
|
}
|
|
|
|
func TestPreviewServer_CacheInvalidationAndOddWidthProxy(t *testing.T) {
|
|
bins := media.FindBinaries("", "")
|
|
if bins.FFmpeg == "" || bins.FFprobe == "" {
|
|
t.Skip("ffmpeg/ffprobe not installed")
|
|
}
|
|
|
|
tmpDir := t.TempDir()
|
|
src := filepath.Join(tmpDir, "odd.mkv")
|
|
// Odd width (321x180) vp8 source forces libx264 transcode in /proxy
|
|
gen := exec.Command(bins.FFmpeg,
|
|
"-hide_banner", "-loglevel", "error",
|
|
"-f", "lavfi", "-i", "testsrc=duration=1:size=321x180:rate=30",
|
|
"-c:v", "libvpx", "-y", src,
|
|
)
|
|
if out, err := gen.CombinedOutput(); err != nil {
|
|
t.Fatalf("failed generating odd-width mkv: %v (%s)", err, string(out))
|
|
}
|
|
|
|
info, err := media.Probe(bins.FFprobe, src)
|
|
if err != nil {
|
|
t.Fatalf("probe failed: %v", err)
|
|
}
|
|
|
|
srv, err := Start(bins.FFmpeg)
|
|
if err != nil {
|
|
t.Fatalf("start failed: %v", err)
|
|
}
|
|
defer srv.Close()
|
|
|
|
id1 := srv.Register(info)
|
|
|
|
// /proxy must succeed even on odd width (321 -> 320)
|
|
pResp, err := http.Get(srv.BaseURL() + "/proxy?id=" + id1 + "&merge=0")
|
|
if err != nil {
|
|
t.Fatalf("GET /proxy failed: %v", err)
|
|
}
|
|
pBody, _ := io.ReadAll(pResp.Body)
|
|
pResp.Body.Close()
|
|
if pResp.StatusCode != http.StatusOK || len(pBody) < 500 {
|
|
t.Fatalf("expected valid MP4 proxy for odd-width source, status=%d len=%d", pResp.StatusCode, len(pBody))
|
|
}
|
|
|
|
// Overwrite same path with different size -> Register must yield a new ID
|
|
if err := os.WriteFile(src, []byte("different-content-size"), 0o644); err != nil {
|
|
t.Fatalf("overwrite failed: %v", err)
|
|
}
|
|
info2 := *info
|
|
info2.SizeBytes = 22
|
|
id2 := srv.Register(&info2)
|
|
if id1 == id2 {
|
|
t.Fatalf("expected different videoId after file modification at same path, both were %q", id1)
|
|
}
|
|
}
|