diff --git a/pkg/grillkafka/assertions.go b/pkg/grillkafka/assertions.go index ac79629..3db7c8b 100644 --- a/pkg/grillkafka/assertions.go +++ b/pkg/grillkafka/assertions.go @@ -12,7 +12,7 @@ import ( func (gk *Kafka) AssertCount(topicName string, expectedCount int) grill.Assertion { return grill.AssertionFunc(func() error { group := fmt.Sprintf("%s_%d_%s", "oh_my_test_helper", rand.Intn(1000), time.Now()) - consumer, err := gk.NewConsumer(group, topicName, time.Second*5) + consumer, err := gk.NewConsumer(group, topicName, time.Second) if err != nil { return err } @@ -33,7 +33,7 @@ func (gk *Kafka) AssertCount(topicName string, expectedCount int) grill.Assertio func (gk *Kafka) AssertMessageCount(topic string, message Message, expectedCount int) grill.Assertion { return grill.AssertionFunc(func() error { group := fmt.Sprintf("%s_%d_%s", "oh_my_test_helper", rand.Intn(1000), time.Now()) - consumer, err := gk.NewConsumer(group, topic, time.Second*5) + consumer, err := gk.NewConsumer(group, topic, time.Second) if err != nil { return err }