Adding upstream version 1.0.0.

Signed-off-by: Daniel Baumann <daniel@debian.org>

file_test.go

@@ -0,0 +1,190 @@
+// Copyright 2014 Martini Authors
+// Copyright 2014 The Macaron Authors
+//
+// Licensed under the Apache License, Version 2.0 (the "License"): you may
+// not use this file except in compliance with the License. You may obtain
+// a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+// License for the specific language governing permissions and limitations
+// under the License.
+
+package binding
+
+import (
+	"bytes"
+	"mime/multipart"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	chi "github.com/go-chi/chi/v5"
+	"github.com/stretchr/testify/assert"
+)
+
+var fileTestCases = []fileTestCase{
+	{
+		description: "Single file",
+		singleFile: &fileInfo{
+			fileName: "message.txt",
+			data:     "All your binding are belong to us",
+		},
+	},
+	{
+		description: "Multiple files",
+		multipleFiles: []*fileInfo{
+			{
+				fileName: "cool-gopher-fact.txt",
+				data:     "Did you know? https://plus.google.com/+MatthewHolt/posts/GmVfd6TPJ51",
+			},
+			{
+				fileName: "gophercon2014.txt",
+				data:     "@bradfitz has a Go time machine: https://twitter.com/mholt6/status/459463953395875840",
+			},
+		},
+	},
+	{
+		description: "Single file and multiple files",
+		singleFile: &fileInfo{
+			fileName: "social media.txt",
+			data:     "Hey, you should follow @mholt6 (Twitter) or +MatthewHolt (Google+)",
+		},
+		multipleFiles: []*fileInfo{
+			{
+				fileName: "thank you!",
+				data:     "Also, thanks to all the contributors of this package!",
+			},
+			{
+				fileName: "btw...",
+				data:     "This tool translates JSON into Go structs: http://mholt.github.io/json-to-go/",
+			},
+		},
+	},
+}
+
+func Test_FileUploads(t *testing.T) {
+	for _, testCase := range fileTestCases {
+		performFileTest(t, MultipartForm, testCase)
+	}
+}
+
+func performFileTest(t *testing.T, binder handlerFunc, testCase fileTestCase) {
+	httpRecorder := httptest.NewRecorder()
+	c := chi.NewRouter()
+
+	fileTestHandler := func(actual BlogPost, _ Errors) {
+		assertFileAsExpected(t, actual.HeaderImage, testCase.singleFile)
+		assert.Len(t, actual.Pictures, len(testCase.multipleFiles))
+
+		for i, expectedFile := range testCase.multipleFiles {
+			if i >= len(actual.Pictures) {
+				break
+			}
+			assertFileAsExpected(t, actual.Pictures[i], expectedFile)
+		}
+	}
+
+	c.Post(testRoute, func(_ http.ResponseWriter, req *http.Request) {
+		var actual BlogPost
+		errs := binder(req, &actual)
+		fileTestHandler(actual, errs)
+	})
+
+	c.ServeHTTP(httpRecorder, buildRequestWithFile(testCase))
+
+	switch httpRecorder.Code {
+	case http.StatusNotFound:
+		panic("Routing is messed up in test fixture (got 404): check methods and paths")
+	case http.StatusInternalServerError:
+		panic("Something bad happened on '" + testCase.description + "'")
+	}
+}
+
+func assertFileAsExpected(t *testing.T, actual *multipart.FileHeader, expected *fileInfo) {
+	if expected == nil && actual == nil {
+		return
+	}
+
+	if expected != nil && actual == nil {
+		assert.NotNil(t, actual)
+		return
+	} else if expected == nil && actual != nil {
+		assert.Nil(t, actual)
+		return
+	}
+
+	assert.EqualValues(t, expected.fileName, actual.Filename)
+	assert.EqualValues(t, expected.data, unpackFileHeaderData(actual))
+}
+
+func buildRequestWithFile(testCase fileTestCase) *http.Request {
+	b := &bytes.Buffer{}
+	w := multipart.NewWriter(b)
+
+	if testCase.singleFile != nil {
+		formFileSingle, err := w.CreateFormFile("header_image", testCase.singleFile.fileName)
+		if err != nil {
+			panic("Could not create FormFile (single file): " + err.Error())
+		}
+		_, _ = formFileSingle.Write([]byte(testCase.singleFile.data))
+	}
+
+	for _, file := range testCase.multipleFiles {
+		formFileMultiple, err := w.CreateFormFile("picture", file.fileName)
+		if err != nil {
+			panic("Could not create FormFile (multiple files): " + err.Error())
+		}
+		_, _ = formFileMultiple.Write([]byte(file.data))
+	}
+
+	err := w.Close()
+	if err != nil {
+		panic("Could not close multipart writer: " + err.Error())
+	}
+
+	req, err := http.NewRequest("POST", testRoute, b)
+	if err != nil {
+		panic("Could not create file upload request: " + err.Error())
+	}
+
+	req.Header.Set("Content-Type", w.FormDataContentType())
+
+	return req
+}
+
+func unpackFileHeaderData(fh *multipart.FileHeader) string {
+	if fh == nil {
+		return ""
+	}
+
+	f, err := fh.Open()
+	if err != nil {
+		panic("Could not open file header:" + err.Error())
+	}
+	defer f.Close()
+
+	var fb bytes.Buffer
+	_, err = fb.ReadFrom(f)
+	if err != nil {
+		panic("Could not read from file header:" + err.Error())
+	}
+
+	return fb.String()
+}
+
+type (
+	fileTestCase struct {
+		description   string
+		singleFile    *fileInfo
+		multipleFiles []*fileInfo
+	}
+
+	fileInfo struct {
+		fileName string
+		data     string
+	}
+)