helios/error_test.go at master · yonasadiel/helios · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
package helios

import (
	"encoding/json"
	"net/http"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestErrorAPI(t *testing.T) {
	App.BeforeTest()

	var err Error = ErrorAPI{
		StatusCode: http.StatusNotFound,
		Code:       "not_found",
		Message:    "Not Found",
	}

	msg := err.GetMessage()

	assert.Equal(t, http.StatusNotFound, err.GetStatusCode(), "Wrong status code")
	assert.Equal(t, "not_found", msg["code"], "Wrong code")
	assert.Equal(t, "Not Found", msg["message"], "Wrong message")
}

func TestErrorForm(t *testing.T) {
	App.BeforeTest()

	type formErrorTestCase struct {
		err                ErrorForm
		expectedStatusCode int
		expectedJSON       string
		expectedIsError    bool
	}
	testCases := []formErrorTestCase{{
		err:                ErrorForm{},
		expectedStatusCode: http.StatusBadRequest,
		expectedJSON:       `{"code":"form_error","message":{"_error":[]}}`,
		expectedIsError:    false,
	}, {
		err:                NewErrorForm(),
		expectedStatusCode: http.StatusBadRequest,
		expectedJSON:       `{"code":"form_error","message":{"_error":[]}}`,
		expectedIsError:    false,
	}, {
		err: ErrorForm{
			FieldError: ErrorFormFieldNested{
				"field1": ErrorFormFieldAtomic{},
				"field2": ErrorFormFieldArray{
					ErrorFormFieldAtomic{},
					ErrorFormFieldAtomic{},
				},
			},
		},
		expectedStatusCode: http.StatusBadRequest,
		expectedJSON:       `{"code":"form_error","message":{"_error":[],"field1":[],"field2":[[],[]]}}`,
		expectedIsError:    false,
	}, {
		err:                ErrorForm{Code: "custom_code"},
		expectedStatusCode: http.StatusBadRequest,
		expectedJSON:       `{"code":"custom_code","message":{"_error":[]}}`,
		expectedIsError:    false,
	}, {
		err: ErrorForm{
			NonFieldError: []string{"err1", "err2"},
		},
		expectedStatusCode: http.StatusBadRequest,
		expectedJSON:       `{"code":"form_error","message":{"_error":["err1","err2"]}}`,
		expectedIsError:    true,
	}, {
		err: ErrorForm{
			FieldError: ErrorFormFieldNested{
				"atomic": ErrorFormFieldAtomic{"err1", "err2"},
				"array": ErrorFormFieldArray{
					ErrorFormFieldNested{
						"field1": ErrorFormFieldAtomic{"err3"},
						"field2": ErrorFormFieldArray{
							ErrorFormFieldAtomic{"err4", "err5"},
						},
					},
					ErrorFormFieldNested{},
					ErrorFormFieldNested{
						"field1": ErrorFormFieldAtomic{"err6"},
					},
				},
			},
		},
		expectedStatusCode: http.StatusBadRequest,
		expectedJSON:       `{"code":"form_error","message":{"_error":[],"array":[{"field1":["err3"],"field2":[["err4","err5"]]},{},{"field1":["err6"]}],"atomic":["err1","err2"]}}`,
		expectedIsError:    true,
	}}
	for i, testCase := range testCases {
		t.Logf("TestErrorForm testcase #%d", i)
		var jsonRepresentation []byte
		var errMashalling error
		var err Error = testCase.err // cast the ErrorForm to Error
		jsonRepresentation, errMashalling = json.Marshal(err.GetMessage())
		assert.Nil(t, errMashalling)
		assert.Equal(t, testCase.expectedStatusCode, err.GetStatusCode())
		assert.Equal(t, testCase.expectedIsError, testCase.err.IsError())
		assert.Equal(t, testCase.expectedJSON, string(jsonRepresentation))
	}
}