38fb28f84f
solves #56. - The expected filename is `404.html`, like GitHub Pages - Each repo/branch can have one `404.html` file at it's root - If a repo does not have a `pages` branch, the 404.html file from the `pages` repository is used - You get status code 404 (unless you request /404.html which returns 200) - The error page is cached --- close #56 Reviewed-on: https://codeberg.org/Codeberg/pages-server/pulls/81 Reviewed-by: 6543 <6543@noreply.codeberg.org> Co-authored-by: crystal <crystal@noreply.codeberg.org> Co-committed-by: crystal <crystal@noreply.codeberg.org>
83 lines
2.4 KiB
Go
83 lines
2.4 KiB
Go
//go:build integration
|
|
// +build integration
|
|
|
|
package integration
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/tls"
|
|
"io"
|
|
"net/http"
|
|
"net/http/cookiejar"
|
|
"testing"
|
|
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestGetRedirect(t *testing.T) {
|
|
log.Printf("== TestGetRedirect ==\n")
|
|
// test custom domain redirect
|
|
resp, err := getTestHTTPSClient().Get("https://calciumdibromid.localhost.mock.directory:4430")
|
|
assert.NoError(t, err)
|
|
if !assert.EqualValues(t, http.StatusTemporaryRedirect, resp.StatusCode) {
|
|
t.FailNow()
|
|
}
|
|
assert.EqualValues(t, "https://www.cabr2.de/", resp.Header["Location"][0])
|
|
assert.EqualValues(t, 0, getSize(resp.Body))
|
|
}
|
|
|
|
func TestGetContent(t *testing.T) {
|
|
log.Printf("== TestGetContent ==\n")
|
|
// test get image
|
|
resp, err := getTestHTTPSClient().Get("https://magiclike.localhost.mock.directory:4430/images/827679288a.jpg")
|
|
assert.NoError(t, err)
|
|
if !assert.EqualValues(t, http.StatusOK, resp.StatusCode) {
|
|
t.FailNow()
|
|
}
|
|
assert.EqualValues(t, "image/jpeg", resp.Header["Content-Type"][0])
|
|
assert.EqualValues(t, "124635", resp.Header["Content-Length"][0])
|
|
assert.EqualValues(t, 124635, getSize(resp.Body))
|
|
|
|
// specify branch
|
|
resp, err = getTestHTTPSClient().Get("https://momar.localhost.mock.directory:4430/pag/@master/")
|
|
assert.NoError(t, err)
|
|
if !assert.EqualValues(t, http.StatusOK, resp.StatusCode) {
|
|
t.FailNow()
|
|
}
|
|
assert.EqualValues(t, "text/html; charset=utf-8", resp.Header["Content-Type"][0])
|
|
assert.True(t, getSize(resp.Body) > 1000)
|
|
}
|
|
|
|
func TestGetNotFound(t *testing.T) {
|
|
log.Printf("== TestGetNotFound ==\n")
|
|
// test custom not found pages
|
|
resp, err := getTestHTTPSClient().Get("https://crystal.localhost.mock.directory:4430/pages-404-demo/blah")
|
|
assert.NoError(t, err)
|
|
if !assert.EqualValues(t, http.StatusNotFound, resp.StatusCode) {
|
|
t.FailNow()
|
|
}
|
|
assert.EqualValues(t, "text/html; charset=utf-8", resp.Header["Content-Type"][0])
|
|
assert.EqualValues(t, "37", resp.Header["Content-Length"][0])
|
|
assert.EqualValues(t, 37, getSize(resp.Body))
|
|
}
|
|
|
|
func getTestHTTPSClient() *http.Client {
|
|
cookieJar, _ := cookiejar.New(nil)
|
|
return &http.Client{
|
|
Jar: cookieJar,
|
|
CheckRedirect: func(_ *http.Request, _ []*http.Request) error {
|
|
return http.ErrUseLastResponse
|
|
},
|
|
Transport: &http.Transport{
|
|
TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
|
|
},
|
|
}
|
|
}
|
|
|
|
func getSize(stream io.Reader) int {
|
|
buf := new(bytes.Buffer)
|
|
_, _ = buf.ReadFrom(stream)
|
|
return buf.Len()
|
|
}
|