Improve ServiceBusConsumerTest assertions

apache · Apr 23, 2024 · 00239e9 · 00239e9
1 parent 823d92e
commit 00239e9
Showing 1 changed file with 2 additions and 11 deletions.
diff --git a/...bus/src/test/java/org/apache/camel/component/azure/servicebus/ServiceBusConsumerTest.java b/...bus/src/test/java/org/apache/camel/component/azure/servicebus/ServiceBusConsumerTest.java
@@ -177,12 +177,7 @@ void consumerPropagatesApplicationPropertiesToMessageHeaders() throws Exception
 
             Message inMessage = exchange.getIn();
             assertThat(inMessage).isNotNull();
-
-            Map<String, Object> expectedMessageHeaders = createExpectedMessageHeaders();
-            expectedMessageHeaders.put(ServiceBusConstants.APPLICATION_PROPERTIES,
-                    Map.of(PROPAGATED_HEADER_KEY, PROPAGATED_HEADER_VALUE));
-            expectedMessageHeaders.put(PROPAGATED_HEADER_KEY, PROPAGATED_HEADER_VALUE);
-            assertThat(inMessage.getHeaders()).isEqualTo(expectedMessageHeaders);
+            assertThat(inMessage.getHeaders()).containsEntry(PROPAGATED_HEADER_KEY, PROPAGATED_HEADER_VALUE);
         }
     }
 
@@ -209,11 +204,7 @@ void consumerFiltersApplicationPropertiesFromMessageHeaders() throws Exception {
 
             Message inMessage = exchange.getIn();
             assertThat(inMessage).isNotNull();
-
-            Map<String, Object> expectedMessageHeaders = createExpectedMessageHeaders();
-            expectedMessageHeaders.put(ServiceBusConstants.APPLICATION_PROPERTIES,
-                    Map.of(PROPAGATED_HEADER_KEY, PROPAGATED_HEADER_VALUE));
-            assertThat(inMessage.getHeaders()).isEqualTo(expectedMessageHeaders);
+            assertThat(inMessage.getHeaders()).doesNotContainKey(PROPAGATED_HEADER_KEY);
         }
     }