135 lines
3.5 KiB
Go
135 lines
3.5 KiB
Go
package staticpages
|
|
|
|
import (
|
|
"bytes"
|
|
"compress/gzip"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"gitlab.com/gitlab-org/gitlab-workhorse/internal/testhelper"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestServingNonExistingFile(t *testing.T) {
|
|
dir := "/path/to/non/existing/directory"
|
|
httpRequest, _ := http.NewRequest("GET", "/file", nil)
|
|
|
|
w := httptest.NewRecorder()
|
|
st := &Static{dir}
|
|
st.ServeExisting("/", CacheDisabled, nil).ServeHTTP(w, httpRequest)
|
|
require.Equal(t, 404, w.Code)
|
|
}
|
|
|
|
func TestServingDirectory(t *testing.T) {
|
|
dir, err := ioutil.TempDir("", "deploy")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer os.RemoveAll(dir)
|
|
|
|
httpRequest, _ := http.NewRequest("GET", "/file", nil)
|
|
w := httptest.NewRecorder()
|
|
st := &Static{dir}
|
|
st.ServeExisting("/", CacheDisabled, nil).ServeHTTP(w, httpRequest)
|
|
require.Equal(t, 404, w.Code)
|
|
}
|
|
|
|
func TestServingMalformedUri(t *testing.T) {
|
|
dir := "/path/to/non/existing/directory"
|
|
httpRequest, _ := http.NewRequest("GET", "/../../../static/file", nil)
|
|
|
|
w := httptest.NewRecorder()
|
|
st := &Static{dir}
|
|
st.ServeExisting("/", CacheDisabled, nil).ServeHTTP(w, httpRequest)
|
|
require.Equal(t, 404, w.Code)
|
|
}
|
|
|
|
func TestExecutingHandlerWhenNoFileFound(t *testing.T) {
|
|
dir := "/path/to/non/existing/directory"
|
|
httpRequest, _ := http.NewRequest("GET", "/file", nil)
|
|
|
|
executed := false
|
|
st := &Static{dir}
|
|
st.ServeExisting("/", CacheDisabled, http.HandlerFunc(func(_ http.ResponseWriter, r *http.Request) {
|
|
executed = (r == httpRequest)
|
|
})).ServeHTTP(nil, httpRequest)
|
|
if !executed {
|
|
t.Error("The handler should get executed")
|
|
}
|
|
}
|
|
|
|
func TestServingTheActualFile(t *testing.T) {
|
|
dir, err := ioutil.TempDir("", "deploy")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer os.RemoveAll(dir)
|
|
|
|
httpRequest, _ := http.NewRequest("GET", "/file", nil)
|
|
|
|
fileContent := "STATIC"
|
|
ioutil.WriteFile(filepath.Join(dir, "file"), []byte(fileContent), 0600)
|
|
|
|
w := httptest.NewRecorder()
|
|
st := &Static{dir}
|
|
st.ServeExisting("/", CacheDisabled, nil).ServeHTTP(w, httpRequest)
|
|
require.Equal(t, 200, w.Code)
|
|
if w.Body.String() != fileContent {
|
|
t.Error("We should serve the file: ", w.Body.String())
|
|
}
|
|
}
|
|
|
|
func testServingThePregzippedFile(t *testing.T, enableGzip bool) {
|
|
dir, err := ioutil.TempDir("", "deploy")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer os.RemoveAll(dir)
|
|
|
|
httpRequest, _ := http.NewRequest("GET", "/file", nil)
|
|
|
|
if enableGzip {
|
|
httpRequest.Header.Set("Accept-Encoding", "gzip, deflate")
|
|
}
|
|
|
|
fileContent := "STATIC"
|
|
|
|
var fileGzipContent bytes.Buffer
|
|
fileGzip := gzip.NewWriter(&fileGzipContent)
|
|
fileGzip.Write([]byte(fileContent))
|
|
fileGzip.Close()
|
|
|
|
ioutil.WriteFile(filepath.Join(dir, "file.gz"), fileGzipContent.Bytes(), 0600)
|
|
ioutil.WriteFile(filepath.Join(dir, "file"), []byte(fileContent), 0600)
|
|
|
|
w := httptest.NewRecorder()
|
|
st := &Static{dir}
|
|
st.ServeExisting("/", CacheDisabled, nil).ServeHTTP(w, httpRequest)
|
|
require.Equal(t, 200, w.Code)
|
|
if enableGzip {
|
|
testhelper.RequireResponseHeader(t, w, "Content-Encoding", "gzip")
|
|
if !bytes.Equal(w.Body.Bytes(), fileGzipContent.Bytes()) {
|
|
t.Error("We should serve the pregzipped file")
|
|
}
|
|
} else {
|
|
require.Equal(t, 200, w.Code)
|
|
testhelper.RequireResponseHeader(t, w, "Content-Encoding")
|
|
if w.Body.String() != fileContent {
|
|
t.Error("We should serve the file: ", w.Body.String())
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestServingThePregzippedFile(t *testing.T) {
|
|
testServingThePregzippedFile(t, true)
|
|
}
|
|
|
|
func TestServingThePregzippedFileWithoutEncoding(t *testing.T) {
|
|
testServingThePregzippedFile(t, false)
|
|
}
|