Add RSocketRequester retrieveAndAwaitOrNull extension

Closes gh-23874
2019-10-27 20:11:08 +01:00
parent 0d8aa854e8
commit 9f242fac66
2 changed files with 18 additions and 0 deletions
--- a/spring-messaging/src/test/kotlin/org/springframework/messaging/rsocket/RSocketRequesterExtensionsTests.kt
+++ b/spring-messaging/src/test/kotlin/org/springframework/messaging/rsocket/RSocketRequesterExtensionsTests.kt
@@ -120,6 +120,15 @@ class RSocketRequesterExtensionsTests {
 		}
 	}

+	@Test
+	fun retrieveAndAwaitOrNull() {
+		val requestSpec = mockk<RSocketRequester.RequestSpec>()
+		every { requestSpec.retrieveMono(match<ParameterizedTypeReference<*>>(stringTypeRefMatcher)) } returns Mono.empty()
+		runBlocking {
+			assertThat(requestSpec.retrieveAndAwaitOrNull<String>()).isNull()
+		}
+	}
+
 	@Test
 	fun retrieveFlow() {
 		val requestSpec = mockk<RSocketRequester.RequestSpec>()