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