1 files changed, 64 insertions, 0 deletions
diff --git a/openecomp-be/lib/openecomp-sdc-notification-lib/openecomp-sdc-notification-core/src/test/java/org/openecomp/sdc/notification/services/impl/NotificationPropagationManagerImplTest.java b/openecomp-be/lib/openecomp-sdc-notification-lib/openecomp-sdc-notification-core/src/test/java/org/openecomp/sdc/notification/services/impl/NotificationPropagationManagerImplTest.java
new file mode 100644
index 0000000000..0eb2b6235c
--- /dev/null
+++ b/openecomp-be/lib/openecomp-sdc-notification-lib/openecomp-sdc-notification-core/src/test/java/org/openecomp/sdc/notification/services/impl/NotificationPropagationManagerImplTest.java
@@ -0,0 +1,64 @@
+package org.openecomp.sdc.notification.services.impl;
+
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Test;
+import org.mockito.ArgumentCaptor;
+import org.mockito.Captor;
+import org.mockito.InjectMocks;
+import org.mockito.Matchers;
+import org.mockito.Mock;
+import org.mockito.MockitoAnnotations;
+import org.mockito.Spy;
+import org.openecomp.sdc.destinationprovider.DestinationProvider;
+import org.openecomp.sdc.destinationprovider.impl.MulticastDestination;
+import org.openecomp.sdc.destinationprovider.impl.UnicastDestination;
+import org.openecomp.sdc.notification.dtos.Event;
+import org.openecomp.sdc.notification.services.PropagationService;
+import org.openecomp.sdc.notification.services.SubscriptionService;
+
+import static org.mockito.Mockito.verify;
+
+
+/**
+ * @author avrahamg
+ * @since July 13, 2017
+ */
+public class NotificationPropagationManagerImplTest {
+    @Mock
+    private PropagationService propagationServiceMock;
+    @Mock
+    private SubscriptionService subscriptionServiceMock;
+    @Mock
+    private Event eventMock;
+    @Captor
+    private ArgumentCaptor<DestinationProvider> destinationProviderCaptor;
+
+    @Spy
+    @InjectMocks
+    private NotificationPropagationManagerImpl notificationPropagationManager;
+
+    @Before
+    public void setUp() throws Exception {
+        MockitoAnnotations.initMocks(this);
+    }
+
+    @Test
+    public void shouldCallPropagationServiceNotifyWithMulticastDestinationWhenNotifySubscribers()
+        throws Exception {
+        notificationPropagationManager.notifySubscribers(eventMock);
+        verify(propagationServiceMock).notify(Matchers.eq(eventMock), destinationProviderCaptor
+            .capture());
+        Assert.assertTrue(destinationProviderCaptor.getValue() instanceof MulticastDestination);
+
+    }
+
+    @Test
+    public void shouldCallPropagationServiceNotifyWithUnicastDestinationWhenDirectNotification()
+        throws Exception {
+        notificationPropagationManager.directNotification(eventMock, "aaa");
+        verify(propagationServiceMock).notify(Matchers.eq(eventMock), destinationProviderCaptor
+            .capture());
+        Assert.assertTrue(destinationProviderCaptor.getValue() instanceof UnicastDestination);
+    }
+}
+\ No newline at end of file