decode_test.go

package httpheader

import (
	"fmt"
	"net/http"
	"net/textproto"
	"reflect"
	"regexp"
	"sort"
	"testing"
	"time"
)

// Event defines a Google Calendar hook event type
type Event string

// GoogleCalendar hook types
const (
	SyncEvent      Event = "sync"
	ExistsEvent    Event = "exists"
	NotExistsEvent Event = "not_exists"
)

// GoogleCalendarPayload a google calendar notice
// https://developers.google.com/calendar/v3/push
type GoogleCalendarPayload struct {
	ChannelID         string    `header:"X-Goog-Channel-ID"`
	ChannelToken      string    `header:"X-Goog-Channel-Token,omitempty"`
	ChannelExpiration time.Time `header:"X-Goog-Channel-Expiration,omitempty"`
	ResourceID        string    `header:"X-Goog-Resource-ID"`
	ResourceURI       string    `header:"X-Goog-Resource-URI"`
	ResourceState     string    `header:"X-Goog-Resource-State"`
	MessageNumber     int       `header:"X-Goog-Message-Number"`
}

func getHeader(e Event) http.Header {
	h := http.Header{}
	h.Add("X-Goog-Channel-ID", "channel-ID-value")
	h.Add("X-Goog-Channel-Token", "channel-token-value")
	h.Add("X-Goog-Channel-Expiration", "Tue, 19 Nov 2013 01:13:52 GMT")
	h.Add("X-Goog-Resource-ID", "identifier-for-the-watched-resource")
	h.Add("X-Goog-Resource-URI", "version-specific-URI-of-the-watched-resource")
	h.Add("X-Goog-Message-Number", "1")
	h.Add("X-Goog-Resource-State", string(e))
	return h
}

func TestDecodeHeader(t *testing.T) {
	type args struct {
		e Event
	}
	tests := []struct {
		name    string
		args    args
		wantErr bool
	}{
		{"Google Calendar sync", args{SyncEvent}, false},
		{"Google Calendar exists", args{ExistsEvent}, false},
		{"Google Calendar no exists", args{NotExistsEvent}, false},
	}

	for i, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			plrun := GoogleCalendarPayload{
				ChannelID:     "channel-ID-value",
				ChannelToken:  "channel-token-value",
				ResourceID:    "identifier-for-the-watched-resource",
				ResourceURI:   "version-specific-URI-of-the-watched-resource",
				MessageNumber: 1,
			}
			plrun.ChannelExpiration, _ = time.Parse(http.TimeFormat, "Tue, 19 Nov 2013 01:13:52 GMT")
			plrun.ResourceState = string(tt.args.e)
			gcp := GoogleCalendarPayload{}
			err := Decode(getHeader(tt.args.e), &gcp)
			if (err != nil) != tt.wantErr {
				t.Errorf("%d. Decode() error = %+v, wantErr %+v", i, err, tt.wantErr)
			}
			if !reflect.DeepEqual(gcp, plrun) {
				t.Errorf("%d. Decode() does not work as expected, \ngot %+v \nwant %+v", i, gcp, plrun)
			}
		})
	}
}

type DecodedArgs []string

func (m *DecodedArgs) DecodeHeader(header http.Header, key string) error {
	baseKey := textproto.CanonicalMIMEHeaderKey(key)
	keyMatch := regexp.MustCompile(fmt.Sprintf(`^%s\.\d+$`, baseKey))
	var args DecodedArgs
	for k := range header {
		if keyMatch.MatchString(textproto.CanonicalMIMEHeaderKey(k)) {
			args = append(args, header.Get(k))
		}
	}
	// TODO: sort args by id
	sort.Strings(args)
	if len(args) > 0 {
		*m = args
	}
	return nil
}

func TestDecodeHeader_Unmarshaler(t *testing.T) {
	type ArgStruct struct {
		Args DecodedArgs `header:"Arg"`
	}
	input := http.Header{
		"Arg.0": []string{"a"},
		"Arg.1": []string{"b"},
		"Arg.2": []string{"c"},
	}
	want := ArgStruct{
		Args: []string{"a", "b", "c"},
	}
	var got ArgStruct

	err := Decode(input, &got)
	if err != nil {
		t.Errorf("want no error, got error: %#v", err)
	}
	if !reflect.DeepEqual(want, got) {
		t.Errorf("Decode returned %#v, want %#v", got, want)
	}
}

func TestDecodeHeader_UnmarshalerWithNilPointer(t *testing.T) {
	s := struct {
		Args *EncodedArgs `header:"Arg"`
	}{}
	err := Decode(http.Header{}, s)
	if err == nil {
		t.Error("want error but got nil")
	}
}

type simpleStruct struct {
	Foo string
	Bar int
}

type fullTypeStruct struct {
	unExport          string
	UnExportTwo       string `header:"-"`
	Bool              bool   `header:"Bool"`
	BoolInt           bool   `header:"Bool-Int,int"`
	String            string
	StringEmpty       string `header:"String-Empty"`
	StringEmptyIgnore string `header:"String-Empty-Ignore,omitempty"`
	Uint              uint
	Uint64            uint64
	Uint8             uint8
	Uint16            uint16
	Uint32            uint32
	Int               int
	Int64             int64
	Int8              int8
	Int16             int16
	Int32             int32
	Float32           float32
	Float64           float64
	Slice             []string
	SliceTwo          []int `header:"Slice-Two"`
	Array             [3]string
	ArrayTwo          [2]int `header:"Array-Two"`
	Interface         interface{}
	Time              time.Time
	TimeUnix          time.Time `header:"Time-Unix,unix"`
	// Point             *string
	Args DecodedArgs `header:"Arg"`
	Foo  simpleStruct
}

func TestDecodeHeader_more_data_type(t *testing.T) {
	timeV := time.Date(2000, 1, 1, 12, 34, 56, 0, time.UTC)
	timeS := "Sat, 01 Jan 2000 12:34:56 GMT"
	timeU := "946730096"
	h := http.Header{
		"UnExportTwo":  []string{"foo"},
		"UnExport-Two": []string{"foo"},
		"Bool":         []string{"true"},
		"Bool-Int":     []string{"1"},
		"String":       []string{"foobar"},
		"String-Empty": []string{""},
		"Uint":         []string{"2"},
		"Uint64":       []string{"3"},
		"Uint8":        []string{"4"},
		"Uint16":       []string{"5"},
		"Uint32":       []string{"6"},
		"Int":          []string{"7"},
		"Int64":        []string{"8"},
		"Int8":         []string{"9"},
		"Int16":        []string{"10"},
		"Int32":        []string{"11"},
		"Float32":      []string{"12.2"},
		"Float64":      []string{"13.2"},
		"Slice":        []string{"a", "b", "c"},
		"Slice-Two":    []string{"1", "2", "3"},
		"Array":        []string{"a", "b", "c"},
		"Array-Two":    []string{"1", "2", "3"},
		"Interface":    []string{"foo", "bar"},
		"Time":         []string{timeS},
		"Time-Unix":    []string{timeU},
		"Point":        []string{"foo"},
		"Arg.0":        []string{"a"},
		"Arg.1":        []string{"b"},
		"Arg.2":        []string{"c"},
		"Foo":          []string{"bar"},
	}
	want := fullTypeStruct{
		unExport:          "",
		UnExportTwo:       "",
		Bool:              true,
		BoolInt:           true,
		String:            "foobar",
		StringEmpty:       "",
		StringEmptyIgnore: "",
		Uint:              2,
		Uint64:            3,
		Uint8:             4,
		Uint16:            5,
		Uint32:            6,
		Int:               7,
		Int64:             8,
		Int8:              9,
		Int16:             10,
		Int32:             11,
		Float32:           12.2,
		Float64:           13.2,
		Slice:             []string{"a", "b", "c"},
		SliceTwo:          []int{1, 2, 3},
		Array:             [3]string{"a", "b", "c"},
		ArrayTwo:          [2]int{1, 2},
		Interface:         interface{}([]string{"foo", "bar"}),
		Time:              timeV,
		TimeUnix:          timeV,
		// Point:             stringPoint("foo"),
		Args: []string{"a", "b", "c"},
		Foo:  simpleStruct{Foo: "bar"},
	}
	var got fullTypeStruct
	err := Decode(h, &got)
	if err != nil {
		t.Errorf("Decode returned error: %#v", err)
	}
	if !reflect.DeepEqual(want, got) {
		t.Errorf("want/got:\n%#v\n%#v", want, got)
	}
}

func TestDecodeHeader_point(t *testing.T) {
	type A struct {
		Point *string
	}
	h := http.Header{}
	h.Set("Point", "foobar")
	want := A{
		Point: stringPoint("foobar"),
	}
	var got A
	err := Decode(h, &got)
	if err != nil {
		t.Errorf("Decode returned error: %#v", err)
	}
	if !reflect.DeepEqual(want, got) {
		t.Errorf("want %#v, but got %#v", want, got)
	}
}
func stringPoint(s string) *string {
	return &s
}

func Test_fillValues_errors(t *testing.T) {
	type args struct {
		sv     reflect.Value
		opts   tagOptions
		valArr []string
	}
	tests := []struct {
		name    string
		args    args
		wantErr bool
	}{
		{
			name: "Uint",
			args: args{
				sv:     reflect.New(reflect.TypeOf(uint(3))),
				opts:   tagOptions{},
				valArr: []string{"a"},
			},
			wantErr: true,
		},
		{
			name: "Uint64",
			args: args{
				sv:     reflect.New(reflect.TypeOf(uint64(3))),
				opts:   tagOptions{},
				valArr: []string{"a"},
			},
			wantErr: true,
		},
		{
			name: "Uint8",
			args: args{
				sv:     reflect.New(reflect.TypeOf(uint8(3))),
				opts:   tagOptions{},
				valArr: []string{"a"},
			},
			wantErr: true,
		},
		{
			name: "Uint16",
			args: args{
				sv:     reflect.New(reflect.TypeOf(uint16(3))),
				opts:   tagOptions{},
				valArr: []string{"a"},
			},
			wantErr: true,
		},
		{
			name: "Uint32",
			args: args{
				sv:     reflect.New(reflect.TypeOf(uint32(3))),
				opts:   tagOptions{},
				valArr: []string{"a"},
			},
			wantErr: true,
		},
		{
			name: "int",
			args: args{
				sv:     reflect.New(reflect.TypeOf(int(3))),
				opts:   tagOptions{},
				valArr: []string{"a"},
			},
			wantErr: true,
		},
		{
			name: "int64",
			args: args{
				sv:     reflect.New(reflect.TypeOf(int64(3))),
				opts:   tagOptions{},
				valArr: []string{"a"},
			},
			wantErr: true,
		},
		{
			name: "int8",
			args: args{
				sv:     reflect.New(reflect.TypeOf(int8(3))),
				opts:   tagOptions{},
				valArr: []string{"a"},
			},
			wantErr: true,
		},
		{
			name: "int16",
			args: args{
				sv:     reflect.New(reflect.TypeOf(int16(3))),
				opts:   tagOptions{},
				valArr: []string{"a"},
			},
			wantErr: true,
		},
		{
			name: "int32",
			args: args{
				sv:     reflect.New(reflect.TypeOf(int32(3))),
				opts:   tagOptions{},
				valArr: []string{"a"},
			},
			wantErr: true,
		},
		{
			name: "float32",
			args: args{
				sv:     reflect.New(reflect.TypeOf(float32(3))),
				opts:   tagOptions{},
				valArr: []string{"a"},
			},
			wantErr: true,
		},
		{
			name: "float64",
			args: args{
				sv:     reflect.New(reflect.TypeOf(float64(3))),
				opts:   tagOptions{},
				valArr: []string{"a"},
			},
			wantErr: true,
		},
		{
			name: "slice",
			args: args{
				sv:     reflect.New(reflect.TypeOf([]int{})),
				opts:   tagOptions{},
				valArr: []string{"a"},
			},
			wantErr: true,
		},
		{
			name: "array",
			args: args{
				sv:     reflect.New(reflect.TypeOf([1]int{})),
				opts:   tagOptions{},
				valArr: []string{"a"},
			},
			wantErr: true,
		},
		{
			name: "time",
			args: args{
				sv:     reflect.New(reflect.TypeOf(time.Time{})),
				opts:   tagOptions{},
				valArr: []string{"a"},
			},
			wantErr: true,
		},
		{
			name: "time unix",
			args: args{
				sv:     reflect.New(reflect.TypeOf(time.Time{})),
				opts:   tagOptions{"unix"},
				valArr: []string{"a"},
			},
			wantErr: true,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if err := fillValues(tt.args.sv, tt.args.opts, tt.args.valArr); (err != nil) != tt.wantErr {
				t.Errorf("fillValues() error = %v, wantErr %v", err, tt.wantErr)
			}
		})
	}
}

func TestDecode_check_header_key_not_present_no_point(t *testing.T) {
	h := http.Header{}
	h.Set("Length", "100")

	var got fullTypeStruct
	err := Decode(h, &got)
	if err != nil {
		t.Errorf("Decode returned error: %#v", err)
	}

	var want fullTypeStruct
	if !reflect.DeepEqual(want, got) {
		t.Errorf("want %#v, but got %#v", want, got)
	}
}

func TestDecode_check_header_key_not_present_point(t *testing.T) {
	type testStruct struct {
		A *string
		B *fullTypeStruct
		C *int
		D *[]string
		E *[2]string
		F interface{}
		G *time.Time
	}
	h := http.Header{}
	h.Set("Length", "100")

	var got testStruct
	err := Decode(h, &got)
	if err != nil {
		t.Errorf("Decode returned error: %#v", err)
	}

	var want testStruct
	if !reflect.DeepEqual(want, got) {
		t.Errorf("want %#v, but got %#v", want, got)
	}
	if got.A != nil || got.B != nil || got.C != nil || got.D != nil || got.E != nil || got.F != nil || got.G != nil {
		t.Error("all fields should be nil")
	}
}

func TestDecode_error(t *testing.T) {
	h := http.Header{
		"Int": []string{"abc"},
	}
	var got fullTypeStruct
	err := Decode(h, &got)
	if err == nil {
		t.Errorf("expect error, got : %#v", got)
	}
}

func TestDecodeHeader_embeddedStructs(t *testing.T) {
	tests := []struct {
		in     http.Header
		decode func(http.Header) (interface{}, error)
		want   interface{}
	}{
		{
			http.Header{"C": []string{"foo"}},
			func(h http.Header) (interface{}, error) {
				var a A
				err := Decode(h, &a)
				return a, err
			},
			A{B{C: "foo"}},
		},
		{
			http.Header{"C": []string{"foo"}},
			func(h http.Header) (interface{}, error) {
				var d D
				err := Decode(h, &d)
				return d, err
			},
			D{B: B{C: ""}, C: "foo"},
		},
		{
			http.Header{"C": []string{"foo", "bar"}},
			func(h http.Header) (interface{}, error) {
				var d D
				err := Decode(h, &d)
				return d, err
			},
			D{B: B{C: "bar"}, C: "foo"},
		},
		{
			http.Header{"C": []string{"foo", "bar"}},
			func(h http.Header) (interface{}, error) {
				var f F
				err := Decode(h, &f)
				return f, err
			},
			F{e{B: B{C: "bar"}, C: "foo"}}, // With unexported embed
		},
	}

	for i, tt := range tests {
		v, err := tt.decode(tt.in)
		if err != nil {
			t.Errorf("%d. Header(%+v) returned error: %v", i, tt.in, err)
		}

		if !reflect.DeepEqual(tt.want, v) {
			t.Errorf("%d. Header(%+v) returned/want:\n%#+v\n%#+v", i, tt.in, v, tt.want)
		}
	}
}