From fab40eefd2275d5ade856871f706a469a9a1cfa2 Mon Sep 17 00:00:00 2001
From: Guillem <you@example.com>
Date: Fri, 2 Feb 2024 11:26:01 -0300
Subject: [PATCH] Add test to assert source metadata

---
 source_test.go | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/source_test.go b/source_test.go
index b07184a..f45fbd7 100644
--- a/source_test.go
+++ b/source_test.go
@@ -71,11 +71,18 @@ func TestSource_Integration_RestartPartial(t *testing.T) {
 	testSourceIntegrationRead(ctx, is, cfgMap, lastPosition, wantRecs, false)
 }
 
+const testAppId = "id-1234"
+
 func generateRabbitmqMsgs(from, to int) []amqp091.Publishing {
 	var msgs []amqp091.Publishing
 
 	for i := from; i <= to; i++ {
 		msg := amqp091.Publishing{
+			MessageId:   fmt.Sprintf("test-msg-id-%d", i),
+			ContentType: "text/plain",
+			// setting testAppId asserts that the metadata is being set
+			AppId: testAppId,
+
 			Body: []byte(fmt.Sprintf("test-payload-%d", i)),
 		}
 
@@ -137,6 +144,9 @@ func testSourceIntegrationRead(
 		wantPayload := string(wantRecord.Body)
 		is.Equal(wantPayload, recPayload)
 
+		is.Equal(wantRecord.MessageId, string(rec.Key.Bytes()))
+		is.Equal(testAppId, rec.Metadata["rabbitmq.appId"])
+
 		positions = append(positions, rec.Position)
 	}