sxp-integration/sxp-ep-provider/src/test/java/org/opendaylight/groupbasedpolicy/sxp/ep/provider/impl/dao/EPForwardingTemplateDaoImplTest.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.sxp.ep.provider.impl.dao;
  10
  11 import com.google.common.base.Optional;
  12 import com.google.common.collect.Lists;
  13 import com.google.common.util.concurrent.Futures;
  14 import com.google.common.util.concurrent.ListenableFuture;
  15 import org.junit.Assert;
  16 import org.junit.Before;
  17 import org.junit.Test;
  18 import org.junit.runner.RunWith;
  19 import org.mockito.InOrder;
  20 import org.mockito.Matchers;
  21 import org.mockito.Mock;
  22 import org.mockito.Mockito;
  23 import org.mockito.runners.MockitoJUnitRunner;
  24 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  25 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
  26 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  27 import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
  28 import org.opendaylight.groupbasedpolicy.sxp.ep.provider.impl.SimpleCachedDao;
  29 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpPrefix;
  30 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Prefix;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.sxp.integration.sxp.ep.provider.model.rev160302.SxpEpMapper;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.sxp.integration.sxp.ep.provider.model.rev160302.SxpEpMapperBuilder;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.sxp.integration.sxp.ep.provider.model.rev160302.sxp.ep.mapper.EndpointForwardingTemplateBySubnet;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.sxp.integration.sxp.ep.provider.model.rev160302.sxp.ep.mapper.EndpointForwardingTemplateBySubnetBuilder;
  35 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  36
  37 /**
  38  * Test for {@link EPForwardingTemplateDaoImpl}.
  39  */
  40 @RunWith(MockitoJUnitRunner.class)
  41 public class EPForwardingTemplateDaoImplTest {
  42
  43     public static final InstanceIdentifier<SxpEpMapper> SXP_MAPPER_PATH = InstanceIdentifier.create(SxpEpMapper.class);
  44     private static final IpPrefix KEY_1 = new IpPrefix(new Ipv4Prefix("1.2.3.4/32"));
  45     private final SxpEpMapper SXP_MAPPER_VALUE;
  46     private final EndpointForwardingTemplateBySubnet EP_FW_TEMPLATE_VALUE;
  47
  48     @Mock
  49     private DataBroker dataBroker;
  50     @Mock
  51     private SimpleCachedDao<IpPrefix, EndpointForwardingTemplateBySubnet> cachedDao;
  52     @Mock
  53     private ReadOnlyTransaction rTx;
  54
  55     private EPForwardingTemplateDaoImpl dao;
  56
  57     public EPForwardingTemplateDaoImplTest() {
  58         EP_FW_TEMPLATE_VALUE = new EndpointForwardingTemplateBySubnetBuilder()
  59                 .setIpPrefix(KEY_1)
  60                 .build();
  61         SXP_MAPPER_VALUE = new SxpEpMapperBuilder()
  62                 .setEndpointForwardingTemplateBySubnet(Lists.newArrayList(EP_FW_TEMPLATE_VALUE))
  63                 .build();
  64     }
  65
  66     @Before
  67     public void setUp() throws Exception {
  68         dao = new EPForwardingTemplateDaoImpl(dataBroker, cachedDao);
  69     }
  70
  71     @Test
  72     public void testRead_absent() throws Exception {
  73         Mockito.when(cachedDao.find(Matchers.<IpPrefix>any())).thenReturn(Optional.<EndpointForwardingTemplateBySubnet>absent());
  74         Mockito.when(dataBroker.newReadOnlyTransaction()).thenReturn(rTx);
  75         Mockito.when(rTx.read(Matchers.eq(LogicalDatastoreType.CONFIGURATION),
  76                 Matchers.<InstanceIdentifier<SxpEpMapper>>any())).thenReturn(
  77                 Futures.<Optional<SxpEpMapper>, ReadFailedException>immediateCheckedFuture(Optional.<SxpEpMapper>absent()));
  78
  79
  80         final ListenableFuture<Optional<EndpointForwardingTemplateBySubnet>> read = dao.read(KEY_1);
  81         Assert.assertTrue(read.isDone());
  82         Assert.assertFalse(read.get().isPresent());
  83     }
  84
  85     @Test
  86     public void testRead_presentCached() throws Exception {
  87         Mockito.when(cachedDao.find(Matchers.<IpPrefix>any())).thenReturn(Optional.of(EP_FW_TEMPLATE_VALUE));
  88
  89         final ListenableFuture<Optional<EndpointForwardingTemplateBySubnet>> read = dao.read(KEY_1);
  90         Assert.assertTrue(read.isDone());
  91         Assert.assertTrue(read.get().isPresent());
  92         Assert.assertEquals(KEY_1, read.get().get().getIpPrefix());
  93     }
  94
  95     @Test
  96     public void testRead_presentDS() throws Exception {
  97         Mockito.when(cachedDao.find(Matchers.<IpPrefix>any())).thenReturn(
  98                 Optional.<EndpointForwardingTemplateBySubnet>absent(),
  99                 Optional.of(EP_FW_TEMPLATE_VALUE));
 100         Mockito.when(cachedDao.isEmpty()).thenReturn(true, false);
 101         Mockito.when(dataBroker.newReadOnlyTransaction()).thenReturn(rTx);
 102         Mockito.when(rTx.read(Matchers.eq(LogicalDatastoreType.CONFIGURATION),
 103                 Matchers.<InstanceIdentifier<SxpEpMapper>>any())).thenReturn(
 104                 Futures.<Optional<SxpEpMapper>, ReadFailedException>immediateCheckedFuture(Optional.of(SXP_MAPPER_VALUE)));
 105
 106         final ListenableFuture<Optional<EndpointForwardingTemplateBySubnet>> read = dao.read(KEY_1);
 107         Assert.assertTrue(read.isDone());
 108         Assert.assertTrue(read.get().isPresent());
 109         Assert.assertEquals(KEY_1, read.get().get().getIpPrefix());
 110
 111         final InOrder inOrder = Mockito.inOrder(cachedDao);
 112         inOrder.verify(cachedDao).invalidateCache();
 113         inOrder.verify(cachedDao).update(KEY_1, EP_FW_TEMPLATE_VALUE);
 114         inOrder.verify(cachedDao).find(KEY_1);
 115         inOrder.verifyNoMoreInteractions();
 116     }
 117
 118     @Test
 119     public void testBuildReadPath() throws Exception {
 120         final InstanceIdentifier<SxpEpMapper> readPath = dao.buildReadPath(KEY_1);
 121         Assert.assertEquals(SXP_MAPPER_PATH, readPath);
 122     }
 123 }