/
app_proxy_test.go
113 lines (98 loc) · 3.5 KB
/
app_proxy_test.go
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
package actions
import (
"encoding/json"
"io"
"net/http"
"net/http/httptest"
"strings"
"testing"
"text/template"
"github.com/gomods/athens/pkg/build"
"github.com/gomods/athens/pkg/config"
"github.com/gomods/athens/pkg/log"
"github.com/gomods/athens/pkg/storage/mem"
"github.com/gorilla/mux"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
type routeTest struct {
method string
path string
body string
test func(t *testing.T, req *http.Request, resp *http.Response)
}
func TestProxyRoutes(t *testing.T) {
r := mux.NewRouter()
s, err := mem.NewStorage()
require.NoError(t, err)
l := log.NoOpLogger()
c, err := config.Load("")
require.NoError(t, err)
c.NoSumPatterns = []string{"*"} // catch all patterns with noSumWrapper to ensure the sumdb handler doesn't make a real http request to the sumdb server.
c.PathPrefix = "/prefix"
subRouter := r.PathPrefix(c.PathPrefix).Subrouter()
err = addProxyRoutes(subRouter, s, l, c)
require.NoError(t, err)
baseURL := "https://athens.azurefd.net" + c.PathPrefix
testCases := []routeTest{
{"GET", "/", "", func(t *testing.T, req *http.Request, resp *http.Response) {
assert.Equal(t, http.StatusOK, resp.StatusCode)
body, err := io.ReadAll(resp.Body)
require.NoError(t, err)
tmp, err := template.New("home").Parse(homepage)
assert.NoError(t, err)
var templateData = make(map[string]string)
templateData["Host"] = req.Host
if !strings.HasPrefix(templateData["Host"], "http://") && !strings.HasPrefix(templateData["Host"], "https://") {
if req.TLS != nil {
templateData["Host"] = "https://" + templateData["Host"]
} else {
templateData["Host"] = "http://" + templateData["Host"]
}
}
templateData["NoSumPatterns"] = strings.Join(c.NoSumPatterns, ",")
var expected strings.Builder
err = tmp.ExecuteTemplate(&expected, "home", templateData)
require.NoError(t, err)
assert.Equal(t, expected.String(), string(body))
}},
{"GET", "/badz", "", func(t *testing.T, req *http.Request, resp *http.Response) {
assert.Equal(t, http.StatusNotFound, resp.StatusCode)
}},
{"GET", "/healthz", "", func(t *testing.T, req *http.Request, resp *http.Response) {
assert.Equal(t, http.StatusOK, resp.StatusCode)
}},
{"GET", "/readyz", "", func(t *testing.T, req *http.Request, resp *http.Response) {
assert.Equal(t, http.StatusOK, resp.StatusCode)
}},
{"GET", "/version", "", func(t *testing.T, req *http.Request, resp *http.Response) {
assert.Equal(t, http.StatusOK, resp.StatusCode)
details := build.Details{}
err := json.NewDecoder(resp.Body).Decode(&details)
require.NoError(t, err)
assert.EqualValues(t, build.Data(), details)
}},
// Default sumdb is sum.golang.org
{"GET", "/sumdb/sum.golang.org/supported", "", func(t *testing.T, req *http.Request, resp *http.Response) {
assert.Equal(t, http.StatusOK, resp.StatusCode)
}},
{"GET", "/sumdb/sum.rust-lang.org/supported", "", func(t *testing.T, req *http.Request, resp *http.Response) {
assert.Equal(t, http.StatusNotFound, resp.StatusCode)
}},
{"GET", "/sumdb/sum.golang.org/lookup/github.com/gomods/athens", "", func(t *testing.T, req *http.Request, resp *http.Response) {
assert.Equal(t, http.StatusForbidden, resp.StatusCode)
}},
}
for _, tc := range testCases {
req := httptest.NewRequest(
tc.method,
baseURL+tc.path,
strings.NewReader(tc.body),
)
t.Run(req.RequestURI, func(t *testing.T) {
w := httptest.NewRecorder()
r.ServeHTTP(w, req)
tc.test(t, req, w.Result())
})
}
}