Commit 505c26ad authored by Janos Guljas's avatar Janos Guljas

fix p2p streamtest TestRecorder_withMiddlewares test

parent be13de1d
...@@ -203,6 +203,16 @@ func TestRecorder_withMiddlewares(t *testing.T) { ...@@ -203,6 +203,16 @@ func TestRecorder_withMiddlewares(t *testing.T) {
}), }),
), ),
streamtest.WithMiddlewares( streamtest.WithMiddlewares(
func(h p2p.HandlerFunc) p2p.HandlerFunc {
return func(peer p2p.Peer, stream p2p.Stream) error {
if err := h(peer, stream); err != nil {
return err
}
// close stream after all previous middlewares wrote to it
// so that the receiving peer can get all the post messages
return stream.Close()
}
},
func(h p2p.HandlerFunc) p2p.HandlerFunc { func(h p2p.HandlerFunc) p2p.HandlerFunc {
return func(peer p2p.Peer, stream p2p.Stream) error { return func(peer p2p.Peer, stream p2p.Stream) error {
if _, err := stream.Write([]byte("pre 1, ")); err != nil { if _, err := stream.Write([]byte("pre 1, ")); err != nil {
...@@ -247,16 +257,6 @@ func TestRecorder_withMiddlewares(t *testing.T) { ...@@ -247,16 +257,6 @@ func TestRecorder_withMiddlewares(t *testing.T) {
return nil return nil
} }
}, },
func(h p2p.HandlerFunc) p2p.HandlerFunc {
return func(peer p2p.Peer, stream p2p.Stream) error {
if err := h(peer, stream); err != nil {
return err
}
// close stream after all previous middlewares wrote to it
// so that the receiving peer can get all the post messages
return stream.Close()
}
},
), ),
) )
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment