opendaylight/topologymanager/implementation/src/test/java/org/opendaylight/controller/topologymanager/internal/TopologyManagerImplTest.java

   1 /*
   2  * Copyright (c) 2013 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.controller.topologymanager.internal;
  10
  11 import java.net.InetAddress;
  12 import java.net.UnknownHostException;
  13 import java.util.ArrayList;
  14 import java.util.HashMap;
  15 import java.util.HashSet;
  16 import java.util.Iterator;
  17 import java.util.List;
  18 import java.util.Map;
  19 import java.util.Set;
  20 import java.util.concurrent.ConcurrentMap;
  21
  22 import org.junit.Assert;
  23 import org.junit.Test;
  24 import org.opendaylight.controller.sal.core.Bandwidth;
  25 import org.opendaylight.controller.sal.core.ConstructionException;
  26 import org.opendaylight.controller.sal.core.Edge;
  27 import org.opendaylight.controller.sal.core.Host;
  28 import org.opendaylight.controller.sal.core.Latency;
  29 import org.opendaylight.controller.sal.core.Node;
  30 import org.opendaylight.controller.sal.core.Node.NodeIDType;
  31 import org.opendaylight.controller.sal.core.NodeConnector;
  32 import org.opendaylight.controller.sal.core.NodeConnector.NodeConnectorIDType;
  33 import org.opendaylight.controller.sal.core.Property;
  34 import org.opendaylight.controller.sal.core.State;
  35 import org.opendaylight.controller.sal.core.UpdateType;
  36 import org.opendaylight.controller.sal.packet.address.EthernetAddress;
  37 import org.opendaylight.controller.sal.topology.TopoEdgeUpdate;
  38 import org.opendaylight.controller.sal.utils.Status;
  39 import org.opendaylight.controller.sal.utils.StatusCode;
  40 import org.opendaylight.controller.sal.utils.NodeConnectorCreator;
  41 import org.opendaylight.controller.sal.utils.NodeCreator;
  42 import org.opendaylight.controller.switchmanager.ISwitchManager;
  43 import org.opendaylight.controller.switchmanager.SpanConfig;
  44 import org.opendaylight.controller.switchmanager.Subnet;
  45 import org.opendaylight.controller.switchmanager.SubnetConfig;
  46 import org.opendaylight.controller.switchmanager.Switch;
  47 import org.opendaylight.controller.switchmanager.SwitchConfig;
  48 import org.opendaylight.controller.topologymanager.TopologyUserLinkConfig;
  49
  50 public class TopologyManagerImplTest {
  51     /**
  52      * Mockup of switch manager that only maintains existence of node
  53      * connector.
  54      */
  55     private final class TestSwitchManager implements ISwitchManager {
  56         private final Set<Node>  nodeSet = new HashSet<Node>();
  57         private final Set<NodeConnector> nodeConnectorSet =
  58             new HashSet<NodeConnector>();
  59
  60         private void addNodeConnectors(NodeConnector ... connectors) {
  61             for (NodeConnector nc: connectors) {
  62                 if (nc != null) {
  63                     nodeSet.add(nc.getNode());
  64                     nodeConnectorSet.add(nc);
  65                 }
  66             }
  67         }
  68
  69         private void addNodeConnectors(TopologyUserLinkConfig ... links) {
  70             for (TopologyUserLinkConfig link: links) {
  71                 NodeConnector src =
  72                     NodeConnector.fromString(link.getSrcNodeConnector());
  73                 NodeConnector dst =
  74                     NodeConnector.fromString(link.getDstNodeConnector());
  75                 addNodeConnectors(src, dst);
  76             }
  77         }
  78
  79         @Override
  80         public Status addSubnet(SubnetConfig configObject) {
  81             return null;
  82         }
  83
  84         @Override
  85         public Status removeSubnet(SubnetConfig configObject) {
  86             return null;
  87         }
  88
  89         @Override
  90         public Status modifySubnet(SubnetConfig configObject) {
  91             return null;
  92         }
  93
  94         @Override
  95         public Status removeSubnet(String name) {
  96             return null;
  97         }
  98
  99         @Override
 100         public List<Switch> getNetworkDevices() {
 101             return null;
 102         }
 103
 104         @Override
 105         public List<SubnetConfig> getSubnetsConfigList() {
 106             return null;
 107         }
 108
 109         @Override
 110         public SubnetConfig getSubnetConfig(String subnet) {
 111             return null;
 112         }
 113
 114         @Override
 115         public Subnet getSubnetByNetworkAddress(InetAddress networkAddress) {
 116             return null;
 117         }
 118
 119         @Override
 120         public Status saveSwitchConfig() {
 121             return null;
 122         }
 123
 124         @Override
 125         public Status addSpanConfig(SpanConfig configObject) {
 126             return null;
 127         }
 128
 129         @Override
 130         public Status removeSpanConfig(SpanConfig cfgObject) {
 131             return null;
 132         }
 133
 134         @Override
 135         public List<SpanConfig> getSpanConfigList() {
 136             return null;
 137         }
 138
 139         @Override
 140         public List<NodeConnector> getSpanPorts(Node node) {
 141             return null;
 142         }
 143
 144         @Override
 145         public void updateSwitchConfig(SwitchConfig cfgObject) {
 146         }
 147
 148         @Override
 149         public Status updateNodeConfig(SwitchConfig switchConfig) {
 150             return null;
 151         }
 152
 153         @Override
 154         public Status removeNodeConfig(String nodeId) {
 155             return null;
 156         }
 157
 158         @Override
 159         public SwitchConfig getSwitchConfig(String nodeId) {
 160             return null;
 161         }
 162
 163         @Override
 164         public Status addPortsToSubnet(String name, List<String> nodeConnectors) {
 165             return null;
 166         }
 167
 168         @Override
 169         public Status removePortsFromSubnet(String name, List<String> nodeConnectors) {
 170             return null;
 171         }
 172
 173         @Override
 174         public Set<Node> getNodes() {
 175             return new HashSet<Node>(nodeSet);
 176         }
 177
 178         @Override
 179         public Map<String, Property> getNodeProps(Node node) {
 180             return new HashMap<String, Property>();
 181         }
 182
 183         @Override
 184         public Property getNodeProp(Node node, String propName) {
 185             return null;
 186         }
 187
 188         @Override
 189         public void setNodeProp(Node node, Property prop) {
 190         }
 191
 192         @Override
 193         public Status removeNodeProp(Node node, String propName) {
 194             return null;
 195         }
 196
 197         @Override
 198         public Status removeNodeAllProps(Node node) {
 199             return null;
 200         }
 201
 202         @Override
 203         public Set<NodeConnector> getUpNodeConnectors(Node node) {
 204             return getNodeConnectors(node);
 205         }
 206
 207         @Override
 208         public Set<NodeConnector> getNodeConnectors(Node node) {
 209             Set<NodeConnector> set = new HashSet<NodeConnector>();
 210             for (NodeConnector nc: nodeConnectorSet) {
 211                 if (nc.getNode().equals(node)) {
 212                     set.add(nc);
 213                 }
 214             }
 215
 216             return set;
 217         }
 218
 219         @Override
 220         public Set<NodeConnector> getPhysicalNodeConnectors(Node node) {
 221             return getNodeConnectors(node);
 222         }
 223
 224         @Override
 225         public Map<String, Property> getNodeConnectorProps(NodeConnector nodeConnector) {
 226             return new HashMap<String, Property>();
 227         }
 228
 229         @Override
 230         public Property getNodeConnectorProp(NodeConnector nodeConnector, String propName) {
 231             return null;
 232         }
 233
 234         @Override
 235         public Status addNodeConnectorProp(NodeConnector nodeConnector, Property prop) {
 236             return null;
 237         }
 238
 239         @Override
 240         public Status removeNodeConnectorProp(NodeConnector nc, String propName) {
 241             return null;
 242         }
 243
 244         @Override
 245         public Status removeNodeConnectorAllProps(NodeConnector nodeConnector) {
 246             return null;
 247         }
 248
 249         @Override
 250         public NodeConnector getNodeConnector(Node node, String nodeConnectorName) {
 251             return null;
 252         }
 253
 254         @Override
 255         public boolean isSpecial(NodeConnector p) {
 256             String type = p.getType();
 257             return (type.equals(NodeConnectorIDType.CONTROLLER)
 258                     || type.equals(NodeConnectorIDType.ALL)
 259                     || type.equals(NodeConnectorIDType.SWSTACK)
 260                     || type.equals(NodeConnectorIDType.HWPATH));
 261         }
 262
 263         @Override
 264         public Boolean isNodeConnectorEnabled(NodeConnector nodeConnector) {
 265             if (doesNodeConnectorExist(nodeConnector)) {
 266                 return Boolean.TRUE;
 267             }
 268             return Boolean.FALSE;
 269         }
 270
 271         @Override
 272         public boolean doesNodeConnectorExist(NodeConnector nc) {
 273             return (nc != null && nodeConnectorSet.contains(nc));
 274         }
 275
 276         @Override
 277         public byte[] getControllerMAC() {
 278             return new byte[6];
 279         }
 280
 281         @Override
 282         public byte[] getNodeMAC(Node node) {
 283             return new byte[6];
 284         }
 285
 286         @Override
 287         public boolean isHostRefreshEnabled() {
 288             return true;
 289         }
 290
 291         @Override
 292         public int getHostRetryCount() {
 293             return 1;
 294         }
 295
 296         @Override
 297         public Property createProperty(String propName, String propValue) {
 298             return null;
 299         }
 300
 301         @Override
 302         public String getNodeDescription(Node node) {
 303             return null;
 304         }
 305     }
 306
 307     /*
 308      * Sets the node, edges and properties for edges here: Edge <SwitchId :
 309      * NodeConnectorId> : <1:1>--><11:11>; <1:2>--><11:12>; <3:3>--><13:13>;
 310      * <3:4>--><13:14>; <5:5>--><15:15>; <5:6>--><15:16>; Method used by two
 311      * tests: testGetNodeEdges and testGetEdges
 312      *
 313      * @param topoManagerImpl
 314      *
 315      * @throws ConstructionException
 316      */
 317     public void setNodeEdges(TopologyManagerImpl topoManagerImpl, TestSwitchManager swMgr)
 318             throws ConstructionException {
 319         topoManagerImpl.nonClusterObjectCreate();
 320
 321         State state;
 322         Bandwidth bw;
 323         Latency l;
 324
 325         Set<Property> props = new HashSet<Property>();
 326         state = new State(State.EDGE_UP);
 327         bw = new Bandwidth(Bandwidth.BW100Gbps);
 328         l = new Latency(Latency.LATENCY100ns);
 329         props.add(state);
 330         props.add(bw);
 331         props.add(l);
 332
 333         for (short i = 1; i < 6; i = (short) (i + 2)) {
 334             List<TopoEdgeUpdate> topoedgeupdateList = new ArrayList<TopoEdgeUpdate>();
 335             NodeConnector headnc1 = NodeConnectorCreator.createOFNodeConnector(
 336                     i, NodeCreator.createOFNode((long) i));
 337             NodeConnector tailnc1 = NodeConnectorCreator
 338                     .createOFNodeConnector((short) (i + 10),
 339                             NodeCreator.createOFNode((long) (i + 10)));
 340             swMgr.addNodeConnectors(tailnc1, headnc1);
 341             Edge e1 = new Edge(tailnc1, headnc1);
 342             TopoEdgeUpdate teu1 = new TopoEdgeUpdate(e1, props,
 343                     UpdateType.ADDED);
 344             topoedgeupdateList.add(teu1);
 345
 346             NodeConnector tailnc2 = NodeConnectorCreator.createOFNodeConnector(
 347                     (short) (i + 1), headnc1.getNode());
 348             NodeConnector headnc2 = NodeConnectorCreator.createOFNodeConnector(
 349                     (short) (i + 11), tailnc1.getNode());
 350             swMgr.addNodeConnectors(tailnc1, headnc2);
 351             Edge e2 = new Edge(tailnc2, headnc2);
 352             TopoEdgeUpdate teu2 = new TopoEdgeUpdate(e2, props,
 353                     UpdateType.ADDED);
 354             topoedgeupdateList.add(teu2);
 355             topoManagerImpl.edgeUpdate(topoedgeupdateList);
 356         }
 357     }
 358
 359     @Test
 360     public void testGetNodeEdges() throws ConstructionException {
 361         TopologyManagerImpl topoManagerImpl = new TopologyManagerImpl();
 362         TestSwitchManager swMgr = new TestSwitchManager();
 363         topoManagerImpl.setSwitchManager(swMgr);
 364         setNodeEdges(topoManagerImpl, swMgr);
 365
 366         Map<Node, Set<Edge>> nodeEdgeMap = topoManagerImpl.getNodeEdges();
 367         for (Iterator<Map.Entry<Node, Set<Edge>>> i = nodeEdgeMap.entrySet()
 368                 .iterator(); i.hasNext();) {
 369             Map.Entry<Node, Set<Edge>> entry = i.next();
 370             Node node = entry.getKey();
 371             Long nodeId = ((Long) node.getID()).longValue();
 372             Assert.assertTrue((node.getType().equals(NodeIDType.OPENFLOW)));
 373
 374             Set<Edge> edges = entry.getValue();
 375             for (Edge edge : edges) {
 376                 Long headNcId = ((Short) edge.getHeadNodeConnector().getID())
 377                         .longValue();
 378                 Long tailNcId = ((Short) edge.getTailNodeConnector().getID())
 379                         .longValue();
 380                 Assert.assertTrue(
 381                         (headNcId.equals(nodeId) && tailNcId.equals(nodeId + 10))
 382                         || (headNcId.equals(nodeId + 11) && tailNcId.equals(nodeId + 1))
 383                         || (headNcId.equals(nodeId + 1) && tailNcId.equals(nodeId - 9))
 384                         || (headNcId.equals(nodeId - 10) && tailNcId.equals(nodeId))
 385                         );
 386             }
 387             i.remove();
 388         }
 389         Assert.assertTrue(nodeEdgeMap.isEmpty());
 390     }
 391
 392     @Test
 393     public void testGetEdges() throws ConstructionException {
 394         TopologyManagerImpl topoManagerImpl = new TopologyManagerImpl();
 395         TestSwitchManager swMgr = new TestSwitchManager();
 396         topoManagerImpl.setSwitchManager(swMgr);
 397         setNodeEdges(topoManagerImpl, swMgr);
 398
 399         Map<Edge, Set<Property>> edgeProperty = topoManagerImpl.getEdges();
 400
 401         for (Iterator<Map.Entry<Edge, Set<Property>>> i = edgeProperty
 402                 .entrySet().iterator(); i.hasNext();) {
 403             Map.Entry<Edge, Set<Property>> entry = i.next();
 404             Edge e = entry.getKey();
 405             NodeConnector headnc = e.getHeadNodeConnector();
 406             NodeConnector tailnc = e.getTailNodeConnector();
 407
 408             Long headNodeId = (Long) headnc.getNode().getID();
 409
 410             Long headNcId = ((Short) headnc.getID()).longValue();
 411             Long tailNcId = ((Short) tailnc.getID()).longValue();
 412
 413             if (headNodeId == 1 || headNodeId == 3 || headNodeId == 5) {
 414                 Assert.assertTrue((headNcId.equals(headNodeId) && tailNcId
 415                         .equals(headNodeId + 10))
 416                         || (headNcId.equals(headNodeId + 10) && tailNcId
 417                                 .equals(headNodeId))
 418                                 || (headNcId.equals(headNodeId + 1) && tailNcId
 419                                         .equals(headNodeId + 11))
 420                                         || (headNcId.equals(headNodeId + 11) && tailNcId
 421                                                 .equals(headNodeId + 1)));
 422             } else if (headNodeId == 11 || headNodeId == 13 || headNodeId == 15) {
 423                 Assert.assertTrue((headNcId.equals(headNodeId) && tailNcId
 424                         .equals(headNodeId - 10))
 425                         || (headNcId.equals(headNodeId) && tailNcId
 426                                 .equals(headNodeId - 10))
 427                                 || (headNcId.equals(headNodeId - 9) && tailNcId
 428                                         .equals(headNodeId + 1))
 429                                         || (headNcId.equals(headNodeId + 1) && tailNcId
 430                                                 .equals(headNodeId - 9)));
 431             }
 432
 433             Set<Property> prop = entry.getValue();
 434             for (Property p : prop) {
 435                 String pName;
 436                 long pValue;
 437                 if (p instanceof Bandwidth) {
 438                     Bandwidth b = (Bandwidth) p;
 439                     pName = Bandwidth.BandwidthPropName;
 440                     pValue = b.getValue();
 441                     Assert.assertTrue(pName.equals(p.getName())
 442                             && pValue == Bandwidth.BW100Gbps);
 443                     continue;
 444                 }
 445                 if (p instanceof Latency) {
 446                     Latency l = (Latency) p;
 447                     pName = Latency.LatencyPropName;
 448                     pValue = l.getValue();
 449                     Assert.assertTrue(pName.equals(p.getName())
 450                             && pValue == Latency.LATENCY100ns);
 451                     continue;
 452                 }
 453                 if (p instanceof State) {
 454                     State state = (State) p;
 455                     pName = State.StatePropName;
 456                     pValue = state.getValue();
 457                     Assert.assertTrue(pName.equals(p.getName())
 458                             && pValue == State.EDGE_UP);
 459                     continue;
 460                 }
 461             }
 462             i.remove();
 463         }
 464         Assert.assertTrue(edgeProperty.isEmpty());
 465     }
 466
 467     @Test
 468     public void testAddDeleteUserLink() {
 469         TopologyUserLinkConfig link1 = new TopologyUserLinkConfig("default1",
 470                 "OF|1@OF|2", "OF|1@OF|2");
 471         TopologyUserLinkConfig link2 = new TopologyUserLinkConfig("default1",
 472                 "OF|10@OF|20", "OF|10@OF|20");
 473         TopologyUserLinkConfig link3 = new TopologyUserLinkConfig("default2",
 474                 "OF|1@OF|2", "OF|1@OF|2");
 475         TopologyUserLinkConfig link4 = new TopologyUserLinkConfig("default20",
 476                 "OF|10@OF|20", "OF|10@OF|20");
 477
 478         TopologyManagerImpl topoManagerImpl = new TopologyManagerImpl();
 479         TestSwitchManager swMgr = new TestSwitchManager();
 480         topoManagerImpl.setSwitchManager(swMgr);
 481         topoManagerImpl.nonClusterObjectCreate();
 482
 483         swMgr.addNodeConnectors(link1, link2, link3, link4);
 484
 485         Assert.assertTrue(topoManagerImpl.addUserLink(link1).isSuccess());
 486         Assert.assertTrue(topoManagerImpl.addUserLink(link2).getCode() == StatusCode.CONFLICT);
 487         Assert.assertTrue(topoManagerImpl.addUserLink(link3).getCode() == StatusCode.CONFLICT);
 488         Assert.assertTrue(topoManagerImpl.addUserLink(link4).isSuccess());
 489
 490         Assert.assertTrue(topoManagerImpl.deleteUserLink(null).getCode() == StatusCode.BADREQUEST);
 491         Assert.assertTrue(topoManagerImpl.deleteUserLink(link1.getName())
 492                 .isSuccess());
 493         Assert.assertTrue(topoManagerImpl.deleteUserLink(link4.getName())
 494                 .isSuccess());
 495         Assert.assertTrue(topoManagerImpl.getUserLinks().isEmpty());
 496
 497         TopologyUserLinkConfig badlink1 =
 498             new TopologyUserLinkConfig("bad1", "OF|1@OF|2", "OF|1@OF|3");
 499         TopologyUserLinkConfig badlink2 =
 500             new TopologyUserLinkConfig("bad2", "OF|10@OF|21", "OF|10@OF|20");
 501         Assert.assertEquals(StatusCode.NOTFOUND,
 502                             topoManagerImpl.addUserLink(badlink1).getCode());
 503         Assert.assertEquals(StatusCode.NOTFOUND,
 504                             topoManagerImpl.addUserLink(badlink2).getCode());
 505     }
 506
 507     @Test
 508     public void testGetUserLink() {
 509         TopologyUserLinkConfig[] link = new TopologyUserLinkConfig[5];
 510         TopologyUserLinkConfig[] reverseLink = new TopologyUserLinkConfig[5];
 511         TopologyManagerImpl topoManagerImpl = new TopologyManagerImpl();
 512         TestSwitchManager swMgr = new TestSwitchManager();
 513         topoManagerImpl.setSwitchManager(swMgr);
 514         topoManagerImpl.nonClusterObjectCreate();
 515
 516         String name = "Test";
 517         String srcSwitchId = null;
 518         String srcNodeConnectorIDType = null;
 519         String srcPort = null;
 520         String srcNodeIDType = null;
 521         String dstNodeIDType = null;
 522         String dstSwitchId = null;
 523         String dstNodeConnectorIDType = null;
 524         String dstPort = null;
 525         String srcNodeConnector = null;
 526         String dstNodeConnector = null;
 527
 528         /* Creating userlinks and checking for their validity */
 529         link[0] = new TopologyUserLinkConfig(name, srcNodeConnector, dstNodeConnector);
 530         Assert.assertTrue(link[0].isValid() == false);
 531
 532         srcNodeConnector = "OF|1@OF|1";
 533         link[0] = new TopologyUserLinkConfig(name, srcNodeConnector, dstNodeConnector);
 534         Assert.assertTrue(link[0].isValid() == false);
 535
 536         dstNodeConnector = "OF|1@OF|2";
 537         link[0] = new TopologyUserLinkConfig(name, srcNodeConnector, dstNodeConnector);
 538         Assert.assertTrue(link[0].isValid() == true);
 539
 540         Integer i;
 541
 542         for (i = 0; i < 5; i++) {
 543             link[i] = new TopologyUserLinkConfig();
 544
 545             name = Integer.toString(i + 1);
 546             srcSwitchId = Integer.toString(i + 1);
 547             srcPort = Integer.toString(i + 1);
 548             dstSwitchId = Integer.toString((i + 1) * 10);
 549             dstPort = Integer.toString((i + 1) * 10);
 550
 551             link[i].setName(name);
 552             srcNodeConnectorIDType = dstNodeConnectorIDType = "INCORRECT";
 553             srcNodeConnector = srcNodeConnectorIDType+"|"+srcSwitchId+"@"+srcNodeConnectorIDType+"|"+srcPort;
 554             dstNodeConnector = dstNodeConnectorIDType+"|"+dstSwitchId+"@"+dstNodeConnectorIDType+"|"+dstPort;
 555
 556             link[i].setSrcNodeConnector(srcNodeConnector);
 557             Assert.assertTrue(link[i].isValid() == false);
 558
 559             srcNodeConnectorIDType = "OF";
 560             srcNodeConnector = srcNodeConnectorIDType+"|"+srcSwitchId+"@"+srcNodeConnectorIDType+"|"+srcPort;
 561             link[i].setSrcNodeConnector(srcNodeConnector);
 562             Assert.assertTrue(link[i].isValid() == false);
 563
 564             dstNodeConnectorIDType = "OF";
 565             dstNodeConnector = dstNodeConnectorIDType+"|"+dstSwitchId+"@"+dstNodeConnectorIDType+"|"+dstPort;
 566             link[i].setDstNodeConnector(dstNodeConnector);
 567             Assert.assertTrue(link[i].isValid() == true);
 568
 569             reverseLink[i] = new TopologyUserLinkConfig(name, dstNodeConnector, srcNodeConnector);
 570
 571             Assert.assertEquals(StatusCode.NOTFOUND,
 572                                 topoManagerImpl.addUserLink(link[i]).getCode());
 573             swMgr.addNodeConnectors(link[i]);
 574             Assert.assertTrue(topoManagerImpl.addUserLink(link[i]).isSuccess());
 575         }
 576
 577         ConcurrentMap<String, TopologyUserLinkConfig> userLinks = topoManagerImpl
 578                 .getUserLinks();
 579         TopologyUserLinkConfig resultLink;
 580
 581         for (i = 0; i < 5; i++) {
 582             resultLink = userLinks.get(((Integer) (i + 1)).toString());
 583
 584             Assert.assertTrue(resultLink.getName().equals(
 585                     reverseLink[i].getName()));
 586             Assert.assertTrue(resultLink.getDstNodeConnector().equals(
 587                     reverseLink[i].getSrcNodeConnector()));
 588             Assert.assertTrue(resultLink.getSrcNodeConnector().equals(
 589                     reverseLink[i].getDstNodeConnector()));
 590         }
 591     }
 592
 593     @Test
 594     public void testHostLinkMethods() throws ConstructionException,
 595     UnknownHostException {
 596         TopologyManagerImpl topoManagerImpl = new TopologyManagerImpl();
 597         TestSwitchManager swMgr = new TestSwitchManager();
 598         topoManagerImpl.setSwitchManager(swMgr);
 599         topoManagerImpl.nonClusterObjectCreate();
 600         int hostCounter = 0;
 601
 602         State state;
 603         Bandwidth bw;
 604         Latency l;
 605         Set<Property> props = new HashSet<Property>();
 606         state = new State(State.EDGE_UP);
 607         bw = new Bandwidth(Bandwidth.BW100Gbps);
 608         l = new Latency(Latency.LATENCY100ns);
 609         props.add(state);
 610         props.add(bw);
 611         props.add(l);
 612
 613         EthernetAddress ea;
 614         InetAddress ip;
 615         Host[] h = new Host[5];
 616         NodeConnector[] nc = new NodeConnector[5];
 617
 618         /*
 619          * Adding host, nodeConnector to hostsDB for the i = 0,1,2,3. No host
 620          * added for i = 4
 621          */
 622         for (int i = 0; i < 5; i++) {
 623             if (hostCounter < 4) {
 624                 ea = new EthernetAddress(new byte[] { (byte) 0x0, (byte) 0x0,
 625                         (byte) 0x0, (byte) 0x0, (byte) 0x0, (byte) i });
 626                 String stringIP = new StringBuilder().append(i + 1).append(".")
 627                         .append(i + 10).append(".").append(i + 20).append(".")
 628                         .append(i + 30).toString();
 629                 ip = InetAddress.getByName(stringIP);
 630                 h[hostCounter] = new Host(ea, ip);
 631             } else {
 632                 h[hostCounter] = null;
 633             }
 634             hostCounter++;
 635             nc[i] = NodeConnectorCreator.createOFNodeConnector((short) (i + 1),
 636                     NodeCreator.createOFNode((long) (i + 1)));
 637             topoManagerImpl
 638             .updateHostLink(nc[i], h[i], UpdateType.ADDED, props);
 639         }
 640
 641         for (int i = 0; i < 5; i++) {
 642             Host host = topoManagerImpl.getHostAttachedToNodeConnector(nc[i]);
 643             if (i == 4)
 644                 Assert.assertTrue(host == null);
 645             else
 646                 Assert.assertTrue(host.equals(h[i]));
 647         }
 648
 649         Set<NodeConnector> ncSet = topoManagerImpl.getNodeConnectorWithHost();
 650         for (int i = 0; i < 5; i++) {
 651             Assert.assertTrue(ncSet.remove(nc[i]));
 652         }
 653         Assert.assertTrue(ncSet.isEmpty());
 654     }
 655
 656     @Test
 657     public void testGetNodesWithNodeConnectorHost()
 658             throws ConstructionException, UnknownHostException {
 659         TopologyManagerImpl topoManagerImpl = new TopologyManagerImpl();
 660         TestSwitchManager swMgr = new TestSwitchManager();
 661         topoManagerImpl.setSwitchManager(swMgr);
 662         topoManagerImpl.nonClusterObjectCreate();
 663         int hostCounter = 0;
 664
 665         State state;
 666         Bandwidth bw;
 667         Latency l;
 668         Set<Property> props = new HashSet<Property>();
 669         state = new State(State.EDGE_UP);
 670         bw = new Bandwidth(Bandwidth.BW100Gbps);
 671         l = new Latency(Latency.LATENCY100ns);
 672         props.add(state);
 673         props.add(bw);
 674         props.add(l);
 675
 676         EthernetAddress ea;
 677         InetAddress ip;
 678         Host[] h = new Host[5];
 679         NodeConnector[] nc = new NodeConnector[5];
 680
 681         /*
 682          * Adding host, nodeconnector, properties of edge to hostsDB for the
 683          * first three nodes only
 684          */
 685         for (int i = 1; i < 6; i++) {
 686             if (i < 4) {
 687                 ea = new EthernetAddress(new byte[] { (byte) 0x0, (byte) 0x0,
 688                         (byte) 0x0, (byte) 0x0, (byte) 0x0, (byte) i });
 689                 String stringIP = new StringBuilder().append(i).append(".")
 690                         .append(i + 10).append(".").append(i + 20).append(".")
 691                         .append(i + 30).toString();
 692                 ip = InetAddress.getByName(stringIP);
 693                 h[hostCounter] = new Host(ea, ip);
 694             } else {
 695                 h[hostCounter] = null;
 696             }
 697             hostCounter++;
 698             nc[i - 1] = NodeConnectorCreator.createOFNodeConnector((short) i,
 699                     NodeCreator.createOFNode((long) i));
 700             topoManagerImpl.updateHostLink(nc[i - 1], h[i - 1],
 701                     UpdateType.ADDED, props);
 702         }
 703
 704         /* Get the nodes which have host connected to its nodeConnector */
 705         Map<Node, Set<NodeConnector>> nodeNCmap = topoManagerImpl
 706                 .getNodesWithNodeConnectorHost();
 707         for (int i = 1; i < 6; i++) {
 708             Node node = nc[i - 1].getNode();
 709             Set<NodeConnector> ncSet = nodeNCmap.get(nc[i - 1].getNode());
 710
 711             Assert.assertTrue(ncSet == nodeNCmap.remove(node));
 712         }
 713
 714         Assert.assertTrue(nodeNCmap.isEmpty());
 715     }
 716 }