| /* |
| * 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); |
| } |
| |
| } |
| } |