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
8 package org.opendaylight.groupbasedpolicy.sxp.mapper.impl.listen;
10 import java.util.Collections;
12 import org.junit.Before;
13 import org.junit.Test;
14 import org.junit.runner.RunWith;
15 import org.mockito.InOrder;
16 import org.mockito.Matchers;
17 import org.mockito.Mock;
18 import org.mockito.Mockito;
19 import org.mockito.runners.MockitoJUnitRunner;
20 import org.opendaylight.controller.md.sal.binding.api.ClusteredDataTreeChangeListener;
21 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
22 import org.opendaylight.controller.md.sal.binding.api.DataObjectModification;
23 import org.opendaylight.controller.md.sal.binding.api.DataTreeIdentifier;
24 import org.opendaylight.controller.md.sal.binding.api.DataTreeModification;
25 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
26 import org.opendaylight.groupbasedpolicy.sxp.mapper.api.DSAsyncDao;
27 import org.opendaylight.groupbasedpolicy.sxp.mapper.api.EPTemplateListener;
28 import org.opendaylight.groupbasedpolicy.sxp.mapper.api.MasterDatabaseBindingListener;
29 import org.opendaylight.groupbasedpolicy.sxp.mapper.api.SimpleCachedDao;
30 import org.opendaylight.groupbasedpolicy.sxp.mapper.api.SxpMapperReactor;
31 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpPrefix;
32 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Prefix;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.sxp.mapper.model.rev160302.sxp.mapper.EndpointForwardingTemplateBySubnet;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.sxp.mapper.model.rev160302.sxp.mapper.EndpointForwardingTemplateBySubnetBuilder;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.sxp.mapper.model.rev160302.sxp.mapper.EndpointPolicyTemplateBySgt;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.sxp.mapper.model.rev160302.sxp.mapper.EndpointPolicyTemplateBySgtBuilder;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.sxp.database.rev160308.Sgt;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.sxp.database.rev160308.master.database.fields.MasterDatabaseBinding;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.sxp.database.rev160308.master.database.fields.MasterDatabaseBindingBuilder;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.sxp.database.rev160308.master.database.fields.MasterDatabaseBindingKey;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.sxp.node.rev160308.SxpNodeIdentity;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.sxp.node.rev160308.network.topology.topology.node.SxpDomains;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.sxp.node.rev160308.network.topology.topology.node.sxp.domains.SxpDomain;
44 import org.opendaylight.yang.gen.v1.urn.opendaylight.sxp.node.rev160308.network.topology.topology.node.sxp.domains.SxpDomainKey;
45 import org.opendaylight.yang.gen.v1.urn.opendaylight.sxp.node.rev160308.sxp.databases.fields.MasterDatabase;
46 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
47 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
48 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeKey;
49 import org.opendaylight.yangtools.concepts.ListenerRegistration;
50 import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
52 import com.google.common.base.Optional;
53 import com.google.common.util.concurrent.Futures;
56 * Test for {@link MasterDatabaseBindingListenerImpl}.
58 @RunWith(MockitoJUnitRunner.class)
59 public class MasterDatabaseBindingListenerImplTest {
61 private static final Sgt SGT_1 = new Sgt(1);
62 private static final IpPrefix IP_PREFIX = new IpPrefix(new Ipv4Prefix("1.2.3.4/32"));
63 private static final KeyedInstanceIdentifier<MasterDatabaseBinding, MasterDatabaseBindingKey> MASTER_DB_PATH =
64 MasterDatabaseBindingListener.SXP_TOPOLOGY_PATH
65 .child(Node.class, new NodeKey(new NodeId("utNodeId")))
66 .augmentation(SxpNodeIdentity.class)
67 .child(SxpDomains.class)
68 .child(SxpDomain.class, new SxpDomainKey("global"))
69 .child(MasterDatabase.class)
70 .child(MasterDatabaseBinding.class, new MasterDatabaseBindingKey(IP_PREFIX));
71 private static final DataTreeIdentifier<MasterDatabaseBinding> MASTER_DB_BINDING_TREE_PATH =
72 new DataTreeIdentifier<>(LogicalDatastoreType.CONFIGURATION, MASTER_DB_PATH);
73 private final MasterDatabaseBinding MASTER_DB_BINDING_VALUE;
76 private DataBroker dataBroker;
78 private SxpMapperReactor sxpMapper;
80 private DSAsyncDao<Sgt, EndpointPolicyTemplateBySgt> epPolicyTemplateDao;
82 private DSAsyncDao<IpPrefix, EndpointForwardingTemplateBySubnet> epForwardingTemplateDao;
84 private SimpleCachedDao<IpPrefix, MasterDatabaseBinding> cachedDao;
86 private ListenerRegistration<? extends EPTemplateListener> listenerRegistration;
88 private DataTreeModification<MasterDatabaseBinding> dataTreeModification;
90 private DataObjectModification<MasterDatabaseBinding> dataObjectModification;
92 private MasterDatabaseBindingListenerImpl listener;
94 public MasterDatabaseBindingListenerImplTest() {
95 MASTER_DB_BINDING_VALUE = new MasterDatabaseBindingBuilder()
96 .setSecurityGroupTag(SGT_1)
97 .setIpPrefix(IP_PREFIX)
103 public void setUp() throws Exception {
104 Mockito.when(dataBroker.registerDataTreeChangeListener(Matchers.<DataTreeIdentifier>any(),
105 Matchers.<ClusteredDataTreeChangeListener>any())).thenReturn(listenerRegistration);
106 listener = new MasterDatabaseBindingListenerImpl(dataBroker, sxpMapper, cachedDao, epPolicyTemplateDao,
107 epForwardingTemplateDao);
111 public void testOnDataTreeChanged() throws Exception {
112 Mockito.when(dataTreeModification.getRootNode()).thenReturn(dataObjectModification);
113 Mockito.when(dataTreeModification.getRootPath()).thenReturn(MASTER_DB_BINDING_TREE_PATH);
114 Mockito.when(dataObjectModification.getDataAfter()).thenReturn(MASTER_DB_BINDING_VALUE);
116 // prepare epPolicy template
117 final EndpointPolicyTemplateBySgt epPolicyTemplate = new EndpointPolicyTemplateBySgtBuilder()
120 Mockito.when(epPolicyTemplateDao.read(Matchers.<Sgt>any())).thenReturn(
121 Futures.immediateFuture(Optional.of(epPolicyTemplate)));
123 // prepare epForwarding template
124 final IpPrefix ipPrefixSubnet = new IpPrefix(new Ipv4Prefix("1.2.3.0/24"));
125 final EndpointForwardingTemplateBySubnet epForwardingTemplate = new EndpointForwardingTemplateBySubnetBuilder()
126 .setIpPrefix(ipPrefixSubnet)
128 Mockito.when(epForwardingTemplateDao.read(Matchers.<IpPrefix>any())).thenReturn(
129 Futures.immediateFuture(Optional.of(epForwardingTemplate)));
131 listener.onDataTreeChanged(Collections.singleton(dataTreeModification));
133 final InOrder inOrder = Mockito.inOrder(cachedDao, epPolicyTemplateDao, epForwardingTemplateDao, sxpMapper);
134 inOrder.verify(cachedDao).update(IP_PREFIX, MASTER_DB_BINDING_VALUE);
135 inOrder.verify(epPolicyTemplateDao).read(SGT_1);
136 inOrder.verify(epForwardingTemplateDao).read(IP_PREFIX);
137 inOrder.verify(sxpMapper).processTemplatesAndSxpMasterDB(epPolicyTemplate, epForwardingTemplate, MASTER_DB_BINDING_VALUE);
138 inOrder.verifyNoMoreInteractions();
142 public void testClose() throws Exception {
143 Mockito.verify(listenerRegistration, Mockito.never()).close();
145 Mockito.verify(listenerRegistration).close();