diff options
| -rw-r--r-- | handler_test.go | 121 |
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 + } +} |