137 lines
3.6 KiB
Go
137 lines
3.6 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"io/ioutil"
|
|
"mime/multipart"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"owl-blogs/app"
|
|
entrytypes "owl-blogs/entry_types"
|
|
"owl-blogs/infra"
|
|
"owl-blogs/test"
|
|
"path"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func getUserToken(service *app.AuthorService) string {
|
|
_, err := service.Create("test", "test")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
token, err := service.CreateToken("test")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return token
|
|
}
|
|
|
|
func TestEditorFormGet(t *testing.T) {
|
|
db := test.NewMockDb()
|
|
owlApp := App(db)
|
|
app := owlApp.FiberApp
|
|
token := getUserToken(owlApp.AuthorService)
|
|
|
|
req := httptest.NewRequest("GET", "/editor/new/Image", nil)
|
|
req.AddCookie(&http.Cookie{Name: "token", Value: token})
|
|
resp, err := app.Test(req)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 200, resp.StatusCode)
|
|
}
|
|
|
|
func TestEditorFormGetNoAuth(t *testing.T) {
|
|
db := test.NewMockDb()
|
|
owlApp := App(db)
|
|
app := owlApp.FiberApp
|
|
|
|
req := httptest.NewRequest("GET", "/editor/new/Image", nil)
|
|
req.AddCookie(&http.Cookie{Name: "token", Value: "invalid"})
|
|
resp, err := app.Test(req)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 302, resp.StatusCode)
|
|
}
|
|
|
|
func TestEditorFormPost(t *testing.T) {
|
|
db := test.NewMockDb()
|
|
owlApp := App(db)
|
|
app := owlApp.FiberApp
|
|
token := getUserToken(owlApp.AuthorService)
|
|
repo := infra.NewEntryRepository(db, owlApp.Registry)
|
|
binRepo := infra.NewBinaryFileRepo(db)
|
|
|
|
fileDir, _ := os.Getwd()
|
|
fileName := "../../test/fixtures/test.png"
|
|
filePath := path.Join(fileDir, fileName)
|
|
|
|
file, err := os.Open(filePath)
|
|
require.NoError(t, err)
|
|
fileBytes, err := ioutil.ReadFile(filePath)
|
|
require.NoError(t, err)
|
|
defer file.Close()
|
|
|
|
body := &bytes.Buffer{}
|
|
writer := multipart.NewWriter(body)
|
|
part, _ := writer.CreateFormFile("image", filepath.Base(file.Name()))
|
|
io.Copy(part, file)
|
|
part, _ = writer.CreateFormField("content")
|
|
io.WriteString(part, "test content")
|
|
writer.Close()
|
|
|
|
req := httptest.NewRequest("POST", "/editor/new/Image", body)
|
|
req.Header.Set("Content-Type", writer.FormDataContentType())
|
|
req.AddCookie(&http.Cookie{Name: "token", Value: token})
|
|
resp, err := app.Test(req)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 302, resp.StatusCode)
|
|
require.Contains(t, resp.Header.Get("Location"), "/posts/")
|
|
|
|
id := strings.Split(resp.Header.Get("Location"), "/")[2]
|
|
entry, err := repo.FindById(id)
|
|
require.NoError(t, err)
|
|
require.Equal(t, "test content", entry.MetaData().(*entrytypes.ImageMetaData).Content)
|
|
imageId := entry.MetaData().(*entrytypes.ImageMetaData).ImageId
|
|
require.NotZero(t, imageId)
|
|
bin, err := binRepo.FindById(imageId)
|
|
require.NoError(t, err)
|
|
require.Equal(t, bin.Name, "test.png")
|
|
require.Equal(t, fileBytes, bin.Data)
|
|
|
|
}
|
|
|
|
func TestEditorFormPostNoAuth(t *testing.T) {
|
|
db := test.NewMockDb()
|
|
owlApp := App(db)
|
|
app := owlApp.FiberApp
|
|
|
|
fileDir, _ := os.Getwd()
|
|
fileName := "../../test/fixtures/test.png"
|
|
filePath := path.Join(fileDir, fileName)
|
|
|
|
file, err := os.Open(filePath)
|
|
require.NoError(t, err)
|
|
defer file.Close()
|
|
|
|
body := &bytes.Buffer{}
|
|
writer := multipart.NewWriter(body)
|
|
part, _ := writer.CreateFormFile("ImageId", filepath.Base(file.Name()))
|
|
io.Copy(part, file)
|
|
part, _ = writer.CreateFormField("Content")
|
|
io.WriteString(part, "test content")
|
|
writer.Close()
|
|
|
|
req := httptest.NewRequest("POST", "/editor/new/Image", body)
|
|
req.Header.Set("Content-Type", writer.FormDataContentType())
|
|
req.AddCookie(&http.Cookie{Name: "token", Value: "invalid"})
|
|
resp, err := app.Test(req)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 302, resp.StatusCode)
|
|
require.Contains(t, resp.Header.Get("Location"), "/auth/login")
|
|
|
|
}
|