109 lines
2.6 KiB
Go
109 lines
2.6 KiB
Go
package forms_test
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"mime/multipart"
|
|
"os"
|
|
"owl-blogs/app"
|
|
"owl-blogs/infra"
|
|
"owl-blogs/test"
|
|
"owl-blogs/web/forms"
|
|
"path"
|
|
"path/filepath"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
type MockData struct {
|
|
Image string `owl:"inputType=file"`
|
|
Content string `owl:"inputType=text"`
|
|
}
|
|
|
|
type MockFormData struct {
|
|
fileHeader *multipart.FileHeader
|
|
}
|
|
|
|
func NewMockFormData() *MockFormData {
|
|
fileDir, _ := os.Getwd()
|
|
fileName := "../../test/fixtures/test.png"
|
|
filePath := path.Join(fileDir, fileName)
|
|
|
|
file, err := os.Open(filePath)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer file.Close()
|
|
|
|
body := &bytes.Buffer{}
|
|
writer := multipart.NewWriter(body)
|
|
part, err := writer.CreateFormFile("ImagePath", filepath.Base(file.Name()))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
io.Copy(part, file)
|
|
writer.Close()
|
|
|
|
multipartForm := multipart.NewReader(body, writer.Boundary())
|
|
formData, err := multipartForm.ReadForm(0)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
fileHeader := formData.File["ImagePath"][0]
|
|
|
|
return &MockFormData{fileHeader: fileHeader}
|
|
}
|
|
|
|
func (f *MockFormData) FormFile(key string) (*multipart.FileHeader, error) {
|
|
return f.fileHeader, nil
|
|
}
|
|
|
|
func (f *MockFormData) FormValue(key string, defaultValue ...string) string {
|
|
return key
|
|
}
|
|
|
|
func TestFieldToFormField(t *testing.T) {
|
|
field := reflect.TypeOf(&MockData{}).Elem().Field(0)
|
|
formField, err := forms.FieldToFormField(field, reflect.Value{})
|
|
require.NoError(t, err)
|
|
require.Equal(t, "Image", formField.Name)
|
|
require.Equal(t, "file", formField.Params.InputType)
|
|
}
|
|
|
|
func TestStructToFields(t *testing.T) {
|
|
fields, err := forms.StructToFormFields(&MockData{})
|
|
require.NoError(t, err)
|
|
require.Len(t, fields, 2)
|
|
require.Equal(t, "Image", fields[0].Name)
|
|
require.Equal(t, "file", fields[0].Params.InputType)
|
|
require.Equal(t, "Content", fields[1].Name)
|
|
require.Equal(t, "text", fields[1].Params.InputType)
|
|
}
|
|
|
|
func TestForm_HtmlForm(t *testing.T) {
|
|
form := forms.NewForm(&MockData{}, nil)
|
|
html, err := form.HtmlForm()
|
|
require.NoError(t, err)
|
|
require.Contains(t, html, "<input type=\"file\" name=\"Image\"")
|
|
require.Contains(t, html, "<input type=\"text\" name=\"Content\"")
|
|
|
|
}
|
|
|
|
func TestFormParseNil(t *testing.T) {
|
|
form := forms.NewForm(&MockData{}, nil)
|
|
_, err := form.Parse(nil)
|
|
require.Error(t, err)
|
|
}
|
|
|
|
func TestFormParse(t *testing.T) {
|
|
binRepo := infra.NewBinaryFileRepo(test.NewMockDb())
|
|
binService := app.NewBinaryFileService(binRepo)
|
|
form := forms.NewForm(&MockData{}, binService)
|
|
data, err := form.Parse(NewMockFormData())
|
|
require.NoError(t, err)
|
|
require.NotZero(t, data.(*MockData).Image)
|
|
require.Equal(t, "Content", data.(*MockData).Content)
|
|
}
|