Beego/pkg/server/web/context/param/parsers_test.go

package param

import (
	"reflect"
	"testing"
	"time"
)

type testDefinition struct {
	strValue       string
	expectedValue  interface{}
	expectedParser paramParser
}

func Test_Parsers(t *testing.T) {

	//ints
	checkParser(testDefinition{"1", 1, intParser{}}, t)
	checkParser(testDefinition{"-1", int64(-1), intParser{}}, t)
	checkParser(testDefinition{"1", uint64(1), intParser{}}, t)

	//floats
	checkParser(testDefinition{"1.0", float32(1.0), floatParser{}}, t)
	checkParser(testDefinition{"-1.0", float64(-1.0), floatParser{}}, t)

	//strings
	checkParser(testDefinition{"AB", "AB", stringParser{}}, t)
	checkParser(testDefinition{"AB", []byte{65, 66}, stringParser{}}, t)

	//bools
	checkParser(testDefinition{"true", true, boolParser{}}, t)
	checkParser(testDefinition{"0", false, boolParser{}}, t)

	//timeParser
	checkParser(testDefinition{"2017-05-30T13:54:53Z", time.Date(2017, 5, 30, 13, 54, 53, 0, time.UTC), timeParser{}}, t)
	checkParser(testDefinition{"2017-05-30", time.Date(2017, 5, 30, 0, 0, 0, 0, time.UTC), timeParser{}}, t)

	//json
	checkParser(testDefinition{`{"X": 5, "Y":"Z"}`, struct {
		X int
		Y string
	}{5, "Z"}, jsonParser{}}, t)

	//slice in query is parsed as comma delimited
	checkParser(testDefinition{`1,2`, []int{1, 2}, sliceParser(intParser{})}, t)

	//slice in body is parsed as json
	checkParser(testDefinition{`["a","b"]`, []string{"a", "b"}, jsonParser{}}, t, MethodParam{in: body})

	//pointers
	var someInt = 1
	checkParser(testDefinition{`1`, &someInt, ptrParser(intParser{})}, t)

	var someStruct = struct{ X int }{5}
	checkParser(testDefinition{`{"X": 5}`, &someStruct, jsonParser{}}, t)

}

func checkParser(def testDefinition, t *testing.T, methodParam ...MethodParam) {
	toType := reflect.TypeOf(def.expectedValue)
	var mp MethodParam
	if len(methodParam) == 0 {
		mp = MethodParam{}
	} else {
		mp = methodParam[0]
	}
	parser := getParser(&mp, toType)

	if reflect.TypeOf(parser) != reflect.TypeOf(def.expectedParser) {
		t.Errorf("Invalid parser for value %v. Expected: %v, actual: %v", def.strValue, reflect.TypeOf(def.expectedParser).Name(), reflect.TypeOf(parser).Name())
		return
	}
	result, err := parser.parse(def.strValue, toType)
	if err != nil {
		t.Errorf("Parsing error for value %v. Expected result: %v, error: %v", def.strValue, def.expectedValue, err)
		return
	}
	convResult, err := safeConvert(reflect.ValueOf(result), toType)
	if err != nil {
		t.Errorf("Conversion error for %v. from value: %v, toType: %v, error: %v", def.strValue, result, toType, err)
		return
	}
	if !reflect.DeepEqual(convResult.Interface(), def.expectedValue) {
		t.Errorf("Parsing error for value %v. Expected result: %v, actual: %v", def.strValue, def.expectedValue, result)
	}
}