From 2a442c1997cbd5b125a47727c86ecb14a27d1c01 Mon Sep 17 00:00:00 2001
From: Rasmus Dahlberg <rasmus.dahlberg@kau.se>
Date: Wed, 25 Nov 2020 13:01:03 +0100
Subject: fixed file typo and added more tests

- Added missing continue in TestNewGetEntries(), and checked all values
in the returned request
- Added TestNewGetProofByHashResponse()
---
 reqres_test.go | 193 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 193 insertions(+)
 create mode 100644 reqres_test.go

(limited to 'reqres_test.go')

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
+}
-- 
cgit v1.2.3