diff --git a/spring-kafka/src/test/java/org/springframework/kafka/support/DefaultKafkaHeaderMapperTests.java b/spring-kafka/src/test/java/org/springframework/kafka/support/DefaultKafkaHeaderMapperTests.java index 5b1fd3a158..d34b873127 100644 --- a/spring-kafka/src/test/java/org/springframework/kafka/support/DefaultKafkaHeaderMapperTests.java +++ b/spring-kafka/src/test/java/org/springframework/kafka/support/DefaultKafkaHeaderMapperTests.java @@ -82,7 +82,8 @@ void testTrustedAndNot() { .build(); RecordHeaders recordHeaders = new RecordHeaders(); mapper.fromHeaders(message.getHeaders(), recordHeaders); - assertThat(recordHeaders.toArray().length).isEqualTo(23); // 21 + json_types + int expectedSize = message.getHeaders().size() - 3; // ID, Timestamp, reply channel + assertThat(recordHeaders.toArray().length).isEqualTo(expectedSize + 1); // json_types header Map headers = new HashMap<>(); mapper.toHeaders(recordHeaders, headers); assertThat(headers.get("foo")).isInstanceOf(byte[].class); @@ -110,7 +111,7 @@ void testTrustedAndNot() { NonTrustedHeaderType ntht = (NonTrustedHeaderType) headers.get("fix"); assertThat(ntht.getHeaderValue()).isNotNull(); assertThat(ntht.getUntrustedType()).isEqualTo(Foo.class.getName()); - assertThat(headers).hasSize(22); + assertThat(headers).hasSize(expectedSize); mapper.addTrustedPackages(getClass().getPackage().getName()); headers = new HashMap<>(); @@ -119,7 +120,7 @@ void testTrustedAndNot() { assertThat(new String((byte[]) headers.get("foo"))).isEqualTo("bar"); assertThat(headers.get("baz")).isEqualTo("qux"); assertThat(headers.get("fix")).isEqualTo(new Foo()); - assertThat(headers).hasSize(22); + assertThat(headers).hasSize(expectedSize); } @Test