1 files changed, 12 insertions, 6 deletions
diff --git a/src/test/java/at/gv/egiz/moazs/App2MzsServiceTest.java b/src/test/java/at/gv/egiz/moazs/App2MzsServiceTest.java
index 01b7745..5579f06 100644
--- a/src/test/java/at/gv/egiz/moazs/App2MzsServiceTest.java
+++ b/src/test/java/at/gv/egiz/moazs/App2MzsServiceTest.java
@@ -1,5 +1,8 @@
 package at.gv.egiz.moazs;
 
+import at.gv.e_government.reference.namespace.zustellung.msg.phase2._20181206_.DeliveryRequestStatusType;
+import at.gv.e_government.reference.namespace.zustellung.msg.phase2._20181206_.DeliveryRequestStatusType.DeliveryRequestStatusTypeBuilder;
+import at.gv.e_government.reference.namespace.zustellung.msg.phase2._20181206_.DeliveryRequestStatusType.Success.SuccessBuilder;
 import at.gv.e_government.reference.namespace.zustellung.msg.phase2._20181206_.ObjectFactory;
 import at.gv.egiz.moazs.pipeline.DeliveryPipeline;
 import at.gv.egiz.moazs.repository.DeliveryRepository;
@@ -42,16 +45,19 @@ public class App2MzsServiceTest {
         @Autowired
         private DeliveryRepository repository;
 
-        private final ObjectFactory factory = new ObjectFactory();
-
         @Bean
         @Primary
         public DeliveryPipeline deliveryPipelineThatAlwaysSucceeds() {
             return appDeliveryId -> {
-                var status = factory.createDeliveryRequestStatusType();
-                var success = factory.createDeliveryRequestStatusTypeSuccess();
-                success.setAppDeliveryID(appDeliveryId);
-                status.setSuccess(success);
+
+                var success = new SuccessBuilder()
+                        .withAppDeliveryID(appDeliveryId)
+                        .build();
+
+                var status = new DeliveryRequestStatusTypeBuilder()
+                        .withSuccess(success)
+                        .build();
+
                 repository.add(status);
             };
         }