Browse Source

Add a missing offset check to a test

May or may not fix another flaky test (e.g.
https://travis-ci.org/Shopify/sarama/jobs/66086521)
Evan Huus 10 years ago
parent
commit
aff914983c
1 changed files with 5 additions and 2 deletions
  1. 5 2
      consumer_test.go

+ 5 - 2
consumer_test.go

@@ -194,8 +194,11 @@ func TestConsumerFunnyOffsets(t *testing.T) {
 		t.Fatal(err)
 		t.Fatal(err)
 	}
 	}
 
 
-	message := <-consumer.Messages()
-	if message.Offset != 3 {
+	if message := <-consumer.Messages(); message.Offset != 3 {
+		t.Error("Incorrect message offset!")
+	}
+
+	if message := <-consumer.Messages(); message.Offset != 5 {
 		t.Error("Incorrect message offset!")
 		t.Error("Incorrect message offset!")
 	}
 	}