@@ -2,6 +2,7 @@ import { EventEmitter } from 'events'
2
2
import tape from 'tape-catch'
3
3
import td from 'testdouble'
4
4
import * as rlp from 'rlp'
5
+ import { ETH as Devp2pETH } from '@ethereumjs/devp2p'
5
6
import { RlpxSender } from '../../../lib/net/protocol'
6
7
7
8
tape ( '[RlpxSender]' , ( t ) => {
@@ -28,7 +29,7 @@ tape('[RlpxSender]', (t) => {
28
29
29
30
t . test ( 'should receive status' , ( t ) => {
30
31
const rlpxProtocol = new EventEmitter ( )
31
- const sender = new RlpxSender ( rlpxProtocol )
32
+ const sender = new RlpxSender ( rlpxProtocol as Devp2pETH )
32
33
sender . on ( 'status' , ( status : any ) => {
33
34
t . equal ( status . id , 5 , 'status received' )
34
35
t . equal ( sender . status . id , 5 , 'status getter' )
@@ -39,7 +40,7 @@ tape('[RlpxSender]', (t) => {
39
40
40
41
t . test ( 'should receive message' , ( t ) => {
41
42
const rlpxProtocol = new EventEmitter ( )
42
- const sender = new RlpxSender ( rlpxProtocol )
43
+ const sender = new RlpxSender ( rlpxProtocol as Devp2pETH )
43
44
sender . on ( 'message' , ( message : any ) => {
44
45
t . equal ( message . code , 1 , 'message received (code)' )
45
46
t . equal ( message . payload , 5 , 'message received (payload)' )
@@ -50,7 +51,7 @@ tape('[RlpxSender]', (t) => {
50
51
51
52
t . test ( 'should catch errors' , ( t ) => {
52
53
const rlpxProtocol = new EventEmitter ( )
53
- const sender = new RlpxSender ( rlpxProtocol )
54
+ const sender = new RlpxSender ( rlpxProtocol as Devp2pETH )
54
55
t . throws ( ( ) => sender . sendStatus ( { id : 5 } ) , / n o t a f u n c t i o n / , 'sendStatus error' )
55
56
t . throws ( ( ) => sender . sendMessage ( 1 , 5 ) , / n o t a f u n c t i o n / , 'sendMessage error' )
56
57
t . end ( )
0 commit comments