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