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
|
package wgcfg
import (
"bytes"
"testing"
)
func TestKeyBasics(t *testing.T) {
pk1, err := NewPrivateKey()
if err != nil {
t.Fatal(err)
}
k1 := pk1.Public()
t.Run("second key", func(t *testing.T) {
// Different keys should be different.
pk2, err := NewPrivateKey()
if err != nil {
t.Fatal(err)
}
k2 := pk2.Public()
if bytes.Equal(k1[:], k2[:]) {
t.Fatalf("k1 %v == k2 %v", k1[:], k2[:])
}
// Check for obvious comparables to make sure we are not generating bad strings somewhere.
if b1, b2 := k1.String(), k2.String(); b1 == b2 {
t.Fatalf("base64-encoded keys match: %s, %s", b1, b2)
}
if pub1, pub2 := pk1.Public().String(), pk2.Public().String(); pub1 == pub2 {
t.Fatalf("base64-encoded public keys match: %s, %s", pub1, pub2)
}
})
}
|