renderers/faas/src/test/java/org/opendaylight/groupbasedpolicy/renderer/faas/FaasTenantManagerListenerCovrgTest.java

   1 package org.opendaylight.groupbasedpolicy.renderer.faas;
   2
   3 import static org.mockito.Matchers.any;
   4 import static org.mockito.Mockito.doNothing;
   5 import static org.mockito.Mockito.mock;
   6 import static org.mockito.Mockito.spy;
   7 import static org.mockito.Mockito.when;
   8
   9 import java.util.concurrent.Executors;
  10 import java.util.concurrent.ScheduledExecutorService;
  11
  12 import com.google.common.util.concurrent.CheckedFuture;
  13 import org.junit.Before;
  14 import org.junit.Test;
  15 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  16 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
  17 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  18 import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;
  19 import org.opendaylight.groupbasedpolicy.renderer.faas.test.DataChangeListenerTester;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.faas.logical.faas.common.rev151013.Uuid;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.TenantId;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.policy.rev140421.tenants.Tenant;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.policy.rev140421.tenants.TenantBuilder;
  24 import org.opendaylight.yangtools.yang.binding.DataObject;
  25 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  26
  27 public class FaasTenantManagerListenerCovrgTest {
  28
  29     private InstanceIdentifier<Tenant> tenantIid;
  30     private FaasPolicyManager faasPolicyManager;
  31     private FaasTenantManagerListener listener;
  32     private TenantId gbpTenantId = new TenantId("gbpTenantId");
  33     private Uuid faasTenantId = new Uuid("b4511aac-ae43-11e5-bf7f-feff819cdc9f");
  34     private final ScheduledExecutorService executor = Executors.newScheduledThreadPool(1);
  35     private DataChangeListenerTester tester;
  36     private DataBroker dataProvider;
  37
  38     @SuppressWarnings("unchecked")
  39     @Before
  40     public void init() {
  41         dataProvider = mock(DataBroker.class);
  42         WriteTransaction wTx = mock(WriteTransaction.class);
  43         CheckedFuture<Void, TransactionCommitFailedException> futureVoid = mock(CheckedFuture.class);
  44         when(wTx.submit()).thenReturn(futureVoid);
  45         doNothing().when(wTx).put(any(LogicalDatastoreType.class), any(InstanceIdentifier.class),
  46                 any(DataObject.class));
  47         when(dataProvider.newWriteOnlyTransaction()).thenReturn(wTx);
  48
  49         tenantIid = mock(InstanceIdentifier.class);
  50         faasPolicyManager = spy(new FaasPolicyManager(dataProvider, executor));
  51         doNothing().when(faasPolicyManager).removeTenantLogicalNetwork(gbpTenantId, faasTenantId);
  52         listener = new FaasTenantManagerListener(faasPolicyManager, gbpTenantId, faasTenantId, executor);
  53         tester = new DataChangeListenerTester(listener);
  54         tester.setRemovedPath(tenantIid);
  55
  56     }
  57
  58     @Test
  59     public void testOnDataChanged() {
  60         Tenant tenant = new TenantBuilder().setId(gbpTenantId).build();
  61         tester.setDataObject(tenantIid, tenant);
  62         tester.callOnDataChanged();
  63         listener.executeEvent(tester.getChangeMock());
  64     }
  65
  66 }