diff options
author | Rasmus Dahlberg <rasmus.dahlberg@kau.se> | 2020-11-25 13:01:03 +0100 |
---|---|---|
committer | Rasmus Dahlberg <rasmus.dahlberg@kau.se> | 2020-11-25 13:01:03 +0100 |
commit | 2a442c1997cbd5b125a47727c86ecb14a27d1c01 (patch) | |
tree | 449a9fa23a4e0c0541c959f788028287b8bb6eda /reqres_test.go | |
parent | 2391eaeb19c810df86af7591af1575a6d6eb5d4b (diff) |
fixed file typo and added more tests
- Added missing continue in TestNewGetEntries(), and checked all values
in the returned request
- Added TestNewGetProofByHashResponse()
Diffstat (limited to 'reqres_test.go')
-rw-r--r-- | reqres_test.go | 193 |
1 files changed, 193 insertions, 0 deletions
diff --git a/reqres_test.go b/reqres_test.go new file mode 100644 index 0000000..e118a4f --- /dev/null +++ b/reqres_test.go @@ -0,0 +1,193 @@ +package stfe + +import ( + "bytes" + "fmt" + "strconv" + "testing" + + "net/http" +) + +// TODO: TestNewAddEntryRequest +func TestNewAddEntryRequest(t *testing.T) { +} + +func TestNewGetEntriesRequest(t *testing.T) { + lp := makeTestLogParameters(t, nil) + for _, table := range []struct { + description string + start string + end string + wantErr bool + }{ + { + description: "bad request: start must be an integer", + start: "start", + end: "10", + wantErr: true, + }, + { + description: "bad request: end must be an integer", + start: "10", + end: "end", + wantErr: true, + }, + { + description: "bad request: start must not be negative", + start: "-1", + end: "10", + wantErr: true, + }, + { + description: "bad request: start must be larger than end", + start: "1", + end: "0", + wantErr: true, + }, + { + description: "ok request but bad response: expected truncated", + start: "0", + end: fmt.Sprintf("%d", testMaxRange), + }, + { + description: "ok request and response", + start: "0", + end: "0", + }, + { + description: "ok request and response", + start: "0", + end: fmt.Sprintf("%d", testMaxRange-1), + }, + } { + r, err := http.NewRequest("GET", "http://example.com/"+lp.Prefix+"/get-entries", nil) + if err != nil { + t.Fatalf("must make http request in test %q: %v", table.description, err) + } + q := r.URL.Query() + q.Add("start", table.start) + q.Add("end", table.end) + r.URL.RawQuery = q.Encode() + + req, err := lp.newGetEntriesRequest(r) + if got, want := err != nil, table.wantErr; got != want { + t.Errorf("got error is %v but wanted %v in test %q: %v", got, want, table.description, err) + } + if err != nil { + continue + } + + if got, want := req.Start, mustParseInt64(t, table.start); got != want { + t.Errorf("got start %d but wanted %d in test %q", got, want, table.description) + } + if got, want := req.End, min(mustParseInt64(t, table.end), req.Start+testMaxRange-1); got != want { + t.Errorf("got end %d but wanted %d in test %q", got, want, table.description) + } + } +} + +func TestNewGetProofByHashRequest(t *testing.T) { + lp := makeTestLogParameters(t, nil) + for _, table := range []struct { + description string + treeSize string + hash string + wantErr bool + }{ + { + description: "bad request: tree size must be an integer", + treeSize: "treeSize", + hash: b64(testNodeHash), + wantErr: true, + }, + { + description: "bad request: tree size must be larger than zero", + treeSize: "0", + hash: b64(testNodeHash), + wantErr: true, + }, + { + description: "bad request: hash is not base64", + treeSize: "1", + hash: "<(^_^)>", + wantErr: true, + }, + { + description: "bad request: invalid node hash (too small)", + treeSize: "1", + hash: b64(testNodeHash[1:]), + wantErr: true, + }, + { + description: "bad request: invalid node hash (too large)", + treeSize: "1", + hash: b64(append(testNodeHash, byte(0))), + wantErr: true, + }, + { + description: "ok request", + treeSize: "1", + hash: b64(testNodeHash), + }, + } { + r, err := http.NewRequest("GET", "http://example.com/"+lp.Prefix+"/get-proof-by-hash", nil) + if err != nil { + t.Fatalf("must make http request in test %q: %v", table.description, err) + } + q := r.URL.Query() + q.Add("tree_size", table.treeSize) + q.Add("hash", table.hash) + r.URL.RawQuery = q.Encode() + + req, err := lp.newGetProofByHashRequest(r) + if got, want := err != nil, table.wantErr; got != want { + t.Errorf("got error is %v but wanted %v in test %q: %v", got, want, table.description, err) + } + if err != nil { + continue + } + + if got, want := req.TreeSize, mustParseInt64(t, table.treeSize); got != want { + t.Errorf("got treeSize %d but wanted %d in test %q", got, want, table.description) + } + if got, want := req.Hash, mustDeb64(t, table.hash); !bytes.Equal(got, want) { + t.Errorf("got hash %X but wanted %X in test %q", got, want, table.description) + } + } +} + +// TODO: TestNewGetConsistencyProofRequest +func TestNewGetConsistencyProofRequest(t *testing.T) { +} + +// TODO: TestNewGetEntryResponse +func TestNewGetEntriesResponse(t *testing.T) { +} + +// TODO: TestNewGetAnchorsResponse +func TestNewGetAnchorsResponse(t *testing.T) { +} + +func mustParseInt64(t *testing.T, num string) int64 { + n, err := strconv.ParseInt(num, 10, 64) + if err != nil { + t.Fatalf("must parse int: %v", err) + } + return n +} + +func mustDeb64(t *testing.T, str string) []byte { + b, err := deb64(str) + if err != nil { + t.Fatalf("must base64 decode: %v", err) + } + return b +} + +func min(a, b int64) int64 { + if a < b { + return a + } + return b +} |