Allow Topology DB to keep track of production switches
[controller.git] / 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 Property createProperty(String propName, String propValue) {
288             return null;
289         }
290
291         @Override
292         public String getNodeDescription(Node node) {
293             return null;
294         }
295     }
296
297     /*
298      * Sets the node, edges and properties for edges here: Edge <SwitchId :
299      * NodeConnectorId> : <1:1>--><11:11>; <1:2>--><11:12>; <3:3>--><13:13>;
300      * <3:4>--><13:14>; <5:5>--><15:15>; <5:6>--><15:16>; Method used by two
301      * tests: testGetNodeEdges and testGetEdges
302      *
303      * @param topoManagerImpl
304      *
305      * @throws ConstructionException
306      */
307     public void setNodeEdges(TopologyManagerImpl topoManagerImpl, TestSwitchManager swMgr)
308             throws ConstructionException {
309         topoManagerImpl.nonClusterObjectCreate();
310
311         State state;
312         Bandwidth bw;
313         Latency l;
314
315         Set<Property> props = new HashSet<Property>();
316         state = new State(State.EDGE_UP);
317         bw = new Bandwidth(Bandwidth.BW100Gbps);
318         l = new Latency(Latency.LATENCY100ns);
319         props.add(state);
320         props.add(bw);
321         props.add(l);
322
323         for (short i = 1; i < 6; i = (short) (i + 2)) {
324             List<TopoEdgeUpdate> topoedgeupdateList = new ArrayList<TopoEdgeUpdate>();
325             NodeConnector headnc1 = NodeConnectorCreator.createOFNodeConnector(
326                     i, NodeCreator.createOFNode((long) i));
327             NodeConnector tailnc1 = NodeConnectorCreator
328                     .createOFNodeConnector((short) (i + 10),
329                             NodeCreator.createOFNode((long) (i + 10)));
330             swMgr.addNodeConnectors(tailnc1, headnc1);
331             Edge e1 = new Edge(tailnc1, headnc1);
332             TopoEdgeUpdate teu1 = new TopoEdgeUpdate(e1, props,
333                     UpdateType.ADDED);
334             topoedgeupdateList.add(teu1);
335
336             NodeConnector tailnc2 = NodeConnectorCreator.createOFNodeConnector(
337                     (short) (i + 1), headnc1.getNode());
338             NodeConnector headnc2 = NodeConnectorCreator.createOFNodeConnector(
339                     (short) (i + 11), tailnc1.getNode());
340             swMgr.addNodeConnectors(tailnc1, headnc2);
341             Edge e2 = new Edge(tailnc2, headnc2);
342             TopoEdgeUpdate teu2 = new TopoEdgeUpdate(e2, props,
343                     UpdateType.ADDED);
344             topoedgeupdateList.add(teu2);
345             topoManagerImpl.edgeUpdate(topoedgeupdateList);
346         }
347     }
348
349     @Test
350     public void testGetNodeEdges() throws ConstructionException {
351         TopologyManagerImpl topoManagerImpl = new TopologyManagerImpl();
352         TestSwitchManager swMgr = new TestSwitchManager();
353         topoManagerImpl.setSwitchManager(swMgr);
354         setNodeEdges(topoManagerImpl, swMgr);
355
356         Map<Node, Set<Edge>> nodeEdgeMap = topoManagerImpl.getNodeEdges();
357         for (Iterator<Map.Entry<Node, Set<Edge>>> i = nodeEdgeMap.entrySet()
358                 .iterator(); i.hasNext();) {
359             Map.Entry<Node, Set<Edge>> entry = i.next();
360             Node node = entry.getKey();
361             Long nodeId = ((Long) node.getID()).longValue();
362             Assert.assertTrue((node.getType().equals(NodeIDType.OPENFLOW)));
363
364             Set<Edge> edges = entry.getValue();
365             for (Edge edge : edges) {
366                 Long headNcId = ((Short) edge.getHeadNodeConnector().getID())
367                         .longValue();
368                 Long tailNcId = ((Short) edge.getTailNodeConnector().getID())
369                         .longValue();
370                 Assert.assertTrue(
371                         (headNcId.equals(nodeId) && tailNcId.equals(nodeId + 10))
372                         || (headNcId.equals(nodeId + 11) && tailNcId.equals(nodeId + 1))
373                         || (headNcId.equals(nodeId + 1) && tailNcId.equals(nodeId - 9))
374                         || (headNcId.equals(nodeId - 10) && tailNcId.equals(nodeId))
375                         );
376             }
377             i.remove();
378         }
379         Assert.assertTrue(nodeEdgeMap.isEmpty());
380     }
381
382     @Test
383     public void testGetEdges() throws ConstructionException {
384         TopologyManagerImpl topoManagerImpl = new TopologyManagerImpl();
385         TestSwitchManager swMgr = new TestSwitchManager();
386         topoManagerImpl.setSwitchManager(swMgr);
387         setNodeEdges(topoManagerImpl, swMgr);
388
389         Map<Edge, Set<Property>> edgeProperty = topoManagerImpl.getEdges();
390
391         for (Iterator<Map.Entry<Edge, Set<Property>>> i = edgeProperty
392                 .entrySet().iterator(); i.hasNext();) {
393             Map.Entry<Edge, Set<Property>> entry = i.next();
394             Edge e = entry.getKey();
395             NodeConnector headnc = e.getHeadNodeConnector();
396             NodeConnector tailnc = e.getTailNodeConnector();
397
398             Long headNodeId = (Long) headnc.getNode().getID();
399
400             Long headNcId = ((Short) headnc.getID()).longValue();
401             Long tailNcId = ((Short) tailnc.getID()).longValue();
402
403             if (headNodeId == 1 || headNodeId == 3 || headNodeId == 5) {
404                 Assert.assertTrue((headNcId.equals(headNodeId) && tailNcId
405                         .equals(headNodeId + 10))
406                         || (headNcId.equals(headNodeId + 10) && tailNcId
407                                 .equals(headNodeId))
408                                 || (headNcId.equals(headNodeId + 1) && tailNcId
409                                         .equals(headNodeId + 11))
410                                         || (headNcId.equals(headNodeId + 11) && tailNcId
411                                                 .equals(headNodeId + 1)));
412             } else if (headNodeId == 11 || headNodeId == 13 || headNodeId == 15) {
413                 Assert.assertTrue((headNcId.equals(headNodeId) && tailNcId
414                         .equals(headNodeId - 10))
415                         || (headNcId.equals(headNodeId) && tailNcId
416                                 .equals(headNodeId - 10))
417                                 || (headNcId.equals(headNodeId - 9) && tailNcId
418                                         .equals(headNodeId + 1))
419                                         || (headNcId.equals(headNodeId + 1) && tailNcId
420                                                 .equals(headNodeId - 9)));
421             }
422
423             Set<Property> prop = entry.getValue();
424             for (Property p : prop) {
425                 String pName;
426                 long pValue;
427                 if (p instanceof Bandwidth) {
428                     Bandwidth b = (Bandwidth) p;
429                     pName = Bandwidth.BandwidthPropName;
430                     pValue = b.getValue();
431                     Assert.assertTrue(pName.equals(p.getName())
432                             && pValue == Bandwidth.BW100Gbps);
433                     continue;
434                 }
435                 if (p instanceof Latency) {
436                     Latency l = (Latency) p;
437                     pName = Latency.LatencyPropName;
438                     pValue = l.getValue();
439                     Assert.assertTrue(pName.equals(p.getName())
440                             && pValue == Latency.LATENCY100ns);
441                     continue;
442                 }
443                 if (p instanceof State) {
444                     State state = (State) p;
445                     pName = State.StatePropName;
446                     pValue = state.getValue();
447                     Assert.assertTrue(pName.equals(p.getName())
448                             && pValue == State.EDGE_UP);
449                     continue;
450                 }
451             }
452             i.remove();
453         }
454         Assert.assertTrue(edgeProperty.isEmpty());
455     }
456
457     @Test
458     public void testAddDeleteUserLink() {
459         TopologyUserLinkConfig link1 = new TopologyUserLinkConfig("default1",
460                 "OF|1@OF|2", "OF|1@OF|3");
461         TopologyUserLinkConfig link2 = new TopologyUserLinkConfig("default1",
462                 "OF|10@OF|20", "OF|10@OF|30");
463         TopologyUserLinkConfig link3 = new TopologyUserLinkConfig("default2",
464                 "OF|1@OF|2", "OF|1@OF|3");
465         TopologyUserLinkConfig link4 = new TopologyUserLinkConfig("default20",
466                 "OF|10@OF|20", "OF|10@OF|30");
467
468         TopologyManagerImpl topoManagerImpl = new TopologyManagerImpl();
469         TestSwitchManager swMgr = new TestSwitchManager();
470         topoManagerImpl.setSwitchManager(swMgr);
471         topoManagerImpl.nonClusterObjectCreate();
472
473         swMgr.addNodeConnectors(link1, link2, link3, link4);
474
475         Assert.assertTrue(topoManagerImpl.addUserLink(link1).isSuccess());
476         Assert.assertTrue(topoManagerImpl.addUserLink(link2).getCode() == StatusCode.CONFLICT);
477         Assert.assertTrue(topoManagerImpl.addUserLink(link3).getCode() == StatusCode.CONFLICT);
478         Assert.assertTrue(topoManagerImpl.addUserLink(link4).isSuccess());
479
480         Assert.assertTrue(topoManagerImpl.deleteUserLink(null).getCode() == StatusCode.BADREQUEST);
481         Assert.assertTrue(topoManagerImpl.deleteUserLink(link1.getName())
482                 .isSuccess());
483         Assert.assertTrue(topoManagerImpl.deleteUserLink(link4.getName())
484                 .isSuccess());
485         Assert.assertTrue(topoManagerImpl.getUserLinks().isEmpty());
486
487         TopologyUserLinkConfig badlink1 =
488             new TopologyUserLinkConfig("bad1", "OF|1@OF|4", "OF|1@OF|5");
489         TopologyUserLinkConfig badlink2 =
490             new TopologyUserLinkConfig("bad2", "OF|10@OF|7", "OF|7@OF|13");
491         Assert.assertEquals(StatusCode.NOTFOUND,
492                             topoManagerImpl.addUserLink(badlink1).getCode());
493         Assert.assertEquals(StatusCode.NOTFOUND,
494                             topoManagerImpl.addUserLink(badlink2).getCode());
495     }
496
497     @Test
498     public void testGetUserLink() {
499         TopologyUserLinkConfig[] link = new TopologyUserLinkConfig[5];
500         TopologyUserLinkConfig[] reverseLink = new TopologyUserLinkConfig[5];
501         TopologyManagerImpl topoManagerImpl = new TopologyManagerImpl();
502         TestSwitchManager swMgr = new TestSwitchManager();
503         topoManagerImpl.setSwitchManager(swMgr);
504         topoManagerImpl.nonClusterObjectCreate();
505
506         String name = "Test";
507         String srcSwitchId = null;
508         String srcNodeConnectorIDType = null;
509         String srcPort = null;
510         String srcNodeIDType = null;
511         String dstNodeIDType = null;
512         String dstSwitchId = null;
513         String dstNodeConnectorIDType = null;
514         String dstPort = null;
515         String srcNodeConnector = null;
516         String dstNodeConnector = null;
517
518         /* Creating userlinks and checking for their validity */
519         link[0] = new TopologyUserLinkConfig(name, srcNodeConnector, dstNodeConnector);
520         Assert.assertTrue(link[0].isValid() == false);
521
522         srcNodeConnector = "OF|1@OF|1";
523         link[0] = new TopologyUserLinkConfig(name, srcNodeConnector, dstNodeConnector);
524         Assert.assertTrue(link[0].isValid() == false);
525
526         dstNodeConnector = "OF|1@OF|2";
527         link[0] = new TopologyUserLinkConfig(name, srcNodeConnector, dstNodeConnector);
528         Assert.assertTrue(link[0].isValid() == true);
529
530         Integer i;
531
532         for (i = 0; i < 5; i++) {
533             link[i] = new TopologyUserLinkConfig();
534
535             name = Integer.toString(i + 1);
536             srcSwitchId = Integer.toString(i + 1);
537             srcPort = Integer.toString(i + 1);
538             dstSwitchId = Integer.toString((i + 1) * 10);
539             dstPort = Integer.toString((i + 1) * 10);
540
541             link[i].setName(name);
542             srcNodeConnectorIDType = dstNodeConnectorIDType = "INCORRECT";
543             srcNodeConnector = srcNodeConnectorIDType+"|"+srcSwitchId+"@"+srcNodeConnectorIDType+"|"+srcPort;
544             dstNodeConnector = dstNodeConnectorIDType+"|"+dstSwitchId+"@"+dstNodeConnectorIDType+"|"+dstPort;
545
546             link[i].setSrcNodeConnector(srcNodeConnector);
547             Assert.assertTrue(link[i].isValid() == false);
548
549             srcNodeConnectorIDType = "OF";
550             srcNodeConnector = srcNodeConnectorIDType+"|"+srcSwitchId+"@"+srcNodeConnectorIDType+"|"+srcPort;
551             link[i].setSrcNodeConnector(srcNodeConnector);
552             Assert.assertTrue(link[i].isValid() == false);
553
554             dstNodeConnectorIDType = "OF";
555             dstNodeConnector = dstNodeConnectorIDType+"|"+dstSwitchId+"@"+dstNodeConnectorIDType+"|"+dstPort;
556             link[i].setDstNodeConnector(dstNodeConnector);
557             Assert.assertTrue(link[i].isValid() == true);
558
559             reverseLink[i] = new TopologyUserLinkConfig(name, dstNodeConnector, srcNodeConnector);
560
561             Assert.assertEquals(StatusCode.NOTFOUND,
562                                 topoManagerImpl.addUserLink(link[i]).getCode());
563             swMgr.addNodeConnectors(link[i]);
564             Assert.assertTrue(topoManagerImpl.addUserLink(link[i]).isSuccess());
565         }
566
567         ConcurrentMap<String, TopologyUserLinkConfig> userLinks = topoManagerImpl
568                 .getUserLinks();
569         TopologyUserLinkConfig resultLink;
570
571         for (i = 0; i < 5; i++) {
572             resultLink = userLinks.get(((Integer) (i + 1)).toString());
573
574             Assert.assertTrue(resultLink.getName().equals(
575                     reverseLink[i].getName()));
576             Assert.assertTrue(resultLink.getDstNodeConnector().equals(
577                     reverseLink[i].getSrcNodeConnector()));
578             Assert.assertTrue(resultLink.getSrcNodeConnector().equals(
579                     reverseLink[i].getDstNodeConnector()));
580         }
581     }
582
583     @Test
584     public void testHostLinkMethods() throws ConstructionException,
585     UnknownHostException {
586         TopologyManagerImpl topoManagerImpl = new TopologyManagerImpl();
587         TestSwitchManager swMgr = new TestSwitchManager();
588         topoManagerImpl.setSwitchManager(swMgr);
589         topoManagerImpl.nonClusterObjectCreate();
590         int hostCounter = 0;
591
592         State state;
593         Bandwidth bw;
594         Latency l;
595         Set<Property> props = new HashSet<Property>();
596         state = new State(State.EDGE_UP);
597         bw = new Bandwidth(Bandwidth.BW100Gbps);
598         l = new Latency(Latency.LATENCY100ns);
599         props.add(state);
600         props.add(bw);
601         props.add(l);
602
603         EthernetAddress ea;
604         InetAddress ip;
605         Host[] h = new Host[5];
606         NodeConnector[] nc = new NodeConnector[5];
607
608         /*
609          * Adding host, nodeConnector to hostsDB for the i = 0,1,2,3. No host
610          * added for i = 4
611          */
612         for (int i = 0; i < 5; i++) {
613             if (hostCounter < 4) {
614                 ea = new EthernetAddress(new byte[] { (byte) 0x0, (byte) 0x0,
615                         (byte) 0x0, (byte) 0x0, (byte) 0x0, (byte) i });
616                 String stringIP = new StringBuilder().append(i + 1).append(".")
617                         .append(i + 10).append(".").append(i + 20).append(".")
618                         .append(i + 30).toString();
619                 ip = InetAddress.getByName(stringIP);
620                 h[hostCounter] = new Host(ea, ip);
621             } else {
622                 h[hostCounter] = null;
623             }
624             hostCounter++;
625             nc[i] = NodeConnectorCreator.createOFNodeConnector((short) (i + 1),
626                     NodeCreator.createOFNode((long) (i + 1)));
627             topoManagerImpl
628             .updateHostLink(nc[i], h[i], UpdateType.ADDED, props);
629         }
630
631         for (int i = 0; i < 5; i++) {
632             Host host = topoManagerImpl.getHostAttachedToNodeConnector(nc[i]);
633             if (i == 4) {
634                 Assert.assertTrue(host == null);
635             } else {
636                 Assert.assertTrue(host.equals(h[i]));
637             }
638         }
639
640         Set<NodeConnector> ncSet = topoManagerImpl.getNodeConnectorWithHost();
641         for (int i = 0; i < 5; i++) {
642             Assert.assertTrue(ncSet.remove(nc[i]));
643         }
644         Assert.assertTrue(ncSet.isEmpty());
645     }
646
647     @Test
648     public void testGetNodesWithNodeConnectorHost()
649             throws ConstructionException, UnknownHostException {
650         TopologyManagerImpl topoManagerImpl = new TopologyManagerImpl();
651         TestSwitchManager swMgr = new TestSwitchManager();
652         topoManagerImpl.setSwitchManager(swMgr);
653         topoManagerImpl.nonClusterObjectCreate();
654         int hostCounter = 0;
655
656         State state;
657         Bandwidth bw;
658         Latency l;
659         Set<Property> props = new HashSet<Property>();
660         state = new State(State.EDGE_UP);
661         bw = new Bandwidth(Bandwidth.BW100Gbps);
662         l = new Latency(Latency.LATENCY100ns);
663         props.add(state);
664         props.add(bw);
665         props.add(l);
666
667         EthernetAddress ea;
668         InetAddress ip;
669         Host[] h = new Host[5];
670         NodeConnector[] nc = new NodeConnector[5];
671
672         /*
673          * Adding host, nodeconnector, properties of edge to hostsDB for the
674          * first three nodes only
675          */
676         for (int i = 1; i < 6; i++) {
677             if (i < 4) {
678                 ea = new EthernetAddress(new byte[] { (byte) 0x0, (byte) 0x0,
679                         (byte) 0x0, (byte) 0x0, (byte) 0x0, (byte) i });
680                 String stringIP = new StringBuilder().append(i).append(".")
681                         .append(i + 10).append(".").append(i + 20).append(".")
682                         .append(i + 30).toString();
683                 ip = InetAddress.getByName(stringIP);
684                 h[hostCounter] = new Host(ea, ip);
685             } else {
686                 h[hostCounter] = null;
687             }
688             hostCounter++;
689             nc[i - 1] = NodeConnectorCreator.createOFNodeConnector((short) i,
690                     NodeCreator.createOFNode((long) i));
691             topoManagerImpl.updateHostLink(nc[i - 1], h[i - 1],
692                     UpdateType.ADDED, props);
693         }
694
695         /* Get the nodes which have host connected to its nodeConnector */
696         Map<Node, Set<NodeConnector>> nodeNCmap = topoManagerImpl
697                 .getNodesWithNodeConnectorHost();
698         for (int i = 1; i < 6; i++) {
699             Node node = nc[i - 1].getNode();
700             Set<NodeConnector> ncSet = nodeNCmap.get(nc[i - 1].getNode());
701
702             Assert.assertTrue(ncSet == nodeNCmap.remove(node));
703         }
704
705         Assert.assertTrue(nodeNCmap.isEmpty());
706     }
707 }