From 7e0f0f84eac2e37edfd177196ed65afa0559f967 Mon Sep 17 00:00:00 2001
From: Rasmus Dahlberg <rasmus@mullvad.net>
Date: Mon, 20 Dec 2021 14:37:43 +0100
Subject: types: Add types and tests

---
 pkg/types/crypto_test.go | 64 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 64 insertions(+)
 create mode 100644 pkg/types/crypto_test.go

(limited to 'pkg/types/crypto_test.go')

diff --git a/pkg/types/crypto_test.go b/pkg/types/crypto_test.go
new file mode 100644
index 0000000..d95d5fa
--- /dev/null
+++ b/pkg/types/crypto_test.go
@@ -0,0 +1,64 @@
+package types
+
+import (
+	"crypto"
+	"crypto/ed25519"
+	"crypto/rand"
+	"io"
+	"testing"
+)
+
+type testSigner struct {
+	PublicKey PublicKey
+	Signature Signature
+	Error     error
+}
+
+func (ts *testSigner) Public() crypto.PublicKey {
+	return ed25519.PublicKey(ts.PublicKey[:])
+}
+
+func (ts *testSigner) Sign(rand io.Reader, digest []byte, opts crypto.SignerOpts) ([]byte, error) {
+	return ts.Signature[:], ts.Error
+}
+
+func newKeyPair(t *testing.T) (crypto.Signer, PublicKey) {
+	vk, sk, err := ed25519.GenerateKey(rand.Reader)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	var pub PublicKey
+	copy(pub[:], vk[:])
+	return sk, pub
+}
+
+func newHashBufferInc(t *testing.T) *Hash {
+	t.Helper()
+
+	var buf Hash
+	for i := 0; i < len(buf); i++ {
+		buf[i] = byte(i)
+	}
+	return &buf
+}
+
+func newSigBufferInc(t *testing.T) *Signature {
+	t.Helper()
+
+	var buf Signature
+	for i := 0; i < len(buf); i++ {
+		buf[i] = byte(i)
+	}
+	return &buf
+}
+
+func newPubBufferInc(t *testing.T) *PublicKey {
+	t.Helper()
+
+	var buf PublicKey
+	for i := 0; i < len(buf); i++ {
+		buf[i] = byte(i)
+	}
+	return &buf
+}
-- 
cgit v1.2.3