http/middleware_test.go at main · aakash-rajur/http · 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
package http

import (
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/mock"
	"net/http"
	"net/http/httptest"
	"testing"
)

func TestMiddlewares_Append(t *testing.T) {
	middlewares := make(Middlewares, 0)

	m1 := func(w http.ResponseWriter, r *http.Request, next Next) {
		next(r)
	}

	m2 := func(w http.ResponseWriter, r *http.Request, next Next) {
		next(r)
	}

	middlewares = middlewares.Append(m1)

	assert.Equalf(t, 1, len(middlewares), "want len(middlewares) == 1, got %v", len(middlewares))

	middlewares = middlewares.Append(m2)

	assert.Equalf(t, 2, len(middlewares), "want len(middlewares) == 2, got %v", len(middlewares))
}

func TestMiddlewares_Chain(t *testing.T) {
	middlewares := make(Middlewares, 0)

	mm1, mm2 := NewMockMiddleware(), NewMockMiddleware()

	middlewares = middlewares.Append(mm1.Middleware())

	middlewares = middlewares.Append(mm2.Middleware())

	final := NewMockHandler(nil)

	handler := middlewares.Chain(final.ServeHTTP)

	assert.NotNil(t, handler)

	rr := httptest.NewRecorder()

	mrw := &ResponseWriter{ResponseWriter: rr}

	mr := httptest.NewRequest(http.MethodGet, "/", nil)

	handler(mrw, mr)

	mm1.AssertCalled(t, "Run", mrw, mr, mock.AnythingOfType("Next"))

	mm1.AssertNumberOfCalls(t, "Run", 1)

	mm2.AssertCalled(t, "Run", mrw, mr, mock.AnythingOfType("Next"))

	mm2.AssertNumberOfCalls(t, "Run", 1)

	final.AssertCalled(t, "ServeHTTP", mrw, mr)

	final.AssertNumberOfCalls(t, "ServeHTTP", 1)
}