test(download): cover AllAnime referer requests

Author: Vittuu

internal/player/download_regression_test.go

@@ -19,6 +19,12 @@ import (
 	"github.com/stretchr/testify/require"
 )
 
+type roundTripFunc func(*http.Request) (*http.Response, error)
+
+func (fn roundTripFunc) RoundTrip(req *http.Request) (*http.Response, error) {
+	return fn(req)
+}
+
 func TestDownloadDirectHTTPWithClientDownloadsMockVideoAndTracksProgress(t *testing.T) {
 	payload := bytes.Repeat([]byte("goanime-video-payload"), 32*1024)
 	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
@@ -52,6 +58,117 @@ func TestDownloadDirectHTTPWithClientDownloadsMockVideoAndTracksProgress(t *test
 	assert.Equal(t, int64(len(payload)), received)
 }
 
+func TestDownloadPartAddsAllAnimeReferer(t *testing.T) {
+	payload := []byte("goanime")
+	var gotReferer string
+	var gotRange string
+
+	client := &http.Client{
+		Transport: roundTripFunc(func(req *http.Request) (*http.Response, error) {
+			gotReferer = req.Header.Get("Referer")
+			gotRange = req.Header.Get("Range")
+
+			return &http.Response{
+				StatusCode: http.StatusPartialContent,
+				Status:     "206 Partial Content",
+				Header:     make(http.Header),
+				Body:       io.NopCloser(bytes.NewReader(payload)),
+				Request:    req,
+			}, nil
+		}),
+	}
+
+	outPath := filepath.Join(t.TempDir(), "episode.mp4")
+	err := downloadPart(
+		"https://allanime.day/video/episode.mp4",
+		0,
+		int64(len(payload)-1),
+		0,
+		client,
+		outPath,
+		&model{},
+	)
+	require.NoError(t, err)
+	assert.Equal(t, "https://allanime.to", gotReferer)
+	assert.Equal(t, "bytes=0-6", gotRange)
+
+	got, err := os.ReadFile(outPath + ".part0")
+	require.NoError(t, err)
+	assert.Equal(t, payload, got)
+}
+
+func TestGetContentLengthAddsAllAnimeReferer(t *testing.T) {
+	const contentLength = "12345"
+	var gotMethod string
+	var gotReferer string
+
+	client := &http.Client{
+		Transport: roundTripFunc(func(req *http.Request) (*http.Response, error) {
+			gotMethod = req.Method
+			gotReferer = req.Header.Get("Referer")
+
+			header := make(http.Header)
+			header.Set("Content-Length", contentLength)
+
+			return &http.Response{
+				StatusCode: http.StatusOK,
+				Status:     "200 OK",
+				Header:     header,
+				Body:       io.NopCloser(strings.NewReader("")),
+				Request:    req,
+			}, nil
+		}),
+	}
+
+	got, err := getContentLength("https://allanime.day/video/episode.mp4", client)
+	require.NoError(t, err)
+	assert.Equal(t, int64(12345), got)
+	assert.Equal(t, http.MethodHead, gotMethod)
+	assert.Equal(t, "https://allanime.to", gotReferer)
+}
+
+func TestGetContentLengthFallbackKeepsAllAnimeReferer(t *testing.T) {
+	var gotRequests []string
+	var gotReferers []string
+	var gotRanges []string
+
+	client := &http.Client{
+		Transport: roundTripFunc(func(req *http.Request) (*http.Response, error) {
+			gotRequests = append(gotRequests, req.Method)
+			gotReferers = append(gotReferers, req.Header.Get("Referer"))
+			gotRanges = append(gotRanges, req.Header.Get("Range"))
+
+			if req.Method == http.MethodHead {
+				return &http.Response{
+					StatusCode: http.StatusMethodNotAllowed,
+					Status:     "405 Method Not Allowed",
+					Header:     make(http.Header),
+					Body:       io.NopCloser(strings.NewReader("")),
+					Request:    req,
+				}, nil
+			}
+
+			header := make(http.Header)
+			header.Set("Content-Length", "1")
+
+			return &http.Response{
+				StatusCode: http.StatusPartialContent,
+				Status:     "206 Partial Content",
+				Header:     header,
+				Body:       io.NopCloser(strings.NewReader("x")),
+				Request:    req,
+			}, nil
+		}),
+	}
+
+	got, err := getContentLength("https://allanime.pro/video/episode.mp4", client)
+	require.NoError(t, err)
+	assert.Equal(t, int64(1), got)
+	assert.Equal(t, []string{http.MethodHead, http.MethodGet}, gotRequests)
+	assert.Equal(t, []string{"https://allanime.to", "https://allanime.to"}, gotReferers)
+	assert.Equal(t, []string{"", "bytes=0-0"}, gotRanges)
+}
+
 func TestDownloadDirectHTTPWithClientReturnsHTTPStatusErrorFromMockCDN(t *testing.T) {
 	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
 		http.Error(w, "missing object", http.StatusNotFound)