Repository navigation
Expand file tree
/
Copy pathprivate_test.go
More file actions
102 lines (90 loc) · 2.71 KB
/
Copy pathprivate_test.go
File metadata and controls
102 lines (90 loc) · 2.71 KB
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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
// Copyright (c) 2024 Andreas Auernhammer. All rights reserved.
// Use of this source code is governed by a license that can be
// found in the LICENSE file.
package minisign
import (
"bytes"
"encoding/base64"
"os"
"testing"
)
var marshalPrivateKeyTests = []struct {
File string
ID uint64
Bytes []byte
}{
{
File: "./internal/testdata/minisign-nopassword-0.key",
ID: 0x3728470A8118E56E,
Bytes: b64("JpjEI/XIKqIVl99tT611AxXwlVjlw2afJC8Nv6o7uuipyNvC3DmgO2csDT+bw1bZR3ss4rd5cXqoq0uftlCJqw=="),
},
{
File: "./internal/testdata/minisign-nopassword-1.key",
ID: 0xD7E531EE76B2FC6F,
Bytes: b64("L24Gi2UbWOb/MBb4MzJLysgC1F1FnE/m72qhb7r5FMlHzHe6M6mCLPMzmj6ln+hI51kqpDqTkIg9VCaToAhZtA=="),
},
}
func TestPrivateKey_Marshal(t *testing.T) {
for i, test := range marshalPrivateKeyTests {
raw, err := os.ReadFile(test.File)
if err != nil {
t.Fatalf("Failed to read private key: %v", err)
}
raw = bytes.ReplaceAll(raw, []byte{'\r', '\n'}, []byte{'\n'})
raw = bytes.TrimRight(raw, "\n")
key := PrivateKey{
id: test.ID,
}
copy(key.bytes[:], test.Bytes)
text, err := key.MarshalText()
if err != nil {
t.Fatalf("Test %d: failed to marshal private key: %v", i, err)
}
if !bytes.Equal(text, raw) {
t.Fatalf("Test %d: failed to marshal private key:\nGot: %v\nWant: %v\n", i, text, raw)
}
}
}
var unmarshalPrivateKeyTests = []struct {
File string
ID uint64
Bytes []byte
}{
{
File: "./internal/testdata/minisign-nopassword-0.key",
ID: 0x3728470A8118E56E,
Bytes: b64("JpjEI/XIKqIVl99tT611AxXwlVjlw2afJC8Nv6o7uuipyNvC3DmgO2csDT+bw1bZR3ss4rd5cXqoq0uftlCJqw=="),
},
{
File: "./internal/testdata/minisign-nopassword-1.key",
ID: 0xD7E531EE76B2FC6F,
Bytes: b64("L24Gi2UbWOb/MBb4MzJLysgC1F1FnE/m72qhb7r5FMlHzHe6M6mCLPMzmj6ln+hI51kqpDqTkIg9VCaToAhZtA=="),
},
}
func TestPrivateKey_Unmarshal(t *testing.T) {
for i, test := range unmarshalPrivateKeyTests {
raw, err := os.ReadFile(test.File)
if err != nil {
t.Fatalf("Test %d: failed to read private key: %v", i, err)
}
var key PrivateKey
if err := key.UnmarshalText(raw); err != nil {
t.Fatalf("Test %d: failed to unmarshal private key: %v\nPrivate key:\n%s", i, err, string(raw))
}
// Print test vector for marshaling:
// t.Logf("\n{\n\tID: htoi(\"%X\"),\n\tBytes: b64(\"%s\"),\n}", key.id, base64.StdEncoding.EncodeToString(key.bytes[:]))
if key.ID() != test.ID {
t.Fatalf("Test %d: ID mismatch: got '%x' - want '%x'", i, key.ID(), test.ID)
}
if !bytes.Equal(key.bytes[:], test.Bytes) {
t.Fatalf("Test %d: private key mismatch: got '%x' - want '%x'", i, key.bytes, test.Bytes)
}
}
}
func b64(s string) []byte {
b, err := base64.StdEncoding.DecodeString(s)
if err != nil {
panic(err)
}
return b
}