renderers/ofoverlay/src/test/java/org/opendaylight/groupbasedpolicy/renderer/ofoverlay/mapper/egressnat/EgressNatMapperTest.java

   1 /*
   2  * Copyright (c) 2015 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.renderer.ofoverlay.mapper.egressnat;
  10
  11
  12 import org.junit.Before;
  13 import org.junit.Test;
  14 import org.mockito.InOrder;
  15 import org.mockito.Mockito;
  16 import org.opendaylight.groupbasedpolicy.dto.EpKey;
  17 import org.opendaylight.groupbasedpolicy.renderer.ofoverlay.OfContext;
  18 import org.opendaylight.groupbasedpolicy.renderer.ofoverlay.OfWriter;
  19 import org.opendaylight.groupbasedpolicy.renderer.ofoverlay.PolicyManager;
  20 import org.opendaylight.groupbasedpolicy.renderer.ofoverlay.endpoint.EndpointManager;
  21 import org.opendaylight.groupbasedpolicy.renderer.ofoverlay.mapper.MapperUtilsTest;
  22 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.IpAddress;
  23 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Ipv4Address;
  24 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev100924.MacAddress;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.endpoint.rev140421.endpoints.Endpoint;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.endpoint.rev140421.endpoints.EndpointL3;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeConnectorId;
  28
  29 import java.util.ArrayList;
  30 import java.util.List;
  31
  32 import static org.mockito.Mockito.*;
  33
  34 public class EgressNatMapperTest extends MapperUtilsTest {
  35
  36     private short tableId;
  37
  38     @Before
  39     public void init() {
  40         tableId = 5;
  41         ctx = mock(OfContext.class);
  42         ofWriter = mock(OfWriter.class);
  43         endpointManager = mock(EndpointManager.class);
  44         policyManager = mock(PolicyManager.class);
  45     }
  46
  47     @Test
  48     public void testSyncFlows() {
  49         EgressNatMapperFlows flows = mock(EgressNatMapperFlows.class);
  50         EgressNatMapper mapper = new EgressNatMapper(ctx, tableId);
  51
  52         // Endpoint
  53         Endpoint endpoint = buildEndpoint(IPV4_0, MAC_0, CONNECTOR_1).build();
  54
  55         // L3 endpoints
  56         EndpointL3 endpointL3 = buildL3Endpoint(IPV4_0, IPV4_1, MAC_0, L2).build();
  57         List<EndpointL3> endpointsL3 = new ArrayList<>();
  58         endpointsL3.add(endpointL3);
  59
  60         when(ctx.getPolicyManager()).thenReturn(policyManager);
  61         when(ctx.getEndpointManager()).thenReturn(endpointManager);
  62         when(endpointManager.getL3EndpointsWithNat()).thenReturn(endpointsL3);
  63         when(endpointManager.getEndpoint(Mockito.any(EpKey.class))).thenReturn(endpoint);
  64
  65         // Verify usage
  66         mapper.syncFlows(flows, endpoint, ofWriter);
  67         verify(flows, times(1)).dropFlow(Mockito.anyInt(), Mockito.anyLong(), eq(ofWriter));
  68         verify(flows, times(1)).natFlows(Mockito.anyShort(), Mockito.any(EndpointL3.class),
  69                 Mockito.anyInt(), eq(ofWriter));
  70
  71         // Verify order
  72         InOrder order = inOrder(ctx, flows);
  73         order.verify(flows, times(1)).dropFlow(Mockito.anyInt(), Mockito.anyLong(), eq(ofWriter));
  74         order.verify(ctx, times(1)).getPolicyManager();
  75         order.verify(flows, times(1)).natFlows(Mockito.anyShort(), Mockito.any(EndpointL3.class),
  76                 Mockito.anyInt(), eq(ofWriter));
  77     }
  78 }