144 lines
3.5 KiB
Go
144 lines
3.5 KiB
Go
package controller
|
|
|
|
import (
|
|
"context"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"testing"
|
|
|
|
"goweb/config"
|
|
"goweb/middleware"
|
|
"goweb/services"
|
|
"goweb/tests"
|
|
|
|
"github.com/eko/gocache/v2/store"
|
|
|
|
"github.com/eko/gocache/v2/marshaler"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/labstack/echo/v4"
|
|
)
|
|
|
|
var (
|
|
c *services.Container
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
// Set the environment to test
|
|
config.SwitchEnvironment(config.EnvTest)
|
|
|
|
// Create a new container
|
|
c = services.NewContainer()
|
|
defer func() {
|
|
if err := c.Shutdown(); err != nil {
|
|
c.Web.Logger.Fatal(err)
|
|
}
|
|
}()
|
|
|
|
// Run tests
|
|
exitVal := m.Run()
|
|
os.Exit(exitVal)
|
|
}
|
|
|
|
func TestController_Redirect(t *testing.T) {
|
|
ctx, _ := tests.NewContext(c.Web, "/abc")
|
|
ctr := NewController(c)
|
|
err := ctr.Redirect(ctx, "home")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "", ctx.Response().Header().Get(echo.HeaderLocation))
|
|
assert.Equal(t, http.StatusFound, ctx.Response().Status)
|
|
}
|
|
|
|
func TestController_RenderPage(t *testing.T) {
|
|
setup := func() (echo.Context, *httptest.ResponseRecorder, Controller, Page) {
|
|
ctx, rec := tests.NewContext(c.Web, "/test/TestController_RenderPage")
|
|
tests.InitSession(ctx)
|
|
ctr := NewController(c)
|
|
|
|
p := NewPage(ctx)
|
|
p.Name = "home"
|
|
p.Layout = "main"
|
|
p.Cache.Enabled = false
|
|
p.Headers["A"] = "b"
|
|
p.Headers["C"] = "d"
|
|
p.StatusCode = http.StatusCreated
|
|
return ctx, rec, ctr, p
|
|
}
|
|
|
|
t.Run("missing name", func(t *testing.T) {
|
|
// Rendering should fail if the Page has no name
|
|
ctx, _, ctr, p := setup()
|
|
p.Name = ""
|
|
err := ctr.RenderPage(ctx, p)
|
|
assert.Error(t, err)
|
|
})
|
|
|
|
t.Run("no page cache", func(t *testing.T) {
|
|
ctx, _, ctr, p := setup()
|
|
err := ctr.RenderPage(ctx, p)
|
|
require.NoError(t, err)
|
|
|
|
// Check status code and headers
|
|
assert.Equal(t, http.StatusCreated, ctx.Response().Status)
|
|
for k, v := range p.Headers {
|
|
assert.Equal(t, v, ctx.Response().Header().Get(k))
|
|
}
|
|
|
|
// Check the template cache
|
|
parsed, err := c.TemplateRenderer.Load("page", p.Name)
|
|
assert.NoError(t, err)
|
|
|
|
// Check that all expected templates were parsed.
|
|
// This includes the name, layout and all components
|
|
expectedTemplates := make(map[string]bool)
|
|
expectedTemplates[p.Name+config.TemplateExt] = true
|
|
expectedTemplates[p.Layout+config.TemplateExt] = true
|
|
components, err := ioutil.ReadDir(c.TemplateRenderer.GetTemplatesPath() + "/components")
|
|
require.NoError(t, err)
|
|
for _, f := range components {
|
|
expectedTemplates[f.Name()] = true
|
|
}
|
|
|
|
for _, v := range parsed.Templates() {
|
|
delete(expectedTemplates, v.Name())
|
|
}
|
|
assert.Empty(t, expectedTemplates)
|
|
})
|
|
|
|
t.Run("page cache", func(t *testing.T) {
|
|
ctx, rec, ctr, p := setup()
|
|
p.Cache.Enabled = true
|
|
p.Cache.Tags = []string{"tag1"}
|
|
err := ctr.RenderPage(ctx, p)
|
|
require.NoError(t, err)
|
|
|
|
// Fetch from the cache
|
|
res, err := marshaler.New(c.Cache).
|
|
Get(context.Background(), p.URL, new(middleware.CachedPage))
|
|
require.NoError(t, err)
|
|
|
|
// Compare the cached page
|
|
cp, ok := res.(*middleware.CachedPage)
|
|
require.True(t, ok)
|
|
assert.Equal(t, p.URL, cp.URL)
|
|
assert.Equal(t, p.Headers, cp.Headers)
|
|
assert.Equal(t, p.StatusCode, cp.StatusCode)
|
|
assert.Equal(t, rec.Body.Bytes(), cp.HTML)
|
|
|
|
// Clear the tag
|
|
err = c.Cache.Invalidate(context.Background(), store.InvalidateOptions{
|
|
Tags: []string{p.Cache.Tags[0]},
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
// Refetch from the cache and expect no results
|
|
_, err = marshaler.New(c.Cache).
|
|
Get(context.Background(), p.URL, new(middleware.CachedPage))
|
|
assert.Error(t, err)
|
|
})
|
|
}
|