mirror of
https://github.com/truewhile/MeBox.git
synced 2026-11-12 19:34:24 +08:00
truewhile
34 lines
1.2 KiB
Go
34 lines
1.2 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestCloud115HLSChildRejectsMismatchedSessionMediaBeforeFetch(t *testing.T) {
|
|
fetched := false
|
|
proxy := &Cloud115HLSProxy{sessions: map[string]*cloud115HLSSession{}}
|
|
proxy.client = &http.Client{Transport: roundTripFunc(func(req *http.Request) (*http.Response, error) {
|
|
fetched = true
|
|
return &http.Response{StatusCode: http.StatusOK, Header: http.Header{"Content-Type": []string{"video/mp2t"}}, Body: io.NopCloser(strings.NewReader("private-segment")), Request: req}, nil
|
|
})}
|
|
proxy.sessions["session-b"] = &cloud115HLSSession{
|
|
ID: "session-b", MediaID: "media-b", ExpiresAt: time.Now().Add(time.Hour),
|
|
entries: map[string]string{"segment": "https://cpats01.115.com/seg.ts"},
|
|
}
|
|
for _, query := range []string{"?media_id=media-a", ""} {
|
|
fetched = false
|
|
req := httptest.NewRequest(http.MethodGet, "/api/cloud115/hls/session-b/segment"+query, nil)
|
|
rec := httptest.NewRecorder()
|
|
err := proxy.ServeChild(context.Background(), rec, req, "session-b", "segment")
|
|
if err == nil || fetched {
|
|
t.Errorf("query %q must reject the wrong/missing media before fetch; err=%v fetched=%v", query, err, fetched)
|
|
}
|
|
}
|
|
}
|