1 files changed, 11 insertions, 3 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 d4975098..0b5f969b 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
@@ -22,6 +22,7 @@ package org.onap.dcaegen2.services.prh.tasks;
 import static org.mockito.Mockito.spy;
 
 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;
@@ -32,12 +33,19 @@ import org.springframework.context.annotation.Primary;
 @Configuration
 public class ScheduleControllerSpy {
 
+
+    @Autowired
+    private DmaapConsumerTask dmaapConsumerTaskImplSpy;
+
+    @Autowired
+    private DmaapPublisherTask dmaapPublisherTaskImplSpy;
+
     @Autowired
-    private DmaapConsumerTask dmaapConsumerTaskSpy;
+    private AAIPublisherTask aaiPublisherTaskImplSpy;
 
     @Bean
     @Primary
-    public ScheduledTask registerSimpleDmaapConsumerTask() {
-        return spy(new ScheduledTask(dmaapConsumerTaskSpy));
+    public ScheduledTasks registerSimpleScheduledTask() {
+        return spy(new ScheduledTasks(dmaapConsumerTaskImplSpy, dmaapPublisherTaskImplSpy, aaiPublisherTaskImplSpy));
     }
 }