apiproxy/client_test.go at master · sourcegraph/apiproxy · 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
105
106
107
108
109
110
111
112
113
114
package apiproxy

import (
	"net/http"
	"regexp"
	"testing"
)

func TestRequestModifyingTransport_NonOverridden(t *testing.T) {
	mockTransport := newMockTransport()
	mockTransport.defaultResponse = &http.Response{}

	transport := &RequestModifyingTransport{
		Transport: mockTransport,
	}

	_, err := transport.RoundTrip(newHTTPGETRequest(t, "http://example.com"))
	if err != nil {
		t.Error("RoundTrip", err)
	}
	if numRequests := len(mockTransport.requests); numRequests != 1 {
		t.Errorf("want numRequests == %d, got %d", 1, numRequests)
	}
}

func TestRequestModifyingTransport_Overridden(t *testing.T) {
	mockTransport := newMockTransport()
	mockTransport.defaultResponse = &http.Response{}

	transport := &RequestModifyingTransport{Transport: mockTransport}
	transport.Override(regexp.MustCompile(`^/foo$`), http.Header{"X-Foo": []string{"bar"}}, false)

	req := newHTTPGETRequest(t, "http://example.com/foo")

	// The override will be applied for all matching requests (not just the
	// first) because runOnlyOnce is false.
	for i := 1; i <= 2; i++ {
		_, err := transport.RoundTrip(req)
		if err != nil {
			t.Error("RoundTrip", err)
		}
		if numRequests := len(mockTransport.requests); numRequests != i {
			t.Errorf("want numRequests == %d, got %d", i, numRequests)
		}
		if want, got := "bar", mockTransport.requests[i-1].Header.Get("X-Foo"); want != got {
			t.Errorf("want X-Foo header %q, got %q", want, got)
		}
	}
}

func TestRequestModifyingTransport_Overridden_RunOnlyOnce(t *testing.T) {
	mockTransport := newMockTransport()
	mockTransport.defaultResponse = &http.Response{}

	transport := &RequestModifyingTransport{Transport: mockTransport}
	transport.Override(regexp.MustCompile(`^/foo$`), http.Header{"X-Foo": []string{"bar"}}, true)

	req := newHTTPGETRequest(t, "http://example.com/foo")

	// The override will be applied the first time.
	_, err := transport.RoundTrip(req)
	if err != nil {
		t.Error("RoundTrip", err)
	}
	if numRequests := len(mockTransport.requests); numRequests != 1 {
		t.Errorf("want numRequests == %d, got %d", 1, numRequests)
	}
	if want, got := "bar", mockTransport.requests[0].Header.Get("X-Foo"); want != got {
		t.Errorf("want X-Foo header %q, got %q", want, got)
	}

	// The override will NOT be applied the second time because runOnlyOnce is true.
	_, err = transport.RoundTrip(req)
	if err != nil {
		t.Error("RoundTrip", err)
	}
	if numRequests := len(mockTransport.requests); numRequests != 2 {
		t.Errorf("want numRequests == %d, got %d", 2, numRequests)
	}
	if got := mockTransport.requests[1].Header.Get("X-Foo"); got != "" {
		t.Errorf("want overrides not to be applied, but they were (want X-Foo header empty, got %q)", got)
	}
}

func TestRequestModifyingTransport_GET_HEAD_Only(t *testing.T) {
	tests := []struct {
		method    string
		overrides bool
	}{
		{"GET", true}, {"HEAD", true},
		{"POST", false}, {"PUT", false}, {"DELETE", false},
	}
	for _, test := range tests {
		mockTransport := newMockTransport()
		mockTransport.defaultResponse = &http.Response{}

		transport := &RequestModifyingTransport{Transport: mockTransport}
		transport.Override(regexp.MustCompile(`^/foo$`), http.Header{"X-Foo": []string{"bar"}}, true)

		req, err := http.NewRequest(test.method, "http://example.com/foo", nil)
		if err != nil {
			t.Fatal("http.NewRequest", err)
		}

		_, err = transport.RoundTrip(req)
		var wantXFooHeader string
		if test.overrides {
			wantXFooHeader = "bar"
		}
		if got := mockTransport.requests[0].Header.Get("X-Foo"); wantXFooHeader != got {
			t.Errorf("%s: want overrides not to be applied, but they were (want X-Foo header %q, got %q)", test.method, wantXFooHeader, got)
		}
	}
}