renderers/ofoverlay/src/test/java/org/opendaylight/groupbasedpolicy/renderer/ofoverlay/mapper/ingressnat/IngressNatMapperTest.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.ingressnat;
  10
  11 import org.junit.Before;
  12 import org.junit.Test;
  13 import org.mockito.InOrder;
  14 import org.mockito.Mockito;
  15 import org.opendaylight.groupbasedpolicy.dto.EgKey;
  16 import org.opendaylight.groupbasedpolicy.dto.EpKey;
  17 import org.opendaylight.groupbasedpolicy.dto.IndexedTenant;
  18 import org.opendaylight.groupbasedpolicy.dto.PolicyInfo;
  19 import org.opendaylight.groupbasedpolicy.renderer.ofoverlay.OfContext;
  20 import org.opendaylight.groupbasedpolicy.renderer.ofoverlay.OfWriter;
  21 import org.opendaylight.groupbasedpolicy.renderer.ofoverlay.PolicyManager;
  22 import org.opendaylight.groupbasedpolicy.renderer.ofoverlay.endpoint.EndpointManager;
  23 import org.opendaylight.groupbasedpolicy.renderer.ofoverlay.flow.OrdinalFactory;
  24 import org.opendaylight.groupbasedpolicy.renderer.ofoverlay.mapper.MapperUtilsTest;
  25 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.IpAddress;
  26 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Ipv4Address;
  27 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev100924.MacAddress;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.EndpointGroupId;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.TenantId;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.endpoint.rev140421.endpoints.Endpoint;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.endpoint.rev140421.endpoints.EndpointL3;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeConnectorId;
  33
  34 import java.util.ArrayList;
  35 import java.util.HashSet;
  36 import java.util.List;
  37 import java.util.Set;
  38
  39 import static org.mockito.Mockito.*;
  40
  41 public class IngressNatMapperTest extends MapperUtilsTest {
  42
  43     private static final String EPG_ID = "dummy epg id";
  44
  45     @Before
  46     public void init() {
  47         ctx = mock(OfContext.class);
  48         ofWriter = mock(OfWriter.class);
  49         endpointManager = mock(EndpointManager.class);
  50         policyManager = mock(PolicyManager.class);
  51         policyInfo = mock(PolicyInfo.class);
  52         tableId = 1;
  53     }
  54
  55     @Test
  56     public void testSyncFlows() throws Exception {
  57
  58         // Endpoints
  59         Endpoint endpoint = endpointBuilder(new IpAddress(new Ipv4Address(IPV4_1)), new MacAddress(MAC_0),
  60                 new NodeConnectorId(CONNECTOR_0), null, null).build();
  61         List<Endpoint> endpoints = new ArrayList<>();
  62         endpoints.add(endpoint);
  63
  64         // L3 endpoints
  65         EndpointL3 endpointL3 = endpointL3Builder(IPV4_1, IPV4_2, MAC_0, L2, false).build();
  66         List<EndpointL3> endpointsL3 = new ArrayList<>();
  67         endpointsL3.add(endpointL3);
  68
  69         // EgKEy set
  70         Set<EgKey> egKeys = new HashSet<>();
  71         egKeys.add(new EgKey(new TenantId(TENANT_ID), new EndpointGroupId(EPG_ID)));
  72
  73         when(ctx.getPolicyManager()).thenReturn(policyManager);
  74         when(ctx.getTenant(Mockito.any(TenantId.class))).thenReturn(indexedTenantBuilder());
  75         when(ctx.getEndpointManager()).thenReturn(endpointManager);
  76         when(endpointManager.getL3EndpointsWithNat()).thenReturn(endpointsL3);
  77         when(endpointManager.getEndpoint(Mockito.any(EpKey.class))).thenReturn(endpoint);
  78         when(endpointManager.getEgKeysForEndpoint(Mockito.any(Endpoint.class))).thenReturn(egKeys);
  79         when(ctx.getCurrentPolicy()).thenReturn(policyInfo);
  80         when(endpointManager.getExtEpsNoLocForGroup(Mockito.any(EgKey.class))).thenReturn(endpoints);
  81
  82         IngressNatMapperFlows flows = mock(IngressNatMapperFlows.class);
  83         IngressNatMapper ingressNatMapper = new IngressNatMapper(ctx, tableId);
  84         ingressNatMapper.syncFlows(flows, endpoint, ofWriter);
  85
  86         // Verify usage
  87         verify(flows, times(1)).baseFlow(Mockito.anyShort(), Mockito.anyInt(), eq(ofWriter));
  88         verify(flows, times(1)).createNatFlow(Mockito.anyShort(), Mockito.any(EndpointL3.class),
  89                 Mockito.any(OrdinalFactory.EndpointFwdCtxOrdinals.class), Mockito.anyInt(), eq(ofWriter));
  90         verify(flows, times(1)).createArpFlow(Mockito.any(IndexedTenant.class), Mockito.any(EndpointL3.class),
  91                 Mockito.anyInt(), eq(ofWriter));
  92         verify(flows, times(1)).createIngressExternalNatFlows(Mockito.anyShort(), Mockito.any(Endpoint.class),
  93                 Mockito.any(OrdinalFactory.EndpointFwdCtxOrdinals.class), Mockito.anyInt(), eq(ofWriter));
  94         verify(flows, times(1)).createIngressExternalArpFlows(Mockito.anyShort(), Mockito.any(Endpoint.class),
  95                 Mockito.any(OrdinalFactory.EndpointFwdCtxOrdinals.class), Mockito.anyInt(), eq(ofWriter));
  96
  97         // Verify order
  98         InOrder order = inOrder(ctx, flows);
  99         order.verify(ctx, times(1)).getPolicyManager();
 100         order.verify(flows, times(1)).baseFlow(Mockito.anyShort(), Mockito.anyInt(), eq(ofWriter));
 101         order.verify(flows, times(1)).createNatFlow(Mockito.anyShort(), Mockito.any(EndpointL3.class),
 102                 Mockito.any(OrdinalFactory.EndpointFwdCtxOrdinals.class), Mockito.anyInt(), eq(ofWriter));
 103         order.verify(ctx, times(1)).getTenant(Mockito.any(TenantId.class));
 104         order.verify(flows, times(1)).createArpFlow(Mockito.any(IndexedTenant.class), Mockito.any(EndpointL3.class),
 105                 Mockito.anyInt(), eq(ofWriter));
 106         order.verify(ctx, times(1)).getCurrentPolicy();
 107         order.verify(ctx, times(1)).getEndpointManager();
 108         order.verify(flows, times(1)).createIngressExternalNatFlows(Mockito.anyShort(), Mockito.any(Endpoint.class),
 109                 Mockito.any(OrdinalFactory.EndpointFwdCtxOrdinals.class), Mockito.anyInt(), eq(ofWriter));
 110         order.verify(flows, times(1)).createIngressExternalArpFlows(Mockito.anyShort(), Mockito.any(Endpoint.class),
 111                 Mockito.any(OrdinalFactory.EndpointFwdCtxOrdinals.class), Mockito.anyInt(), eq(ofWriter));
 112     }
 113 }