rename (#465)

[bytom/vapor.git] / p2p / discover / dht / net_test.go

diff --git a/p2p/discover/dht/net_test.go b/p2p/discover/dht/net_test.go
index eacdaa8..9b063de 100644 (file)
--- a/p2p/discover/dht/net_test.go
+++ b/p2p/discover/dht/net_test.go
@@ -8,8 +8,8 @@ import (
        "testing"
        "time"
 
-       "github.com/vapor/common"
-       "github.com/vapor/crypto/ed25519"
+       "github.com/bytom/vapor/common"
+       "github.com/bytom/vapor/p2p/signlib"
 )
 
 func TestNetwork_Lookup(t *testing.T) {
@@ -19,8 +19,12 @@ func TestNetwork_Lookup(t *testing.T) {
        }
        defer os.RemoveAll(tmpDir)
 
-       pubKey, _, err := ed25519.GenerateKey(nil)
-       network, err := newNetwork(lookupTestnet, pubKey, tmpDir, nil)
+       privKey, err := signlib.NewPrivKey()
+       if err != nil {
+               t.Fatal(err)
+       }
+
+       network, err := newNetwork(lookupTestnet, privKey.XPub(), tmpDir, nil)
        if err != nil {
                t.Fatal(err)
        }
@@ -328,7 +332,8 @@ func mine(target NodeID) {
        var dists [hashBits + 1][]NodeID
        found := 0
        for found < bucketSize*10 {
-               pubKey, _, _ := ed25519.GenerateKey(nil)
+               privKey, _ := signlib.NewPrivKey()
+               pubKey := privKey.XPub()
                sha := hash(pubKey[:])
                ld := logdist(targetSha, sha)
                if len(dists[ld]) < bucketSize {