groupbasedpolicy/src/test/java/org/opendaylight/groupbasedpolicy/base_endpoint/BaseEndpointRpcRegistryTest.java

   1 /*
   2  * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8
   9 package org.opendaylight.groupbasedpolicy.base_endpoint;
  10
  11 import com.google.common.util.concurrent.CheckedFuture;
  12
  13 import org.junit.Assert;
  14 import org.junit.Before;
  15 import org.junit.Test;
  16 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  17 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
  18 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  19 import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;
  20 import org.opendaylight.controller.sal.binding.api.RpcProviderRegistry;
  21 import org.opendaylight.groupbasedpolicy.api.BaseEndpointRendererAugmentation;
  22 import org.opendaylight.groupbasedpolicy.base_endpoint.BaseEndpointRpcRegistry;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.base_endpoint.rev160427.common.endpoint.fields.NetworkContainmentBuilder;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.base_endpoint.rev160427.common.endpoint.fields.network.containment.containment.NetworkDomainContainmentBuilder;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.base_endpoint.rev160427.register.endpoint.input.AddressEndpointRegKey;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.base_endpoint.rev160427.register.endpoint.input.ContainmentEndpointRegKey;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.*;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.base_endpoint.rev160427.RegisterEndpointInput;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.base_endpoint.rev160427.RegisterEndpointInputBuilder;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.base_endpoint.rev160427.UnregisterEndpointInput;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.base_endpoint.rev160427.UnregisterEndpointInputBuilder;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.base_endpoint.rev160427.register.endpoint.input.AddressEndpointRegBuilder;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.base_endpoint.rev160427.register.endpoint.input.ContainmentEndpointRegBuilder;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.base_endpoint.rev160427.unregister.endpoint.input.AddressEndpointUnregBuilder;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.base_endpoint.rev160427.unregister.endpoint.input.ContainmentEndpointUnregBuilder;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.forwarding.l2_l3.rev160427.L2FloodDomain;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.forwarding.l2_l3.rev160427.MacAddressType;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.forwarding.l2_l3.rev160427.Subnet;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.forwarding.rev160427.AddressType;
  40 import org.opendaylight.yangtools.yang.binding.DataObject;
  41 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  42
  43 import java.util.ArrayList;
  44
  45 import static org.mockito.Matchers.any;
  46 import static org.mockito.Matchers.eq;
  47 import static org.mockito.Mockito.*;
  48 import static org.mockito.internal.verification.VerificationModeFactory.times;
  49
  50 public class BaseEndpointRpcRegistryTest {
  51
  52     private static final String MAC_ADDRESS = "01:23:45:67:89:AB";
  53     private static final String TENANT = "admin";
  54     private static final String DOMAIN = "test.domain";
  55     private static final String CONTEXT_ID = "testContext";
  56     private static final String FLOOD_DOMAIN = "testFloodDomain";
  57
  58     private DataBroker dataProvider;
  59     private BaseEndpointRendererAugmentation baseEndpointRendererAugmentation;
  60     private BaseEndpointRpcRegistry baseEndpointRpcRegistry;
  61
  62     @Before
  63     public void init() {
  64         dataProvider = mock(DataBroker.class);
  65         baseEndpointRendererAugmentation = mock(BaseEndpointRendererAugmentation.class);
  66         WriteTransaction wt = newWriteTransactionMock();
  67         RpcProviderRegistry rpcRegistry = mock(RpcProviderRegistry.class);
  68
  69         baseEndpointRpcRegistry = new BaseEndpointRpcRegistry(dataProvider, rpcRegistry);
  70     }
  71
  72     @Test
  73     public void testConstructor() throws Exception {
  74         RpcProviderRegistry rpcRegistry = mock(RpcProviderRegistry.class);
  75         BaseEndpointRpcRegistry registry = new BaseEndpointRpcRegistry(dataProvider, rpcRegistry);
  76         registry.close();
  77     }
  78
  79     @Test
  80     public void testRegister() throws Exception {
  81         baseEndpointRpcRegistry.register(baseEndpointRendererAugmentation);
  82         Assert.assertEquals(1, BaseEndpointRpcRegistry.registeredRenderers.size());
  83
  84         baseEndpointRpcRegistry.unregister(baseEndpointRendererAugmentation);
  85         Assert.assertEquals(0, BaseEndpointRpcRegistry.registeredRenderers.size());
  86     }
  87
  88     @Test
  89     public void testRegisterEndpoint() throws Exception {
  90         WriteTransaction wt = newWriteTransactionMock();
  91
  92         baseEndpointRpcRegistry.registerEndpoint(createRegisterEndpointInputVariablesForTest());
  93
  94         verify(wt, times(2)).put(eq(LogicalDatastoreType.OPERATIONAL), any(InstanceIdentifier.class),
  95                 any(DataObject.class), eq(true));
  96     }
  97
  98     private RegisterEndpointInput createRegisterEndpointInputVariablesForTest() throws Exception {
  99         RegisterEndpointInputBuilder registerEndpointInputBuilder = new RegisterEndpointInputBuilder();
 100         long timestamp = System.currentTimeMillis();
 101
 102         registerEndpointInputBuilder.setAddressEndpointReg(new ArrayList<>());
 103         registerEndpointInputBuilder.setContainmentEndpointReg(new ArrayList<>());
 104
 105         registerEndpointInputBuilder.getAddressEndpointReg().add(
 106                 new AddressEndpointRegBuilder().setTimestamp(timestamp)
 107                     .setContextId(new ContextId(CONTEXT_ID))
 108                     .setContextType(L2FloodDomain.class)
 109                     .setTenant(new TenantId(TENANT))
 110                     .setAddress(MAC_ADDRESS)
 111                     .setAddressType(MacAddressType.class)
 112                     .setAddressType(AddressType.class)
 113                     .setNetworkContainment(new NetworkContainmentBuilder().setContainment(new NetworkDomainContainmentBuilder()
 114                         .setNetworkDomainId(new NetworkDomainId(DOMAIN)).setNetworkDomainType(Subnet.class).build()).build())
 115                     .setKey(new AddressEndpointRegKey(MAC_ADDRESS,MacAddressType.class, new ContextId(CONTEXT_ID), L2FloodDomain.class))
 116                     .setTimestamp(timestamp).build());
 117
 118         registerEndpointInputBuilder.getContainmentEndpointReg().add(
 119                 new ContainmentEndpointRegBuilder()
 120                     .setTimestamp(timestamp)
 121                     .setContextId(new L2FloodDomainId(FLOOD_DOMAIN))
 122                     .setContextType(L2FloodDomain.class)
 123                     .setTenant(new TenantId(TENANT))
 124                     .setNetworkContainment(new NetworkContainmentBuilder().setContainment(new NetworkDomainContainmentBuilder()
 125                         .setNetworkDomainId(new NetworkDomainId(DOMAIN)).setNetworkDomainType(Subnet.class).build()).build())
 126                     .setKey(new ContainmentEndpointRegKey(new L2FloodDomainId(FLOOD_DOMAIN),L2FloodDomain.class))
 127                     .build());
 128
 129         return registerEndpointInputBuilder.build();
 130     }
 131
 132     @Test
 133     public void testUnregisterEndpoint() throws Exception {
 134         WriteTransaction wt = newWriteTransactionMock();
 135
 136         UnregisterEndpointInput unregisterEndpointInput = unregisterEndpointInput();
 137
 138         baseEndpointRpcRegistry.unregisterEndpoint(unregisterEndpointInput);
 139
 140         verify(wt, times(2)).delete(eq(LogicalDatastoreType.OPERATIONAL), any(InstanceIdentifier.class));
 141     }
 142
 143     private UnregisterEndpointInput unregisterEndpointInput() {
 144         UnregisterEndpointInputBuilder builder = new UnregisterEndpointInputBuilder();
 145
 146         builder.setAddressEndpointUnreg(new ArrayList<>());
 147         builder.setContainmentEndpointUnreg(new ArrayList<>());
 148
 149         builder.getAddressEndpointUnreg().add(new AddressEndpointUnregBuilder().setContextId(new ContextId(CONTEXT_ID))
 150             .setContextType(L2FloodDomain.class)
 151             .setAddress(MAC_ADDRESS)
 152             .setAddressType(MacAddressType.class)
 153             .build());
 154
 155         builder.getContainmentEndpointUnreg().add(new ContainmentEndpointUnregBuilder()
 156             .setContextId(new ContextId(CONTEXT_ID)).setContextType(L2FloodDomain.class).build());
 157
 158         return builder.build();
 159     }
 160
 161     private WriteTransaction newWriteTransactionMock() {
 162         WriteTransaction wt = mock(WriteTransaction.class);
 163         CheckedFuture<Void, TransactionCommitFailedException> f = mock(CheckedFuture.class);
 164
 165         when(dataProvider.newWriteOnlyTransaction()).thenReturn(wt);
 166         when(wt.submit()).thenReturn(f);
 167         return wt;
 168     }
 169
 170 }