Refactor NBINotifications and serviceHandlerImpl

[transportpce.git] / nbinotifications / src / test / java / org / opendaylight / transportpce / nbinotifications / producer / PublisherTest.java

diff --git a/nbinotifications/src/test/java/org/opendaylight/transportpce/nbinotifications/producer/PublisherTest.java b/nbinotifications/src/test/java/org/opendaylight/transportpce/nbinotifications/producer/PublisherTest.java
index b3a5b2edbeb5b5a858eb377f012b8c3f70c3c669..86e33307bda1b2366405c2557a3f47219481f334 100644 (file)
--- a/nbinotifications/src/test/java/org/opendaylight/transportpce/nbinotifications/producer/PublisherTest.java
+++ b/nbinotifications/src/test/java/org/opendaylight/transportpce/nbinotifications/producer/PublisherTest.java
@@ -19,37 +19,58 @@ import org.junit.Before;
 import org.junit.Test;
 import org.opendaylight.transportpce.common.converter.JsonStringConverter;
 import org.opendaylight.transportpce.nbinotifications.serialization.ConfigConstants;
+import org.opendaylight.transportpce.nbinotifications.serialization.NotificationAlarmServiceSerializer;
 import org.opendaylight.transportpce.nbinotifications.serialization.NotificationServiceSerializer;
 import org.opendaylight.transportpce.test.AbstractTest;
-import org.opendaylight.yang.gen.v1.nbi.notifications.rev201130.NotificationService;
+import org.opendaylight.yang.gen.v1.nbi.notifications.rev210813.NotificationAlarmService;
+import org.opendaylight.yang.gen.v1.nbi.notifications.rev210813.NotificationProcessService;
 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
 import org.opendaylight.yangtools.yang.data.codec.gson.JSONCodecFactorySupplier;
 
 public class PublisherTest extends AbstractTest {
-    private JsonStringConverter<NotificationService> converter;
-    private Publisher publisher;
-    private MockProducer<String, NotificationService> mockProducer;
+    private JsonStringConverter<NotificationProcessService> converterService;
+    private JsonStringConverter<NotificationAlarmService> converterAlarm;
+    private Publisher<NotificationProcessService> publisherService;
+    private Publisher<NotificationAlarmService> publisherAlarm;
+    private MockProducer<String, NotificationProcessService> mockProducer;
+    private MockProducer<String, NotificationAlarmService> mockAlarmProducer;
 
     @Before
     public void setUp() {
-        NotificationServiceSerializer serializer = new NotificationServiceSerializer();
-        Map<String, Object> properties = Map.of(ConfigConstants.CONVERTER , serializer);
-        serializer.configure(properties, false);
-        mockProducer =  new MockProducer<>(true, new StringSerializer(), serializer);
-        converter = new JsonStringConverter<NotificationService>(
-                getDataStoreContextUtil().getBindingDOMCodecServices());
-        publisher = new Publisher("test",mockProducer);
+        NotificationServiceSerializer serializerService = new NotificationServiceSerializer();
+        NotificationAlarmServiceSerializer serializerAlarm = new NotificationAlarmServiceSerializer();
+        Map<String, Object> properties = Map.of(ConfigConstants.CONVERTER, serializerService);
+        Map<String, Object> propertiesAlarm = Map.of(ConfigConstants.CONVERTER, serializerAlarm);
+        serializerService.configure(properties, false);
+        serializerAlarm.configure(propertiesAlarm, false);
+        mockProducer = new MockProducer<>(true, new StringSerializer(), serializerService);
+        mockAlarmProducer = new MockProducer<>(true, new StringSerializer(), serializerAlarm);
+        converterService = new JsonStringConverter<>(getDataStoreContextUtil().getBindingDOMCodecServices());
+        converterAlarm = new JsonStringConverter<>(getDataStoreContextUtil().getBindingDOMCodecServices());
+        publisherService = new Publisher<>("test", mockProducer);
+        publisherAlarm = new Publisher<>("test", mockAlarmProducer);
     }
 
     @Test
-    public void sendEventShouldBeSuccessful() throws IOException {
+    public void sendEventServiceShouldBeSuccessful() throws IOException {
         String json = Files.readString(Paths.get("src/test/resources/event.json"));
-        NotificationService notificationService = converter
-                .createDataObjectFromJsonString(YangInstanceIdentifier.of(NotificationService.QNAME),
+        NotificationProcessService notificationProcessService = converterService
+                .createDataObjectFromJsonString(YangInstanceIdentifier.of(NotificationProcessService.QNAME),
                         json, JSONCodecFactorySupplier.RFC7951);
-        publisher.sendEvent(notificationService);
+        publisherService.sendEvent(notificationProcessService, notificationProcessService.getConnectionType().name());
         assertEquals("We should have one message", 1, mockProducer.history().size());
-        assertEquals("Key should be test", "test",mockProducer.history().get(0).key());
+        assertEquals("Key should be test", "test", mockProducer.history().get(0).key());
     }
 
+    @Test
+    public void sendEventAlarmShouldBeSuccessful() throws IOException {
+        String json = Files.readString(Paths.get("src/test/resources/event_alarm_service.json"));
+        NotificationAlarmService notificationAlarmService = converterAlarm
+                .createDataObjectFromJsonString(YangInstanceIdentifier.of(NotificationAlarmService.QNAME),
+                        json, JSONCodecFactorySupplier.RFC7951);
+        publisherAlarm.sendEvent(notificationAlarmService, "alarm"
+                + notificationAlarmService.getConnectionType().getName());
+        assertEquals("We should have one message", 1, mockAlarmProducer.history().size());
+        assertEquals("Key should be test", "test", mockAlarmProducer.history().get(0).key());
+    }
 }