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
|
package stfe
import (
"net/http"
"net/http/httptest"
"testing"
"github.com/system-transparency/stfe/pkg/types"
)
// TestHandlers check that the expected handlers are configured
func TestHandlers(t *testing.T) {
endpoints := map[types.Endpoint]bool{
types.EndpointAddLeaf: false,
types.EndpointAddCosignature: false,
types.EndpointGetTreeHeadLatest: false,
types.EndpointGetTreeHeadToSign: false,
types.EndpointGetTreeHeadCosigned: false,
types.EndpointGetConsistencyProof: false,
types.EndpointGetProofByHash: false,
types.EndpointGetLeaves: false,
}
i := &Instance{
Config: testConfig,
}
for _, handler := range i.Handlers() {
if _, ok := endpoints[handler.Endpoint]; !ok {
t.Errorf("got unexpected endpoint: %s", handler.Endpoint)
}
endpoints[handler.Endpoint] = true
}
for endpoint, ok := range endpoints {
if !ok {
t.Errorf("endpoint %s is not configured", endpoint)
}
}
}
// TestServeHTTP checks that invalid HTTP methods are rejected
func TestServeHTTP(t *testing.T) {
i := &Instance{
Config: testConfig,
}
for _, handler := range i.Handlers() {
// Prepare invalid HTTP request
method := http.MethodPost
if method == handler.Method {
method = http.MethodGet
}
url := handler.Endpoint.Path("http://example.com", i.Prefix)
req, err := http.NewRequest(method, url, nil)
if err != nil {
t.Fatalf("must create HTTP request: %v", err)
}
w := httptest.NewRecorder()
// Check that it is rejected
handler.ServeHTTP(w, req)
if got, want := w.Code, http.StatusMethodNotAllowed; got != want {
t.Errorf("got HTTP code %v but wanted %v for endpoint %q", got, want, handler.Endpoint)
}
}
}
func TestPath(t *testing.T) {
instance := &Instance{
Config: Config{
Prefix: "testonly",
},
}
handler := Handler{
Instance: instance,
Handler: addLeaf,
Endpoint: types.EndpointAddLeaf,
Method: http.MethodPost,
}
if got, want := handler.Path(), "testonly/st/v0/add-leaf"; got != want {
t.Errorf("got path %v but wanted %v", got, want)
}
}
|
