mirror of
https://github.com/truewhile/MeBox.git
synced 2026-09-28 03:06:38 +08:00
truewhile
75 lines
2.4 KiB
Go
75 lines
2.4 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/truewhile/MeBox/internal/model"
|
|
)
|
|
|
|
// 同源转发必须原样保留 Range 语义：把 206 改写成 200 会让浏览器误判响应长度，
|
|
// 拖动进度条时反复重新拉流。
|
|
func TestProxyMediaDirectForwardsRangeAndStatus(t *testing.T) {
|
|
var gotRange string
|
|
var gotUA string
|
|
upstream := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
gotRange = r.Header.Get("Range")
|
|
gotUA = r.Header.Get("User-Agent")
|
|
w.Header().Set("Content-Type", "video/mp4")
|
|
w.Header().Set("Accept-Ranges", "bytes")
|
|
w.Header().Set("Content-Range", "bytes 10-19/100")
|
|
w.WriteHeader(http.StatusPartialContent)
|
|
_, _ = w.Write([]byte("0123456789"))
|
|
}))
|
|
defer upstream.Close()
|
|
|
|
svc := testStrmService(t)
|
|
media := &model.Media{STRMURL: upstream.URL + "/video.mp4"}
|
|
|
|
rec := httptest.NewRecorder()
|
|
req := httptest.NewRequest(http.MethodGet, "/api/stream/media-1?proxy=1", nil)
|
|
req.Header.Set("Range", "bytes=10-19")
|
|
req.Header.Set("User-Agent", "MeBoxTest/1.0")
|
|
|
|
if err := svc.ProxyMediaDirect(context.Background(), rec, req, media); err != nil {
|
|
t.Fatalf("ProxyMediaDirect: %v", err)
|
|
}
|
|
if rec.Code != http.StatusPartialContent {
|
|
t.Fatalf("status = %d, want 206", rec.Code)
|
|
}
|
|
if gotRange != "bytes=10-19" {
|
|
t.Fatalf("upstream Range = %q, want bytes=10-19", gotRange)
|
|
}
|
|
if gotUA != "MeBoxTest/1.0" {
|
|
t.Fatalf("upstream User-Agent = %q, want MeBoxTest/1.0", gotUA)
|
|
}
|
|
if rec.Header().Get("Content-Range") != "bytes 10-19/100" {
|
|
t.Fatalf("Content-Range = %q", rec.Header().Get("Content-Range"))
|
|
}
|
|
if rec.Body.String() != "0123456789" {
|
|
t.Fatalf("body = %q", rec.Body.String())
|
|
}
|
|
}
|
|
|
|
// 本地文件本身就是同源资源，不应该被代理（调用方按原静态文件逻辑处理）。
|
|
func TestProxyMediaDirectSkipsLocalFile(t *testing.T) {
|
|
svc := testStrmService(t)
|
|
dir := t.TempDir()
|
|
path := filepath.Join(dir, "movie.mkv")
|
|
writeFile(t, path, "not-a-video")
|
|
|
|
rec := httptest.NewRecorder()
|
|
req := httptest.NewRequest(http.MethodGet, "/api/stream/media-1?proxy=1", nil)
|
|
err := svc.ProxyMediaDirect(context.Background(), rec, req, &model.Media{Path: path})
|
|
if !errors.Is(err, ErrStrmProxyNotApplicable) {
|
|
t.Fatalf("err = %v, want ErrStrmProxyNotApplicable", err)
|
|
}
|
|
if rec.Body.Len() != 0 {
|
|
t.Fatalf("local file must not be proxied, body = %q", rec.Body.String())
|
|
}
|
|
}
|