aboutsummaryrefslogtreecommitdiff
path: root/key/gen_test.go
blob: 7f86b4e528a4155b8c40de27c738fc96baa2cd99 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
package key

import (
	"bytes"
	"encoding"
	"encoding/base64"
	"testing"
)

type testInterface interface {
	encoding.BinaryUnmarshaler
	encoding.BinaryMarshaler
	Validate() error
}

func TestKeys(t *testing.T) {
	var testCases = []struct {
		caption  string
		key      testInterface
		rawValue string
	}{
		{
			caption:  "sig",
			key:      new(Sig),
			rawValue: "RWRbOC0bBf7abaGwGtq45KLDK63tgcF7CO4qTZSlTKCSbZTYlDfFm/DISQ60u+/jEzrk22suvXXAEsxQTe2xUOfV90get1YRGQo=",
		},
		{
			caption:  "pub",
			key:      new(Pub),
			rawValue: "RWRbOC0bBf7abfanaXuTYfCa6+YO69Kxyz8RD5nL/3Ta7umY6iOwnBrG",
		},
		{
			caption:  "sec",
			key:      new(Sec),
			rawValue: "RWRCSwAAAACzJBN2gC5//jVvDiV76rs4m2aKXkljqDpbOC0bBf7abZhV/Zygr6b0KIbSI56JQutwzsQeouxnnHuVTZp3IW4M9qdpe5Nh8Jrr5g7r0rHLPxEPmcv/dNru6ZjqI7CcGsY=",
		},
	}
	for _, tc := range testCases {
		t.Run(tc.caption, func(t *testing.T) {
			raw, err := base64.StdEncoding.DecodeString(tc.rawValue)
			if err != nil {
				t.Fatal(err)
			}
			if err := tc.key.UnmarshalBinary(raw); err != nil {
				t.Fatal(err)
			}
			if err := tc.key.Validate(); err != nil {
				t.Fatal(err)
			}
			out, err := tc.key.MarshalBinary()
			if err != nil {
				t.Fatal(err)
			}
			if !bytes.Equal(raw, out) {
				t.Errorf("want %v, got %v", raw, out)
			}
		})
	}
}

type mocDeriver struct{}

func (mocDeriver) Derive(_ []byte, _, length int) ([]byte, error) {
	return make([]byte, length), nil
}

func TestPublic(t *testing.T) {
	pub, sec, err := Generate(mocDeriver{})
	if err != nil {
		t.Fatal(err)
	}
	p := sec.Public()
	if *p != *pub {
		t.Errorf("got %v, want %v", p, pub)
	}
}

func TestSign(t *testing.T) {
	pub, sec, err := Generate(mocDeriver{})
	if err != nil {
		t.Fatal(err)
	}
	msg := []byte{}
	sig := sec.Sign(msg)
	if err := sig.Verify(msg, pub); err != nil {
		t.Error(err)
	}
}