Toshio Koide | ebdbb62 | 2014-02-12 20:28:38 -0800 | [diff] [blame] | 1 | package net.onrc.onos.intent; |
| 2 | |
Toshio Koide | 0c9106d | 2014-02-19 15:26:38 -0800 | [diff] [blame] | 3 | import net.onrc.onos.ofcontroller.networkgraph.NetworkGraph; |
Pavlin Radoslavov | 6397a7f | 2014-02-18 14:56:52 -0800 | [diff] [blame] | 4 | import net.onrc.onos.ofcontroller.networkgraph.NetworkGraphImpl; |
Toshio Koide | ebdbb62 | 2014-02-12 20:28:38 -0800 | [diff] [blame] | 5 | import net.onrc.onos.ofcontroller.networkgraph.Link; |
| 6 | import net.onrc.onos.ofcontroller.networkgraph.LinkImpl; |
Toshio Koide | 0c9106d | 2014-02-19 15:26:38 -0800 | [diff] [blame] | 7 | import net.onrc.onos.ofcontroller.networkgraph.Port; |
Toshio Koide | ebdbb62 | 2014-02-12 20:28:38 -0800 | [diff] [blame] | 8 | import net.onrc.onos.ofcontroller.networkgraph.Switch; |
| 9 | import net.onrc.onos.ofcontroller.networkgraph.SwitchImpl; |
| 10 | |
Toshio Koide | 066506e | 2014-02-20 19:52:09 -0800 | [diff] [blame] | 11 | /** |
Toshio Koide | fe1d5d9 | 2014-02-26 20:09:48 -0800 | [diff] [blame] | 12 | * A mock class of NetworkGraph. |
| 13 | * This class should be used only by test codes. |
| 14 | * |
Toshio Koide | 066506e | 2014-02-20 19:52:09 -0800 | [diff] [blame] | 15 | * @author Toshio Koide (t-koide@onlab.us) |
| 16 | */ |
Pavlin Radoslavov | 6397a7f | 2014-02-18 14:56:52 -0800 | [diff] [blame] | 17 | public class MockNetworkGraph extends NetworkGraphImpl { |
Toshio Koide | fe1d5d9 | 2014-02-26 20:09:48 -0800 | [diff] [blame] | 18 | public static Long LOCAL_PORT = 0xFFFEL; |
| 19 | public SwitchImpl sw1, sw2, sw3, sw4; |
| 20 | |
Toshio Koide | 0c9106d | 2014-02-19 15:26:38 -0800 | [diff] [blame] | 21 | class DetachableLinkImpl extends LinkImpl { |
| 22 | public DetachableLinkImpl(NetworkGraph graph, Port srcPort, Port dstPort) { |
| 23 | super(graph, srcPort, dstPort); |
| 24 | } |
| 25 | |
| 26 | public void detachFromGraph() { |
| 27 | unsetFromPorts(); |
Toshio Koide | 066506e | 2014-02-20 19:52:09 -0800 | [diff] [blame] | 28 | } |
Toshio Koide | 0c9106d | 2014-02-19 15:26:38 -0800 | [diff] [blame] | 29 | } |
Toshio Koide | ebdbb62 | 2014-02-12 20:28:38 -0800 | [diff] [blame] | 30 | public Switch addSwitch(Long switchId) { |
| 31 | SwitchImpl sw = new SwitchImpl(this, switchId); |
Pavlin Radoslavov | 6d224ee | 2014-02-18 16:43:15 -0800 | [diff] [blame] | 32 | this.putSwitch(sw); |
Toshio Koide | ebdbb62 | 2014-02-12 20:28:38 -0800 | [diff] [blame] | 33 | return sw; |
Toshio Koide | ebdbb62 | 2014-02-12 20:28:38 -0800 | [diff] [blame] | 34 | } |
| 35 | |
| 36 | public Link addLink(Long srcDpid, Long srcPortNo, Long dstDpid, Long dstPortNo) { |
Toshio Koide | 0c9106d | 2014-02-19 15:26:38 -0800 | [diff] [blame] | 37 | return new DetachableLinkImpl( |
Toshio Koide | ebdbb62 | 2014-02-12 20:28:38 -0800 | [diff] [blame] | 38 | this, |
| 39 | getSwitch(srcDpid).getPort(srcPortNo), |
| 40 | getSwitch(dstDpid).getPort(dstPortNo)); |
| 41 | } |
| 42 | |
| 43 | public Link[] addBidirectionalLinks(Long srcDpid, Long srcPortNo, Long dstDpid, Long dstPortNo) { |
| 44 | Link[] links = new Link[2]; |
| 45 | links[0] = addLink(srcDpid, srcPortNo, dstDpid, dstPortNo); |
| 46 | links[1] = addLink(dstDpid, dstPortNo, srcDpid, srcPortNo); |
| 47 | |
| 48 | return links; |
| 49 | } |
Toshio Koide | 066506e | 2014-02-20 19:52:09 -0800 | [diff] [blame] | 50 | |
Toshio Koide | fe1d5d9 | 2014-02-26 20:09:48 -0800 | [diff] [blame] | 51 | /** |
| 52 | * create sample topology of 4 switches and 5 bidirectional links. |
| 53 | * <pre> |
| 54 | * [1] --- [2] |
| 55 | * | / | |
| 56 | * | / | |
| 57 | * [4] --- [3] |
| 58 | * </pre> |
| 59 | */ |
| 60 | public void createSampleTopology1() { |
| 61 | sw1 = (SwitchImpl) addSwitch(1L); |
| 62 | sw1.addPort(LOCAL_PORT); |
| 63 | sw2 = (SwitchImpl) addSwitch(2L); |
| 64 | sw2.addPort(LOCAL_PORT); |
| 65 | sw3 = (SwitchImpl) addSwitch(3L); |
| 66 | sw3.addPort(LOCAL_PORT); |
| 67 | sw4 = (SwitchImpl) addSwitch(4L); |
| 68 | sw4.addPort(LOCAL_PORT); |
Toshio Koide | ebdbb62 | 2014-02-12 20:28:38 -0800 | [diff] [blame] | 69 | |
Toshio Koide | fe1d5d9 | 2014-02-26 20:09:48 -0800 | [diff] [blame] | 70 | sw1.addPort(12L); // sw1 -> sw2 |
| 71 | sw1.addPort(14L); // sw1 -> sw4 |
| 72 | sw2.addPort(21L); // sw2 -> sw1 |
| 73 | sw2.addPort(23L); // sw2 -> sw3 |
| 74 | sw2.addPort(24L); // sw2 -> sw4 |
| 75 | sw3.addPort(32L); // sw3 -> sw2 |
| 76 | sw3.addPort(34L); // sw3 -> sw4 |
| 77 | sw4.addPort(41L); // sw4 -> sw1 |
| 78 | sw4.addPort(42L); // sw4 -> sw2 |
| 79 | sw4.addPort(43L); // sw4 -> sw3 |
Toshio Koide | ebdbb62 | 2014-02-12 20:28:38 -0800 | [diff] [blame] | 80 | |
Toshio Koide | fe1d5d9 | 2014-02-26 20:09:48 -0800 | [diff] [blame] | 81 | addBidirectionalLinks(1L, 12L, 2L, 21L); |
| 82 | addBidirectionalLinks(2L, 23L, 3L, 32L); |
| 83 | addBidirectionalLinks(3L, 34L, 4L, 43L); |
| 84 | addBidirectionalLinks(4L, 41L, 1L, 14L); |
| 85 | addBidirectionalLinks(2L, 24L, 4L, 42L); |
Toshio Koide | ebdbb62 | 2014-02-12 20:28:38 -0800 | [diff] [blame] | 86 | |
| 87 | // set capacity of all links to 1000Mbps |
| 88 | for (Link link: getLinks()) { |
| 89 | ((LinkImpl)link).setCapacity(1000.0); |
| 90 | } |
| 91 | } |
Toshio Koide | 0c9106d | 2014-02-19 15:26:38 -0800 | [diff] [blame] | 92 | |
| 93 | public void removeLink(Long srcDpid, Long srcPortNo, Long dstDpid, Long dstPortNo) { |
| 94 | DetachableLinkImpl link = (DetachableLinkImpl)getSwitch(srcDpid).getPort(srcPortNo).getOutgoingLink(); |
Pavlin Radoslavov | 7c8f69a | 2014-02-19 19:01:45 -0800 | [diff] [blame] | 95 | if (link.getDstSwitch().getDpid().equals(dstDpid) && link.getDstPort().getNumber().equals(dstPortNo)) { |
Toshio Koide | 0c9106d | 2014-02-19 15:26:38 -0800 | [diff] [blame] | 96 | link.detachFromGraph(); |
| 97 | } |
| 98 | } |
Toshio Koide | ebdbb62 | 2014-02-12 20:28:38 -0800 | [diff] [blame] | 99 | } |