diff options
Diffstat (limited to 'framework/src/onos/providers/host/src/test/java/org/onosproject/provider/host/impl/HostLocationProviderTest.java')
| -rw-r--r-- | framework/src/onos/providers/host/src/test/java/org/onosproject/provider/host/impl/HostLocationProviderTest.java | 369 |
1 files changed, 369 insertions, 0 deletions
diff --git a/framework/src/onos/providers/host/src/test/java/org/onosproject/provider/host/impl/HostLocationProviderTest.java b/framework/src/onos/providers/host/src/test/java/org/onosproject/provider/host/impl/HostLocationProviderTest.java new file mode 100644 index 00000000..b9d90976 --- /dev/null +++ b/framework/src/onos/providers/host/src/test/java/org/onosproject/provider/host/impl/HostLocationProviderTest.java @@ -0,0 +1,369 @@ +/* + * Copyright 2014-2015 Open Networking Laboratory + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.onosproject.provider.host.impl; + +import com.google.common.collect.ImmutableSet; +import org.junit.After; +import org.junit.Before; +import org.junit.Test; +import org.onlab.osgi.ComponentContextAdapter; +import org.onlab.packet.ARP; +import org.onlab.packet.ChassisId; +import org.onlab.packet.Ethernet; +import org.onlab.packet.IpAddress; +import org.onlab.packet.MacAddress; +import org.onlab.packet.VlanId; +import org.onosproject.cfg.ComponentConfigAdapter; +import org.onosproject.core.ApplicationId; +import org.onosproject.core.CoreService; +import org.onosproject.core.DefaultApplicationId; +import org.onosproject.net.ConnectPoint; +import org.onosproject.net.DefaultDevice; +import org.onosproject.net.DefaultHost; +import org.onosproject.net.DefaultPort; +import org.onosproject.net.Device; +import org.onosproject.net.DeviceId; +import org.onosproject.net.Host; +import org.onosproject.net.HostId; +import org.onosproject.net.HostLocation; +import org.onosproject.net.device.DeviceEvent; +import org.onosproject.net.device.DeviceListener; +import org.onosproject.net.device.DeviceServiceAdapter; +import org.onosproject.net.flow.TrafficTreatment; +import org.onosproject.net.host.HostDescription; +import org.onosproject.net.host.HostProvider; +import org.onosproject.net.host.HostProviderRegistry; +import org.onosproject.net.host.HostProviderService; +import org.onosproject.net.host.HostServiceAdapter; +import org.onosproject.net.packet.DefaultInboundPacket; +import org.onosproject.net.packet.InboundPacket; +import org.onosproject.net.packet.OutboundPacket; +import org.onosproject.net.packet.PacketContext; +import org.onosproject.net.packet.PacketProcessor; +import org.onosproject.net.packet.PacketServiceAdapter; +import org.onosproject.net.provider.AbstractProviderService; +import org.onosproject.net.provider.ProviderId; +import org.onosproject.net.topology.Topology; +import org.onosproject.net.topology.TopologyServiceAdapter; + +import java.nio.ByteBuffer; +import java.util.Collections; +import java.util.Dictionary; +import java.util.Hashtable; +import java.util.Set; + +import static org.easymock.EasyMock.*; +import static org.junit.Assert.*; +import static org.onlab.packet.VlanId.vlanId; +import static org.onosproject.net.Device.Type.SWITCH; +import static org.onosproject.net.DeviceId.deviceId; +import static org.onosproject.net.HostId.hostId; +import static org.onosproject.net.PortNumber.portNumber; +import static org.onosproject.net.device.DeviceEvent.Type.*; + +public class HostLocationProviderTest { + + private static final Integer INPORT = 10; + private static final String DEV1 = "of:1"; + private static final String DEV2 = "of:2"; + private static final String DEV3 = "of:3"; + + private static final VlanId VLAN = vlanId(); + private static final MacAddress MAC = MacAddress.valueOf("00:00:11:00:00:01"); + private static final MacAddress BCMAC = MacAddress.valueOf("ff:ff:ff:ff:ff:ff"); + private static final byte[] IP = new byte[]{10, 0, 0, 1}; + + private static final IpAddress IP_ADDRESS = + IpAddress.valueOf(IpAddress.Version.INET, IP); + private static final HostLocation LOCATION = + new HostLocation(deviceId(DEV1), portNumber(INPORT), 0L); + + private static final DefaultHost HOST = + new DefaultHost(ProviderId.NONE, hostId(MAC), MAC, + vlanId(VlanId.UNTAGGED), LOCATION, + ImmutableSet.of(IP_ADDRESS)); + + private static final ComponentContextAdapter CTX_FOR_REMOVE = + new ComponentContextAdapter() { + @Override + public Dictionary getProperties() { + Hashtable<String, String> props = new Hashtable<>(); + props.put("hostRemovalEnabled", "true"); + return props; + } + }; + + public static final ComponentContextAdapter CTX_FOR_NO_REMOVE = + new ComponentContextAdapter() { + @Override + public Dictionary getProperties() { + return new Hashtable(); + } + }; + + private final HostLocationProvider provider = new HostLocationProvider(); + private final TestHostRegistry hostRegistry = new TestHostRegistry(); + private final TestTopologyService topoService = new TestTopologyService(); + private final TestDeviceService deviceService = new TestDeviceService(); + private final TestHostService hostService = new TestHostService(); + private final TestPacketService packetService = new TestPacketService(); + + private PacketProcessor testProcessor; + private CoreService coreService; + private TestHostProviderService providerService; + + private ApplicationId appId = + new DefaultApplicationId(100, "org.onosproject.provider.host"); + + @Before + public void setUp() { + + coreService = createMock(CoreService.class); + expect(coreService.registerApplication(appId.name())) + .andReturn(appId).anyTimes(); + replay(coreService); + + provider.cfgService = new ComponentConfigAdapter(); + provider.coreService = coreService; + + provider.providerRegistry = hostRegistry; + provider.topologyService = topoService; + provider.packetService = packetService; + provider.deviceService = deviceService; + provider.hostService = hostService; + + provider.activate(CTX_FOR_NO_REMOVE); + } + + @Test + public void basics() { + assertNotNull("registration expected", providerService); + assertEquals("incorrect provider", provider, providerService.provider()); + } + + @Test + public void events() { + // new host + testProcessor.process(new TestPacketContext(DEV1)); + assertNotNull("new host expected", providerService.added); + assertNull("host motion unexpected", providerService.moved); + + // the host moved to new switch + testProcessor.process(new TestPacketContext(DEV2)); + assertNotNull("host motion expected", providerService.moved); + + // the host was misheard on a spine + testProcessor.process(new TestPacketContext(DEV3)); + assertNull("host misheard on spine switch", providerService.spine); + } + + @Test + public void removeHostByDeviceRemove() { + provider.modified(CTX_FOR_REMOVE); + testProcessor.process(new TestPacketContext(DEV1)); + Device device = new DefaultDevice(ProviderId.NONE, deviceId(DEV1), SWITCH, + "m", "h", "s", "n", new ChassisId(0L)); + deviceService.listener.event(new DeviceEvent(DEVICE_REMOVED, device)); + assertEquals("incorrect remove count", 1, providerService.removeCount); + } + + @Test + public void removeHostByDeviceOffline() { + provider.modified(CTX_FOR_REMOVE); + testProcessor.process(new TestPacketContext(DEV1)); + Device device = new DefaultDevice(ProviderId.NONE, deviceId(DEV1), SWITCH, + "m", "h", "s", "n", new ChassisId(0L)); + deviceService.listener.event(new DeviceEvent(DEVICE_AVAILABILITY_CHANGED, device)); + assertEquals("incorrect remove count", 1, providerService.removeCount); + } + + @Test + public void removeHostByDevicePortDown() { + provider.modified(CTX_FOR_REMOVE); + testProcessor.process(new TestPacketContext(DEV1)); + Device device = new DefaultDevice(ProviderId.NONE, deviceId(DEV1), SWITCH, + "m", "h", "s", "n", new ChassisId(0L)); + deviceService.listener.event(new DeviceEvent(PORT_UPDATED, device, + new DefaultPort(device, portNumber(INPORT), + false))); + assertEquals("incorrect remove count", 1, providerService.removeCount); + } + + + @After + public void tearDown() { + provider.deactivate(); + provider.coreService = null; + provider.providerRegistry = null; + } + + private class TestHostRegistry implements HostProviderRegistry { + + @Override + public HostProviderService register(HostProvider provider) { + providerService = new TestHostProviderService(provider); + return providerService; + } + + @Override + public void unregister(HostProvider provider) { + } + + @Override + public Set<ProviderId> getProviders() { + return null; + } + + } + + private class TestHostProviderService + extends AbstractProviderService<HostProvider> + implements HostProviderService { + + DeviceId added = null; + DeviceId moved = null; + DeviceId spine = null; + public int removeCount; + + protected TestHostProviderService(HostProvider provider) { + super(provider); + } + + @Override + public void hostDetected(HostId hostId, HostDescription hostDescription) { + DeviceId descr = hostDescription.location().deviceId(); + if (added == null) { + added = descr; + } else if ((moved == null) && !descr.equals(added)) { + moved = descr; + } else { + spine = descr; + } + } + + @Override + public void hostVanished(HostId hostId) { + removeCount++; + } + + } + + private class TestPacketService extends PacketServiceAdapter { + @Override + public void addProcessor(PacketProcessor processor, int priority) { + testProcessor = processor; + } + } + + + private class TestTopologyService extends TopologyServiceAdapter { + @Override + public boolean isInfrastructure(Topology topology, + ConnectPoint connectPoint) { + //simulate DPID3 as an infrastructure switch + if ((connectPoint.deviceId()).equals(deviceId(DEV3))) { + return true; + } + return false; + } + } + + private class TestPacketContext implements PacketContext { + + private final String deviceId; + + public TestPacketContext(String deviceId) { + this.deviceId = deviceId; + } + + @Override + public long time() { + return 0; + } + + @Override + public InboundPacket inPacket() { + ARP arp = new ARP(); + arp.setSenderProtocolAddress(IP) + .setSenderHardwareAddress(MAC.toBytes()) + .setTargetHardwareAddress(BCMAC.toBytes()) + .setTargetProtocolAddress(IP); + + Ethernet eth = new Ethernet(); + eth.setEtherType(Ethernet.TYPE_ARP) + .setVlanID(VLAN.toShort()) + .setSourceMACAddress(MAC.toBytes()) + .setDestinationMACAddress(BCMAC) + .setPayload(arp); + ConnectPoint receivedFrom = new ConnectPoint(deviceId(deviceId), + portNumber(INPORT)); + return new DefaultInboundPacket(receivedFrom, eth, + ByteBuffer.wrap(eth.serialize())); + } + + @Override + public OutboundPacket outPacket() { + return null; + } + + @Override + public TrafficTreatment.Builder treatmentBuilder() { + return null; + } + + @Override + public void send() { + + } + + @Override + public boolean block() { + return false; + } + + @Override + public boolean isHandled() { + return false; + } + } + + private class TestDeviceService extends DeviceServiceAdapter { + private DeviceListener listener; + + @Override + public void addListener(DeviceListener listener) { + this.listener = listener; + } + + @Override + public Iterable<Device> getDevices() { + return Collections.emptyList(); + } + } + + private class TestHostService extends HostServiceAdapter { + @Override + public Set<Host> getConnectedHosts(ConnectPoint connectPoint) { + return ImmutableSet.of(HOST); + } + + @Override + public Set<Host> getConnectedHosts(DeviceId deviceId) { + return ImmutableSet.of(HOST); + } + + } +} |