@@ -14,7 +14,7 @@ import (
14
14
"github.com/celestiaorg/go-header/headertest"
15
15
)
16
16
17
- // TestSubscriber tests the header Module's implementation of Subscriber .
17
+ // TestSubscriber a simple test to check if the subscriber can receive headers from the network .
18
18
func TestSubscriber (t * testing.T ) {
19
19
ctx , cancel := context .WithTimeout (context .Background (), time .Second * 15 )
20
20
defer cancel ()
@@ -73,23 +73,24 @@ func TestSubscriber(t *testing.T) {
73
73
}
74
74
75
75
// subscribe
76
- _ , err = p2pSub2 .Subscribe ()
76
+ senderSubscription , err : = p2pSub2 .Subscribe ()
77
77
require .NoError (t , err )
78
78
79
79
subscription , err := p2pSub1 .Subscribe ()
80
80
require .NoError (t , err )
81
81
82
82
expectedHeader := suite .GenDummyHeaders (1 )[0 ]
83
- bin , err := expectedHeader .MarshalBinary ()
84
- require .NoError (t , err )
85
-
86
- err = p2pSub2 .topic .Publish (ctx , bin , pubsub .WithReadiness (pubsub .MinTopicSize (1 )))
83
+ err = p2pSub2 .Broadcast (ctx , expectedHeader , pubsub .WithReadiness (pubsub .MinTopicSize (1 )))
87
84
require .NoError (t , err )
88
85
89
86
// get next Header from network
90
87
header , err := subscription .NextHeader (ctx )
91
88
require .NoError (t , err )
89
+ assert .Equal (t , expectedHeader .Height (), header .Height ())
90
+ assert .Equal (t , expectedHeader .Hash (), header .Hash ())
92
91
92
+ header , err = senderSubscription .NextHeader (ctx )
93
+ require .NoError (t , err )
93
94
assert .Equal (t , expectedHeader .Height (), header .Height ())
94
95
assert .Equal (t , expectedHeader .Hash (), header .Hash ())
95
96
}
0 commit comments