diff options
Diffstat (limited to 'dex/nodetable_test.go')
-rw-r--r-- | dex/nodetable_test.go | 17 |
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)) } |