test good

transferwise · Aug 6, 2024 · e8846b8 · e8846b8
1 parent e4dc338
commit e8846b8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tw-tkms-starter/src/test/java/com/transferwise/kafka/tkms/test/TestMessageDecorator.java b/tw-tkms-starter/src/test/java/com/transferwise/kafka/tkms/test/TestMessageDecorator.java
@@ -13,10 +13,10 @@ public class TestMessageDecorator implements ITkmsMessageDecorator {
   @Override
   public List<Header> getHeaders(TkmsMessage message) {
     var h1 = new Header().setKey("tool").setValue("jambi".getBytes(StandardCharsets.UTF_8));
-    if (message.getKey() == null) {
-      return List.of();
+    if (message.getValue() != null && new String(message.getValue(), StandardCharsets.UTF_8).startsWith("Here from")) {
+      return List.of(h1);
     }
-    return List.of(h1);
+    return List.of();
   }
 
 }