1 files changed, 78 insertions, 0 deletions
diff --git a/pkg/oaddr/oaddr_test.go b/pkg/oaddr/oaddr_test.go
new file mode 100644
index 0000000..a37d4de
--- /dev/null
+++ b/pkg/oaddr/oaddr_test.go
@@ -0,0 +1,78 @@
+package oaddr
+
+import (
+	"crypto"
+	"crypto/ed25519"
+	"testing"
+
+	"sauteed-onions.org/onion-csr/internal/testonly"
+)
+
+const (
+	testPriv = "a4007fabb23fae0f50fc45481553bf7d5d26b9fd8d76142c572606a6ebd7a2c1"
+	testPub  = "da67efa9e06e724d999f8e0409b4a5a08aebd26005b63bef90d51a241d631cfd"
+	testAddr = "3jt67kpanzze3gm7rycatnffucfoxutaaw3dx34q2uncihlddt6tq3ad.onion"
+)
+
+func TestNew(t *testing.T) {
+	for _, table := range []struct {
+		desc string
+		pub  []byte
+		want OnionAddress
+	}{
+		{"too short key", testonly.DecodeHex(t, testPub)[1:], newAddr(t, testPub)},
+		{"too long key", append(testonly.DecodeHex(t, testPub), 0xff), newAddr(t, testPub)},
+		{"valid", testonly.DecodeHex(t, testPub), newAddr(t, testPub)},
+	} {
+		addr, err := New(table.pub)
+		if got, want := err != nil, table.desc != "valid"; got != want {
+			t.Errorf("%s: got error %v but wanted %v: %v", table.desc, got, want, err)
+		}
+		if err != nil {
+			continue
+		}
+		if got, want := addr, table.want; got != want {
+			t.Errorf("%s: got address\n%x\nbut wanted\n%x", table.desc, got[:], want[:])
+		}
+	}
+}
+
+func TestNewFromSigner(t *testing.T) {
+	for _, table := range []struct {
+		desc string
+		priv crypto.Signer
+		want OnionAddress
+	}{
+		{"rsa key", testonly.RSAPriv(t), OnionAddress{}},
+		{"valid", testonly.Ed25519Priv(t, testPriv), newAddr(t, testPub)},
+	} {
+		addr, err := NewFromSigner(table.priv)
+		if got, want := err != nil, table.desc != "valid"; got != want {
+			t.Errorf("%s: got error %v but wanted %v: %v", table.desc, got, want, err)
+		}
+		if err != nil {
+			continue
+		}
+		if got, want := addr, table.want; got != want {
+			t.Errorf("%s: got address\n%x\nbut wanted\n%x", table.desc, got[:], want[:])
+		}
+	}
+}
+
+func TestString(t *testing.T) {
+	addr := newAddr(t, testPub)
+	want := testAddr
+	if got, want := addr.String(), want; got != want {
+		t.Errorf("got address\n%s\nbut wanted\n%s", got, want)
+	}
+}
+
+func newAddr(t *testing.T, pub string) (addr OnionAddress) {
+	t.Helper()
+	b := testonly.DecodeHex(t, pub)
+	if got, want := len(b), ed25519.PublicKeySize; got != want {
+		t.Fatalf("invalid public key size: %d", got)
+	}
+	copy(addr[:], b)
+	return
+}