dex: polish network related function

1 files changed, 12 insertions, 5 deletions

diff --git a/dex/nodetable_test.go b/dex/nodetable_test.go
index 5b3f7de57..2e44eabe5 100644
--- a/dex/nodetable_test.go
+++ b/dex/nodetable_test.go
@@ -1,11 +1,13 @@
 package dex
 
 import (
-	"math/rand"
+	"crypto/ecdsa"
 	"testing"
 	"time"
 
 	"github.com/dexon-foundation/dexon/common"
+	"github.com/dexon-foundation/dexon/crypto"
+	"github.com/dexon-foundation/dexon/p2p/discover"
 	"github.com/dexon-foundation/dexon/p2p/enode"
 )
 
@@ -85,9 +87,14 @@ func TestNodeTable(t *testing.T) {
 	}
 }
 
-func randomID() (id enode.ID) {
-	for i := range id {
-		id[i] = byte(rand.Intn(255))
+func randomID() enode.ID {
+	var err error
+	var privkey *ecdsa.PrivateKey
+	for {
+		privkey, err = crypto.GenerateKey()
+		if err == nil {
+			break
+		}
 	}
-	return id
+	return discover.PubkeyID(&(privkey.PublicKey))
 }