arnaucube
arnaucube/slowlorisdbPublic Self-hosted repository
slowlorisdb/core/keys_test.go
core/keys_test.goRaw
 1package core
 2
 3import (
 4	"testing"
 5
 6	"github.com/stretchr/testify/assert"
 7)
 8
 9func TestNewKey(t *testing.T) {
10	_, err := NewKey()
11	assert.Nil(t, err)
12}
13func TestAddress(t *testing.T) {
14	privK, err := NewKey()
15	assert.Nil(t, err)
16
17	addr := AddressFromPrivK(privK)
18	assert.NotEqual(t, addr, Address{})
19}
20
21func TestSignAndVerify(t *testing.T) {
22	privK, err := NewKey()
23	assert.Nil(t, err)
24
25	// Sign
26	m := []byte("test")
27	sig, err := Sign(privK, m)
28	assert.Nil(t, err)
29
30	// Verify
31	verified := VerifySignature(&privK.PublicKey, m, *sig)
32	assert.True(t, verified)
33}