2 * Copyright (c) 2015 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
9 package org.opendaylight.groupbasedpolicy.renderer.ofoverlay.flow;
11 import static org.mockito.Matchers.any;
12 import static org.mockito.Mockito.mock;
13 import static org.mockito.Mockito.never;
14 import static org.mockito.Mockito.times;
15 import static org.mockito.Mockito.verify;
17 import org.junit.Assert;
18 import org.junit.Before;
19 import org.junit.Test;
20 import org.opendaylight.groupbasedpolicy.renderer.ofoverlay.OfWriter;
21 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.IpAddress;
22 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Ipv4Address;
23 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev100924.MacAddress;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.table.Flow;
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.EndpointBuilder;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.endpoint.rev140421.endpoints.EndpointL3;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.endpoint.rev140421.endpoints.EndpointL3Builder;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.l3endpoint.rev151217.NatAddress;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.l3endpoint.rev151217.NatAddressBuilder;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.ofoverlay.rev140528.OfOverlayContext;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.ofoverlay.rev140528.OfOverlayContextBuilder;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeConnectorId;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
36 import com.google.common.collect.ImmutableSet;
38 public class ExternalMapperTest extends FlowTableTest {
40 private ExternalMapper mapper;
42 private short tableId;
43 private NodeId nodeId;
44 private OfWriter ofWriter;
46 private NatAddress natAddr = new NatAddressBuilder()
47 .setNatAddress(new IpAddress(new Ipv4Address("192.168.111.52")))
50 private EndpointL3 natL3Ep = new EndpointL3Builder()
53 .setIpAddress(new IpAddress(new Ipv4Address("10.0.0.3")))
54 .addAugmentation(NatAddress.class, natAddr)
57 private Endpoint l2Ep = new EndpointBuilder()
59 .setMacAddress(new MacAddress("00:00:00:00:00:03"))
61 .setNetworkContainment(sub)
66 public void initialisation() {
69 nodeId = mock(NodeId.class);
70 ofWriter = mock(OfWriter.class);
71 mapper = new ExternalMapper(ctx, tableId);
75 public void consturctorTest() {
76 Assert.assertEquals(tableId, mapper.getTableId());
80 public void syncTest() throws Exception {
81 ctx.addTenant(baseTenant().build());
82 endpointManager.addL3Endpoint(natL3Ep);
83 l2Ep = new EndpointBuilder(l2Ep)
84 .addAugmentation(OfOverlayContext.class,new OfOverlayContextBuilder()
88 endpointManager.addEndpoint(l2Ep);
89 switchManager.addSwitch(nodeId,null,ImmutableSet.<NodeConnectorId>of(new NodeConnectorId("openflow:1:1")), null);
90 mapper.sync(nodeId, ofWriter);
91 verify(ofWriter, times(4)).writeFlow(any(NodeId.class), any(Short.class), any(Flow.class));
95 public void syncTestNoExternalPorts() throws Exception {
96 // we still need ExternalMapper flows (default output and default drop) to be generated
97 mapper.sync(nodeId, ofWriter);
98 verify(ofWriter, times(2)).writeFlow(any(NodeId.class), any(Short.class), any(Flow.class));