mirror of
https://github.com/1f349/site-hosting.git
synced 2025-04-13 23:26:02 +01:00
101 lines
3.1 KiB
Go
101 lines
3.1 KiB
Go
package serve
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"fmt"
|
|
"github.com/1f349/bluebell/database"
|
|
"github.com/1f349/bluebell/logger"
|
|
"github.com/charmbracelet/log"
|
|
"github.com/spf13/afero"
|
|
"github.com/stretchr/testify/assert"
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func init() {
|
|
logger.Logger.SetLevel(log.DebugLevel)
|
|
}
|
|
|
|
type fakeServeDB struct {
|
|
branch string
|
|
}
|
|
|
|
//goland:noinspection HttpUrlsUsage
|
|
const httpPrefix = "http://"
|
|
|
|
func (f *fakeServeDB) GetLastUpdatedByDomainBranch(_ context.Context, params database.GetLastUpdatedByDomainBranchParams) (time.Time, error) {
|
|
if params.Domain == "example.com" && params.Branch == "@"+f.branch {
|
|
return time.Now(), nil
|
|
}
|
|
return time.Time{}, sql.ErrNoRows
|
|
}
|
|
|
|
func TestHandler_ServeHTTP(t *testing.T) {
|
|
for _, branch := range []string{"", "test", "dev"} {
|
|
t.Run(branch+" branch", func(t *testing.T) {
|
|
serveTest(t, "example.com", branch, "example.com/@"+branch+"/index.html")
|
|
serveTest(t, "example.com/hello-world", branch, "example.com/@"+branch+"/hello-world.html")
|
|
serveTest(t, "example.com/hello-world", branch, "example.com/@"+branch+"/hello-world/index.html")
|
|
serveTest(t, "example.com/hello-world", branch, "example.com/@"+branch+"/hello-world")
|
|
})
|
|
|
|
for _, testPath := range []string{"/", "/posts/test", "/this-is-definitely-a-page-that-I-really-need-to-view-right-now"} {
|
|
t.Run("switch to "+branch+" branch", func(t *testing.T) {
|
|
h := New(afero.NewMemMapFs(), &fakeServeDB{})
|
|
|
|
req := httptest.NewRequest(http.MethodGet, httpPrefix+"example.com"+testPath+"?__bluebell-switch-beta="+branch, nil)
|
|
rec := httptest.NewRecorder()
|
|
h.ServeHTTP(rec, req)
|
|
|
|
res := rec.Result()
|
|
assert.Equal(t, http.StatusFound, rec.Code)
|
|
assert.NotNil(t, res.Body)
|
|
all, err := io.ReadAll(res.Body)
|
|
assert.NoError(t, err)
|
|
assert.Contains(t, string(all), "<a href=\"/?__bluebell-no-cache=")
|
|
assert.Contains(t, string(all), "\">Found</a>.\n\n")
|
|
|
|
cookies := res.Cookies()
|
|
assert.Len(t, cookies, 1)
|
|
assert.Equal(t, cookies[0].Name, "__bluebell-site-beta")
|
|
assert.Equal(t, cookies[0].Value, branch)
|
|
assert.Equal(t, cookies[0].Path, "/")
|
|
assert.Equal(t, cookies[0].HttpOnly, true)
|
|
assert.Equal(t, cookies[0].SameSite, http.SameSiteLaxMode)
|
|
})
|
|
}
|
|
}
|
|
}
|
|
|
|
func serveTest(t *testing.T, address string, branch string, name string) {
|
|
t.Run(fmt.Sprintf("serveTest \"%s\" (%s) -> \"%s\"", address, branch, name), func(t *testing.T) {
|
|
fs := afero.NewMemMapFs()
|
|
assert.NoError(t, fs.MkdirAll(filepath.Dir(name), os.ModePerm))
|
|
assert.NoError(t, afero.WriteFile(fs, name, []byte("Hello World\n"), 0666))
|
|
h := New(fs, &fakeServeDB{branch: branch})
|
|
|
|
req := httptest.NewRequest(http.MethodGet, httpPrefix+address, nil)
|
|
if branch != "" {
|
|
req.AddCookie(&http.Cookie{
|
|
Name: "__bluebell-site-beta",
|
|
Value: branch,
|
|
})
|
|
}
|
|
rec := httptest.NewRecorder()
|
|
h.ServeHTTP(rec, req)
|
|
|
|
res := rec.Result()
|
|
assert.Equal(t, http.StatusOK, res.StatusCode)
|
|
assert.NotNil(t, res.Body)
|
|
all, err := io.ReadAll(res.Body)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "Hello World\n", string(all))
|
|
})
|
|
}
|