2 * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
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
9 package org.opendaylight.groupbasedpolicy.sxp.ep.provider.impl.listen;
11 import com.google.common.base.Optional;
12 import com.google.common.util.concurrent.Futures;
13 import java.util.Collections;
14 import org.junit.Before;
15 import org.junit.Test;
16 import org.junit.runner.RunWith;
17 import org.mockito.InOrder;
18 import org.mockito.Matchers;
19 import org.mockito.Mock;
20 import org.mockito.Mockito;
21 import org.mockito.runners.MockitoJUnitRunner;
22 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
23 import org.opendaylight.controller.md.sal.binding.api.DataObjectModification;
24 import org.opendaylight.controller.md.sal.binding.api.ClusteredDataTreeChangeListener;
25 import org.opendaylight.controller.md.sal.binding.api.DataTreeIdentifier;
26 import org.opendaylight.controller.md.sal.binding.api.DataTreeModification;
27 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
28 import org.opendaylight.groupbasedpolicy.sxp.ep.provider.impl.DSAsyncDao;
29 import org.opendaylight.groupbasedpolicy.sxp.ep.provider.impl.EPTemplateListener;
30 import org.opendaylight.groupbasedpolicy.sxp.ep.provider.impl.MasterDatabaseBindingListener;
31 import org.opendaylight.groupbasedpolicy.sxp.ep.provider.impl.SimpleCachedDao;
32 import org.opendaylight.groupbasedpolicy.sxp.ep.provider.impl.SxpMapperReactor;
33 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpPrefix;
34 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Prefix;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.sxp.ep.provider.model.rev160302.sxp.ep.mapper.EndpointForwardingTemplateBySubnet;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.sxp.ep.provider.model.rev160302.sxp.ep.mapper.EndpointForwardingTemplateBySubnetBuilder;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.sxp.ep.provider.model.rev160302.sxp.ep.mapper.EndpointPolicyTemplateBySgt;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.sxp.ep.provider.model.rev160302.sxp.ep.mapper.EndpointPolicyTemplateBySgtBuilder;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.sxp.database.rev160308.Sgt;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.sxp.database.rev160308.master.database.fields.MasterDatabaseBinding;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.sxp.database.rev160308.master.database.fields.MasterDatabaseBindingBuilder;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.sxp.database.rev160308.master.database.fields.MasterDatabaseBindingKey;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.sxp.node.rev160308.SxpNodeIdentity;
44 import org.opendaylight.yang.gen.v1.urn.opendaylight.sxp.node.rev160308.network.topology.topology.node.SxpDomains;
45 import org.opendaylight.yang.gen.v1.urn.opendaylight.sxp.node.rev160308.network.topology.topology.node.sxp.domains.SxpDomain;
46 import org.opendaylight.yang.gen.v1.urn.opendaylight.sxp.node.rev160308.network.topology.topology.node.sxp.domains.SxpDomainKey;
47 import org.opendaylight.yang.gen.v1.urn.opendaylight.sxp.node.rev160308.sxp.databases.fields.MasterDatabase;
48 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
49 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
50 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeKey;
51 import org.opendaylight.yangtools.concepts.ListenerRegistration;
52 import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
55 * Test for {@link MasterDatabaseBindingListenerImpl}.
57 @RunWith(MockitoJUnitRunner.class)
58 public class MasterDatabaseBindingListenerImplTest {
60 private static final Sgt SGT_1 = new Sgt(1);
61 private static final IpPrefix IP_PREFIX = new IpPrefix(new Ipv4Prefix("1.2.3.4/32"));
62 private static final KeyedInstanceIdentifier<MasterDatabaseBinding, MasterDatabaseBindingKey> MASTER_DB_PATH =
63 MasterDatabaseBindingListener.SXP_TOPOLOGY_PATH
64 .child(Node.class, new NodeKey(new NodeId("utNodeId")))
65 .augmentation(SxpNodeIdentity.class)
66 .child(SxpDomains.class)
67 .child(SxpDomain.class, new SxpDomainKey("global"))
68 .child(MasterDatabase.class)
69 .child(MasterDatabaseBinding.class, new MasterDatabaseBindingKey(IP_PREFIX));
70 private static final DataTreeIdentifier<MasterDatabaseBinding> MASTER_DB_BINDING_TREE_PATH =
71 new DataTreeIdentifier<>(LogicalDatastoreType.CONFIGURATION, MASTER_DB_PATH);
72 private final MasterDatabaseBinding MASTER_DB_BINDING_VALUE;
75 private DataBroker dataBroker;
77 private SxpMapperReactor sxpMapper;
79 private DSAsyncDao<Sgt, EndpointPolicyTemplateBySgt> epPolicyTemplateDao;
81 private DSAsyncDao<IpPrefix, EndpointForwardingTemplateBySubnet> epForwardingTemplateDao;
83 private SimpleCachedDao<IpPrefix, MasterDatabaseBinding> cachedDao;
85 private ListenerRegistration<? extends EPTemplateListener> listenerRegistration;
87 private DataTreeModification<MasterDatabaseBinding> dataTreeModification;
89 private DataObjectModification<MasterDatabaseBinding> dataObjectModification;
91 private MasterDatabaseBindingListenerImpl listener;
93 public MasterDatabaseBindingListenerImplTest() {
94 MASTER_DB_BINDING_VALUE = new MasterDatabaseBindingBuilder()
95 .setSecurityGroupTag(SGT_1)
96 .setIpPrefix(IP_PREFIX)
102 public void setUp() throws Exception {
103 Mockito.when(dataBroker.registerDataTreeChangeListener(Matchers.<DataTreeIdentifier>any(),
104 Matchers.<ClusteredDataTreeChangeListener>any())).thenReturn(listenerRegistration);
105 listener = new MasterDatabaseBindingListenerImpl(dataBroker, sxpMapper, cachedDao, epPolicyTemplateDao,
106 epForwardingTemplateDao);
110 public void testOnDataTreeChanged() throws Exception {
111 Mockito.when(dataTreeModification.getRootNode()).thenReturn(dataObjectModification);
112 Mockito.when(dataTreeModification.getRootPath()).thenReturn(MASTER_DB_BINDING_TREE_PATH);
113 Mockito.when(dataObjectModification.getDataAfter()).thenReturn(MASTER_DB_BINDING_VALUE);
115 // prepare epPolicy template
116 final EndpointPolicyTemplateBySgt epPolicyTemplate = new EndpointPolicyTemplateBySgtBuilder()
119 Mockito.when(epPolicyTemplateDao.read(Matchers.<Sgt>any())).thenReturn(
120 Futures.immediateFuture(Optional.of(epPolicyTemplate)));
122 // prepare epForwarding template
123 final IpPrefix ipPrefixSubnet = new IpPrefix(new Ipv4Prefix("1.2.3.0/24"));
124 final EndpointForwardingTemplateBySubnet epForwardingTemplate = new EndpointForwardingTemplateBySubnetBuilder()
125 .setIpPrefix(ipPrefixSubnet)
127 Mockito.when(epForwardingTemplateDao.read(Matchers.<IpPrefix>any())).thenReturn(
128 Futures.immediateFuture(Optional.of(epForwardingTemplate)));
130 listener.onDataTreeChanged(Collections.singleton(dataTreeModification));
132 final InOrder inOrder = Mockito.inOrder(cachedDao, epPolicyTemplateDao, epForwardingTemplateDao, sxpMapper);
133 inOrder.verify(cachedDao).update(IP_PREFIX, MASTER_DB_BINDING_VALUE);
134 inOrder.verify(epPolicyTemplateDao).read(SGT_1);
135 inOrder.verify(epForwardingTemplateDao).read(IP_PREFIX);
136 inOrder.verify(sxpMapper).processTemplatesAndSxpMasterDB(epPolicyTemplate, epForwardingTemplate, MASTER_DB_BINDING_VALUE);
137 inOrder.verifyNoMoreInteractions();
141 public void testClose() throws Exception {
142 Mockito.verify(listenerRegistration, Mockito.never()).close();
144 Mockito.verify(listenerRegistration).close();