From 1978b39e07436ca6fbd1ad3a693aa0c0ac5469d9 Mon Sep 17 00:00:00 2001
From: OZGCloud <ozgcloud@mgm-tp.com>
Date: Mon, 13 Nov 2023 08:42:12 +0100
Subject: [PATCH] cleanup

---
 .../binaryfile/GrpcFileUploadUtilsTest.java      | 16 ++++++----------
 1 file changed, 6 insertions(+), 10 deletions(-)

diff --git a/kop-common-lib/src/test/java/de/itvsh/kop/common/binaryfile/GrpcFileUploadUtilsTest.java b/kop-common-lib/src/test/java/de/itvsh/kop/common/binaryfile/GrpcFileUploadUtilsTest.java
index dcf6dd0..840dfc9 100644
--- a/kop-common-lib/src/test/java/de/itvsh/kop/common/binaryfile/GrpcFileUploadUtilsTest.java
+++ b/kop-common-lib/src/test/java/de/itvsh/kop/common/binaryfile/GrpcFileUploadUtilsTest.java
@@ -93,30 +93,26 @@ class GrpcFileUploadUtilsTest {
 
 		@Test
 		void shouldSetOnReadyHandler() {
-			callService();
+			fileSender.send();
 
 			verify(requestObserver).setOnReadyHandler(any());
 		}
 
 		@Test
 		void shouldCallSendNextOnReady() {
-			callService();
+			fileSender.send();
 
-			verify(requestObserver).setOnReadyHandler(runnableCaptor.capture());
+			verify(requestObserver, timeout(500)).setOnReadyHandler(runnableCaptor.capture());
 			runnableCaptor.getValue().run();
 
 			verify(fileSender, timeout(500)).sendNext();
 		}
 
 		@Test
-		void shouldReturnFuture() {
-			var result = callService();
-
-			assertThat(result).isNotNull();
-		}
+		void shouldReturnSenderWithFuture() {
+			var result = fileSender.send();
 
-		private FileSender<TestRequestType, TestResponseType> callService() {
-			return fileSender.send();
+			assertThat(result).isNotNull().extracting(FileSender::getResultFuture).isNotNull();
 		}
 	}
 
-- 
GitLab