From 63b6bb3fa10dd33b069c15151bf706a313970550 Mon Sep 17 00:00:00 2001
From: Ilya Kuramshin <iliax@proton.me>
Date: Mon, 17 Oct 2022 16:24:14 +0400
Subject: [PATCH] ISSUE-2202: smart filters headers test added (#2763)

---
 .../provectus/kafka/ui/emitter/MessageFiltersTest.java    | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/kafka-ui-api/src/test/java/com/provectus/kafka/ui/emitter/MessageFiltersTest.java b/kafka-ui-api/src/test/java/com/provectus/kafka/ui/emitter/MessageFiltersTest.java
index 426ba98020..94a377c9c8 100644
--- a/kafka-ui-api/src/test/java/com/provectus/kafka/ui/emitter/MessageFiltersTest.java
+++ b/kafka-ui-api/src/test/java/com/provectus/kafka/ui/emitter/MessageFiltersTest.java
@@ -13,6 +13,7 @@ import java.time.OffsetDateTime;
 import java.time.temporal.ChronoUnit;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.Map;
 import java.util.function.Predicate;
 import org.apache.commons.lang3.RandomStringUtils;
 import org.junit.jupiter.api.Nested;
@@ -80,6 +81,13 @@ class MessageFiltersTest {
       assertFalse(f.test(msg().offset(200L)));
     }
 
+    @Test
+    void canCheckHeaders() {
+      var f = groovyScriptFilter("headers.size() == 2 && headers['k1'] == 'v1'");
+      assertTrue(f.test(msg().headers(Map.of("k1", "v1", "k2", "v2"))));
+      assertFalse(f.test(msg().headers(Map.of("k1", "unexpected", "k2", "v2"))));
+    }
+
     @Test
     void canCheckTimestampMs() {
       var ts = OffsetDateTime.now();