p2p: tmconn->conn and types->p2p

p2p/key_test.go

@@ -0,0 +1,50 @@
+package p2p
+
+import (
+	"bytes"
+	"os"
+	"path/filepath"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	cmn "github.com/tendermint/tmlibs/common"
+)
+
+func TestLoadOrGenNodeKey(t *testing.T) {
+	filePath := filepath.Join(os.TempDir(), cmn.RandStr(12)+"_peer_id.json")
+
+	nodeKey, err := LoadOrGenNodeKey(filePath)
+	assert.Nil(t, err)
+
+	nodeKey2, err := LoadOrGenNodeKey(filePath)
+	assert.Nil(t, err)
+
+	assert.Equal(t, nodeKey, nodeKey2)
+}
+
+//----------------------------------------------------------
+
+func padBytes(bz []byte, targetBytes int) []byte {
+	return append(bz, bytes.Repeat([]byte{0xFF}, targetBytes-len(bz))...)
+}
+
+func TestPoWTarget(t *testing.T) {
+
+	targetBytes := 20
+	cases := []struct {
+		difficulty uint
+		target     []byte
+	}{
+		{0, padBytes([]byte{}, targetBytes)},
+		{1, padBytes([]byte{127}, targetBytes)},
+		{8, padBytes([]byte{0}, targetBytes)},
+		{9, padBytes([]byte{0, 127}, targetBytes)},
+		{10, padBytes([]byte{0, 63}, targetBytes)},
+		{16, padBytes([]byte{0, 0}, targetBytes)},
+		{17, padBytes([]byte{0, 0, 127}, targetBytes)},
+	}
+
+	for _, c := range cases {
+		assert.Equal(t, MakePoWTarget(c.difficulty, 20*8), c.target)
+	}
+}