60 lines
1.4 KiB
Go
60 lines
1.4 KiB
Go
|
package form
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/labstack/echo/v4"
|
||
|
|
"github.com/mikestefanello/pagoda/pkg/tests"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestContextFuncs(t *testing.T) {
|
||
|
|
e := echo.New()
|
||
|
|
|
||
|
|
type example struct {
|
||
|
|
Name string `form:"name"`
|
||
|
|
}
|
||
|
|
|
||
|
|
t.Run("get empty context", func(t *testing.T) {
|
||
|
|
// Empty context, still return a form
|
||
|
|
ctx, _ := tests.NewContext(e, "/")
|
||
|
|
form := Get[example](ctx)
|
||
|
|
assert.NotNil(t, form)
|
||
|
|
})
|
||
|
|
|
||
|
|
t.Run("set bad request", func(t *testing.T) {
|
||
|
|
// Set with a bad request
|
||
|
|
req := httptest.NewRequest(http.MethodPost, "/", strings.NewReader("abc=abc"))
|
||
|
|
ctx := e.NewContext(req, httptest.NewRecorder())
|
||
|
|
var form example
|
||
|
|
err := Set(ctx, &form)
|
||
|
|
assert.Error(t, err)
|
||
|
|
})
|
||
|
|
|
||
|
|
t.Run("set", func(t *testing.T) {
|
||
|
|
// Set and parse the values
|
||
|
|
req := httptest.NewRequest(http.MethodPost, "/", strings.NewReader("name=abc"))
|
||
|
|
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
|
||
|
|
ctx := e.NewContext(req, httptest.NewRecorder())
|
||
|
|
var form example
|
||
|
|
err := Set(ctx, &form)
|
||
|
|
require.NoError(t, err)
|
||
|
|
assert.Equal(t, "abc", form.Name)
|
||
|
|
|
||
|
|
// Get again and expect the values were stored
|
||
|
|
got := Get[example](ctx)
|
||
|
|
require.NotNil(t, got)
|
||
|
|
assert.Equal(t, "abc", form.Name)
|
||
|
|
|
||
|
|
// Clear
|
||
|
|
Clear(ctx)
|
||
|
|
got = Get[example](ctx)
|
||
|
|
require.NotNil(t, got)
|
||
|
|
assert.Empty(t, got.Name)
|
||
|
|
})
|
||
|
|
}
|