From 778672e6eadc2d56caa78dfb82abd5ddfcc0de80 Mon Sep 17 00:00:00 2001
From: Oleg Zhurakousky <ozhurakousky@vmware.com>
Date: Wed, 10 Mar 2021 15:38:13 +0100
Subject: [PATCH] Add test assertions to validate header propagation for
 rsocket message

---
 .../springframework/cloud/function/rsocket/MessagingTests.java  | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/spring-cloud-function-rsocket/src/test/java/org/springframework/cloud/function/rsocket/MessagingTests.java b/spring-cloud-function-rsocket/src/test/java/org/springframework/cloud/function/rsocket/MessagingTests.java
index 03752b142..936354d67 100644
--- a/spring-cloud-function-rsocket/src/test/java/org/springframework/cloud/function/rsocket/MessagingTests.java
+++ b/spring-cloud-function-rsocket/src/test/java/org/springframework/cloud/function/rsocket/MessagingTests.java
@@ -269,6 +269,8 @@ public class MessagingTests {
 		@Bean
 		public Function<Message<Map<String, Object>>, Message<Map<String, Object>>> echoMessageMap() {
 			return v -> {
+				assertThat(v.getHeaders()).containsKey("rsocketFrameType");
+				assertThat(v.getHeaders()).containsKey("lookupDestination");
 				return v;
 			};
 		}