diff options
| author | Sonic <sonic@dexon.org> | 2019-02-12 16:16:13 +0800 |
|---|---|---|
| committer | Wei-Ning Huang <w@byzantine-lab.io> | 2019-06-12 17:27:22 +0800 |
| commit | 5ea856585a994390c0f22f11868086a985c1f3f7 (patch) | |
| tree | 46e344ef50c5f41ba82503894f5aeeda6f017c1d /dex/handler_test.go | |
| parent | 003088b6787dae1b718785ddddfe361d2fe5c505 (diff) | |
| download | go-tangerine-5ea856585a994390c0f22f11868086a985c1f3f7.tar go-tangerine-5ea856585a994390c0f22f11868086a985c1f3f7.tar.gz go-tangerine-5ea856585a994390c0f22f11868086a985c1f3f7.tar.bz2 go-tangerine-5ea856585a994390c0f22f11868086a985c1f3f7.tar.lz go-tangerine-5ea856585a994390c0f22f11868086a985c1f3f7.tar.xz go-tangerine-5ea856585a994390c0f22f11868086a985c1f3f7.tar.zst go-tangerine-5ea856585a994390c0f22f11868086a985c1f3f7.zip | |
dex: Add a flag to GetBlockHeadersMsg and GetBlockBodiesMsg (#188)
* dex: Add a flag to GetBlockHeadersMsg and GetBlockBodiesMsg
So that we can dispatch the response msg to fetcher or downloader
easily.
* fixup! dex: Add a flag to GetBlockHeadersMsg and GetBlockBodiesMsg
Diffstat (limited to 'dex/handler_test.go')
| -rw-r--r-- | dex/handler_test.go | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/dex/handler_test.go b/dex/handler_test.go index d8398bd2d..75a57c125 100644 --- a/dex/handler_test.go +++ b/dex/handler_test.go @@ -205,7 +205,7 @@ func testGetBlockHeaders(t *testing.T, protocol int) { } // Send the hash request and verify the response p2p.Send(peer.app, 0x03, tt.query) - if err := p2p.ExpectMsg(peer.app, 0x04, headers); err != nil { + if err := p2p.ExpectMsg(peer.app, 0x04, headersData{Headers: headers}); err != nil { t.Errorf("test %d: headers mismatch: %v", i, err) } // If the test used number origins, repeat with hashes as the too @@ -214,7 +214,7 @@ func testGetBlockHeaders(t *testing.T, protocol int) { tt.query.Origin.Hash, tt.query.Origin.Number = origin.Hash(), 0 p2p.Send(peer.app, 0x03, tt.query) - if err := p2p.ExpectMsg(peer.app, 0x04, headers); err != nil { + if err := p2p.ExpectMsg(peer.app, 0x04, headersData{Headers: headers}); err != nil { t.Errorf("test %d: headers mismatch: %v", i, err) } } @@ -287,8 +287,8 @@ func testGetBlockBodies(t *testing.T, protocol int) { } } // Send the hash request and verify the response - p2p.Send(peer.app, 0x05, hashes) - if err := p2p.ExpectMsg(peer.app, 0x06, bodies); err != nil { + p2p.Send(peer.app, 0x05, []interface{}{downloaderReq, hashes}) + if err := p2p.ExpectMsg(peer.app, 0x06, blockBodiesData{Flag: downloaderReq, Bodies: bodies}); err != nil { t.Errorf("test %d: bodies mismatch: %v", i, err) } } |