|
@@ -221,8 +221,8 @@ func TestStreamReaderDialDetectUnsupport(t *testing.T) {
|
|
|
// TestStream tests that streamReader and streamWriter can build stream to
|
|
// TestStream tests that streamReader and streamWriter can build stream to
|
|
|
// send messages between each other.
|
|
// send messages between each other.
|
|
|
func TestStream(t *testing.T) {
|
|
func TestStream(t *testing.T) {
|
|
|
- recvc := make(chan raftpb.Message)
|
|
|
|
|
- propc := make(chan raftpb.Message)
|
|
|
|
|
|
|
+ recvc := make(chan raftpb.Message, streamBufSize)
|
|
|
|
|
+ propc := make(chan raftpb.Message, streamBufSize)
|
|
|
msgapp := raftpb.Message{
|
|
msgapp := raftpb.Message{
|
|
|
Type: raftpb.MsgApp,
|
|
Type: raftpb.MsgApp,
|
|
|
From: 2,
|
|
From: 2,
|
|
@@ -294,10 +294,10 @@ func TestStream(t *testing.T) {
|
|
|
select {
|
|
select {
|
|
|
case m = <-tt.wc:
|
|
case m = <-tt.wc:
|
|
|
case <-time.After(time.Second):
|
|
case <-time.After(time.Second):
|
|
|
- t.Errorf("#%d: failed to receive message from the channel", i)
|
|
|
|
|
|
|
+ t.Fatalf("#%d: failed to receive message from the channel", i)
|
|
|
}
|
|
}
|
|
|
if !reflect.DeepEqual(m, tt.m) {
|
|
if !reflect.DeepEqual(m, tt.m) {
|
|
|
- t.Errorf("#%d: message = %+v, want %+v", i, m, tt.m)
|
|
|
|
|
|
|
+ t.Fatalf("#%d: message = %+v, want %+v", i, m, tt.m)
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|