ONOS-2740,ONOS-2741,from ONOS-3032 - to ONOS 3071 , OSPF Protocol Implementation Unit Tests

Change-Id: I18972a0712fbd63798f92da7b4c48381b4a38519

+/*
+* Copyright 2016 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.ospf.controller.impl;
+
+import org.jboss.netty.buffer.ChannelBuffer;
+import org.jboss.netty.buffer.ChannelBuffers;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+import org.onlab.packet.Ip4Address;
+import org.onosproject.ospf.controller.DeviceInformation;
+import org.onosproject.ospf.controller.OspfLinkTed;
+import org.onosproject.ospf.controller.OspfLsa;
+import org.onosproject.ospf.controller.area.OspfAreaImpl;
+import org.onosproject.ospf.controller.area.OspfInterfaceImpl;
+import org.onosproject.ospf.protocol.lsa.OpaqueLsaHeader;
+import org.onosproject.ospf.protocol.lsa.TlvHeader;
+import org.onosproject.ospf.protocol.lsa.subtypes.OspfLsaLink;
+import org.onosproject.ospf.protocol.lsa.tlvtypes.LinkTlv;
+import org.onosproject.ospf.protocol.lsa.types.OpaqueLsa10;
+import org.onosproject.ospf.protocol.lsa.types.RouterLsa;
+
+import java.util.List;
+import java.util.Map;
+
+import static org.hamcrest.CoreMatchers.nullValue;
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.is;
+import static org.hamcrest.Matchers.notNullValue;
+
+/**
+ * Unit test class for TopologyForDeviceAndLinkImpl.
+ */
+public class TopologyForDeviceAndLinkImplTest {
+    private final byte[] packet = {0, 9, 0, 4, 0, 0, 0, 1,
+            0, 9, 0, 4, 0, 0, 0, 1,
+            0, 1, 0, 4, 0, 0, 0, 1,
+            0, 2, 0, 4, 0, 0, 0, 1,
+            0, 3, 0, 4, 0, 0, 0, 1,
+            0, 4, 0, 4, 0, 0, 0, 1,
+            0, 6, 0, 4, 0, 0, 0, 1,
+            0, 7, 0, 4, 0, 0, 0, 1,
+            0, 8, 0, 4, 0, 0, 0, 1,
+    };
+    private TopologyForDeviceAndLinkImpl topologyForDeviceAndLink;
+    private Map result;
+    private LinkTlv linkTlv;
+    private TlvHeader header;
+    private ChannelBuffer channelBuffer;
+
+    @Before
+    public void setUp() throws Exception {
+        topologyForDeviceAndLink = new TopologyForDeviceAndLinkImpl();
+    }
+
+    @After
+    public void tearDown() throws Exception {
+        topologyForDeviceAndLink = null;
+    }
+
+    /**
+     * Tests deviceInformationMap() method.
+     */
+    @Test
+    public void testDeviceInformationMap() throws Exception {
+        result = topologyForDeviceAndLink.deviceInformationMap();
+        assertThat(result.size(), is(0));
+    }
+
+    /**
+     * Tests setDeviceInformationMap() method.
+     */
+    @Test
+    public void testSetDeviceInformationMap() throws Exception {
+        topologyForDeviceAndLink.setDeviceInformationMap("1.1.1.1", new DeviceInformationImpl());
+        result = topologyForDeviceAndLink.deviceInformationMap();
+        assertThat(result.size(), is(1));
+    }
+
+    /**
+     * Tests deviceInformation() method.
+     */
+    @Test
+    public void testDeviceInformation() throws Exception {
+        topologyForDeviceAndLink.setDeviceInformationMap("1.1.1.1", new DeviceInformationImpl());
+        DeviceInformation deviceInformation = topologyForDeviceAndLink.deviceInformation("1.1.1.1");
+        assertThat(deviceInformation, is(notNullValue()));
+    }
+
+    /**
+     * Tests removeDeviceInformationMap() method.
+     */
+    @Test
+    public void testRemoveDeviceInformationMap() throws Exception {
+        topologyForDeviceAndLink.setDeviceInformationMap("1.1.1.1", new DeviceInformationImpl());
+        topologyForDeviceAndLink.deviceInformation("1.1.1.1");
+        result = topologyForDeviceAndLink.deviceInformationMap();
+        topologyForDeviceAndLink.removeDeviceInformationMap("1.1.1.1");
+        assertThat(result.size(), is(0));
+    }
+
+    /**
+     * Tests linkInformationMap() method.
+     */
+    @Test
+    public void testLinkInformationMap() throws Exception {
+        result = topologyForDeviceAndLink.linkInformationMap();
+        assertThat(result.size(), is(0));
+    }
+
+    /**
+     * Tests setLinkInformationMap() method.
+     */
+    @Test
+    public void testSetLinkInformationMap() throws Exception {
+        topologyForDeviceAndLink.setLinkInformationMap("1.1.1.1", new LinkInformationImpl());
+        result = topologyForDeviceAndLink.linkInformationMap();
+        assertThat(result.size(), is(1));
+    }
+
+    /**
+     * Tests removeLinkInformationMap() method.
+     */
+    @Test
+    public void testRemoveLinkInformationMap() throws Exception {
+        topologyForDeviceAndLink.setLinkInformationMap("1.1.1.1", new LinkInformationImpl());
+        topologyForDeviceAndLink.removeLinkInformationMap("1.1.1.1");
+        result = topologyForDeviceAndLink.linkInformationMap();
+        assertThat(result.size(), is(0));
+    }
+
+    /**
+     * Tests getOspfLinkTedHashMap() method.
+     */
+    @Test
+    public void testGetOspfLinkTedHashMap() throws Exception {
+        OspfLinkTed ospfLinkTed = topologyForDeviceAndLink.getOspfLinkTedHashMap("1.1.1.1");
+        assertThat(ospfLinkTed, is(nullValue()));
+    }
+
+    /**
+     * Tests addLocalDevice() method.
+     */
+    @Test
+    public void testAddLocalDevice() throws Exception {
+        OspfAreaImpl ospfArea = new OspfAreaImpl();
+        ospfArea.setRouterId(Ip4Address.valueOf("5.5.5.5"));
+        topologyForDeviceAndLink.addLocalDevice(createOspfLsa(), new OspfInterfaceImpl(), ospfArea);
+        topologyForDeviceAndLink.addLocalDevice(createOspfLsa1(), new OspfInterfaceImpl(), ospfArea);
+        assertThat(topologyForDeviceAndLink, is(notNullValue()));
+    }
+
+    /**
+     * Tests addLocalLink() method.
+     */
+    @Test
+    public void testAddLocalLink() throws Exception {
+        Ip4Address linkData = Ip4Address.valueOf("1.1.1.1");
+        Ip4Address linkSrc = Ip4Address.valueOf("2.2.2.2");
+        Ip4Address linkDest = Ip4Address.valueOf("3.3.3.3");
+        boolean opaqueEnabled = true;
+        boolean linkSrcIdNotRouterId = true;
+        topologyForDeviceAndLink.addLocalLink("10.10.10.10", linkData, linkSrc,
+                                              linkDest, opaqueEnabled, linkSrcIdNotRouterId);
+        assertThat(topologyForDeviceAndLink, is(notNullValue()));
+    }
+
+    /**
+     * Tests removeLinks() method.
+     */
+    @Test
+    public void testRemoveLinks() throws Exception {
+        Ip4Address linkData = Ip4Address.valueOf("1.1.1.1");
+        Ip4Address linkSrc = Ip4Address.valueOf("2.2.2.2");
+        Ip4Address linkDest = Ip4Address.valueOf("3.3.3.3");
+        boolean opaqueEnabled = true;
+        boolean linkSrcIdNotRouterId = true;
+        topologyForDeviceAndLink.addLocalLink("10.10.10.10", linkData, linkSrc,
+                                              linkDest, opaqueEnabled, linkSrcIdNotRouterId);
+        topologyForDeviceAndLink.removeLinks(Ip4Address.valueOf("10.10.10.10"));
+        assertThat(topologyForDeviceAndLink, is(notNullValue()));
+    }
+
+    /**
+     * Tests updateLinkInformation() method.
+     */
+    @Test
+    public void testUpdateLinkInformation() throws Exception {
+        OspfAreaImpl ospfArea = new OspfAreaImpl();
+        ospfArea.setRouterId(Ip4Address.valueOf("5.5.5.5"));
+        topologyForDeviceAndLink.updateLinkInformation(createOspfLsa(), ospfArea);
+        assertThat(topologyForDeviceAndLink, is(notNullValue()));
+    }
+
+    /**
+     * Tests getDeleteRouterInformation() method.
+     */
+    @Test
+    public void testGetDeleteRouterInformation() throws Exception {
+        OspfAreaImpl ospfArea = new OspfAreaImpl();
+        ospfArea.setRouterId(Ip4Address.valueOf("5.5.5.5"));
+        topologyForDeviceAndLink.updateLinkInformation(createOspfLsa(), ospfArea);
+        List list = topologyForDeviceAndLink.getDeleteRouterInformation(createOspfLsa(), ospfArea);
+        assertThat(list, is(notNullValue()));
+    }
+
+    /**
+     * Utility for test methods.
+     */
+    private OspfLsa createOspfLsa() {
+        RouterLsa routerLsa = new RouterLsa();
+        routerLsa.setLsType(1);
+        routerLsa.setAdvertisingRouter(Ip4Address.valueOf("6.6.6.6"));
+        OspfLsaLink ospfLsaLink = new OspfLsaLink();
+        ospfLsaLink.setLinkData("192.168.7.77");
+        ospfLsaLink.setLinkId("9.9.9.9");
+        ospfLsaLink.setLinkType(1);
+        OspfLsaLink ospfLsaLink0 = new OspfLsaLink();
+        ospfLsaLink0.setLinkData("7.7.7.7");
+        ospfLsaLink0.setLinkId("7.7.7.7");
+        ospfLsaLink0.setLinkType(2);
+        OspfLsaLink ospfLsaLink120 = new OspfLsaLink();
+        ospfLsaLink120.setLinkData("192.168.7.77");
+        ospfLsaLink120.setLinkId("1.1.1.1");
+        ospfLsaLink120.setLinkType(1);
+        OspfLsaLink lsaLink = new OspfLsaLink();
+        lsaLink.setLinkData("192.168.7.77");
+        lsaLink.setLinkId("14.14.14.14");
+        lsaLink.setLinkType(2);
+        routerLsa.addRouterLink(lsaLink);
+        routerLsa.addRouterLink(ospfLsaLink);
+        routerLsa.addRouterLink(ospfLsaLink0);
+        routerLsa.addRouterLink(ospfLsaLink120);
+        return routerLsa;
+    }
+
+    /**
+     * Utility for test methods.
+     */
+    private OspfLsa createOspfLsa1() throws Exception {
+        OpaqueLsa10 opaqueLsa10 = new OpaqueLsa10(new OpaqueLsaHeader());
+        opaqueLsa10.setLsType(10);
+        header = new TlvHeader();
+        header.setTlvLength(8);
+        header.setTlvType(9);
+        linkTlv = new LinkTlv(header);
+        channelBuffer = ChannelBuffers.copiedBuffer(packet);
+        linkTlv.readFrom(channelBuffer);
+        opaqueLsa10.addValue(linkTlv);
+        return opaqueLsa10;
+    }
+}