From 42b2e56501b61d84f758588002fbeb05805cac89 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mi=C5=82osz=20Sm=C3=B3=C5=82ka?= <milosz.smolka@gmail.com>
Date: Wed, 28 Aug 2024 15:47:28 +0200
Subject: [PATCH] Add an assert

---
 pkg/kafka/pubsub_test.go | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pkg/kafka/pubsub_test.go b/pkg/kafka/pubsub_test.go
index b6d027b..3d9a688 100644
--- a/pkg/kafka/pubsub_test.go
+++ b/pkg/kafka/pubsub_test.go
@@ -254,6 +254,7 @@ func TestCtxValuesAfterRetry(t *testing.T) {
 	require.NoError(t, err)
 
 	receivedMessage, ok := readAfterRetries(messages, 1, time.Second)
+	assert.True(t, ok)
 
 	expectedPartitionsOffsets := map[int32]int64{}
 	partition, ok := kafka.MessagePartitionFromCtx(receivedMessage.Context())