1 files changed, 8 insertions, 4 deletions
diff --git a/src/test/java/at/gv/egiz/moazs/DeliveryRequestAugmenterTest.java b/src/test/java/at/gv/egiz/moazs/DeliveryRequestAugmenterTest.java
index 4bd64ef..424dce1 100644
--- a/src/test/java/at/gv/egiz/moazs/DeliveryRequestAugmenterTest.java
+++ b/src/test/java/at/gv/egiz/moazs/DeliveryRequestAugmenterTest.java
@@ -1,5 +1,6 @@
 package at.gv.egiz.moazs;
 
+import at.gv.egiz.moazs.mzs.MzsValidator;
 import at.gv.egiz.moazs.preprocess.ConfigUtil;
 import at.gv.egiz.moazs.preprocess.DeliveryRequestAugmenter;
 import at.gv.zustellung.app2mzs.xsd.ConfigType;
@@ -8,7 +9,6 @@ import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.mockito.Mock;
-import org.mockito.Mockito;
 import org.mockito.junit.MockitoJUnitRunner;
 
 import java.util.Map;
@@ -28,9 +28,13 @@ public class DeliveryRequestAugmenterTest {
     @Mock
     private ConfigUtil configUtil;
 
+    @Mock
+    private MzsValidator validator;
+
     @Before
     public void setupMock() {
-        when(configUtil.isComplete(Mockito.any())).thenReturn(true);
+        when(validator.isConfigProfileComplete(any())).thenReturn(true);
+        when(validator.isTnvzComplete(any())).thenReturn(true);
         when(configUtil.merge(any(), any())).thenCallRealMethod();
     }
 
@@ -105,11 +109,11 @@ public class DeliveryRequestAugmenterTest {
     }
 
     private DeliveryRequestAugmenter createAugmenter(ConfigType fallback) {
-        return new DeliveryRequestAugmenter(Map.of("default", fallback), configUtil);
+        return new DeliveryRequestAugmenter(Map.of("default", fallback), configUtil, validator);
     }
 
     private DeliveryRequestAugmenter createAugmenter(Map<String, ConfigType> profiles) {
-        return new DeliveryRequestAugmenter(profiles, configUtil);
+        return new DeliveryRequestAugmenter(profiles, configUtil, validator);
     }
 
     private ConfigType createConfig(String url, Boolean performTnvz) {