params_test.go

package revel

import (
	"bytes"
	"fmt"
	"io/ioutil"
	"net/http"
	"reflect"
	"testing"
)

// Params: Testing Multipart forms

const (
	MULTIPART_BOUNDARY  = "A"
	MULTIPART_FORM_DATA = `--A
Content-Disposition: form-data; name="text1"

data1
--A
Content-Disposition: form-data; name="text2"

data2
--A
Content-Disposition: form-data; name="text2"

data3
--A
Content-Disposition: form-data; name="file1"; filename="test.txt"
Content-Type: text/plain

content1
--A
Content-Disposition: form-data; name="file2[]"; filename="test.txt"
Content-Type: text/plain

content2
--A
Content-Disposition: form-data; name="file2[]"; filename="favicon.ico"
Content-Type: image/x-icon

xyz
--A
Content-Disposition: form-data; name="file3[0]"; filename="test.txt"
Content-Type: text/plain

content3
--A
Content-Disposition: form-data; name="file3[1]"; filename="favicon.ico"
Content-Type: image/x-icon

zzz
--A--
`
)

// The values represented by the form data.
type fh struct {
	filename string
	content  []byte
}

var (
	expectedValues = map[string][]string{
		"text1": {"data1"},
		"text2": {"data2", "data3"},
	}
	expectedFiles = map[string][]fh{
		"file1":    {fh{"test.txt", []byte("content1")}},
		"file2[]":  {fh{"test.txt", []byte("content2")}, fh{"favicon.ico", []byte("xyz")}},
		"file3[0]": {fh{"test.txt", []byte("content3")}},
		"file3[1]": {fh{"favicon.ico", []byte("zzz")}},
	}
)

func getMultipartRequest() *http.Request {
	req, _ := http.NewRequest("POST", "http://localhost/path",
		bytes.NewBufferString(MULTIPART_FORM_DATA))
	req.Header.Set(
		"Content-Type", fmt.Sprintf("multipart/form-data; boundary=%s", MULTIPART_BOUNDARY))
	req.Header.Set(
		"Content-Length", fmt.Sprintf("%d", len(MULTIPART_FORM_DATA)))
	return req
}

func TestMultipartForm(t *testing.T) {
	params := ParseParams(NewRequest(getMultipartRequest()))

	if !reflect.DeepEqual(expectedValues, map[string][]string(params.Values)) {
		t.Errorf("Param values: (expected) %v != %v (actual)",
			expectedValues, map[string][]string(params.Values))
	}

	actualFiles := make(map[string][]fh)
	for key, fileHeaders := range params.Files {
		for _, fileHeader := range fileHeaders {
			file, _ := fileHeader.Open()
			content, _ := ioutil.ReadAll(file)
			actualFiles[key] = append(actualFiles[key], fh{fileHeader.Filename, content})
		}
	}

	if !reflect.DeepEqual(expectedFiles, actualFiles) {
		t.Errorf("Param files: (expected) %v != %v (actual)", expectedFiles, actualFiles)
	}
}

func TestResolveAcceptLanguage(t *testing.T) {
	request := buildHttpRequestWithAcceptLanguage("")
	if result := ResolveAcceptLanguage(request); result != nil {
		t.Errorf("Expected Accept-Language to resolve to an empty string but it was '%s'", result)
	}

	request = buildHttpRequestWithAcceptLanguage("en-GB,en;q=0.8,nl;q=0.6")
	if result := ResolveAcceptLanguage(request); len(result) != 3 {
		t.Errorf("Unexpected Accept-Language values length of %d (expected %d)", len(result), 3)
	} else {
		if result[0].Language != "en-GB" {
			t.Errorf("Expected '%s' to be most qualified but instead it's '%s'", "en-GB", result[0].Language)
		}
		if result[1].Language != "en" {
			t.Errorf("Expected '%s' to be most qualified but instead it's '%s'", "en", result[1].Language)
		}
		if result[2].Language != "nl" {
			t.Errorf("Expected '%s' to be most qualified but instead it's '%s'", "nl", result[2].Language)
		}
	}

	request = buildHttpRequestWithAcceptLanguage("en;q=0.8,nl;q=0.6,en-AU;q=malformed")
	if result := ResolveAcceptLanguage(request); len(result) != 3 {
		t.Errorf("Unexpected Accept-Language values length of %d (expected %d)", len(result), 3)
	} else {
		if result[0].Language != "en-AU" {
			t.Errorf("Expected '%s' to be most qualified but instead it's '%s'", "en-AU", result[0].Language)
		}
	}
}

func BenchmarkResolveAcceptLanguage(b *testing.B) {
	for i := 0; i < b.N; i++ {
		request := buildHttpRequestWithAcceptLanguage("en-GB,en;q=0.8,nl;q=0.6,fr;q=0.5,de-DE;q=0.4,no-NO;q=0.4,ru;q=0.2")
		ResolveAcceptLanguage(request)
	}
}

func buildHttpRequestWithAcceptLanguage(acceptLanguage string) *http.Request {
	request, _ := http.NewRequest("POST", "http://localhost/path", nil)
	request.Header.Set("Accept-Language", acceptLanguage)
	return request
}