1 files changed, 17 insertions, 6 deletions
diff --git a/prh-app-server/src/test/java/org/onap/dcaegen2/services/prh/tasks/ScheduleControllerSpy.java b/prh-app-server/src/test/java/org/onap/dcaegen2/services/prh/tasks/ScheduleControllerSpy.java
index b12b3d39..8d41ec63 100644
--- a/prh-app-server/src/test/java/org/onap/dcaegen2/services/prh/tasks/ScheduleControllerSpy.java
+++ b/prh-app-server/src/test/java/org/onap/dcaegen2/services/prh/tasks/ScheduleControllerSpy.java
@@ -25,6 +25,7 @@ import static org.mockito.Mockito.spy;
 import java.util.Map;
 
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Qualifier;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 import org.springframework.context.annotation.Primary;
@@ -38,8 +39,16 @@ public class ScheduleControllerSpy {
     @Autowired
     private DmaapConsumerTask dmaapConsumerTaskImplSpy;
 
+    @Qualifier("ReadyPublisherTask")
     @Autowired
-    private DmaapPublisherTask dmaapPublisherTaskImplSpy;
+    private DmaapPublisherTask dmaapReadyPublisherTaskImplSpy;
+
+    @Qualifier("UpdatePublisherTask")
+    @Autowired
+    private DmaapPublisherTask dmaapUpdatePublisherTaskImplSpy;
+
+    @Autowired
+    private AaiQueryTask aaiQueryTaskImplSpy;
 
     @Autowired
     private AaiProducerTask aaiPublisherTaskImplSpy;
@@ -54,10 +63,12 @@ public class ScheduleControllerSpy {
     @Primary
     public ScheduledTasks registerSimpleScheduledTask() {
         return spy(new ScheduledTasks(
-            dmaapConsumerTaskImplSpy,
-            dmaapPublisherTaskImplSpy,
-            aaiPublisherTaskImplSpy,
-            bbsActionsTaskImplSpy,
-            mdcContextMap));
+                dmaapConsumerTaskImplSpy,
+                dmaapReadyPublisherTaskImplSpy,
+                dmaapUpdatePublisherTaskImplSpy,
+                aaiQueryTaskImplSpy,
+                aaiPublisherTaskImplSpy,
+                bbsActionsTaskImplSpy,
+                mdcContextMap));
     }
 }