move test helpers to testdata package

1 files changed, 47 insertions, 0 deletions

diff --git a/server/testdata/helper.go b/server/testdata/helper.go
new file mode 100644
index 0000000..6874616
--- /dev/null
+++ b/server/testdata/helper.go
@@ -0,0 +1,47 @@
+package testdata
+
+import (
+	"testing"
+
+	"github.com/google/trillian"
+	"github.com/google/trillian/types"
+)
+
+// NewGetLatestSignedLogRootResponse creates a new trillian STH.  Revision,
+// Metadata, Proof, KeyHint, and LogRootSignature are unsset.
+func NewGetLatestSignedLogRootResponse(t *testing.T, timestamp, size uint64, hash []byte) *trillian.GetLatestSignedLogRootResponse {
+	t.Helper()
+	return &trillian.GetLatestSignedLogRootResponse{
+		SignedLogRoot: marshalSignedLogRoot(t, &types.LogRootV1{
+			TreeSize:       size,
+			RootHash:       hash,
+			TimestampNanos: timestamp,
+			Revision:       0,   // not used by stfe
+			Metadata:       nil, // not used by stfe
+		}),
+		Proof: nil, // not used by stfe
+	}
+}
+
+// TruncatedSignedLogRootResponse creates a truncated signed log root response
+// that cannot be unmarshalled, i.e., SignedLogRoot.LogRoot is invalid.
+func TruncatedSignedLogRootResponse(t *testing.T) *trillian.GetLatestSignedLogRootResponse {
+	t.Helper()
+	slrr := NewGetLatestSignedLogRootResponse(t, 0, 0, make([]byte, 32))
+	slrr.SignedLogRoot.LogRoot = slrr.SignedLogRoot.LogRoot[1:]
+	return slrr
+}
+
+// marshalSignedLogRoot must marshal a signed log root
+func marshalSignedLogRoot(t *testing.T, lr *types.LogRootV1) *trillian.SignedLogRoot {
+	t.Helper()
+	rootBytes, err := lr.MarshalBinary()
+	if err != nil {
+		t.Fatalf("failed to marshal root in test: %v", err)
+	}
+	return &trillian.SignedLogRoot{
+		KeyHint:          nil, // not used by stfe
+		LogRoot:          rootBytes,
+		LogRootSignature: nil, // not used by stfe
+	}
+}