1 files changed, 7 insertions, 11 deletions

diff --git a/src/test/java/at/gv/egiz/moazs/SingleThreadedDeliveryPipelineTest.java b/src/test/java/at/gv/egiz/moazs/SingleThreadedDeliveryPipelineTest.java
index 4154db9..e245eb1 100644
--- a/src/test/java/at/gv/egiz/moazs/SingleThreadedDeliveryPipelineTest.java
+++ b/src/test/java/at/gv/egiz/moazs/SingleThreadedDeliveryPipelineTest.java
@@ -3,8 +3,6 @@ package at.gv.egiz.moazs;
 import at.gv.egiz.moazs.process.DeliveryPipeline;
 import at.gv.egiz.moazs.process.SingleThreadedDeliveryPipeline;
 import at.gv.egiz.moazs.client.TnvzHelper;
-import at.gv.egiz.moazs.repository.BinaryRepository;
-import at.gv.egiz.moazs.repository.InMemoryBinaryRepository;
 import at.gv.egiz.moazs.scheme.*;
 import at.gv.egiz.moazs.client.ClientFactory;
 import at.gv.egiz.moazs.repository.DeliveryRepository;
@@ -39,8 +37,7 @@ import static org.mockito.Mockito.*;
 @RunWith(MockitoJUnitRunner.class)
 public class SingleThreadedDeliveryPipelineTest {
 
-    private DeliveryRepository repository = new InMemoryDeliveryRepository();
-    private BinaryRepository binaryRepository = new InMemoryBinaryRepository(30);
+    private DeliveryRepository repository = new InMemoryDeliveryRepository(100);
 
     @Mock
     private TnvzHelper tnvzHelper;
@@ -67,7 +64,6 @@ public class SingleThreadedDeliveryPipelineTest {
     public void setup() {
         pipeline = new SingleThreadedDeliveryPipeline(
                 repository,
-                binaryRepository,
                 tnvzHelper,
                 converter,
                 clientFactory,
@@ -82,7 +78,7 @@ public class SingleThreadedDeliveryPipelineTest {
         pipeline.processRequest(appDeliveryID);
 
         verifyZeroInteractions(tnvzHelper);
-        var response = repository.getResponse(getResponseID(appDeliveryID)).get();
+        var response = repository.retrieveResponse(getResponseID(appDeliveryID)).get();
         var actualStatus = (DeliveryRequestStatusType) response.getResponse();
 
         assertThat(actualStatus).isEqualTo(expectedStatus);
@@ -98,7 +94,7 @@ public class SingleThreadedDeliveryPipelineTest {
         pipeline.processRequest(appDeliveryID);
 
         var responseID = getResponseID(appDeliveryID);
-        var response = repository.getResponse(responseID).get();
+        var response = repository.retrieveResponse(responseID).get();
         var status = (DeliveryRequestStatusType) response.getResponse();
         var actualCode = status.getError().getErrorInfo().getCode();
 
@@ -113,7 +109,7 @@ public class SingleThreadedDeliveryPipelineTest {
         var expectedStatus = setupMocks(appDeliveryID, true);
 
         pipeline.processRequest(appDeliveryID);
-        var response = repository.getResponse(getResponseID(appDeliveryID)).get();
+        var response = repository.retrieveResponse(getResponseID(appDeliveryID)).get();
         var actualStatus = (DeliveryRequestStatusType) response.getResponse();
 
         assertThat(actualStatus).isEqualTo(expectedStatus);
@@ -126,7 +122,7 @@ public class SingleThreadedDeliveryPipelineTest {
         doThrow(moaZSException("Signature Invalid!")).when(verifier).verify(any());
 
         pipeline.processRequest(appDeliveryID);
-        var response = repository.getResponse(getResponseID(appDeliveryID)).get();
+        var response = repository.retrieveResponse(getResponseID(appDeliveryID)).get();
         var status = (DeliveryRequestStatusType) response.getResponse();
         var actualCode = status.getError().getErrorInfo().getCode();
 
@@ -141,8 +137,8 @@ public class SingleThreadedDeliveryPipelineTest {
         var status = setupStatus(appDeliveryId);
         var signedStatus = new byte[0];
         var identification = IdentificationType.identificationTypeBuilder().build();
-        repository.add(mzsRequest);
-        binaryRepository.add(getResponseID(appDeliveryId), signedStatus);
+        repository.store(mzsRequest);
+        repository.store(getResponseID(appDeliveryId), signedStatus);
 
         when(converter.convert(eq(mzsRequest)       )).thenReturn(msgRequest);
         when(converter.convert(eq(mzsRequest), any())).thenReturn(msgRequest);