diff options
| author | Sonic <sonic@dexon.org> | 2019-01-31 19:40:39 +0800 |
|---|---|---|
| committer | Wei-Ning Huang <w@dexon.org> | 2019-01-31 19:40:39 +0800 |
| commit | ccec7e715e9e31d631668b6ae8451c9ac04f36bd (patch) | |
| tree | f7dcc202b9f1879d1da810a6f315757bb93cdb73 /dex/protocol_test.go | |
| parent | bae9b4372d2f5cfa5c24182d173c86872797576b (diff) | |
| download | dexon-ccec7e715e9e31d631668b6ae8451c9ac04f36bd.tar dexon-ccec7e715e9e31d631668b6ae8451c9ac04f36bd.tar.gz dexon-ccec7e715e9e31d631668b6ae8451c9ac04f36bd.tar.bz2 dexon-ccec7e715e9e31d631668b6ae8451c9ac04f36bd.tar.lz dexon-ccec7e715e9e31d631668b6ae8451c9ac04f36bd.tar.xz dexon-ccec7e715e9e31d631668b6ae8451c9ac04f36bd.tar.zst dexon-ccec7e715e9e31d631668b6ae8451c9ac04f36bd.zip | |
p2p, dex: rework connection management (#183)
* p2p, dex: rework connection management
* dex: refresh our node record periodically
* dex: don't send new record event if no new record
Diffstat (limited to 'dex/protocol_test.go')
| -rw-r--r-- | dex/protocol_test.go | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/dex/protocol_test.go b/dex/protocol_test.go index 5c68785e8..74778cdda 100644 --- a/dex/protocol_test.go +++ b/dex/protocol_test.go @@ -18,7 +18,6 @@ package dex import ( "crypto/ecdsa" - "encoding/hex" "fmt" "reflect" "sync" @@ -36,6 +35,7 @@ import ( "github.com/dexon-foundation/dexon/crypto" "github.com/dexon-foundation/dexon/dex/downloader" "github.com/dexon-foundation/dexon/p2p" + "github.com/dexon-foundation/dexon/p2p/enode" "github.com/dexon-foundation/dexon/p2p/enr" "github.com/dexon-foundation/dexon/rlp" ) @@ -553,11 +553,15 @@ func TestSendVote(t *testing.T) { }, } + pm.peers.label2Nodes = make(map[peerLabel]map[string]*enode.Node) for i, tt := range testPeers { p, _ := newTestPeer(fmt.Sprintf("peer #%d", i), dex64, pm, true) if tt.label != nil { - b := crypto.FromECDSAPub(p.Node().Pubkey()) - pm.peers.addDirectPeer(hex.EncodeToString(b), *tt.label) + if pm.peers.label2Nodes[*tt.label] == nil { + pm.peers.label2Nodes[*tt.label] = make(map[string]*enode.Node) + } + pm.peers.label2Nodes[*tt.label][p.ID().String()] = p.Node() + pm.peers.addDirectPeer(p.ID().String(), *tt.label) } wg.Add(1) go checkvote(p, tt.isReceiver) |