X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=artifactbroker%2Freception%2Fsrc%2Ftest%2Fjava%2Forg%2Fonap%2Fpolicy%2Fdistribution%2Freception%2Fhandling%2FDummyArtifactForwarder.java;h=ee3d1665d1172ce0e26d2eaa9d11cc2b61447e55;hb=41373a9619308bdbab6a31c02fe3d54e91ac6ee3;hp=57859278a707084f2108b986ac6bbd9ea1efef7a;hpb=0c2c5169090e61fb00d8c10b0e539acaf39d6d22;p=multicloud%2Fframework.git

diff --git a/artifactbroker/reception/src/test/java/org/onap/policy/distribution/reception/handling/DummyArtifactForwarder.java b/artifactbroker/reception/src/test/java/org/onap/policy/distribution/reception/handling/DummyArtifactForwarder.java
index 5785927..ee3d166 100644
--- a/artifactbroker/reception/src/test/java/org/onap/policy/distribution/reception/handling/DummyArtifactForwarder.java
+++ b/artifactbroker/reception/src/test/java/org/onap/policy/distribution/reception/handling/DummyArtifactForwarder.java
@@ -24,7 +24,7 @@ import java.util.ArrayList;
 import java.util.Collection;
 import org.onap.policy.distribution.forwarding.ArtifactForwarder;
 import org.onap.policy.distribution.forwarding.ArtifactForwardingException;
-import org.onap.sdc.api.notification.IArtifactInfo;
+import org.onap.policy.distribution.model.PolicyInput;
 
 /**
  * Class to create a dummy forwarder for test cases in AbstractReceptionHandlerTest.
@@ -33,21 +33,16 @@ import org.onap.sdc.api.notification.IArtifactInfo;
  */
 public class DummyArtifactForwarder implements ArtifactForwarder {
     private int numberOfPoliciesReceived = 0;
-    private Collection<IArtifactInfo> policiesReceived = new ArrayList<>();
 
     @Override
-    public void forward(final Collection<IArtifactInfo> policies) throws ArtifactForwardingException {
-        numberOfPoliciesReceived += policies.size();
-        policiesReceived.addAll(policies);
+    public void forward(final PolicyInput policy) throws ArtifactForwardingException {
+        numberOfPoliciesReceived += 1;
     }
 
     public int getNumberOfPoliciesReceived() {
         return numberOfPoliciesReceived;
     }
 
-    public boolean receivedPolicy(final IArtifactInfo policy) {
-        return policiesReceived.contains(policy);
-    }
 
     @Override
     public void configure(String parameterGroupName) {}