v3.0.0 (#18)

test/testfunc.go

@@ -0,0 +1,159 @@
+// Package test contains useful functions used across tests.
+package test
+
+import (
+	"bytes"
+	"fmt"
+	"io"
+	"mime/multipart"
+	"os"
+	"path"
+	"path/filepath"
+	"runtime"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+	"github.com/thecodingmachine/gotenberg/internal/pkg/rand"
+)
+
+// HTMLMultipartForm returns the body
+// for a multipate/form-data request with all
+// files under "html" folder.
+func HTMLMultipartForm(t *testing.T) (*bytes.Buffer, string) {
+	return multipartForm(t, "html")
+}
+
+// MarkdownMultipartForm returns the body
+// for a multipate/form-data request with all
+// files under "markdown" folder.
+func MarkdownMultipartForm(t *testing.T) (*bytes.Buffer, string) {
+	return multipartForm(t, "markdown")
+}
+
+// OfficeMultipartForm returns the body
+// for a multipate/form-data request with all
+// files under "office" folder.
+func OfficeMultipartForm(t *testing.T) (*bytes.Buffer, string) {
+	return multipartForm(t, "office")
+}
+
+// PDFMultipartForm returns the body
+// for a multipate/form-data request with all
+// files under "pdf" folder.
+func PDFMultipartForm(t *testing.T) (*bytes.Buffer, string) {
+	return multipartForm(t, "pdf")
+}
+
+func multipartForm(t *testing.T, kind string) (*bytes.Buffer, string) {
+	body := &bytes.Buffer{}
+	writer := multipart.NewWriter(body)
+	defer writer.Close()
+	dirPath := abs(t, kind, "")
+	fpaths := make(map[string]string)
+	filepath.Walk(dirPath, func(path string, info os.FileInfo, _ error) error {
+		if info.IsDir() {
+			return nil
+		}
+		fpaths[info.Name()] = abs(t, kind, info.Name())
+		return nil
+	})
+	for filename, fpath := range fpaths {
+		file, err := os.Open(fpath)
+		require.Nil(t, err)
+		part, err := writer.CreateFormFile("foo", filename)
+		require.Nil(t, err)
+		_, err = io.Copy(part, file)
+		require.Nil(t, err)
+	}
+	return body, writer.FormDataContentType()
+}
+
+// HTMLTestDirPath creates a copy
+// of "html" folder in test/testdata.
+func HTMLTestDirPath(t *testing.T) string {
+	return copyDir(t, "html")
+}
+
+// MarkdownTestDirPath creates a copy
+// of "markdown" folder in test/testdata.
+func MarkdownTestDirPath(t *testing.T) string {
+	return copyDir(t, "markdown")
+}
+
+// OfficeTestDirPath creates a copy
+// of "office" folder in test/testdata.
+func OfficeTestDirPath(t *testing.T) string {
+	return copyDir(t, "office")
+}
+
+// PDFTestDirPath creates a copy
+// of "pdf" folder in test/testdata.
+func PDFTestDirPath(t *testing.T) string {
+	return copyDir(t, "pdf")
+}
+
+func copyDir(t *testing.T, kind string) string {
+	tmpDirPath, err := rand.Get()
+	require.Nil(t, err)
+	err = os.MkdirAll(tmpDirPath, 0755)
+	require.Nil(t, err)
+	dirPath := abs(t, kind, "")
+	filepath.Walk(dirPath, func(path string, info os.FileInfo, _ error) error {
+		if info.IsDir() {
+			return nil
+		}
+		fpath := abs(t, kind, info.Name())
+		in, err := os.Open(fpath)
+		require.Nil(t, err)
+		defer in.Close()
+		tmpFpath := fmt.Sprintf("%s/%s", tmpDirPath, info.Name())
+		out, err := os.Create(tmpFpath)
+		require.Nil(t, err)
+		defer out.Close()
+		err = out.Chmod(0644)
+		require.Nil(t, err)
+		_, err = io.Copy(out, in)
+		require.Nil(t, err)
+		_, err = out.Seek(0, 0)
+		require.Nil(t, err)
+		return nil
+	})
+	return tmpDirPath
+}
+
+// HTMLTestFilePath returns the absolute file path
+// of a file in "html" folder in test/testdata.
+func HTMLTestFilePath(t *testing.T, filename string) string {
+	return abs(t, "html", filename)
+}
+
+// MarkdownTestFilePath returns the absolute file path
+// of a file in "markdown" folder in test/testdata.
+func MarkdownTestFilePath(t *testing.T, filename string) string {
+	return abs(t, "markdown", filename)
+}
+
+// OfficeTestFilePath returns the absolute file path
+// of a file in "office" folder in test/testdata.
+func OfficeTestFilePath(t *testing.T, filename string) string {
+	return abs(t, "office", filename)
+}
+
+// PDFTestFilePath returns the absolute file path
+// of a file in "pdf" folder in test/testdata.
+func PDFTestFilePath(t *testing.T, filename string) string {
+	return abs(t, "pdf", filename)
+}
+
+func abs(t *testing.T, kind, filename string) string {
+	_, gofilename, _, ok := runtime.Caller(0)
+	require.Equal(t, ok, true, "got no caller information")
+	if filename == "" {
+		path, err := filepath.Abs(fmt.Sprintf("%s/testdata/%s", path.Dir(gofilename), kind))
+		require.Nil(t, err, `getting the absolute path of "%s"`, kind)
+		return path
+	}
+	path, err := filepath.Abs(fmt.Sprintf("%s/testdata/%s/%s", path.Dir(gofilename), kind, filename))
+	require.Nil(t, err, `getting the absolute path of "%s"`, filename)
+	return path
+}