add check empty in the integration test

Signed-off-by: myan <myan@redhat.com>
cloudevents · May 23, 2023 · 69ea310 · 69ea310
1 parent 1ab33b9
commit 69ea310
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 6 deletions.
diff --git a/test/integration/kafka_sarama/kafka_test.go b/test/integration/kafka_sarama/kafka_test.go
@@ -34,9 +34,14 @@ func TestSendEvent(t *testing.T) {
 			return protocolFactory(t)
 		}, eventIn, func(e event.Event) {
 			got := test.ConvertEventExtensionsToString(t, e)
-			eventIn.SetExtension(KAFKA_OFFSET, got.Extensions()[KAFKA_OFFSET])
-			eventIn.SetExtension(KAFKA_PARTITION, got.Extensions()[KAFKA_PARTITION])
-			eventIn.SetExtension(KAFKA_TOPIC, got.Extensions()[KAFKA_TOPIC])
+			extensions := got.Extensions()
+			require.NotEmpty(t, extensions[KAFKA_OFFSET])
+			require.NotEmpty(t, extensions[KAFKA_PARTITION])
+			require.NotEmpty(t, extensions[KAFKA_TOPIC])
+
+			eventIn.SetExtension(KAFKA_OFFSET, extensions[KAFKA_OFFSET])
+			eventIn.SetExtension(KAFKA_PARTITION, extensions[KAFKA_PARTITION])
+			eventIn.SetExtension(KAFKA_TOPIC, extensions[KAFKA_TOPIC])
 			test.AssertEventEquals(t, eventIn, got)
 		})
 	})

diff --git a/test/integration/kafka_sarama_binding/kafka_test.go b/test/integration/kafka_sarama_binding/kafka_test.go
@@ -59,9 +59,13 @@ func TestSendBinaryMessageToBinary(t *testing.T) {
 			eventOut := MustToEvent(t, context.Background(), out)
 			assert.Equal(t, binding.EncodingBinary, out.ReadEncoding())
 			got := ConvertEventExtensionsToString(t, eventOut)
-			eventIn.SetExtension(KAFKA_OFFSET, got.Extensions()[KAFKA_OFFSET])
-			eventIn.SetExtension(KAFKA_PARTITION, got.Extensions()[KAFKA_PARTITION])
-			eventIn.SetExtension(KAFKA_TOPIC, got.Extensions()[KAFKA_TOPIC])
+			extensions := got.Extensions()
+			require.NotEmpty(t, extensions[KAFKA_OFFSET])
+			require.NotEmpty(t, extensions[KAFKA_PARTITION])
+			require.NotEmpty(t, extensions[KAFKA_TOPIC])
+			eventIn.SetExtension(KAFKA_OFFSET, extensions[KAFKA_OFFSET])
+			eventIn.SetExtension(KAFKA_PARTITION, extensions[KAFKA_PARTITION])
+			eventIn.SetExtension(KAFKA_TOPIC, extensions[KAFKA_TOPIC])
 			AssertEventEquals(t, eventIn, got)
 		})
 	})