2021-12-18 14:24:46 -08:00
|
|
|
package controller
|
|
|
|
|
|
|
|
import (
|
2021-12-19 12:50:50 -08:00
|
|
|
"context"
|
|
|
|
"io/ioutil"
|
2021-12-18 14:24:46 -08:00
|
|
|
"net/http"
|
|
|
|
"net/http/httptest"
|
|
|
|
"os"
|
|
|
|
"strings"
|
|
|
|
"testing"
|
|
|
|
|
|
|
|
"goweb/config"
|
2021-12-19 12:50:50 -08:00
|
|
|
"goweb/middleware"
|
2021-12-19 10:11:23 -08:00
|
|
|
"goweb/msg"
|
2021-12-18 14:24:46 -08:00
|
|
|
"goweb/services"
|
|
|
|
|
2021-12-19 12:50:50 -08:00
|
|
|
"github.com/eko/gocache/v2/store"
|
|
|
|
|
|
|
|
"github.com/eko/gocache/v2/marshaler"
|
|
|
|
|
2021-12-19 10:11:23 -08:00
|
|
|
"github.com/gorilla/sessions"
|
|
|
|
"github.com/labstack/echo-contrib/session"
|
|
|
|
|
|
|
|
"github.com/go-playground/validator/v10"
|
|
|
|
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
|
2021-12-18 14:24:46 -08:00
|
|
|
"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()
|
2021-12-19 11:56:00 -08:00
|
|
|
defer func() {
|
|
|
|
if err := c.Shutdown(); err != nil {
|
|
|
|
c.Web.Logger.Fatal(err)
|
|
|
|
}
|
|
|
|
}()
|
2021-12-18 14:24:46 -08:00
|
|
|
|
|
|
|
// Run tests
|
|
|
|
exitVal := m.Run()
|
|
|
|
os.Exit(exitVal)
|
|
|
|
}
|
|
|
|
|
2021-12-19 12:50:50 -08:00
|
|
|
func newContext(url string) (echo.Context, *httptest.ResponseRecorder) {
|
2021-12-18 15:08:04 -08:00
|
|
|
req := httptest.NewRequest(http.MethodGet, url, strings.NewReader(""))
|
2021-12-19 12:50:50 -08:00
|
|
|
rec := httptest.NewRecorder()
|
|
|
|
return c.Web.NewContext(req, rec), rec
|
2021-12-18 14:24:46 -08:00
|
|
|
}
|
2021-12-19 10:11:23 -08:00
|
|
|
|
|
|
|
func initSesssion(t *testing.T, ctx echo.Context) {
|
|
|
|
// Simulate an HTTP request through the session middleware to initiate the session
|
|
|
|
mw := session.Middleware(sessions.NewCookieStore([]byte("secret")))
|
|
|
|
handler := mw(echo.NotFoundHandler)
|
|
|
|
assert.Error(t, handler(ctx))
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestController_Redirect(t *testing.T) {
|
2021-12-19 12:50:50 -08:00
|
|
|
ctx, _ := newContext("/abc")
|
2021-12-19 10:11:23 -08:00
|
|
|
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_SetValidationErrorMessages(t *testing.T) {
|
|
|
|
type example struct {
|
|
|
|
Name string `validate:"required" label:"Label test"`
|
|
|
|
}
|
|
|
|
e := example{}
|
|
|
|
v := validator.New()
|
|
|
|
err := v.Struct(e)
|
|
|
|
require.Error(t, err)
|
|
|
|
|
2021-12-19 12:50:50 -08:00
|
|
|
ctx, _ := newContext("/")
|
2021-12-19 10:11:23 -08:00
|
|
|
initSesssion(t, ctx)
|
|
|
|
ctr := NewController(c)
|
|
|
|
ctr.SetValidationErrorMessages(ctx, err, e)
|
|
|
|
|
|
|
|
msgs := msg.Get(ctx, msg.TypeDanger)
|
|
|
|
require.Len(t, msgs, 1)
|
|
|
|
assert.Equal(t, "<strong>Label test</strong> is required.", msgs[0])
|
|
|
|
}
|
2021-12-19 12:50:50 -08:00
|
|
|
|
|
|
|
func TestController_RenderPage(t *testing.T) {
|
|
|
|
setup := func() (echo.Context, *httptest.ResponseRecorder, Controller, Page) {
|
|
|
|
ctx, rec := newContext("/test/TestController_RenderPage")
|
|
|
|
initSesssion(t, 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
|
2021-12-19 17:09:01 -08:00
|
|
|
parsed, err := c.Templates.Load("controller", p.Name)
|
|
|
|
assert.NoError(t, err)
|
2021-12-19 12:50:50 -08:00
|
|
|
|
|
|
|
// 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
|
2021-12-19 17:09:01 -08:00
|
|
|
components, err := ioutil.ReadDir(c.Templates.GetTemplatesPath() + "/components")
|
2021-12-19 12:50:50 -08:00
|
|
|
require.NoError(t, err)
|
|
|
|
for _, f := range components {
|
|
|
|
expectedTemplates[f.Name()] = true
|
|
|
|
}
|
2021-12-19 17:09:01 -08:00
|
|
|
|
|
|
|
for _, v := range parsed.Templates() {
|
2021-12-19 12:50:50 -08:00
|
|
|
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)
|
|
|
|
})
|
|
|
|
}
|