mirror of
https://github.com/go-gitea/gitea.git
synced 2026-09-19 19:23:39 +09:00
c04802b6b3
Signed-off-by: silverwind <me@silverwind.io> Co-authored-by: Claude (Opus 5) <noreply@anthropic.com> Co-authored-by: wxiaoguang <wxiaoguang@gmail.com>
77 lines
3.1 KiB
Go
77 lines
3.1 KiB
Go
// Copyright 2024 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package integration
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
"strings"
|
|
"testing"
|
|
|
|
"gitea.dev/modules/setting"
|
|
"gitea.dev/modules/test"
|
|
"gitea.dev/routers"
|
|
"gitea.dev/routers/web"
|
|
"gitea.dev/tests"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestRepoDownloadArchive(t *testing.T) {
|
|
defer tests.PrepareTestEnv(t)()
|
|
defer test.MockVariableValue(&setting.EnableGzip, true)()
|
|
defer test.MockVariableValue(&web.GzipMinSize, 10)()
|
|
defer test.MockVariableValue(&testWebRoutes, routers.NormalRoutes())()
|
|
|
|
t.Run("NoDuplicateCompression", func(t *testing.T) {
|
|
req := NewRequest(t, "GET", "/user2/repo1/archive/master.zip")
|
|
req.Header.Set("Accept-Encoding", "gzip")
|
|
resp := MakeRequest(t, req, http.StatusOK)
|
|
bs, err := io.ReadAll(resp.Body)
|
|
assert.NoError(t, err)
|
|
assert.Empty(t, resp.Header().Get("Content-Encoding"))
|
|
assert.Len(t, bs, 320)
|
|
})
|
|
|
|
t.Run("Conditional", func(t *testing.T) {
|
|
defer test.MockVariableValue(&setting.Repository.StreamArchives, true)()
|
|
archiveURL := "/user2/glob/archive/master.tar.gz"
|
|
etag := MakeRequest(t, NewRequest(t, "GET", archiveURL), http.StatusOK).Header().Get("ETag")
|
|
assert.True(t, strings.HasPrefix(etag, `W/"`))
|
|
MakeRequest(t, NewRequest(t, "GET", archiveURL).SetHeader("If-None-Match", etag), http.StatusNotModified)
|
|
MakeRequest(t, NewRequest(t, "GET", archiveURL+"?path=aaa.doc").SetHeader("If-None-Match", etag), http.StatusOK)
|
|
assert.NotEqual(t, etag, MakeRequest(t, NewRequest(t, "GET", "/user2/repo1/archive/master.tar.gz"), http.StatusOK).Header().Get("ETag"))
|
|
|
|
restoreStreaming := test.MockVariableValue(&setting.Repository.StreamArchives, false)
|
|
MakeRequest(t, NewRequest(t, "GET", archiveURL).SetHeader("If-None-Match", etag), http.StatusOK)
|
|
restoreStreaming()
|
|
|
|
defer test.MockVariableValue(&setting.Repository.PrefixArchiveFiles, !setting.Repository.PrefixArchiveFiles)()
|
|
MakeRequest(t, NewRequest(t, "GET", archiveURL).SetHeader("If-None-Match", etag), http.StatusOK)
|
|
})
|
|
|
|
t.Run("SubPath", func(t *testing.T) {
|
|
// When using "archiving and caching" approach, archiving with paths will always use streaming and never be cached
|
|
defer test.MockVariableValue(&setting.Repository.StreamArchives, false)() // this can be removed if there is always streaming mode
|
|
req := NewRequest(t, "GET", "/user2/glob/archive/master.tar.gz?path=aaa.doc&path=x/y")
|
|
resp := MakeRequest(t, req, http.StatusOK)
|
|
content, err := test.ReadAllTarGzContent(resp.Body)
|
|
require.NoError(t, err)
|
|
assert.Empty(t, content["glob/a.txt"])
|
|
assert.NotEmpty(t, content["glob/aaa.doc"])
|
|
assert.Empty(t, content["glob/x/b.txt"])
|
|
assert.NotEmpty(t, content["glob/x/y/a.txt"])
|
|
|
|
req = NewRequest(t, "GET", "/user2/glob/archive/master.tar.gz")
|
|
resp = MakeRequest(t, req, http.StatusOK)
|
|
content, err = test.ReadAllTarGzContent(resp.Body)
|
|
require.NoError(t, err)
|
|
assert.NotEmpty(t, content["glob/a.txt"])
|
|
assert.NotEmpty(t, content["glob/aaa.doc"])
|
|
assert.NotEmpty(t, content["glob/x/b.txt"])
|
|
assert.NotEmpty(t, content["glob/x/y/a.txt"])
|
|
})
|
|
}
|