1package node
 2
 3import (
 4	"io/ioutil"
 5	"testing"
 6
 7	"github.com/arnaucube/slowlorisdb/core"
 8	"github.com/arnaucube/slowlorisdb/db"
 9	"github.com/stretchr/testify/assert"
10)
11
12func TestNode(t *testing.T) {
13	dir, err := ioutil.TempDir("", "db")
14	assert.Nil(t, err)
15	db, err := db.New(dir)
16	assert.Nil(t, err)
17
18	node, err := NewNode(db)
19	assert.Nil(t, err)
20
21	assert.Equal(t, node.Addr, core.AddressFromPrivK(node.PrivK))
22}
23
24func TestNodeSignature(t *testing.T) {
25	dir, err := ioutil.TempDir("", "db")
26	assert.Nil(t, err)
27	db, err := db.New(dir)
28	assert.Nil(t, err)
29
30	node, err := NewNode(db)
31	assert.Nil(t, err)
32
33	m := []byte("test")
34	sig, err := node.Sign(m)
35	assert.Nil(t, err)
36	pubK := node.PrivK.PublicKey
37	assert.True(t, core.VerifySignature(&pubK, m, *sig))
38}