Add tests for internal/hook/request.go

internal/hook/request_test.go

@@ -0,0 +1,164 @@
+package hook
+
+import (
+ "encoding/json"
+ "reflect"
+ "testing"
+)
+
+var parseJSONPayloadTests = []struct {
+ body []byte
+ payload map[string]interface{}
+ ok bool
+}{
+ {[]byte(`[1,2,3]`), map[string]interface{}{"root": []interface{}{json.Number("1"), json.Number("2"), json.Number("3")}}, true},
+ {[]byte(` [1,2,3]`), map[string]interface{}{"root": []interface{}{json.Number("1"), json.Number("2"), json.Number("3")}}, true},
+ {[]byte(`{"key": "value"}`), map[string]interface{}{"key": "value"}, true},
+ {[]byte(`[1, {"]`), map[string]interface{}(nil), false},
+ {[]byte(`{"key": "value}`), map[string]interface{}(nil), false},
+}
+
+func TestParseJSONPayload(t *testing.T) {
+ for _, tt := range parseJSONPayloadTests {
+ r := Request{
+ Body: tt.body,
+ }
+ err := r.ParseJSONPayload()
+ if (err == nil) != tt.ok {
+ t.Errorf("unexpected result given %q: %s\n", string(tt.body), err)
+ }
+
+ if !reflect.DeepEqual(tt.payload, r.Payload) {
+ t.Errorf("failed to parse json %q:\nexpected %#v,\ngot %#v", string(tt.body), tt.payload, r.Payload)
+ }
+ }
+}
+
+var parseHeadersTests = []struct {
+ headers map[string][]string
+ expectedHeaders map[string]interface{}
+}{
+ {
+ map[string][]string{"header1": {"12"}},
+ map[string]interface{}{"header1": "12"},
+ },
+ {
+ map[string][]string{"header1": {"12", "34"}},
+ map[string]interface{}{"header1": "12"},
+ },
+ {
+ map[string][]string{"header1": {}},
+ map[string]interface{}{},
+ },
+}
+
+func TestParseHeaders(t *testing.T) {
+ for _, tt := range parseHeadersTests {
+ r := Request{}
+ r.ParseHeaders(tt.headers)
+
+ if !reflect.DeepEqual(tt.expectedHeaders, r.Headers) {
+ t.Errorf("failed to parse headers %#v:\nexpected %#v,\ngot %#v", tt.headers, tt.expectedHeaders, r.Headers)
+ }
+ }
+}
+
+var parseQueryTests = []struct {
+ query map[string][]string
+ expectedQuery map[string]interface{}
+}{
+ {
+ map[string][]string{"query1": {"12"}},
+ map[string]interface{}{"query1": "12"},
+ },
+ {
+ map[string][]string{"query1": {"12", "34"}},
+ map[string]interface{}{"query1": "12"},
+ },
+ {
+ map[string][]string{"query1": {}},
+ map[string]interface{}{},
+ },
+}
+
+func TestParseQuery(t *testing.T) {
+ for _, tt := range parseQueryTests {
+ r := Request{}
+ r.ParseQuery(tt.query)
+
+ if !reflect.DeepEqual(tt.expectedQuery, r.Query) {
+ t.Errorf("failed to parse query %#v:\nexpected %#v,\ngot %#v", tt.query, tt.expectedQuery, r.Query)
+ }
+ }
+}
+
+var parseFormPayloadTests = []struct {
+ body []byte
+ expectedPayload map[string]interface{}
+ ok bool
+}{
+ {
+ []byte("x=1&y=2"),
+ map[string]interface{}{"x": "1", "y": "2"},
+ true,
+ },
+ {
+ []byte("x=1&y=2&y=3"),
+ map[string]interface{}{"x": "1", "y": "2"},
+ true,
+ },
+ {
+ []byte(";"),
+ map[string]interface{}(nil),
+ false,
+ },
+}
+
+func TestParseFormPayload(t *testing.T) {
+ for _, tt := range parseFormPayloadTests {
+ r := Request{
+ Body: tt.body,
+ }
+ err := r.ParseFormPayload()
+ if (err == nil) != tt.ok {
+ t.Errorf("unexpected result given %q: %s\n", string(tt.body), err)
+ }
+
+ if !reflect.DeepEqual(tt.expectedPayload, r.Payload) {
+ t.Errorf("failed to parse form payload %q:\nexpected %#v,\ngot %#v", string(tt.body), tt.expectedPayload, r.Payload)
+ }
+ }
+}
+
+var parseXMLPayloadTests = []struct {
+ body []byte
+ expectedPayload map[string]interface{}
+ ok bool
+}{
+ {
+ []byte("<x>1</x>"),
+ map[string]interface{}{"x": "1"},
+ true,
+ },
+ {
+ []byte("<x>1<x>"),
+ map[string]interface{}(nil),
+ false,
+ },
+}
+
+func TestParseXMLPayload(t *testing.T) {
+ for _, tt := range parseXMLPayloadTests {
+ r := Request{
+ Body: tt.body,
+ }
+ err := r.ParseXMLPayload()
+ if (err == nil) != tt.ok {
+ t.Errorf("unexpected result given %q: %s\n", string(tt.body), err)
+ }
+
+ if !reflect.DeepEqual(tt.expectedPayload, r.Payload) {
+ t.Errorf("failed to parse xml %q:\nexpected %#v,\ngot %#v", string(tt.body), tt.expectedPayload, r.Payload)
+ }
+ }
+}