BUG-6650: ep-ip/sgt, implement and wire template provider

[groupbasedpolicy.git] / sxp-integration / sxp-ise-adapter / src / test / java / org / opendaylight / groupbasedpolicy / sxp_ise_adapter / impl / GbpIseAdapterProviderTest.java

diff --git a/sxp-integration/sxp-ise-adapter/src/test/java/org/opendaylight/groupbasedpolicy/sxp_ise_adapter/impl/GbpIseAdapterProviderTest.java b/sxp-integration/sxp-ise-adapter/src/test/java/org/opendaylight/groupbasedpolicy/sxp_ise_adapter/impl/GbpIseAdapterProviderTest.java
index a7febc9b649c12a46228adbfd6cfcc3eaf8dd37e..c4c79d24825e91da67a1dabd842168f794284947 100644 (file)
--- a/sxp-integration/sxp-ise-adapter/src/test/java/org/opendaylight/groupbasedpolicy/sxp_ise_adapter/impl/GbpIseAdapterProviderTest.java
+++ b/sxp-integration/sxp-ise-adapter/src/test/java/org/opendaylight/groupbasedpolicy/sxp_ise_adapter/impl/GbpIseAdapterProviderTest.java
@@ -20,6 +20,8 @@ import org.mockito.Mockito;
 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
 import org.opendaylight.controller.md.sal.binding.api.DataTreeIdentifier;
 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker;
+import org.opendaylight.groupbasedpolicy.sxp.ep.provider.api.EPPolicyTemplateProviderRegistry;
+import org.opendaylight.groupbasedpolicy.sxp.ep.provider.spi.SxpEpProviderProvider;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.gbp.sxp.ise.adapter.model.rev160630.gbp.sxp.ise.adapter.IseSourceConfig;
 import org.opendaylight.yangtools.concepts.ListenerRegistration;
 import org.powermock.core.classloader.annotations.PrepareForTest;
@@ -48,12 +50,20 @@ public class GbpIseAdapterProviderTest {
     private GbpIseSgtHarvesterImpl harvester;
     @Mock
     private GbpIseConfigListenerImpl listener;
+    @Mock
+    private SxpEpProviderProvider sxpEpProvider;
+    @Mock
+    private EPPolicyTemplateProviderRegistry templateProviderRegistry;
+    @Mock
+    private EPPolicyTemplateProviderIseImpl templateProvider;
 
     private GbpIseAdapterProvider provider;
 
     @Before
     public void setUp() throws Exception {
-        provider = new GbpIseAdapterProvider(dataBroker, broker);
+        Mockito.when(sxpEpProvider.getEPPolicyTemplateProviderRegistry())
+                .thenReturn(templateProviderRegistry);
+        provider = new GbpIseAdapterProvider(dataBroker, broker, sxpEpProvider);
         Mockito.verify(broker).registerProvider(provider);
     }
 
@@ -68,13 +78,15 @@ public class GbpIseAdapterProviderTest {
         whenNew(SgtToEPTemplateGeneratorImpl.class).withAnyArguments().thenReturn(generator2);
         whenNew(GbpIseSgtHarvesterImpl.class).withArguments(generator1, generator2).thenReturn(harvester);
         whenNew(GbpIseConfigListenerImpl.class).withAnyArguments().thenReturn(listener);
+        whenNew(EPPolicyTemplateProviderIseImpl.class).withNoArguments().thenReturn(templateProvider);
         provider.onSessionInitiated(providerContext);
 
+        Mockito.verify(templateProviderRegistry).registerTemplateProvider(templateProvider);
         // check if all expected object got constructed and wired
         verifyNew(SgtToEpgGeneratorImpl.class).withArguments(dataBroker);
         verifyNew(SgtToEPTemplateGeneratorImpl.class).withArguments(dataBroker);
         verifyNew(GbpIseSgtHarvesterImpl.class).withArguments(generator1, generator2);
-        verifyNew(GbpIseConfigListenerImpl.class).withArguments(dataBroker, harvester);
+        verifyNew(GbpIseConfigListenerImpl.class).withArguments(dataBroker, harvester, templateProvider);
 
         // close provider check
         provider.close();