瀏覽代碼

Add timeout to async_producer_test

Maxim Vladimirskiy 7 年之前
父節點
當前提交
1596275bc3
共有 1 個文件被更改,包括 4 次插入1 次删除
  1. 4 1
      async_producer_test.go

+ 4 - 1
async_producer_test.go

@@ -131,9 +131,12 @@ func TestAsyncProducer(t *testing.T) {
 			if msg.Metadata.(int) != i {
 			if msg.Metadata.(int) != i {
 				t.Error("Message metadata did not match")
 				t.Error("Message metadata did not match")
 			}
 			}
+		case <-time.After(time.Second):
+			t.Errorf("Timeout waiting for msg #%d", i)
+			goto done
 		}
 		}
 	}
 	}
-
+done:
 	closeProducer(t, producer)
 	closeProducer(t, producer)
 	leader.Close()
 	leader.Close()
 	seedBroker.Close()
 	seedBroker.Close()