aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorRasmus Dahlberg <rasmus.dahlberg@kau.se>2020-11-16 17:32:24 +0100
committerRasmus Dahlberg <rasmus.dahlberg@kau.se>2020-11-16 17:32:24 +0100
commitf83c1d74b1ef5775b2d6b63e69b5ada5f1fc003d (patch)
tree1f1c306118532654bb65637b29b31f8002e68f10
parent7f06b2de5440a3523594daa2d74cf277c8af89cc (diff)
added get-consistency-proof handler tests
Ensures that the respective error handling functions are invoked.
-rw-r--r--handler_test.go121
1 files changed, 121 insertions, 0 deletions
diff --git a/handler_test.go b/handler_test.go
index 8ae0987..5992119 100644
--- a/handler_test.go
+++ b/handler_test.go
@@ -382,6 +382,115 @@ func TestGetSth(t *testing.T) {
}
}
+func TestGetConsistencyProof(t *testing.T) {
+ fixedProof := [][]byte{
+ make([]byte, 32),
+ make([]byte, 32),
+ }
+ for _, table := range []struct {
+ description string
+ breq *GetConsistencyProofRequest
+ trsp *trillian.GetConsistencyProofResponse
+ terr error
+ wantCode int
+ wantErrText string
+ }{
+ {
+ description: "bad request parameters",
+ breq: &GetConsistencyProofRequest{
+ First: 2,
+ Second: 1,
+ },
+ wantCode: http.StatusBadRequest,
+ wantErrText: http.StatusText(http.StatusBadRequest) + "\n",
+ },
+ {
+ description: "empty trillian response",
+ breq: &GetConsistencyProofRequest{
+ First: 1,
+ Second: 2,
+ },
+ terr: fmt.Errorf("back-end failure"),
+ wantCode: http.StatusInternalServerError,
+ wantErrText: http.StatusText(http.StatusInternalServerError) + "\n",
+ },
+ {
+ description: "valid request and response",
+ breq: &GetConsistencyProofRequest{
+ First: 1,
+ Second: 2,
+ },
+ trsp: makeTrillianGetConsistencyProofResponse(t, fixedProof),
+ wantCode: http.StatusOK,
+ },
+ } {
+ func() { // run deferred functions at the end of each iteration
+ th := newTestHandler(t, nil)
+ defer th.mockCtrl.Finish()
+
+ url := "http://example.com" + th.instance.LogParameters.Prefix + "/get-consistency-proof"
+ req, err := http.NewRequest("GET", url, nil)
+ if err != nil {
+ t.Fatalf("failed creating http request: %v", err)
+ }
+ q := req.URL.Query()
+ q.Add("first", fmt.Sprintf("%d", table.breq.First))
+ q.Add("second", fmt.Sprintf("%d", table.breq.Second))
+ req.URL.RawQuery = q.Encode()
+
+ w := httptest.NewRecorder()
+ if table.trsp != nil || table.terr != nil {
+ th.client.EXPECT().GetConsistencyProof(testdata.NewDeadlineMatcher(), gomock.Any()).Return(table.trsp, table.terr)
+ }
+ th.getHandler(t, "get-consistency-proof").ServeHTTP(w, req)
+ if w.Code != table.wantCode {
+ t.Errorf("GET(%s)=%d, want http status code %d", url, w.Code, table.wantCode)
+ }
+ body := w.Body.String()
+ if w.Code != http.StatusOK {
+ if body != table.wantErrText {
+ t.Errorf("GET(%s)=%q, want text %q", url, body, table.wantErrText)
+ }
+ return
+ }
+
+ // status code is http.StatusOK, check response
+ var data []byte
+ if err := json.Unmarshal([]byte(body), &data); err != nil {
+ t.Errorf("failed unmarshaling json: %v, wanted ok", err)
+ return
+ }
+ var item StItem
+ if err := item.Unmarshal(data); err != nil {
+ t.Errorf("failed unmarshaling StItem: %v, wanted ok", err)
+ return
+ }
+ if item.Format != StFormatConsistencyProofV1 {
+ t.Errorf("invalid StFormat: got %v, want %v", item.Format, StFormatInclusionProofV1)
+ }
+ proof := item.ConsistencyProofV1
+ if !bytes.Equal(proof.LogId, th.instance.LogParameters.LogId) {
+ t.Errorf("want log id %X, got %X", proof.LogId, th.instance.LogParameters.LogId)
+ }
+ if got, want := proof.TreeSize1, uint64(table.breq.First); got != want {
+ t.Errorf("want tree size %d, got %d", want, got)
+ }
+ if got, want := proof.TreeSize2, uint64(table.breq.Second); got != want {
+ t.Errorf("want tree size %d, got %d", want, got)
+ }
+ if got, want := len(proof.ConsistencyPath), len(fixedProof); got != want {
+ t.Errorf("want proof length %d, got %d", want, got)
+ return
+ }
+ for i, nh := range proof.ConsistencyPath {
+ if !bytes.Equal(nh.Data, fixedProof[i]) {
+ t.Errorf("want proof[%d]=%X, got %X", i, fixedProof[i], nh.Data)
+ }
+ }
+ }()
+ }
+}
+
func TestGetProofByHash(t *testing.T) {
fixedProof := [][]byte{
make([]byte, 32),
@@ -574,3 +683,15 @@ func makeTrillianGetInclusionProofByHashResponse(t *testing.T, index int64, path
SignedLogRoot: nil, // not used by stfe
}
}
+
+// makeTrillianGetInclusionProofByHashResponse
+func makeTrillianGetConsistencyProofResponse(t *testing.T, path [][]byte) *trillian.GetConsistencyProofResponse {
+ t.Helper()
+ return &trillian.GetConsistencyProofResponse{
+ Proof: &trillian.Proof{
+ LeafIndex: 0, // not used by consistency proofs
+ Hashes: path,
+ },
+ SignedLogRoot: nil, // not used by stfe
+ }
+}