瀏覽代碼

Merge pull request #2970 from yichengq/fix-stream-test

rafthttp: use buffered channel as recv/prop chan
Xiang Li 10 年之前
父節點
當前提交
684c721307
共有 1 個文件被更改,包括 4 次插入4 次删除
  1. 4 4
      rafthttp/stream_test.go

+ 4 - 4
rafthttp/stream_test.go

@@ -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)
 		}
 		}
 	}
 	}
 }
 }