1 | """Module to test the main napp file.""" |
||
2 | # pylint: disable=import-error,no-name-in-module,wrong-import-order |
||
3 | # pylint: disable=import-outside-toplevel,attribute-defined-outside-init |
||
4 | 1 | import asyncio |
|
5 | 1 | import pytest |
|
6 | 1 | import time |
|
7 | 1 | from datetime import timedelta |
|
8 | 1 | from unittest.mock import MagicMock, create_autospec, patch, call, Mock |
|
9 | 1 | import tenacity |
|
10 | 1 | from kytos.core.common import EntityStatus |
|
11 | 1 | from kytos.core.helpers import now |
|
12 | 1 | from kytos.core.events import KytosEvent |
|
13 | 1 | from kytos.core.exceptions import (KytosSetTagRangeError, |
|
14 | KytosTagtypeNotSupported) |
||
15 | 1 | from kytos.core.interface import Interface |
|
16 | 1 | from kytos.core.link import Link |
|
17 | 1 | from kytos.core.switch import Switch |
|
18 | 1 | from kytos.lib.helpers import (get_interface_mock, get_link_mock, |
|
19 | get_controller_mock, get_switch_mock, |
||
20 | get_test_client) |
||
21 | 1 | from napps.kytos.topology.exceptions import RestoreError |
|
22 | |||
23 | |||
24 | 1 | @pytest.mark.parametrize("liveness_status, status", |
|
25 | [("up", EntityStatus.UP), |
||
26 | ("down", EntityStatus.DOWN)]) |
||
27 | 1 | def test_handle_link_liveness_status(liveness_status, status) -> None: |
|
28 | """Test handle link liveness.""" |
||
29 | 1 | from napps.kytos.topology.main import Main |
|
30 | 1 | napp = Main(get_controller_mock()) |
|
31 | 1 | napp.notify_topology_update = MagicMock() |
|
32 | 1 | napp.notify_link_status_change = MagicMock() |
|
33 | |||
34 | 1 | link = MagicMock(id="some_id", status=status) |
|
35 | 1 | napp.handle_link_liveness_status(link, liveness_status) |
|
36 | |||
37 | 1 | link.extend_metadata.assert_called_with({"liveness_status": |
|
38 | liveness_status}) |
||
39 | 1 | assert napp.notify_topology_update.call_count == 1 |
|
40 | 1 | assert napp.notify_link_status_change.call_count == 1 |
|
41 | 1 | reason = f"liveness_{liveness_status}" |
|
42 | 1 | napp.notify_link_status_change.assert_called_with(link, reason=reason) |
|
43 | |||
44 | |||
45 | # pylint: disable=too-many-public-methods |
||
46 | 1 | class TestMain: |
|
47 | """Test the Main class.""" |
||
48 | |||
49 | # pylint: disable=too-many-public-methods, protected-access,C0302 |
||
50 | |||
51 | 1 | def setup_method(self): |
|
52 | """Execute steps before each tests.""" |
||
53 | 1 | patch('kytos.core.helpers.run_on_thread', lambda x: x).start() |
|
54 | # pylint: disable=import-outside-toplevel |
||
55 | 1 | from napps.kytos.topology.main import Main |
|
56 | 1 | Main.get_topo_controller = MagicMock() |
|
57 | 1 | controller = get_controller_mock() |
|
58 | 1 | self.napp = Main(controller) |
|
59 | 1 | self.api_client = get_test_client(controller, self.napp) |
|
60 | 1 | self.base_endpoint = 'kytos/topology/v3' |
|
61 | |||
62 | 1 | def test_get_event_listeners(self): |
|
63 | """Verify all event listeners registered.""" |
||
64 | 1 | expected_events = [ |
|
65 | 'kytos/core.shutdown', |
||
66 | 'kytos/core.shutdown.kytos/topology', |
||
67 | '.*.topo_controller.upsert_switch', |
||
68 | '.*.of_lldp.network_status.updated', |
||
69 | '.*.interface.is.nni', |
||
70 | '.*.connection.lost', |
||
71 | '.*.switch.interfaces.created', |
||
72 | '.*.topology.switch.interface.created', |
||
73 | '.*.switch.interface.deleted', |
||
74 | '.*.switch.interface.link_down', |
||
75 | '.*.switch.interface.link_up', |
||
76 | '.*.switch.(new|reconnected)', |
||
77 | 'kytos/.*.liveness.(up|down|disabled)', |
||
78 | '.*.switch.port.created', |
||
79 | 'kytos/topology.notify_link_up_if_status', |
||
80 | 'topology.interruption.(start|end)', |
||
81 | 'kytos/core.interface_tags', |
||
82 | ] |
||
83 | 1 | actual_events = self.napp.listeners() |
|
84 | 1 | assert sorted(expected_events) == sorted(actual_events) |
|
85 | |||
86 | 1 | async def test_get_topology(self): |
|
87 | """Test get_topology.""" |
||
88 | 1 | dpid_a = "00:00:00:00:00:00:00:01" |
|
89 | 1 | dpid_b = "00:00:00:00:00:00:00:02" |
|
90 | 1 | expected = { |
|
91 | "topology": { |
||
92 | "switches": { |
||
93 | "00:00:00:00:00:00:00:01": { |
||
94 | "metadata": { |
||
95 | "lat": "0.0", |
||
96 | "lng": "-30.0" |
||
97 | } |
||
98 | }, |
||
99 | "00:00:00:00:00:00:00:02": { |
||
100 | "metadata": { |
||
101 | "lat": "0.0", |
||
102 | "lng": "-30.0" |
||
103 | } |
||
104 | } |
||
105 | }, |
||
106 | "links": { |
||
107 | "cf0f4071be4": { |
||
108 | "id": "cf0f4071be4" |
||
109 | } |
||
110 | } |
||
111 | } |
||
112 | } |
||
113 | |||
114 | 1 | mock_switch_a = get_switch_mock(dpid_a, 0x04) |
|
115 | 1 | mock_switch_b = get_switch_mock(dpid_b, 0x04) |
|
116 | 1 | mock_interface_a = get_interface_mock('s1-eth1', 1, mock_switch_a) |
|
117 | 1 | mock_interface_b = get_interface_mock('s2-eth1', 1, mock_switch_b) |
|
118 | |||
119 | 1 | mock_link = get_link_mock(mock_interface_a, mock_interface_b) |
|
120 | 1 | mock_link.id = 'cf0f4071be4' |
|
121 | 1 | mock_switch_a.id = dpid_a |
|
122 | 1 | mock_switch_a.as_dict.return_value = {'metadata': {'lat': '0.0', |
|
123 | 'lng': '-30.0'}} |
||
124 | 1 | mock_switch_b.id = dpid_b |
|
125 | 1 | mock_switch_b.as_dict.return_value = {'metadata': {'lat': '0.0', |
|
126 | 'lng': '-30.0'}} |
||
127 | |||
128 | 1 | self.napp.controller.switches = {dpid_a: mock_switch_a, |
|
129 | dpid_b: mock_switch_b} |
||
130 | |||
131 | 1 | self.napp.controller.links = {"cf0f4071be4": mock_link} |
|
132 | 1 | mock_link.as_dict.return_value = {"id": "cf0f4071be4"} |
|
133 | 1 | endpoint = f"{self.base_endpoint}/" |
|
134 | 1 | response = await self.api_client.get(endpoint) |
|
135 | 1 | assert response.status_code == 200 |
|
136 | 1 | assert response.json() == expected |
|
137 | |||
138 | 1 | def test_load_topology(self): |
|
139 | """Test load_topology.""" |
||
140 | 1 | mock_buffers_put = MagicMock() |
|
141 | 1 | self.napp.controller.buffers.app.put = mock_buffers_put |
|
142 | 1 | link_id = \ |
|
143 | 'cf0f4071be426b3f745027f5d22bc61f8312ae86293c9b28e7e66015607a9260' |
||
144 | 1 | dpid_a = '00:00:00:00:00:00:00:01' |
|
145 | 1 | dpid_b = '00:00:00:00:00:00:00:02' |
|
146 | 1 | topology = { |
|
147 | "topology": { |
||
148 | "links": { |
||
149 | link_id: { |
||
150 | "enabled": True, |
||
151 | "id": link_id, |
||
152 | "endpoint_a": {"id": f"{dpid_a}:2"}, |
||
153 | "endpoint_b": {"id": f"{dpid_b}:2"}, |
||
154 | } |
||
155 | }, |
||
156 | "switches": { |
||
157 | dpid_a: { |
||
158 | "dpid": dpid_a, |
||
159 | "enabled": True, |
||
160 | "metadata": {}, |
||
161 | "id": dpid_a, |
||
162 | "interfaces": { |
||
163 | f"{dpid_a}:2": { |
||
164 | "enabled": True, |
||
165 | "metadata": {}, |
||
166 | "lldp": True, |
||
167 | "port_number": 2, |
||
168 | "name": "s1-eth2", |
||
169 | } |
||
170 | }, |
||
171 | }, |
||
172 | dpid_b: { |
||
173 | "dpid": dpid_b, |
||
174 | "enabled": True, |
||
175 | "metadata": {}, |
||
176 | "id": dpid_b, |
||
177 | "interfaces": { |
||
178 | f"{dpid_b}:2": { |
||
179 | "enabled": True, |
||
180 | "metadata": {}, |
||
181 | "lldp": True, |
||
182 | "port_number": 2, |
||
183 | "name": "s2-eth2", |
||
184 | } |
||
185 | }, |
||
186 | }, |
||
187 | }, |
||
188 | } |
||
189 | } |
||
190 | 1 | switches_expected = [dpid_a, dpid_b] |
|
191 | 1 | interfaces_expected = [f'{dpid_a}:2', f'{dpid_b}:2'] |
|
192 | 1 | links_expected = [link_id] |
|
193 | 1 | self.napp.topo_controller.get_topology.return_value = topology |
|
194 | 1 | self.napp.load_topology() |
|
195 | 1 | assert switches_expected == list(self.napp.controller.switches.keys()) |
|
196 | 1 | interfaces = [] |
|
197 | 1 | for switch in self.napp.controller.switches.values(): |
|
198 | 1 | for iface in switch.interfaces.values(): |
|
199 | 1 | interfaces.append(iface.id) |
|
200 | 1 | assert interfaces_expected == interfaces |
|
201 | 1 | assert links_expected == list(self.napp.controller.links.keys()) |
|
202 | 1 | assert mock_buffers_put.call_args[1] == {"timeout": 1} |
|
203 | |||
204 | 1 | @patch('napps.kytos.topology.main.Main._load_switch') |
|
205 | 1 | @patch('napps.kytos.topology.main.Main._load_link') |
|
206 | 1 | def test_load_topology_does_nothing(self, *args): |
|
207 | """Test _load_network_status doing nothing.""" |
||
208 | 1 | (mock_load_link, mock_load_switch) = args |
|
209 | 1 | self.napp.topo_controller.get_topology.return_value = { |
|
210 | "topology": {"switches": {}, "links": {}} |
||
211 | } |
||
212 | 1 | self.napp.topo_controller.load_topology() |
|
213 | 1 | assert mock_load_link.call_count == 0 |
|
214 | 1 | assert mock_load_switch.call_count == 0 |
|
215 | |||
216 | 1 | @patch('napps.kytos.topology.main.Main._load_switch') |
|
217 | 1 | @patch('napps.kytos.topology.main.log') |
|
218 | 1 | def test_load_topology_fail_switch(self, *args): |
|
219 | """Test load_topology failure in switch.""" |
||
220 | 1 | (mock_log, mock_load_switch) = args |
|
221 | 1 | topology = { |
|
222 | 'topology': { |
||
223 | 'links': {}, |
||
224 | 'switches': { |
||
225 | '1': {} |
||
226 | } |
||
227 | } |
||
228 | } |
||
229 | 1 | mock_log.error.return_value = True |
|
230 | 1 | self.napp.topo_controller.get_topology.return_value = topology |
|
231 | 1 | mock_load_switch.side_effect = AttributeError('xpto') |
|
232 | 1 | self.napp.load_topology() |
|
233 | 1 | error = 'Error loading switch: xpto' |
|
234 | 1 | mock_log.error.assert_called_with(error) |
|
235 | |||
236 | 1 | @patch('napps.kytos.topology.main.Main._load_link') |
|
237 | 1 | @patch('napps.kytos.topology.main.log') |
|
238 | 1 | def test_load_topology_fail_link(self, *args): |
|
239 | """Test load_topology failure in link.""" |
||
240 | 1 | (mock_log, mock_load_link) = args |
|
241 | 1 | topology = { |
|
242 | 'topology': { |
||
243 | 'switches': {}, |
||
244 | 'links': { |
||
245 | '1': {} |
||
246 | } |
||
247 | } |
||
248 | } |
||
249 | 1 | mock_log.error.return_value = True |
|
250 | 1 | self.napp.topo_controller.get_topology.return_value = topology |
|
251 | 1 | mock_load_link.side_effect = AttributeError('xpto') |
|
252 | 1 | self.napp.load_topology() |
|
253 | 1 | error = 'Error loading link 1: xpto' |
|
254 | 1 | mock_log.error.assert_called_with(error) |
|
255 | |||
256 | 1 | @patch('napps.kytos.topology.main.Main.load_interfaces_tags_values') |
|
257 | 1 | @patch('napps.kytos.topology.main.KytosEvent') |
|
258 | 1 | def test_load_switch(self, *args): |
|
259 | """Test _load_switch.""" |
||
260 | 1 | (mock_event, mock_load_tags) = args |
|
261 | 1 | mock_buffers_put = MagicMock() |
|
262 | 1 | self.napp.controller.buffers.app.put = mock_buffers_put |
|
263 | 1 | dpid_a = "00:00:00:00:00:00:00:01" |
|
264 | 1 | dpid_x = "00:00:00:00:00:00:00:XX" |
|
265 | 1 | iface_a = f'{dpid_a}:1' |
|
266 | 1 | switch_attrs = { |
|
267 | 'dpid': dpid_a, |
||
268 | 'enabled': True, |
||
269 | 'id': dpid_a, |
||
270 | 'metadata': {}, |
||
271 | 'interfaces': { |
||
272 | iface_a: { |
||
273 | 'enabled': True, |
||
274 | 'active': True, |
||
275 | 'lldp': True, |
||
276 | 'id': iface_a, |
||
277 | 'switch': dpid_a, |
||
278 | 'metadata': {}, |
||
279 | 'name': 's2-eth1', |
||
280 | 'port_number': 1 |
||
281 | } |
||
282 | } |
||
283 | } |
||
284 | 1 | self.napp._load_switch(dpid_a, switch_attrs) |
|
285 | |||
286 | 1 | assert len(self.napp.controller.switches) == 1 |
|
287 | 1 | assert dpid_a in self.napp.controller.switches |
|
288 | 1 | assert dpid_x not in self.napp.controller.switches |
|
289 | 1 | switch = self.napp.controller.switches[dpid_a] |
|
290 | 1 | interface_details = self.napp.topo_controller.get_interfaces_details |
|
291 | 1 | interface_details.assert_called_once_with([iface_a]) |
|
292 | 1 | mock_load_tags.assert_called() |
|
293 | |||
294 | 1 | assert switch.id == dpid_a |
|
295 | 1 | assert switch.dpid == dpid_a |
|
296 | 1 | assert switch.is_enabled() |
|
297 | 1 | assert not switch.is_active() |
|
298 | |||
299 | 1 | assert len(switch.interfaces) == 1 |
|
300 | 1 | assert 1 in switch.interfaces |
|
301 | 1 | assert 2 not in switch.interfaces |
|
302 | 1 | mock_event.assert_called() |
|
303 | 1 | mock_buffers_put.assert_called() |
|
304 | 1 | assert mock_buffers_put.call_args[1] == {"timeout": 1} |
|
305 | |||
306 | 1 | interface = switch.interfaces[1] |
|
307 | 1 | assert interface.id == iface_a |
|
308 | 1 | assert interface.switch.id == dpid_a |
|
309 | 1 | assert interface.port_number == 1 |
|
310 | 1 | assert interface.is_enabled() |
|
311 | 1 | assert not interface.is_active() |
|
312 | 1 | assert interface.lldp |
|
313 | 1 | assert interface.uni |
|
314 | 1 | assert not interface.nni |
|
315 | |||
316 | 1 | def test_load_switch_attrs(self): |
|
317 | """Test _load_switch.""" |
||
318 | 1 | dpid_b = "00:00:00:00:00:00:00:02" |
|
319 | 1 | iface_b = f'{dpid_b}:1' |
|
320 | 1 | switch_attrs = { |
|
321 | "active": True, |
||
322 | "connection": "127.0.0.1:43230", |
||
323 | "data_path": "XX Human readable desc of dp", |
||
324 | "dpid": "00:00:00:00:00:00:00:02", |
||
325 | "enabled": False, |
||
326 | "hardware": "Open vSwitch", |
||
327 | "id": "00:00:00:00:00:00:00:02", |
||
328 | "interfaces": { |
||
329 | "00:00:00:00:00:00:00:02:1": { |
||
330 | "active": True, |
||
331 | "enabled": False, |
||
332 | "id": "00:00:00:00:00:00:00:02:1", |
||
333 | "link": "", |
||
334 | "lldp": False, |
||
335 | "mac": "de:58:c3:30:b7:b7", |
||
336 | "metadata": {}, |
||
337 | "name": "s2-eth1", |
||
338 | "nni": False, |
||
339 | "port_number": 1, |
||
340 | "speed": 1250000000, |
||
341 | "switch": "00:00:00:00:00:00:00:02", |
||
342 | "type": "interface", |
||
343 | "uni": True |
||
344 | }, |
||
345 | }, |
||
346 | "manufacturer": "Nicira, Inc.", |
||
347 | "metadata": {}, |
||
348 | "name": "00:00:00:00:00:00:00:04", |
||
349 | "ofp_version": "0x04", |
||
350 | "serial": "XX serial number", |
||
351 | "software": "2.10.7", |
||
352 | "type": "switch" |
||
353 | } |
||
354 | |||
355 | 1 | assert len(self.napp.controller.switches) == 0 |
|
356 | 1 | self.napp._load_switch(dpid_b, switch_attrs) |
|
357 | 1 | assert len(self.napp.controller.switches) == 1 |
|
358 | 1 | assert dpid_b in self.napp.controller.switches |
|
359 | |||
360 | 1 | switch = self.napp.controller.switches[dpid_b] |
|
361 | 1 | assert switch.id == dpid_b |
|
362 | 1 | assert switch.dpid == dpid_b |
|
363 | 1 | assert not switch.is_enabled() |
|
364 | 1 | assert not switch.is_active() |
|
365 | 1 | assert switch.description['manufacturer'] == 'Nicira, Inc.' |
|
366 | 1 | assert switch.description['hardware'] == 'Open vSwitch' |
|
367 | 1 | assert switch.description['software'] == '2.10.7' |
|
368 | 1 | assert switch.description['serial'] == 'XX serial number' |
|
369 | 1 | exp_data_path = 'XX Human readable desc of dp' |
|
370 | 1 | assert switch.description['data_path'] == exp_data_path |
|
371 | |||
372 | 1 | assert len(switch.interfaces) == 1 |
|
373 | 1 | assert 1 in switch.interfaces |
|
374 | 1 | assert 2 not in switch.interfaces |
|
375 | |||
376 | 1 | interface = switch.interfaces[1] |
|
377 | 1 | assert interface.id == iface_b |
|
378 | 1 | assert interface.switch.id == dpid_b |
|
379 | 1 | assert interface.port_number == 1 |
|
380 | 1 | assert not interface.is_enabled() |
|
381 | 1 | assert not interface.lldp |
|
382 | 1 | assert interface.uni |
|
383 | 1 | assert not interface.nni |
|
384 | |||
385 | 1 | def test_load_interfaces_tags_values(self): |
|
386 | """Test load_interfaces_tags_values.""" |
||
387 | 1 | dpid_a = "00:00:00:00:00:00:00:01" |
|
388 | 1 | mock_switch_a = get_switch_mock(dpid_a, 0x04) |
|
389 | 1 | mock_interface_a = get_interface_mock('s1-eth1', 1, mock_switch_a) |
|
390 | 1 | mock_interface_a.id = dpid_a + ':1' |
|
391 | 1 | mock_switch_a.interfaces = {1: mock_interface_a} |
|
392 | 1 | ava_tags = {'vlan': [[10, 4095]]} |
|
393 | 1 | tag_ranges = {'vlan': [[5, 4095]]} |
|
394 | 1 | special_available_tags = {'vlan': ["untagged", "any"]} |
|
395 | 1 | special_tags = {'vlan': ["untagged", "any"]} |
|
396 | 1 | interface_details = [{ |
|
397 | "id": mock_interface_a.id, |
||
398 | "available_tags": ava_tags, |
||
399 | "tag_ranges": tag_ranges, |
||
400 | "special_available_tags": special_available_tags, |
||
401 | "special_tags": special_tags |
||
402 | }] |
||
403 | 1 | self.napp.load_interfaces_tags_values(mock_switch_a, |
|
404 | interface_details) |
||
405 | 1 | set_method = mock_interface_a.set_available_tags_tag_ranges |
|
406 | 1 | set_method.assert_called_once_with( |
|
407 | ava_tags, tag_ranges, |
||
408 | special_available_tags, special_tags |
||
409 | ) |
||
410 | |||
411 | 1 | def test_handle_on_interface_tags(self): |
|
412 | """test_handle_on_interface_tags.""" |
||
413 | 1 | dpid_a = "00:00:00:00:00:00:00:01" |
|
414 | 1 | available_tags = {'vlan': [[200, 3000]]} |
|
415 | 1 | tag_ranges = {'vlan': [[20, 20], [200, 3000]]} |
|
416 | 1 | special_available_tags = {'vlan': ["untagged", "any"]} |
|
417 | 1 | mock_switch_a = get_switch_mock(dpid_a, 0x04) |
|
418 | 1 | mock_interface_a = get_interface_mock('s1-eth1', 1, mock_switch_a) |
|
419 | 1 | mock_interface_a.available_tags = available_tags |
|
420 | 1 | mock_interface_a.tag_ranges = tag_ranges |
|
421 | 1 | mock_interface_a.special_available_tags = special_available_tags |
|
422 | 1 | mock_interface_a.special_tags = special_available_tags |
|
423 | 1 | self.napp.handle_on_interface_tags(mock_interface_a) |
|
424 | 1 | tp_controller = self.napp.topo_controller |
|
425 | 1 | args = tp_controller.upsert_interface_details.call_args[0] |
|
426 | 1 | assert args[0] == '00:00:00:00:00:00:00:01:1' |
|
427 | 1 | assert args[1] == {'vlan': [[200, 3000]]} |
|
428 | 1 | assert args[2] == {'vlan': [[20, 20], [200, 3000]]} |
|
429 | |||
430 | 1 | def test_load_link(self): |
|
431 | """Test _load_link.""" |
||
432 | 1 | dpid_a = "00:00:00:00:00:00:00:01" |
|
433 | 1 | dpid_b = "00:00:00:00:00:00:00:02" |
|
434 | 1 | link_id = '4d42dc08522' |
|
435 | 1 | mock_switch_a = get_switch_mock(dpid_a, 0x04) |
|
436 | 1 | mock_switch_b = get_switch_mock(dpid_b, 0x04) |
|
437 | 1 | mock_interface_a = get_interface_mock('s1-eth1', 1, mock_switch_a) |
|
438 | 1 | mock_interface_a.id = dpid_a + ':1' |
|
439 | 1 | mock_interface_a.available_tags = [1, 2, 3] |
|
440 | 1 | mock_interface_a.link = None |
|
441 | 1 | mock_interface_b = get_interface_mock('s2-eth1', 1, mock_switch_b) |
|
442 | 1 | mock_interface_b.id = dpid_b + ':1' |
|
443 | 1 | mock_interface_b.available_tags = [1, 2, 3] |
|
444 | 1 | mock_interface_b.link = None |
|
445 | 1 | mock_switch_a.interfaces = {1: mock_interface_a} |
|
446 | 1 | mock_switch_b.interfaces = {1: mock_interface_b} |
|
447 | 1 | self.napp.controller.switches[dpid_a] = mock_switch_a |
|
448 | 1 | self.napp.controller.switches[dpid_b] = mock_switch_b |
|
449 | 1 | link_attrs = { |
|
450 | 'enabled': True, |
||
451 | 'id': link_id, |
||
452 | 'metadata': {}, |
||
453 | 'endpoint_a': { |
||
454 | 'id': mock_interface_a.id |
||
455 | }, |
||
456 | 'endpoint_b': { |
||
457 | 'id': mock_interface_b.id |
||
458 | } |
||
459 | } |
||
460 | |||
461 | 1 | self.napp._load_link(link_attrs) |
|
462 | |||
463 | 1 | assert len(self.napp.controller.links) == 1 |
|
464 | 1 | link = list(self.napp.controller.links.values())[0] |
|
465 | |||
466 | 1 | assert link.endpoint_a.id == mock_interface_a.id |
|
467 | 1 | assert link.endpoint_b.id == mock_interface_b.id |
|
468 | 1 | assert mock_interface_a.nni |
|
469 | 1 | assert mock_interface_b.nni |
|
470 | 1 | assert mock_interface_a.update_link.call_count == 1 |
|
471 | 1 | assert mock_interface_b.update_link.call_count == 1 |
|
472 | |||
473 | # test enable/disable |
||
474 | 1 | link_id = '4d42dc08522' |
|
475 | 1 | mock_interface_a = get_interface_mock('s1-eth1', 1, mock_switch_a) |
|
476 | 1 | mock_interface_b = get_interface_mock('s2-eth1', 1, mock_switch_b) |
|
477 | 1 | mock_link = get_link_mock(mock_interface_a, mock_interface_b) |
|
478 | 1 | mock_link.id = link_id |
|
479 | |||
480 | 1 | self.napp.controller.get_link_or_create = MagicMock() |
|
481 | 1 | mock_get_link_or_create = self.napp.controller.get_link_or_create |
|
482 | 1 | mock_get_link_or_create.return_value = (mock_link, True) |
|
483 | |||
484 | 1 | def test_fail_load_link(self): |
|
485 | """Test fail load_link.""" |
||
486 | 1 | self.napp.controller.get_link_or_create = MagicMock() |
|
487 | 1 | mock_get_link_or_create = self.napp.controller.get_link_or_create |
|
488 | 1 | dpid_a = '00:00:00:00:00:00:00:01' |
|
489 | 1 | dpid_b = '00:00:00:00:00:00:00:02' |
|
490 | 1 | link_id = '4d42dc08522' |
|
491 | 1 | mock_switch_a = get_switch_mock(dpid_a) |
|
492 | 1 | mock_switch_b = get_switch_mock(dpid_b) |
|
493 | 1 | mock_interface_a_1 = get_interface_mock('s1-eth1', 1, mock_switch_a) |
|
494 | 1 | mock_interface_b_1 = get_interface_mock('s2-eth1', 1, mock_switch_b) |
|
495 | 1 | mock_link = get_link_mock(mock_interface_a_1, mock_interface_b_1) |
|
496 | 1 | mock_link.id = link_id |
|
497 | 1 | self.napp.controller.links = {link_id: mock_link} |
|
498 | 1 | mock_get_link_or_create.return_value = mock_link |
|
499 | |||
500 | 1 | link_attrs_fail = { |
|
501 | 'enabled': True, |
||
502 | 'id': link_id, |
||
503 | 'metadata': {}, |
||
504 | 'endpoint_a': { |
||
505 | 'id': f"{dpid_a}:999", |
||
506 | }, |
||
507 | 'endpoint_b': { |
||
508 | 'id': f"{dpid_b}:999", |
||
509 | } |
||
510 | } |
||
511 | 1 | with pytest.raises(RestoreError): |
|
512 | 1 | self.napp._load_link(link_attrs_fail) |
|
513 | |||
514 | 1 | link_attrs_fail = { |
|
515 | 'enabled': True, |
||
516 | 'id': link_id, |
||
517 | 'endpoint_a': { |
||
518 | 'id': f"{dpid_a}:1", |
||
519 | }, |
||
520 | 'endpoint_b': { |
||
521 | 'id': f"{dpid_b}:1", |
||
522 | } |
||
523 | } |
||
524 | 1 | with pytest.raises(RestoreError): |
|
525 | 1 | self.napp._load_link(link_attrs_fail) |
|
526 | |||
527 | 1 | @patch('napps.kytos.topology.main.Main.notify_switch_links_status') |
|
528 | 1 | @patch('napps.kytos.topology.main.Main.notify_topology_update') |
|
529 | 1 | async def test_enable_switch(self, mock_notify_topo, mock_sw_l_status): |
|
530 | """Test enable_switch.""" |
||
531 | 1 | dpid = "00:00:00:00:00:00:00:01" |
|
532 | 1 | mock_switch = get_switch_mock(dpid) |
|
533 | 1 | self.napp.controller.switches = {dpid: mock_switch} |
|
534 | |||
535 | 1 | endpoint = f"{self.base_endpoint}/switches/{dpid}/enable" |
|
536 | 1 | response = await self.api_client.post(endpoint) |
|
537 | 1 | assert response.status_code == 201 |
|
538 | 1 | assert mock_switch.enable.call_count == 1 |
|
539 | 1 | self.napp.topo_controller.enable_switch.assert_called_once_with(dpid) |
|
540 | 1 | mock_notify_topo.assert_called() |
|
541 | 1 | mock_sw_l_status.assert_called() |
|
542 | |||
543 | # fail case |
||
544 | 1 | mock_switch.enable.call_count = 0 |
|
545 | 1 | dpid = "00:00:00:00:00:00:00:02" |
|
546 | 1 | endpoint = f"{self.base_endpoint}/switches/{dpid}/enable" |
|
547 | 1 | response = await self.api_client.post(endpoint) |
|
548 | assert response.status_code == 404 |
||
549 | assert mock_switch.enable.call_count == 0 |
||
550 | |||
551 | 1 | @patch('napps.kytos.topology.main.Main.notify_link_enabled_state') |
|
552 | 1 | @patch('napps.kytos.topology.main.Main.notify_switch_links_status') |
|
553 | 1 | @patch('napps.kytos.topology.main.Main.notify_topology_update') |
|
554 | 1 | async def test_disable_switch(self, *args): |
|
555 | """Test disable_switch.""" |
||
556 | 1 | mock_notify_topo, mock_sw_l_status, mock_noti_link = args |
|
557 | 1 | dpid = "00:00:00:00:00:00:00:01" |
|
558 | 1 | mock_switch = get_switch_mock(dpid) |
|
559 | 1 | interface = Mock() |
|
560 | 1 | interface.link.is_enabled = lambda: True |
|
561 | 1 | interface.link.link_lock = MagicMock() |
|
562 | 1 | mock_switch.interfaces = {1: interface} |
|
563 | 1 | self.napp.controller.switches = {dpid: mock_switch} |
|
564 | |||
565 | 1 | endpoint = f"{self.base_endpoint}/switches/{dpid}/disable" |
|
566 | 1 | response = await self.api_client.post(endpoint) |
|
567 | 1 | assert response.status_code == 201 |
|
568 | 1 | assert mock_switch.disable.call_count == 1 |
|
569 | 1 | assert mock_noti_link.call_count == 1 |
|
570 | 1 | assert mock_noti_link.call_args[0][0] == interface.link |
|
571 | 1 | assert mock_noti_link.call_args[0][1] == "disabled" |
|
572 | 1 | assert interface.link.disable.call_count == 1 |
|
573 | 1 | assert self.napp.topo_controller.bulk_disable_links.call_count == 1 |
|
574 | 1 | self.napp.topo_controller.disable_switch.assert_called_once_with(dpid) |
|
575 | 1 | mock_notify_topo.assert_called() |
|
576 | 1 | mock_sw_l_status.assert_called() |
|
577 | |||
578 | # fail case |
||
579 | 1 | mock_switch.disable.call_count = 0 |
|
580 | 1 | dpid = "00:00:00:00:00:00:00:02" |
|
581 | 1 | endpoint = f"{self.base_endpoint}/switches/{dpid}/disable" |
|
582 | 1 | response = await self.api_client.post(endpoint) |
|
583 | assert response.status_code == 404 |
||
584 | assert mock_switch.disable.call_count == 0 |
||
585 | |||
586 | 1 | async def test_get_switch_metadata(self): |
|
587 | """Test get_switch_metadata.""" |
||
588 | 1 | dpid = "00:00:00:00:00:00:00:01" |
|
589 | 1 | mock_switch = get_switch_mock(dpid) |
|
590 | 1 | mock_switch.metadata = "A" |
|
591 | 1 | self.napp.controller.switches = {dpid: mock_switch} |
|
592 | |||
593 | 1 | endpoint = f"{self.base_endpoint}/switches/{dpid}/metadata" |
|
594 | 1 | response = await self.api_client.get(endpoint) |
|
595 | 1 | assert response.status_code == 200 |
|
596 | 1 | assert response.json() == {"metadata": mock_switch.metadata} |
|
597 | |||
598 | # fail case |
||
599 | 1 | dpid = "00:00:00:00:00:00:00:02" |
|
600 | 1 | endpoint = f"{self.base_endpoint}/switches/{dpid}/metadata" |
|
601 | 1 | response = await self.api_client.get(endpoint) |
|
602 | assert response.status_code == 404 |
||
603 | |||
604 | 1 | @patch('napps.kytos.topology.main.Main.notify_metadata_changes') |
|
605 | 1 | async def test_add_switch_metadata( |
|
606 | self, mock_metadata_changes |
||
607 | ): |
||
608 | """Test add_switch_metadata.""" |
||
609 | 1 | self.napp.controller.loop = asyncio.get_running_loop() |
|
610 | 1 | dpid = "00:00:00:00:00:00:00:01" |
|
611 | 1 | mock_switch = get_switch_mock(dpid) |
|
612 | 1 | self.napp.controller.switches = {dpid: mock_switch} |
|
613 | 1 | payload = {"data": "A"} |
|
614 | |||
615 | 1 | endpoint = f"{self.base_endpoint}/switches/{dpid}/metadata" |
|
616 | 1 | response = await self.api_client.post(endpoint, json=payload) |
|
617 | 1 | assert response.status_code == 201 |
|
618 | |||
619 | 1 | mock_metadata_changes.assert_called() |
|
620 | 1 | self.napp.topo_controller.add_switch_metadata.assert_called_once_with( |
|
621 | dpid, payload |
||
622 | ) |
||
623 | |||
624 | # fail case |
||
625 | 1 | dpid = "00:00:00:00:00:00:00:02" |
|
626 | 1 | endpoint = f"{self.base_endpoint}/switches/{dpid}/metadata" |
|
627 | 1 | response = await self.api_client.post(endpoint, json=payload) |
|
628 | assert response.status_code == 404 |
||
629 | |||
630 | 1 | async def test_add_switch_metadata_wrong_format(self): |
|
631 | """Test add_switch_metadata_wrong_format.""" |
||
632 | 1 | self.napp.controller.loop = asyncio.get_running_loop() |
|
633 | 1 | dpid = "00:00:00:00:00:00:00:01" |
|
634 | 1 | payload = 'A' |
|
635 | |||
636 | 1 | endpoint = f"{self.base_endpoint}/switches/{dpid}/metadata" |
|
637 | 1 | response = await self.api_client.post(endpoint, json=payload) |
|
638 | assert response.status_code == 400 |
||
639 | |||
640 | payload = None |
||
641 | response = await self.api_client.post(endpoint, json=payload) |
||
642 | assert response.status_code == 415 |
||
643 | |||
644 | 1 | @patch('napps.kytos.topology.main.Main.notify_metadata_changes') |
|
645 | 1 | async def test_delete_switch_metadata( |
|
646 | self, mock_metadata_changes |
||
647 | ): |
||
648 | """Test delete_switch_metadata.""" |
||
649 | 1 | self.napp.controller.loop = asyncio.get_running_loop() |
|
650 | 1 | dpid = "00:00:00:00:00:00:00:01" |
|
651 | 1 | mock_switch = get_switch_mock(dpid) |
|
652 | 1 | mock_switch.metadata = {"A": "A"} |
|
653 | 1 | self.napp.controller.switches = {dpid: mock_switch} |
|
654 | |||
655 | 1 | key = "A" |
|
656 | 1 | endpoint = f"{self.base_endpoint}/switches/{dpid}/metadata/{key}" |
|
657 | 1 | response = await self.api_client.delete(endpoint) |
|
658 | |||
659 | 1 | assert response.status_code == 200 |
|
660 | 1 | assert mock_metadata_changes.call_count == 1 |
|
661 | 1 | del_key_mock = self.napp.topo_controller.delete_switch_metadata_key |
|
662 | 1 | del_key_mock.assert_called_with( |
|
663 | dpid, key |
||
664 | ) |
||
665 | |||
666 | # 404, Metadata not found |
||
667 | 1 | mock_switch.metadata = {} |
|
668 | 1 | endpoint = f"{self.base_endpoint}/switches/{dpid}/metadata/{key}" |
|
669 | 1 | response = await self.api_client.delete(endpoint) |
|
670 | assert response.status_code == 404 |
||
671 | |||
672 | # 404, Switch not found |
||
673 | key = "A" |
||
674 | dpid = "00:00:00:00:00:00:00:02" |
||
675 | endpoint = f"{self.base_endpoint}/switches/{dpid}/metadata/{key}" |
||
676 | response = await self.api_client.delete(endpoint) |
||
677 | assert mock_metadata_changes.call_count == 1 |
||
678 | assert response.status_code == 404 |
||
679 | |||
680 | # pylint: disable=too-many-statements |
||
681 | 1 | @patch('napps.kytos.topology.main.Main.notify_topology_update') |
|
682 | 1 | async def test_enable_interfaces(self, mock_notify_topo): |
|
683 | """Test enable_interfaces.""" |
||
684 | 1 | dpid = '00:00:00:00:00:00:00:01' |
|
685 | 1 | mock_switch = get_switch_mock(dpid) |
|
686 | 1 | mock_interface_1 = get_interface_mock('s1-eth1', 1, mock_switch) |
|
687 | 1 | mock_interface_1.link = Mock() |
|
688 | 1 | mock_interface_1.link.link_lock = MagicMock() |
|
689 | 1 | mock_interface_1.link._enabled = True |
|
690 | 1 | mock_interface_2 = get_interface_mock('s1-eth2', 2, mock_switch) |
|
691 | 1 | mock_interface_2.link = Mock() |
|
692 | 1 | mock_interface_2.link.link_lock = MagicMock() |
|
693 | 1 | mock_interface_2.link._enabled = False |
|
694 | 1 | mock_switch.interfaces = {1: mock_interface_1, 2: mock_interface_2} |
|
695 | |||
696 | # Switch not found |
||
697 | 1 | interface_id = '00:00:00:00:00:00:00:01:1' |
|
698 | 1 | endpoint = f"{self.base_endpoint}/interfaces/{interface_id}/enable" |
|
699 | 1 | response = await self.api_client.post(endpoint) |
|
700 | assert response.status_code == 404 |
||
701 | |||
702 | # Switch not enabled |
||
703 | 1 | mock_switch.is_enabled = lambda: False |
|
704 | self.napp.controller.switches = {dpid: mock_switch} |
||
705 | endpoint = f"{self.base_endpoint}/interfaces/{interface_id}/enable" |
||
706 | response = await self.api_client.post(endpoint) |
||
707 | assert response.status_code == 409 |
||
708 | |||
709 | # Success |
||
710 | 1 | mock_switch.is_enabled = lambda: True |
|
711 | endpoint = f"{self.base_endpoint}/interfaces/{interface_id}/enable" |
||
712 | response = await self.api_client.post(endpoint) |
||
713 | 1 | assert response.status_code == 200 |
|
714 | 1 | assert mock_interface_1.enable.call_count == 1 |
|
715 | 1 | assert mock_interface_2.enable.call_count == 0 |
|
716 | 1 | self.napp.topo_controller.enable_interface.assert_called_with( |
|
717 | interface_id |
||
718 | ) |
||
719 | 1 | mock_notify_topo.assert_called() |
|
720 | |||
721 | 1 | mock_interface_1.enable.call_count = 0 |
|
722 | 1 | mock_interface_2.enable.call_count = 0 |
|
723 | 1 | endpoint = f"{self.base_endpoint}/interfaces/switch/{dpid}/enable" |
|
724 | 1 | response = await self.api_client.post(endpoint) |
|
725 | 1 | assert response.status_code == 200 |
|
726 | 1 | self.napp.topo_controller.upsert_switch.assert_called_with( |
|
727 | mock_switch.id, mock_switch.as_dict() |
||
728 | ) |
||
729 | 1 | assert mock_interface_1.enable.call_count == 1 |
|
730 | 1 | assert mock_interface_2.enable.call_count == 1 |
|
731 | |||
732 | # test interface not found |
||
733 | 1 | interface_id = '00:00:00:00:00:00:00:01:3' |
|
734 | 1 | mock_interface_1.enable.call_count = 0 |
|
735 | 1 | mock_interface_2.enable.call_count = 0 |
|
736 | 1 | endpoint = f"{self.base_endpoint}/interfaces/{interface_id}/enable" |
|
737 | 1 | response = await self.api_client.post(endpoint) |
|
738 | assert response.status_code == 404 |
||
739 | assert mock_interface_1.enable.call_count == 0 |
||
740 | assert mock_interface_2.enable.call_count == 0 |
||
741 | |||
742 | # test switch not found |
||
743 | dpid = '00:00:00:00:00:00:00:02' |
||
744 | endpoint = f"{self.base_endpoint}/interfaces/{interface_id}/enable" |
||
745 | response = await self.api_client.post(endpoint) |
||
746 | assert response.status_code == 404 |
||
747 | assert mock_interface_1.enable.call_count == 0 |
||
748 | assert mock_interface_2.enable.call_count == 0 |
||
749 | |||
750 | 1 | @patch('napps.kytos.topology.main.Main.notify_link_enabled_state') |
|
751 | 1 | @patch('napps.kytos.topology.main.Main.notify_topology_update') |
|
752 | 1 | async def test_disable_interfaces(self, mock_notify_topo, mock_noti_link): |
|
753 | """Test disable_interfaces.""" |
||
754 | 1 | interface_id = '00:00:00:00:00:00:00:01:1' |
|
755 | 1 | dpid = '00:00:00:00:00:00:00:01' |
|
756 | 1 | mock_switch = get_switch_mock(dpid) |
|
757 | 1 | mock_interface_1 = get_interface_mock('s1-eth1', 1, mock_switch) |
|
758 | 1 | mock_interface_1.link = Mock() |
|
759 | 1 | mock_interface_1.link.is_enabled = lambda: True |
|
760 | 1 | mock_interface_1.link.link_lock = MagicMock() |
|
761 | 1 | mock_interface_2 = get_interface_mock('s1-eth2', 2, mock_switch) |
|
762 | 1 | mock_interface_2.link = Mock() |
|
763 | 1 | mock_interface_2.link.is_enabled = lambda: False |
|
764 | 1 | mock_interface_2.link.link_lock = MagicMock() |
|
765 | 1 | mock_switch.interfaces = {1: mock_interface_1, 2: mock_interface_2} |
|
766 | 1 | self.napp.controller.switches = {dpid: mock_switch} |
|
767 | |||
768 | 1 | endpoint = f"{self.base_endpoint}/interfaces/{interface_id}/disable" |
|
769 | 1 | response = await self.api_client.post(endpoint) |
|
770 | 1 | assert response.status_code == 200 |
|
771 | |||
772 | 1 | self.napp.topo_controller.disable_interface.assert_called_with( |
|
773 | interface_id |
||
774 | ) |
||
775 | 1 | assert mock_interface_1.disable.call_count == 1 |
|
776 | 1 | assert mock_interface_2.disable.call_count == 0 |
|
777 | 1 | assert mock_interface_1.link.disable.call_count == 1 |
|
778 | 1 | assert mock_noti_link.call_count == 1 |
|
779 | 1 | assert mock_noti_link.call_args[0][0] == mock_interface_1.link |
|
780 | 1 | assert mock_noti_link.call_args[0][1] == "disabled" |
|
781 | 1 | assert self.napp.topo_controller.disable_interface.call_count == 1 |
|
782 | 1 | mock_notify_topo.assert_called() |
|
783 | |||
784 | 1 | mock_interface_1.disable.call_count = 0 |
|
785 | 1 | mock_interface_2.disable.call_count = 0 |
|
786 | |||
787 | 1 | endpoint = f"{self.base_endpoint}/interfaces/switch/{dpid}/disable" |
|
788 | 1 | response = await self.api_client.post(endpoint) |
|
789 | 1 | assert response.status_code == 200 |
|
790 | |||
791 | 1 | self.napp.topo_controller.upsert_switch.assert_called_with( |
|
792 | mock_switch.id, mock_switch.as_dict() |
||
793 | ) |
||
794 | 1 | assert mock_interface_1.disable.call_count == 1 |
|
795 | 1 | assert mock_interface_1.link.disable.call_count == 2 |
|
796 | 1 | assert mock_interface_2.disable.call_count == 1 |
|
797 | 1 | assert mock_noti_link.call_count == 2 |
|
798 | 1 | assert mock_noti_link.call_args[0][0] == mock_interface_1.link |
|
799 | 1 | assert mock_noti_link.call_args[0][1] == "disabled" |
|
800 | 1 | bulk_controller = self.napp.topo_controller.bulk_disable_links |
|
801 | 1 | assert bulk_controller.call_count == 2 |
|
802 | 1 | assert len(bulk_controller.call_args[0][0]) == 1 |
|
803 | |||
804 | # test interface not found |
||
805 | 1 | interface_id = '00:00:00:00:00:00:00:01:3' |
|
806 | 1 | mock_interface_1.disable.call_count = 0 |
|
807 | 1 | mock_interface_2.disable.call_count = 0 |
|
808 | 1 | endpoint = f"{self.base_endpoint}/interfaces/{interface_id}/disable" |
|
809 | 1 | response = await self.api_client.post(endpoint) |
|
810 | |||
811 | assert response.status_code == 404 |
||
812 | assert mock_interface_1.disable.call_count == 0 |
||
813 | assert mock_interface_2.disable.call_count == 0 |
||
814 | |||
815 | # test switch not found |
||
816 | dpid = '00:00:00:00:00:00:00:02' |
||
817 | endpoint = f"{self.base_endpoint}/interfaces/switch/{dpid}/disable" |
||
818 | response = await self.api_client.post(endpoint) |
||
819 | assert response.status_code == 404 |
||
820 | assert mock_interface_1.disable.call_count == 0 |
||
821 | assert mock_interface_2.disable.call_count == 0 |
||
822 | |||
823 | 1 | async def test_get_interface_metadata(self): |
|
824 | """Test get_interface_metada.""" |
||
825 | 1 | interface_id = '00:00:00:00:00:00:00:01:1' |
|
826 | 1 | dpid = '00:00:00:00:00:00:00:01' |
|
827 | 1 | mock_switch = get_switch_mock(dpid) |
|
828 | 1 | mock_interface = get_interface_mock('s1-eth1', 1, mock_switch) |
|
829 | 1 | mock_interface.metadata = {"A": "B"} |
|
830 | 1 | mock_switch.interfaces = {1: mock_interface} |
|
831 | 1 | self.napp.controller.switches = {dpid: mock_switch} |
|
832 | |||
833 | 1 | endpoint = f"{self.base_endpoint}/interfaces/{interface_id}/metadata" |
|
834 | 1 | response = await self.api_client.get(endpoint) |
|
835 | 1 | assert response.status_code == 200 |
|
836 | 1 | assert response.json() == {"metadata": mock_interface.metadata} |
|
837 | |||
838 | # fail case switch not found |
||
839 | 1 | interface_id = '00:00:00:00:00:00:00:02:1' |
|
840 | 1 | endpoint = f"{self.base_endpoint}/interfaces/{interface_id}/metadata" |
|
841 | 1 | response = await self.api_client.get(endpoint) |
|
842 | assert response.status_code == 404 |
||
843 | |||
844 | # fail case interface not found |
||
845 | interface_id = '00:00:00:00:00:00:00:01:2' |
||
846 | endpoint = f"{self.base_endpoint}/interfaces/{interface_id}/metadata" |
||
847 | response = await self.api_client.get(endpoint) |
||
848 | assert response.status_code == 404 |
||
849 | |||
850 | 1 | @patch('napps.kytos.topology.main.Main.notify_metadata_changes') |
|
851 | 1 | async def test_add_interface_metadata( |
|
852 | self, mock_metadata_changes |
||
853 | ): |
||
854 | """Test add_interface_metadata.""" |
||
855 | 1 | self.napp.controller.loop = asyncio.get_running_loop() |
|
856 | 1 | interface_id = '00:00:00:00:00:00:00:01:1' |
|
857 | 1 | dpid = '00:00:00:00:00:00:00:01' |
|
858 | 1 | mock_switch = get_switch_mock(dpid) |
|
859 | 1 | mock_interface = get_interface_mock('s1-eth1', 1, mock_switch) |
|
860 | 1 | mock_interface.metadata = {"metada": "A"} |
|
861 | 1 | mock_switch.interfaces = {1: mock_interface} |
|
862 | 1 | self.napp.controller.switches = {dpid: mock_switch} |
|
863 | 1 | payload = {"metada": "A"} |
|
864 | 1 | endpoint = f"{self.base_endpoint}/interfaces/{interface_id}/metadata" |
|
865 | 1 | response = await self.api_client.post(endpoint, json=payload) |
|
866 | 1 | assert response.status_code == 201 |
|
867 | 1 | mock_metadata_changes.assert_called() |
|
868 | |||
869 | # fail case switch not found |
||
870 | 1 | interface_id = '00:00:00:00:00:00:00:02:1' |
|
871 | 1 | endpoint = f"{self.base_endpoint}/interfaces/{interface_id}/metadata" |
|
872 | 1 | response = await self.api_client.post(endpoint, json=payload) |
|
873 | assert response.status_code == 404 |
||
874 | |||
875 | # fail case interface not found |
||
876 | interface_id = '00:00:00:00:00:00:00:01:2' |
||
877 | endpoint = f"{self.base_endpoint}/interfaces/{interface_id}/metadata" |
||
878 | response = await self.api_client.post(endpoint, json=payload) |
||
879 | assert response.status_code == 404 |
||
880 | |||
881 | 1 | async def test_add_interface_metadata_wrong_format(self): |
|
882 | """Test add_interface_metadata_wrong_format.""" |
||
883 | 1 | self.napp.controller.loop = asyncio.get_running_loop() |
|
884 | 1 | interface_id = "00:00:00:00:00:00:00:01:1" |
|
885 | 1 | endpoint = f"{self.base_endpoint}/interfaces/{interface_id}/metadata" |
|
886 | 1 | response = await self.api_client.post(endpoint, json='A') |
|
887 | assert response.status_code == 400 |
||
888 | response = await self.api_client.post(endpoint, json=None) |
||
889 | assert response.status_code == 415 |
||
890 | |||
891 | 1 | async def test_delete_interface_metadata(self): |
|
892 | """Test delete_interface_metadata.""" |
||
893 | 1 | self.napp.controller.loop = asyncio.get_running_loop() |
|
894 | 1 | interface_id = '00:00:00:00:00:00:00:01:1' |
|
895 | 1 | dpid = '00:00:00:00:00:00:00:01' |
|
896 | 1 | mock_switch = get_switch_mock(dpid) |
|
897 | 1 | mock_interface = get_interface_mock('s1-eth1', 1, mock_switch) |
|
898 | 1 | mock_interface.remove_metadata.side_effect = [True, False] |
|
899 | 1 | mock_interface.metadata = {"A": "A"} |
|
900 | 1 | mock_switch.interfaces = {1: mock_interface} |
|
901 | 1 | self.napp.controller.switches = {'00:00:00:00:00:00:00:01': |
|
902 | mock_switch} |
||
903 | |||
904 | 1 | key = 'A' |
|
905 | 1 | url = f"{self.base_endpoint}/interfaces/{interface_id}/metadata/{key}" |
|
906 | 1 | response = await self.api_client.delete(url) |
|
907 | 1 | assert response.status_code == 200 |
|
908 | |||
909 | 1 | del_key_mock = self.napp.topo_controller.delete_interface_metadata_key |
|
910 | 1 | del_key_mock.assert_called_once_with(interface_id, key) |
|
911 | |||
912 | # fail case switch not found |
||
913 | 1 | key = 'A' |
|
914 | 1 | interface_id = '00:00:00:00:00:00:00:02:1' |
|
915 | 1 | url = f"{self.base_endpoint}/interfaces/{interface_id}/metadata/{key}" |
|
916 | 1 | response = await self.api_client.delete(url) |
|
917 | assert response.status_code == 404 |
||
918 | |||
919 | # fail case interface not found |
||
920 | key = 'A' |
||
921 | interface_id = '00:00:00:00:00:00:00:01:2' |
||
922 | url = f"{self.base_endpoint}/interfaces/{interface_id}/metadata/{key}" |
||
923 | response = await self.api_client.delete(url) |
||
924 | assert response.status_code == 404 |
||
925 | |||
926 | # fail case metadata not found |
||
927 | key = 'B' |
||
928 | interface_id = '00:00:00:00:00:00:00:01:1' |
||
929 | url = f"{self.base_endpoint}/interfaces/{interface_id}/metadata/{key}" |
||
930 | response = await self.api_client.delete(url) |
||
931 | assert response.status_code == 404 |
||
932 | |||
933 | 1 | @patch('napps.kytos.topology.main.Main.notify_link_enabled_state') |
|
934 | 1 | @patch('napps.kytos.topology.main.Main.notify_topology_update') |
|
935 | 1 | async def test_enable_link(self, mock_notify_topo, mock_noti_link): |
|
936 | """Test enable_link.""" |
||
937 | 1 | mock_link = MagicMock(Link) |
|
938 | 1 | mock_link.link_lock = MagicMock() |
|
939 | 1 | link_id = "1" |
|
940 | 1 | mock_link.id = link_id |
|
941 | 1 | mock_link.is_enabled = lambda: False |
|
942 | 1 | mock_link.endpoint_a = MagicMock(is_enabled=lambda: True) |
|
943 | 1 | mock_link.endpoint_b = MagicMock(is_enabled=lambda: True) |
|
944 | 1 | self.napp.controller.links = {'1': mock_link} |
|
945 | |||
946 | # 409, endpoint is/are disabled |
||
947 | 1 | mock_link.endpoint_a.is_enabled = lambda: False |
|
948 | 1 | mock_link.endpoint_b.is_enabled = lambda: False |
|
949 | 1 | endpoint = f"{self.base_endpoint}/links/{link_id}/enable" |
|
950 | 1 | response = await self.api_client.post(endpoint) |
|
951 | assert response.status_code == 409 |
||
952 | |||
953 | 1 | mock_link.endpoint_a.is_enabled = lambda: True |
|
954 | endpoint = f"{self.base_endpoint}/links/{link_id}/enable" |
||
955 | response = await self.api_client.post(endpoint) |
||
956 | assert response.status_code == 409 |
||
957 | |||
958 | # Success |
||
959 | 1 | mock_link.endpoint_b.is_enabled = lambda: True |
|
960 | endpoint = f"{self.base_endpoint}/links/{link_id}/enable" |
||
961 | response = await self.api_client.post(endpoint) |
||
962 | 1 | assert response.status_code == 201 |
|
963 | 1 | assert mock_noti_link.call_count == 1 |
|
964 | 1 | assert mock_noti_link.call_args[0][0] == mock_link |
|
965 | 1 | assert mock_noti_link.call_args[0][1] == "enabled" |
|
966 | 1 | mock_notify_topo.assert_called() |
|
967 | |||
968 | # 404, link not found |
||
969 | 1 | link_id = "2" |
|
970 | 1 | endpoint = f"{self.base_endpoint}/links/{link_id}/enable" |
|
971 | 1 | response = await self.api_client.post(endpoint) |
|
972 | assert response.status_code == 404 |
||
973 | assert mock_noti_link.call_count == 1 |
||
974 | |||
975 | 1 | @patch('napps.kytos.topology.main.Main.notify_link_enabled_state') |
|
976 | 1 | @patch('napps.kytos.topology.main.Main.notify_topology_update') |
|
977 | 1 | async def test_disable_link(self, mock_notify_topo, mock_notify): |
|
978 | """Test disable_link.""" |
||
979 | 1 | mock_link = MagicMock(Link) |
|
980 | 1 | mock_link.link_lock = MagicMock() |
|
981 | 1 | mock_link.is_enabled = lambda: True |
|
982 | 1 | self.napp.controller.links = {'1': mock_link} |
|
983 | |||
984 | 1 | link_id = "1" |
|
985 | 1 | endpoint = f"{self.base_endpoint}/links/{link_id}/disable" |
|
986 | 1 | response = await self.api_client.post(endpoint) |
|
987 | 1 | assert response.status_code == 201 |
|
988 | 1 | assert mock_notify_topo.call_count == 1 |
|
989 | 1 | assert mock_notify.call_count == 1 |
|
990 | 1 | assert mock_notify.call_args[0][0] == mock_link |
|
991 | 1 | assert mock_notify.call_args[0][1] == "disabled" |
|
992 | 1 | assert mock_link.disable.call_count == 1 |
|
993 | 1 | assert self.napp.topo_controller.disable_link.call_count == 1 |
|
994 | |||
995 | # fail case |
||
996 | 1 | link_id = "2" |
|
997 | 1 | endpoint = f"{self.base_endpoint}/links/{link_id}/disable" |
|
998 | 1 | response = await self.api_client.post(endpoint) |
|
999 | assert response.status_code == 404 |
||
1000 | |||
1001 | 1 | def test_handle_lldp_status_updated(self): |
|
1002 | """Test handle_lldp_status_updated.""" |
||
1003 | 1 | event = MagicMock() |
|
1004 | 1 | self.napp.controller.buffers.app.put = MagicMock() |
|
1005 | |||
1006 | 1 | dpid_a = "00:00:00:00:00:00:00:01" |
|
1007 | 1 | dpid_b = "00:00:00:00:00:00:00:02" |
|
1008 | 1 | dpids = [dpid_a, dpid_b] |
|
1009 | 1 | interface_ids = [f"{dpid}:1" for dpid in dpids] |
|
1010 | |||
1011 | 1 | mock_switch_a = get_switch_mock(dpid_a, 0x04) |
|
1012 | 1 | mock_switch_b = get_switch_mock(dpid_b, 0x04) |
|
1013 | 1 | self.napp.controller.switches = {dpid_a: mock_switch_a, |
|
1014 | dpid_b: mock_switch_b} |
||
1015 | |||
1016 | 1 | event.content = {"interface_ids": interface_ids, "state": "disabled"} |
|
1017 | 1 | self.napp.handle_lldp_status_updated(event) |
|
1018 | |||
1019 | 1 | mock_put = self.napp.controller.buffers.app.put |
|
1020 | 1 | assert mock_put.call_count == len(interface_ids) |
|
1021 | |||
1022 | 1 | def test_handle_topo_controller_upsert_switch(self): |
|
1023 | """Test handle_topo_controller_upsert_switch.""" |
||
1024 | 1 | event = MagicMock() |
|
1025 | 1 | self.napp.handle_topo_controller_upsert_switch(event) |
|
1026 | 1 | mock = self.napp.topo_controller.upsert_switch |
|
1027 | 1 | mock.assert_called_with(event.id, event.as_dict()) |
|
1028 | |||
1029 | 1 | async def test_get_link_metadata(self): |
|
1030 | """Test get_link_metadata.""" |
||
1031 | 1 | mock_link = MagicMock(Link) |
|
1032 | 1 | mock_link.link_lock = MagicMock() |
|
1033 | 1 | mock_link.metadata = "A" |
|
1034 | 1 | self.napp.controller.links = {'1': mock_link} |
|
1035 | 1 | msg_success = {"metadata": "A"} |
|
1036 | |||
1037 | 1 | link_id = "1" |
|
1038 | 1 | endpoint = f"{self.base_endpoint}/links/{link_id}/metadata" |
|
1039 | 1 | response = await self.api_client.get(endpoint) |
|
1040 | 1 | assert response.status_code == 200 |
|
1041 | 1 | assert msg_success == response.json() |
|
1042 | |||
1043 | # fail case |
||
1044 | 1 | link_id = "2" |
|
1045 | 1 | endpoint = f"{self.base_endpoint}/links/{link_id}/metadata" |
|
1046 | 1 | response = await self.api_client.get(endpoint) |
|
1047 | assert response.status_code == 404 |
||
1048 | |||
1049 | 1 | @patch('napps.kytos.topology.main.Main.notify_topology_update') |
|
1050 | 1 | @patch('napps.kytos.topology.main.Main.notify_metadata_changes') |
|
1051 | 1 | async def test_add_link_metadata( |
|
1052 | self, |
||
1053 | mock_metadata_changes, |
||
1054 | mock_topology_update |
||
1055 | ): |
||
1056 | """Test add_link_metadata.""" |
||
1057 | 1 | self.napp.controller.loop = asyncio.get_running_loop() |
|
1058 | 1 | mock_link = MagicMock(Link) |
|
1059 | 1 | mock_link.link_lock = MagicMock() |
|
1060 | 1 | mock_link.metadata = "A" |
|
1061 | 1 | self.napp.controller.links = {'1': mock_link} |
|
1062 | 1 | payload = {"metadata": "A"} |
|
1063 | 1 | link_id = 1 |
|
1064 | |||
1065 | 1 | endpoint = f"{self.base_endpoint}/links/{link_id}/metadata" |
|
1066 | 1 | response = await self.api_client.post(endpoint, json=payload) |
|
1067 | 1 | assert response.status_code == 201 |
|
1068 | 1 | mock_metadata_changes.assert_called() |
|
1069 | 1 | mock_topology_update.assert_called() |
|
1070 | |||
1071 | # fail case |
||
1072 | 1 | link_id = 2 |
|
1073 | 1 | endpoint = f"{self.base_endpoint}/links/{link_id}/metadata" |
|
1074 | 1 | response = await self.api_client.post(endpoint, json=payload) |
|
1075 | assert response.status_code == 404 |
||
1076 | |||
1077 | 1 | async def test_add_link_metadata_wrong_format(self): |
|
1078 | """Test add_link_metadata_wrong_format.""" |
||
1079 | 1 | self.napp.controller.loop = asyncio.get_running_loop() |
|
1080 | 1 | link_id = 'cf0f4071be426b3f745027f5d22' |
|
1081 | 1 | payload = "A" |
|
1082 | 1 | endpoint = f"{self.base_endpoint}/links/{link_id}/metadata" |
|
1083 | 1 | response = await self.api_client.post(endpoint, json=payload) |
|
1084 | assert response.status_code == 400 |
||
1085 | |||
1086 | payload = None |
||
1087 | response = await self.api_client.post(endpoint, json=payload) |
||
1088 | assert response.status_code == 415 |
||
1089 | |||
1090 | 1 | @patch('napps.kytos.topology.main.Main.notify_topology_update') |
|
1091 | 1 | @patch('napps.kytos.topology.main.Main.notify_metadata_changes') |
|
1092 | 1 | async def test_delete_link_metadata( |
|
1093 | self, |
||
1094 | mock_metadata_changes, |
||
1095 | mock_topology_update |
||
1096 | ): |
||
1097 | """Test delete_link_metadata.""" |
||
1098 | 1 | mock_link = MagicMock(Link) |
|
1099 | 1 | mock_link.link_lock = MagicMock() |
|
1100 | 1 | mock_link.metadata = {"A": "A"} |
|
1101 | 1 | mock_link.remove_metadata.side_effect = [True, False] |
|
1102 | 1 | self.napp.controller.links = {'1': mock_link} |
|
1103 | |||
1104 | 1 | link_id = 1 |
|
1105 | 1 | key = 'A' |
|
1106 | 1 | endpoint = f"{self.base_endpoint}/links/{link_id}/metadata/{key}" |
|
1107 | 1 | response = await self.api_client.delete(endpoint) |
|
1108 | 1 | assert response.status_code == 200 |
|
1109 | 1 | del_mock = self.napp.topo_controller.delete_link_metadata_key |
|
1110 | 1 | del_mock.assert_called_once_with(mock_link.id, key) |
|
1111 | 1 | mock_metadata_changes.assert_called() |
|
1112 | 1 | mock_topology_update.assert_called() |
|
1113 | |||
1114 | # fail case link not found |
||
1115 | 1 | link_id = 2 |
|
1116 | 1 | key = 'A' |
|
1117 | 1 | endpoint = f"{self.base_endpoint}/links/{link_id}/metadata/{key}" |
|
1118 | 1 | response = await self.api_client.delete(endpoint) |
|
1119 | assert response.status_code == 404 |
||
1120 | |||
1121 | # fail case metadata not found |
||
1122 | link_id = 1 |
||
1123 | key = 'B' |
||
1124 | endpoint = f"{self.base_endpoint}/links/{link_id}/metadata/{key}" |
||
1125 | response = await self.api_client.delete(endpoint) |
||
1126 | assert response.status_code == 404 |
||
1127 | |||
1128 | 1 | @patch('napps.kytos.topology.main.Main.notify_topology_update') |
|
1129 | 1 | def test_handle_new_switch(self, mock_notify_topology_update): |
|
1130 | """Test handle_new_switch.""" |
||
1131 | 1 | mock_event = MagicMock() |
|
1132 | 1 | mock_switch = create_autospec(Switch) |
|
1133 | 1 | mock_event.content['switch'] = mock_switch |
|
1134 | 1 | self.napp.handle_new_switch(mock_event) |
|
1135 | 1 | mock = self.napp.topo_controller.upsert_switch |
|
1136 | 1 | mock.assert_called_once_with(mock_event.content['switch'].id, |
|
1137 | mock_event.content['switch'].as_dict()) |
||
1138 | 1 | mock_notify_topology_update.assert_called() |
|
1139 | |||
1140 | 1 | @patch('napps.kytos.topology.main.Main.notify_topology_update') |
|
1141 | 1 | def test_handle_connection_lost(self, mock_notify_topology_update): |
|
1142 | """Test handle connection_lost.""" |
||
1143 | 1 | mock_event = MagicMock() |
|
1144 | 1 | mock_switch = create_autospec(Switch) |
|
1145 | 1 | mock_switch.return_value = True |
|
1146 | 1 | mock_event.content['source'] = mock_switch |
|
1147 | 1 | self.napp.handle_connection_lost(mock_event) |
|
1148 | 1 | mock_notify_topology_update.assert_called() |
|
1149 | |||
1150 | 1 | @patch('napps.kytos.topology.main.Main.handle_interface_link_down') |
|
1151 | 1 | @patch('napps.kytos.topology.main.Main.handle_interface_link_up') |
|
1152 | 1 | def test_handle_interface_created(self, mock_link_up, mock_link_down): |
|
1153 | """Test handle_interface_created.""" |
||
1154 | 1 | mock_event = MagicMock() |
|
1155 | 1 | mock_interface = create_autospec(Interface) |
|
1156 | 1 | mock_interface.id = "1" |
|
1157 | 1 | mock_event.content = {'interface': mock_interface} |
|
1158 | 1 | self.napp.handle_interface_created(mock_event) |
|
1159 | 1 | mock_link_up.assert_called() |
|
1160 | 1 | mock_link_down.assert_not_called() |
|
1161 | |||
1162 | 1 | @patch('napps.kytos.topology.main.Main.handle_interface_link_down') |
|
1163 | 1 | @patch('napps.kytos.topology.main.Main.handle_interface_link_up') |
|
1164 | 1 | def test_handle_interface_created_inactive(self, mock_link_up, |
|
1165 | mock_link_down): |
||
1166 | """Test handle_interface_created inactive.""" |
||
1167 | 1 | mock_event = MagicMock() |
|
1168 | 1 | mock_interface = create_autospec(Interface) |
|
1169 | 1 | mock_interface.id = "1" |
|
1170 | 1 | mock_event.content = {'interface': mock_interface} |
|
1171 | 1 | mock_interface.is_active.return_value = False |
|
1172 | 1 | self.napp.handle_interface_created(mock_event) |
|
1173 | 1 | mock_link_up.assert_not_called() |
|
1174 | 1 | mock_link_down.assert_called() |
|
1175 | |||
1176 | 1 | def test_handle_interfaces_created(self): |
|
1177 | """Test handle_interfaces_created.""" |
||
1178 | 1 | buffers_app_mock = MagicMock() |
|
1179 | 1 | self.napp.controller.buffers.app = buffers_app_mock |
|
1180 | 1 | mock_switch = create_autospec(Switch) |
|
1181 | 1 | mock_event = MagicMock() |
|
1182 | 1 | mock_interface = create_autospec(Interface) |
|
1183 | 1 | mock_interface.id = "1" |
|
1184 | 1 | mock_interface.switch = mock_switch |
|
1185 | 1 | mock_interface_two = create_autospec(Interface) |
|
1186 | 1 | mock_interface_two.id = "2" |
|
1187 | 1 | mock_event.content = {'interfaces': [mock_interface, |
|
1188 | mock_interface_two]} |
||
1189 | 1 | self.napp.handle_interfaces_created(mock_event) |
|
1190 | 1 | upsert_mock = self.napp.topo_controller.upsert_switch |
|
1191 | 1 | upsert_mock.assert_called_with(mock_switch.id, mock_switch.as_dict()) |
|
1192 | 1 | assert self.napp.controller.buffers.app.put.call_count == 2 |
|
1193 | |||
1194 | 1 | @patch('napps.kytos.topology.main.Main.handle_interface_link_down') |
|
1195 | 1 | def test_handle_interface_down(self, mock_handle_interface_link_down): |
|
1196 | """Test handle interface down.""" |
||
1197 | 1 | mock_event = MagicMock() |
|
1198 | 1 | mock_interface = create_autospec(Interface) |
|
1199 | 1 | mock_event.content['interface'] = mock_interface |
|
1200 | 1 | self.napp.handle_interface_down(mock_event) |
|
1201 | 1 | mock_handle_interface_link_down.assert_called() |
|
1202 | |||
1203 | 1 | @patch('napps.kytos.topology.main.Main.handle_interface_down') |
|
1204 | 1 | def test_interface_deleted(self, mock_handle_interface_link_down): |
|
1205 | """Test interface deleted.""" |
||
1206 | 1 | mock_event = MagicMock() |
|
1207 | 1 | self.napp.handle_interface_deleted(mock_event) |
|
1208 | 1 | mock_handle_interface_link_down.assert_called() |
|
1209 | |||
1210 | 1 | @patch('napps.kytos.topology.main.Main.notify_topology_update') |
|
1211 | 1 | def test_interface_link_up(self, mock_notify_topology_update): |
|
1212 | """Test interface link_up.""" |
||
1213 | 1 | self.napp.controller.buffers.app.put = MagicMock() |
|
1214 | |||
1215 | 1 | tnow = time.time() |
|
1216 | 1 | mock_switch_a = create_autospec(Switch) |
|
1217 | 1 | mock_switch_a.is_active.return_value = True |
|
1218 | 1 | mock_switch_b = create_autospec(Switch) |
|
1219 | 1 | mock_switch_b.is_active.return_value = True |
|
1220 | 1 | mock_interface_a = create_autospec(Interface) |
|
1221 | 1 | mock_interface_a.switch = mock_switch_a |
|
1222 | 1 | mock_interface_a.is_active.return_value = False |
|
1223 | 1 | mock_interface_b = create_autospec(Interface) |
|
1224 | 1 | mock_interface_b.switch = mock_switch_b |
|
1225 | 1 | mock_interface_b.is_active.return_value = True |
|
1226 | 1 | mock_link = create_autospec(Link) |
|
1227 | 1 | mock_link.link_lock = MagicMock() |
|
1228 | 1 | mock_link.get_metadata.return_value = tnow |
|
1229 | 1 | mock_link.is_active.return_value = False |
|
1230 | 1 | mock_link.endpoint_a = mock_interface_a |
|
1231 | 1 | mock_link.endpoint_b = mock_interface_b |
|
1232 | 1 | mock_link.status = EntityStatus.UP |
|
1233 | 1 | mock_interface_a.link = mock_link |
|
1234 | 1 | mock_interface_b.link = mock_link |
|
1235 | 1 | event = KytosEvent("kytos.of_core.switch.interface.down") |
|
1236 | 1 | self.napp.handle_interface_link_up(mock_interface_a, event) |
|
1237 | 1 | mock_notify_topology_update.assert_called() |
|
1238 | 1 | assert mock_link.id in self.napp.link_status_change |
|
1239 | 1 | mock_link.activate.assert_called() |
|
1240 | 1 | self.napp.controller.buffers.app.put.assert_not_called() |
|
1241 | |||
1242 | 1 | mock_interface_a.is_active.return_value = True |
|
1243 | 1 | event = KytosEvent("kytos.of_core.switch.interface.down") |
|
1244 | 1 | self.napp.handle_interface_link_up(mock_interface_a, event) |
|
1245 | |||
1246 | 1 | assert mock_link.id in self.napp.link_status_change |
|
1247 | 1 | link_status_info = self.napp.link_status_change[mock_link.id] |
|
1248 | 1 | mock_link.activate.assert_called() |
|
1249 | 1 | assert self.napp.controller.buffers.app.put.call_count == 1 |
|
1250 | 1 | ev = "kytos/topology.notify_link_up_if_status" |
|
1251 | 1 | assert self.napp.controller.buffers.app.put.call_args[0][0].name == ev |
|
1252 | |||
1253 | 1 | mock_link.is_active.return_value = True |
|
1254 | 1 | orig_change_time = link_status_info["last_status_change"] |
|
1255 | |||
1256 | 1 | self.napp.handle_interface_link_up(mock_interface_a, event) |
|
1257 | |||
1258 | 1 | link_status_info = self.napp.link_status_change[mock_link.id] |
|
1259 | 1 | new_change_time = link_status_info["last_status_change"] |
|
1260 | 1 | assert orig_change_time == new_change_time |
|
1261 | |||
1262 | 1 | @patch('napps.kytos.topology.main.Main.notify_topology_update') |
|
1263 | 1 | @patch('napps.kytos.topology.main.Main.notify_link_status_change') |
|
1264 | 1 | def test_interface_link_down(self, *args): |
|
1265 | """Test interface link down.""" |
||
1266 | 1 | mock_status_change, mock_topology_update = args |
|
1267 | |||
1268 | 1 | mock_interface = create_autospec(Interface) |
|
1269 | 1 | mock_link = create_autospec(Link) |
|
1270 | 1 | mock_link.link_lock = MagicMock() |
|
1271 | 1 | mock_link.is_active.return_value = True |
|
1272 | 1 | mock_interface.link = mock_link |
|
1273 | 1 | event = KytosEvent("kytos.of_core.switch.interface.link_up") |
|
1274 | 1 | self.napp.handle_interface_link_down(mock_interface, event) |
|
1275 | 1 | mock_topology_update.assert_called() |
|
1276 | 1 | mock_status_change.assert_called() |
|
1277 | |||
1278 | 1 | @patch('napps.kytos.topology.main.Main.notify_topology_update') |
|
1279 | 1 | @patch('napps.kytos.topology.main.Main.notify_link_status_change') |
|
1280 | 1 | def test_interface_link_down_unordered_event(self, *args): |
|
1281 | """Test interface link down unordered event.""" |
||
1282 | 1 | (mock_status_change, mock_topology_update) = args |
|
1283 | |||
1284 | 1 | mock_interface = create_autospec(Interface) |
|
1285 | 1 | mock_interface.id = "1" |
|
1286 | 1 | event_2 = KytosEvent("kytos.of_core.switch.interface.down") |
|
1287 | 1 | event_1 = KytosEvent("kytos.of_core.switch.interface.up") |
|
1288 | 1 | assert event_1.timestamp > event_2.timestamp |
|
1289 | 1 | self.napp._intfs_updated_at[mock_interface.id] = event_1.timestamp |
|
1290 | 1 | self.napp.handle_interface_link_down(mock_interface, event_2) |
|
1291 | 1 | mock_topology_update.assert_not_called() |
|
1292 | 1 | mock_status_change.assert_not_called() |
|
1293 | |||
1294 | 1 | @patch('napps.kytos.topology.main.Main.notify_topology_update') |
|
1295 | 1 | @patch('napps.kytos.topology.main.Main.notify_link_status_change') |
|
1296 | 1 | def test_interface_link_up_unordered_event(self, *args): |
|
1297 | """Test interface link up unordered event.""" |
||
1298 | 1 | (mock_status_change, mock_topology_update) = args |
|
1299 | |||
1300 | 1 | mock_interface = create_autospec(Interface) |
|
1301 | 1 | mock_interface.id = "1" |
|
1302 | 1 | event_2 = KytosEvent("kytos.of_core.switch.interface.up") |
|
1303 | 1 | event_1 = KytosEvent("kytos.of_core.switch.interface.down") |
|
1304 | 1 | assert event_1.timestamp > event_2.timestamp |
|
1305 | 1 | self.napp._intfs_updated_at[mock_interface.id] = event_1.timestamp |
|
1306 | 1 | self.napp.handle_interface_link_up(mock_interface, event_2) |
|
1307 | 1 | mock_topology_update.assert_not_called() |
|
1308 | 1 | mock_status_change.assert_not_called() |
|
1309 | |||
1310 | 1 | @patch('napps.kytos.topology.main.Main.notify_topology_update') |
|
1311 | 1 | @patch('napps.kytos.topology.main.Main.notify_link_status_change') |
|
1312 | 1 | def test_handle_link_down(self, *args): |
|
1313 | """Test interface link down.""" |
||
1314 | 1 | (mock_status_change, mock_topology_update) = args |
|
1315 | |||
1316 | 1 | mock_interface = create_autospec(Interface) |
|
1317 | 1 | mock_link = create_autospec(Link) |
|
1318 | 1 | mock_link.link_lock = MagicMock() |
|
1319 | 1 | mock_link.is_active.return_value = True |
|
1320 | 1 | mock_interface.link = mock_link |
|
1321 | 1 | self.napp.handle_link_down(mock_interface) |
|
1322 | 1 | mock_interface.deactivate.assert_not_called() |
|
1323 | 1 | mock_link.deactivate.assert_called() |
|
1324 | 1 | assert mock_topology_update.call_count == 1 |
|
1325 | 1 | mock_status_change.assert_called() |
|
1326 | |||
1327 | 1 | @patch('napps.kytos.topology.main.Main.notify_topology_update') |
|
1328 | 1 | def test_handle_link_down_not_active(self, args): |
|
1329 | """Test interface link down with link not active.""" |
||
1330 | 1 | mock_topology_update = args |
|
1331 | 1 | self.napp.controller.buffers.app.put = MagicMock() |
|
1332 | |||
1333 | 1 | mock_interface = create_autospec(Interface) |
|
1334 | 1 | mock_link = create_autospec(Link) |
|
1335 | 1 | mock_link.link_lock = MagicMock() |
|
1336 | 1 | mock_link.is_active.return_value = False |
|
1337 | 1 | mock_link.get_metadata.return_value = False |
|
1338 | 1 | mock_interface.link = mock_link |
|
1339 | 1 | self.napp.link_up = set() |
|
1340 | 1 | self.napp.link_status_change[mock_link.id] = {} |
|
1341 | 1 | self.napp.handle_link_down(mock_interface) |
|
1342 | 1 | mock_topology_update.assert_called() |
|
1343 | 1 | self.napp.controller.buffers.app.put.assert_not_called() |
|
1344 | |||
1345 | 1 | @patch('napps.kytos.topology.main.Main.notify_topology_update') |
|
1346 | 1 | @patch('napps.kytos.topology.main.Main.notify_link_status_change') |
|
1347 | 1 | def test_handle_link_down_not_active_last_status(self, *args): |
|
1348 | """Test interface link down with link not active.""" |
||
1349 | 1 | (mock_status_change, mock_topology_update) = args |
|
1350 | |||
1351 | 1 | mock_interface = create_autospec(Interface) |
|
1352 | 1 | mock_link = create_autospec(Link) |
|
1353 | 1 | mock_link.link_lock = MagicMock() |
|
1354 | 1 | mock_link.is_active.return_value = False |
|
1355 | 1 | mock_link.get_metadata.return_value = True |
|
1356 | 1 | mock_interface.link = mock_link |
|
1357 | 1 | self.napp.handle_link_down(mock_interface) |
|
1358 | 1 | mock_topology_update.assert_called() |
|
1359 | 1 | mock_status_change.assert_called() |
|
1360 | |||
1361 | 1 | @patch('napps.kytos.topology.main.Main.notify_topology_update') |
|
1362 | 1 | def test_handle_link_up(self, mock_notify_topology_update): |
|
1363 | """Test handle link up.""" |
||
1364 | 1 | mock_switch_a = create_autospec(Switch) |
|
1365 | 1 | mock_switch_a.is_active.return_value = True |
|
1366 | 1 | mock_interface = create_autospec(Interface) |
|
1367 | 1 | mock_interface.switch = mock_switch_a |
|
1368 | 1 | mock_interface.is_active.return_value = True |
|
1369 | 1 | mock_link = MagicMock(status=EntityStatus.UP) |
|
1370 | 1 | mock_link.is_active.return_value = True |
|
1371 | 1 | mock_interface.link = mock_link |
|
1372 | 1 | self.napp.handle_link_up(mock_interface) |
|
1373 | 1 | mock_interface.activate.assert_not_called() |
|
1374 | 1 | mock_notify_topology_update.assert_called() |
|
1375 | 1 | assert self.napp.controller.buffers.app.put.call_count == 2 |
|
1376 | 1 | ev = "kytos/topology.notify_link_up_if_status" |
|
1377 | 1 | assert self.napp.controller.buffers.app.put.call_args[0][0].name == ev |
|
1378 | |||
1379 | 1 | @patch('time.sleep') |
|
1380 | 1 | @patch('napps.kytos.topology.main.Main.notify_topology_update') |
|
1381 | 1 | @patch('napps.kytos.topology.main.Main.notify_link_status_change') |
|
1382 | 1 | def test_handle_link_up_intf_down(self, *args): |
|
1383 | """Test handle link up but one intf down.""" |
||
1384 | 1 | (mock_status_change, mock_topology_update, _) = args |
|
1385 | |||
1386 | 1 | mock_switch = create_autospec(Switch) |
|
1387 | 1 | mock_interface = create_autospec(Interface) |
|
1388 | 1 | mock_interface.switch = mock_switch |
|
1389 | 1 | mock_link = MagicMock() |
|
1390 | 1 | mock_link.endpoint_a.is_active.return_value = False |
|
1391 | 1 | mock_link.is_active.return_value = False |
|
1392 | 1 | mock_interface.link = mock_link |
|
1393 | 1 | self.napp.handle_link_up(mock_interface) |
|
1394 | 1 | mock_interface.activate.assert_not_called() |
|
1395 | 1 | assert mock_topology_update.call_count == 1 |
|
1396 | 1 | mock_status_change.assert_not_called() |
|
1397 | |||
1398 | 1 | @patch('napps.kytos.topology.main.Main.notify_link_up_if_status') |
|
1399 | 1 | def test_add_links(self, mock_notify_link_up_if_status): |
|
1400 | """Test add_links.""" |
||
1401 | 1 | mock_link = MagicMock() |
|
1402 | 1 | self.napp.controller.get_link_or_create = MagicMock() |
|
1403 | 1 | mock_get_link_or_create = self.napp.controller.get_link_or_create |
|
1404 | 1 | mock_get_link_or_create.return_value = (mock_link, True) |
|
1405 | 1 | mock_event = MagicMock() |
|
1406 | 1 | mock_intf_a = MagicMock() |
|
1407 | 1 | mock_intf_b = MagicMock() |
|
1408 | 1 | mock_event.content = { |
|
1409 | "interface_a": mock_intf_a, |
||
1410 | "interface_b": mock_intf_b |
||
1411 | } |
||
1412 | 1 | self.napp.add_links(mock_event) |
|
1413 | 1 | assert mock_link.id in self.napp.link_status_change |
|
1414 | 1 | mock_get_link_or_create.assert_called() |
|
1415 | 1 | mock_notify_link_up_if_status.assert_called() |
|
1416 | |||
1417 | 1 | def test_notify_switch_enabled(self): |
|
1418 | """Test notify switch enabled.""" |
||
1419 | 1 | dpid = "00:00:00:00:00:00:00:01" |
|
1420 | 1 | mock_buffers_put = MagicMock() |
|
1421 | 1 | self.napp.controller.buffers.app.put = mock_buffers_put |
|
1422 | 1 | self.napp.notify_switch_enabled(dpid) |
|
1423 | 1 | mock_buffers_put.assert_called() |
|
1424 | |||
1425 | 1 | def test_notify_switch_disabled(self): |
|
1426 | """Test notify switch disabled.""" |
||
1427 | 1 | dpid = "00:00:00:00:00:00:00:01" |
|
1428 | 1 | mock_buffers_put = MagicMock() |
|
1429 | 1 | self.napp.controller.buffers.app.put = mock_buffers_put |
|
1430 | 1 | self.napp.notify_switch_disabled(dpid) |
|
1431 | 1 | mock_buffers_put.assert_called() |
|
1432 | |||
1433 | 1 | def test_notify_topology_update(self): |
|
1434 | """Test notify_topology_update.""" |
||
1435 | 1 | mock_buffers_put = MagicMock() |
|
1436 | 1 | self.napp.controller.buffers.app.put = mock_buffers_put |
|
1437 | 1 | self.napp.notify_topology_update() |
|
1438 | 1 | mock_buffers_put.assert_called() |
|
1439 | |||
1440 | 1 | def test_notify_link_status_change(self): |
|
1441 | """Test notify link status change.""" |
||
1442 | 1 | mock_buffers_put = MagicMock() |
|
1443 | 1 | self.napp.controller.buffers.app.put = mock_buffers_put |
|
1444 | 1 | mock_link = create_autospec(Link) |
|
1445 | 1 | mock_link.id = 'test_link' |
|
1446 | 1 | mock_link.status_reason = frozenset() |
|
1447 | 1 | mock_link.status = EntityStatus.UP |
|
1448 | |||
1449 | # Check when switching to up |
||
1450 | 1 | self.napp.notify_link_status_change(mock_link, 'test') |
|
1451 | 1 | assert mock_buffers_put.call_count == 1 |
|
1452 | 1 | args, _ = mock_buffers_put.call_args |
|
1453 | 1 | event = args[0] |
|
1454 | 1 | assert event.content['link'] is mock_link |
|
1455 | 1 | assert event.content['reason'] == 'test' |
|
1456 | 1 | assert event.name == 'kytos/topology.link_up' |
|
1457 | |||
1458 | # Check result when no change |
||
1459 | 1 | self.napp.notify_link_status_change(mock_link, 'test2') |
|
1460 | 1 | assert mock_buffers_put.call_count == 1 |
|
1461 | |||
1462 | # Check when switching to down |
||
1463 | 1 | mock_link.status_reason = frozenset({'disabled'}) |
|
1464 | 1 | mock_link.status = EntityStatus.DOWN |
|
1465 | 1 | self.napp.notify_link_status_change(mock_link, 'test3') |
|
1466 | 1 | assert mock_buffers_put.call_count == 2 |
|
1467 | 1 | args, _ = mock_buffers_put.call_args |
|
1468 | 1 | event = args[0] |
|
1469 | 1 | assert event.content['link'] is mock_link |
|
1470 | 1 | assert event.content['reason'] == 'test3' |
|
1471 | 1 | assert event.name == 'kytos/topology.link_down' |
|
1472 | |||
1473 | 1 | def test_notify_metadata_changes(self): |
|
1474 | """Test notify metadata changes.""" |
||
1475 | 1 | mock_buffers_put = MagicMock() |
|
1476 | 1 | self.napp.controller.buffers.app.put = mock_buffers_put |
|
1477 | 1 | count = 0 |
|
1478 | 1 | for spec in [Switch, Interface, Link]: |
|
1479 | 1 | mock_obj = create_autospec(spec) |
|
1480 | 1 | mock_obj.metadata = {"some_key": "some_value"} |
|
1481 | 1 | self.napp.notify_metadata_changes(mock_obj, 'added') |
|
1482 | 1 | assert mock_buffers_put.call_count == count+1 |
|
1483 | 1 | count += 1 |
|
1484 | 1 | with pytest.raises(ValueError): |
|
1485 | 1 | self.napp.notify_metadata_changes(MagicMock(), 'added') |
|
1486 | |||
1487 | 1 | def test_notify_port_created(self): |
|
1488 | """Test notify port created.""" |
||
1489 | 1 | mock_buffers_put = MagicMock() |
|
1490 | 1 | self.napp.controller.buffers.app.put = mock_buffers_put |
|
1491 | 1 | event = KytosEvent("some_event") |
|
1492 | 1 | expected_name = "kytos/topology.port.created" |
|
1493 | 1 | self.napp.notify_port_created(event) |
|
1494 | 1 | assert mock_buffers_put.call_count == 1 |
|
1495 | 1 | assert mock_buffers_put.call_args_list[0][0][0].name == expected_name |
|
1496 | |||
1497 | 1 | def test_handle_link_liveness_disabled(self) -> None: |
|
1498 | """Test handle_link_liveness_disabled.""" |
||
1499 | 1 | interfaces = [MagicMock(id=f"intf{n}") for n in range(4)] |
|
1500 | 1 | links = { |
|
1501 | "link1": MagicMock(id="link1", |
||
1502 | endpoint_a=interfaces[0], |
||
1503 | endpoint_b=interfaces[1]), |
||
1504 | "link2": MagicMock(id="link2", |
||
1505 | endpoint_a=interfaces[2], |
||
1506 | endpoint_b=interfaces[3]), |
||
1507 | } |
||
1508 | 1 | self.napp.controller.links = links |
|
1509 | 1 | self.napp.notify_topology_update = MagicMock() |
|
1510 | 1 | self.napp.notify_link_status_change = MagicMock() |
|
1511 | |||
1512 | 1 | self.napp.handle_link_liveness_disabled(interfaces) |
|
1513 | |||
1514 | 1 | assert self.napp.notify_topology_update.call_count == 1 |
|
1515 | 1 | assert self.napp.notify_link_status_change.call_count == len(links) |
|
1516 | |||
1517 | 1 | def test_link_status_hook_link_up_timer(self) -> None: |
|
1518 | """Test status hook link up timer.""" |
||
1519 | 1 | last_change = time.time() - self.napp.link_up_timer + 5 |
|
1520 | 1 | link = MagicMock(metadata={"last_status_change": last_change}) |
|
1521 | 1 | self.napp.link_status_change[link.id] = { |
|
1522 | "last_status_change": last_change, |
||
1523 | } |
||
1524 | 1 | link.is_active.return_value = True |
|
1525 | 1 | link.is_enabled.return_value = True |
|
1526 | 1 | res = self.napp.link_status_hook_link_up_timer(link) |
|
1527 | 1 | assert res == EntityStatus.DOWN |
|
1528 | |||
1529 | 1 | last_change = time.time() - self.napp.link_up_timer |
|
1530 | 1 | self.napp.link_status_change[link.id] = { |
|
1531 | "last_status_change": last_change, |
||
1532 | } |
||
1533 | 1 | res = self.napp.link_status_hook_link_up_timer(link) |
|
1534 | 1 | assert res is None |
|
1535 | |||
1536 | 1 | @patch('napps.kytos.topology.main.Main.notify_link_status_change') |
|
1537 | 1 | @patch('napps.kytos.topology.main.Main.notify_topology_update') |
|
1538 | 1 | @patch('time.sleep') |
|
1539 | 1 | def test_notify_link_up_if_status( |
|
1540 | self, |
||
1541 | mock_sleep, |
||
1542 | mock_notify_topo, |
||
1543 | mock_notify_link, |
||
1544 | ) -> None: |
||
1545 | """Test notify link up if status.""" |
||
1546 | |||
1547 | 1 | link = MagicMock(status=EntityStatus.UP) |
|
1548 | 1 | self.napp.link_status_change[link.id] = { |
|
1549 | "notified_up_at": now(), |
||
1550 | } |
||
1551 | 1 | assert not self.napp.notify_link_up_if_status(link, "link up") |
|
1552 | 1 | link.update_metadata.assert_not_called() |
|
1553 | 1 | mock_notify_topo.assert_not_called() |
|
1554 | 1 | mock_notify_link.assert_not_called() |
|
1555 | |||
1556 | 1 | link = MagicMock(status=EntityStatus.UP) |
|
1557 | 1 | orig_time = now() - timedelta(seconds=60) |
|
1558 | 1 | self.napp.link_status_change[link.id] = { |
|
1559 | "notified_up_at": orig_time, |
||
1560 | } |
||
1561 | 1 | assert not self.napp.notify_link_up_if_status(link, "link up") |
|
1562 | 1 | link_status_info = self.napp.link_status_change[link.id] |
|
1563 | 1 | new_time = link_status_info["notified_up_at"] |
|
1564 | 1 | assert new_time != orig_time |
|
1565 | 1 | mock_notify_topo.assert_called() |
|
1566 | 1 | mock_notify_link.assert_called() |
|
1567 | |||
1568 | 1 | assert mock_sleep.call_count == 2 |
|
1569 | |||
1570 | 1 | @patch('napps.kytos.topology.main.Main.notify_link_status_change') |
|
1571 | 1 | def test_notify_switch_links_status(self, mock_notify_link_status_change): |
|
1572 | """Test switch links notification when switch status change""" |
||
1573 | 1 | buffers_app_mock = MagicMock() |
|
1574 | 1 | self.napp.controller.buffers.app = buffers_app_mock |
|
1575 | 1 | dpid = "00:00:00:00:00:00:00:01" |
|
1576 | 1 | mock_switch = get_switch_mock(dpid) |
|
1577 | 1 | link1 = MagicMock() |
|
1578 | 1 | link1.endpoint_a.switch = mock_switch |
|
1579 | 1 | self.napp.controller.links = {1: link1} |
|
1580 | |||
1581 | 1 | self.napp.notify_switch_links_status(mock_switch, "link enabled") |
|
1582 | 1 | assert self.napp.controller.buffers.app.put.call_count == 1 |
|
1583 | |||
1584 | 1 | self.napp.notify_switch_links_status(mock_switch, "link disabled") |
|
1585 | 1 | assert self.napp.controller.buffers.app.put.call_count == 1 |
|
1586 | 1 | assert mock_notify_link_status_change.call_count == 1 |
|
1587 | |||
1588 | # Without notification |
||
1589 | 1 | link1.endpoint_a.switch = None |
|
1590 | 1 | self.napp.notify_switch_links_status(mock_switch, "link enabled") |
|
1591 | 1 | assert self.napp.controller.buffers.app.put.call_count == 1 |
|
1592 | |||
1593 | 1 | View Code Duplication | @patch('napps.kytos.topology.main.Main.notify_topology_update') |
0 ignored issues
–
show
Duplication
introduced
by
![]() |
|||
1594 | 1 | @patch('napps.kytos.topology.main.Main.notify_link_status_change') |
|
1595 | 1 | def test_interruption_start( |
|
1596 | self, |
||
1597 | mock_notify_link_status_change, |
||
1598 | mock_notify_topology_update |
||
1599 | ): |
||
1600 | """Tests processing of received interruption start events.""" |
||
1601 | 1 | link_a = MagicMock() |
|
1602 | 1 | link_b = MagicMock() |
|
1603 | 1 | link_c = MagicMock() |
|
1604 | 1 | self.napp.controller.links = { |
|
1605 | 'link_a': link_a, |
||
1606 | 'link_b': link_b, |
||
1607 | 'link_c': link_c, |
||
1608 | } |
||
1609 | 1 | event = KytosEvent( |
|
1610 | "topology.interruption.start", |
||
1611 | { |
||
1612 | 'type': 'test_interruption', |
||
1613 | 'switches': [ |
||
1614 | ], |
||
1615 | 'interfaces': [ |
||
1616 | ], |
||
1617 | 'links': [ |
||
1618 | 'link_a', |
||
1619 | 'link_c', |
||
1620 | ], |
||
1621 | } |
||
1622 | ) |
||
1623 | 1 | self.napp.handle_interruption_start(event) |
|
1624 | 1 | mock_notify_link_status_change.assert_has_calls( |
|
1625 | [ |
||
1626 | call(link_a, 'test_interruption'), |
||
1627 | call(link_c, 'test_interruption'), |
||
1628 | ] |
||
1629 | ) |
||
1630 | 1 | assert mock_notify_link_status_change.call_count == 2 |
|
1631 | 1 | mock_notify_topology_update.assert_called_once() |
|
1632 | |||
1633 | 1 | View Code Duplication | @patch('napps.kytos.topology.main.Main.notify_topology_update') |
0 ignored issues
–
show
|
|||
1634 | 1 | @patch('napps.kytos.topology.main.Main.notify_link_status_change') |
|
1635 | 1 | def test_interruption_end( |
|
1636 | self, |
||
1637 | mock_notify_link_status_change, |
||
1638 | mock_notify_topology_update |
||
1639 | ): |
||
1640 | """Tests processing of received interruption end events.""" |
||
1641 | 1 | link_a = MagicMock() |
|
1642 | 1 | link_b = MagicMock() |
|
1643 | 1 | link_c = MagicMock() |
|
1644 | 1 | self.napp.controller.links = { |
|
1645 | 'link_a': link_a, |
||
1646 | 'link_b': link_b, |
||
1647 | 'link_c': link_c, |
||
1648 | } |
||
1649 | 1 | event = KytosEvent( |
|
1650 | "topology.interruption.start", |
||
1651 | { |
||
1652 | 'type': 'test_interruption', |
||
1653 | 'switches': [ |
||
1654 | ], |
||
1655 | 'interfaces': [ |
||
1656 | ], |
||
1657 | 'links': [ |
||
1658 | 'link_a', |
||
1659 | 'link_c', |
||
1660 | ], |
||
1661 | } |
||
1662 | ) |
||
1663 | 1 | self.napp.handle_interruption_end(event) |
|
1664 | 1 | mock_notify_link_status_change.assert_has_calls( |
|
1665 | [ |
||
1666 | call(link_a, 'test_interruption'), |
||
1667 | call(link_c, 'test_interruption'), |
||
1668 | ] |
||
1669 | ) |
||
1670 | 1 | assert mock_notify_link_status_change.call_count == 2 |
|
1671 | 1 | mock_notify_topology_update.assert_called_once() |
|
1672 | |||
1673 | 1 | async def test_set_tag_range(self): |
|
1674 | """Test set_tag_range""" |
||
1675 | 1 | self.napp.controller.loop = asyncio.get_running_loop() |
|
1676 | 1 | interface_id = '00:00:00:00:00:00:00:01:1' |
|
1677 | 1 | dpid = '00:00:00:00:00:00:00:01' |
|
1678 | 1 | mock_switch = get_switch_mock(dpid) |
|
1679 | 1 | mock_interface = get_interface_mock('s1-eth1', 1, mock_switch) |
|
1680 | 1 | mock_interface.set_tag_ranges = MagicMock() |
|
1681 | 1 | self.napp.handle_on_interface_tags = MagicMock() |
|
1682 | 1 | self.napp.controller.get_interface_by_id = MagicMock() |
|
1683 | 1 | self.napp.controller.get_interface_by_id.return_value = mock_interface |
|
1684 | 1 | payload = { |
|
1685 | "tag_type": "vlan", |
||
1686 | "tag_ranges": [[20, 20], [200, 3000]] |
||
1687 | } |
||
1688 | 1 | url = f"{self.base_endpoint}/interfaces/{interface_id}/tag_ranges" |
|
1689 | 1 | response = await self.api_client.post(url, json=payload) |
|
1690 | 1 | assert response.status_code == 200 |
|
1691 | |||
1692 | 1 | args = mock_interface.set_tag_ranges.call_args[0] |
|
1693 | 1 | assert args[0] == payload['tag_ranges'] |
|
1694 | 1 | assert args[1] == payload['tag_type'] |
|
1695 | 1 | assert self.napp.handle_on_interface_tags.call_count == 1 |
|
1696 | |||
1697 | 1 | async def test_set_tag_range_not_found(self): |
|
1698 | """Test set_tag_range. Not found""" |
||
1699 | 1 | self.napp.controller.loop = asyncio.get_running_loop() |
|
1700 | 1 | interface_id = '00:00:00:00:00:00:00:01:1' |
|
1701 | 1 | self.napp.controller.get_interface_by_id = MagicMock() |
|
1702 | 1 | self.napp.controller.get_interface_by_id.return_value = None |
|
1703 | 1 | payload = { |
|
1704 | "tag_type": "vlan", |
||
1705 | "tag_ranges": [[20, 20], [200, 3000]] |
||
1706 | } |
||
1707 | 1 | url = f"{self.base_endpoint}/interfaces/{interface_id}/tag_ranges" |
|
1708 | 1 | response = await self.api_client.post(url, json=payload) |
|
1709 | assert response.status_code == 404 |
||
1710 | |||
1711 | 1 | View Code Duplication | async def test_set_tag_range_tag_error(self): |
0 ignored issues
–
show
|
|||
1712 | """Test set_tag_range TagRangeError""" |
||
1713 | 1 | self.napp.controller.loop = asyncio.get_running_loop() |
|
1714 | 1 | interface_id = '00:00:00:00:00:00:00:01:1' |
|
1715 | 1 | dpid = '00:00:00:00:00:00:00:01' |
|
1716 | 1 | mock_switch = get_switch_mock(dpid) |
|
1717 | 1 | mock_interface = get_interface_mock('s1-eth1', 1, mock_switch) |
|
1718 | 1 | mock_interface.set_tag_ranges = MagicMock() |
|
1719 | 1 | mock_interface.set_tag_ranges.side_effect = KytosSetTagRangeError("") |
|
1720 | 1 | mock_interface.notify_interface_tags = MagicMock() |
|
1721 | 1 | self.napp.controller.get_interface_by_id = MagicMock() |
|
1722 | 1 | self.napp.controller.get_interface_by_id.return_value = mock_interface |
|
1723 | 1 | payload = { |
|
1724 | "tag_type": "vlan", |
||
1725 | "tag_ranges": [[20, 20], [200, 3000]] |
||
1726 | } |
||
1727 | 1 | url = f"{self.base_endpoint}/interfaces/{interface_id}/tag_ranges" |
|
1728 | 1 | response = await self.api_client.post(url, json=payload) |
|
1729 | assert response.status_code == 400 |
||
1730 | assert mock_interface.notify_interface_tags.call_count == 0 |
||
1731 | |||
1732 | 1 | View Code Duplication | async def test_set_tag_range_type_error(self): |
0 ignored issues
–
show
|
|||
1733 | """Test set_tag_range TagRangeError""" |
||
1734 | 1 | self.napp.controller.loop = asyncio.get_running_loop() |
|
1735 | 1 | interface_id = '00:00:00:00:00:00:00:01:1' |
|
1736 | 1 | dpid = '00:00:00:00:00:00:00:01' |
|
1737 | 1 | mock_switch = get_switch_mock(dpid) |
|
1738 | 1 | mock_interface = get_interface_mock('s1-eth1', 1, mock_switch) |
|
1739 | 1 | mock_interface.set_tag_ranges = MagicMock() |
|
1740 | 1 | mock_interface.set_tag_ranges.side_effect = KytosTagtypeNotSupported( |
|
1741 | "" |
||
1742 | ) |
||
1743 | 1 | self.napp.handle_on_interface_tags = MagicMock() |
|
1744 | 1 | self.napp.controller.get_interface_by_id = MagicMock() |
|
1745 | 1 | self.napp.controller.get_interface_by_id.return_value = mock_interface |
|
1746 | 1 | payload = { |
|
1747 | "tag_type": "wrong_tag_type", |
||
1748 | "tag_ranges": [[20, 20], [200, 3000]] |
||
1749 | } |
||
1750 | 1 | url = f"{self.base_endpoint}/interfaces/{interface_id}/tag_ranges" |
|
1751 | 1 | response = await self.api_client.post(url, json=payload) |
|
1752 | assert response.status_code == 400 |
||
1753 | assert self.napp.handle_on_interface_tags.call_count == 0 |
||
1754 | |||
1755 | 1 | View Code Duplication | async def test_delete_tag_range(self): |
0 ignored issues
–
show
|
|||
1756 | """Test delete_tag_range""" |
||
1757 | 1 | self.napp.controller.loop = asyncio.get_running_loop() |
|
1758 | 1 | interface_id = '00:00:00:00:00:00:00:01:1' |
|
1759 | 1 | dpid = '00:00:00:00:00:00:00:01' |
|
1760 | 1 | mock_switch = get_switch_mock(dpid) |
|
1761 | 1 | mock_interface = get_interface_mock('s1-eth1', 1, mock_switch) |
|
1762 | 1 | mock_interface.remove_tag_ranges = MagicMock() |
|
1763 | 1 | self.napp.handle_on_interface_tags = MagicMock() |
|
1764 | 1 | self.napp.controller.get_interface_by_id = MagicMock() |
|
1765 | 1 | self.napp.controller.get_interface_by_id.return_value = mock_interface |
|
1766 | 1 | url = f"{self.base_endpoint}/interfaces/{interface_id}/tag_ranges" |
|
1767 | 1 | response = await self.api_client.delete(url) |
|
1768 | 1 | assert response.status_code == 200 |
|
1769 | 1 | assert mock_interface.remove_tag_ranges.call_count == 1 |
|
1770 | |||
1771 | 1 | View Code Duplication | async def test_delete_tag_range_not_found(self): |
0 ignored issues
–
show
|
|||
1772 | """Test delete_tag_range. Not found""" |
||
1773 | 1 | self.napp.controller.loop = asyncio.get_running_loop() |
|
1774 | 1 | interface_id = '00:00:00:00:00:00:00:01:1' |
|
1775 | 1 | dpid = '00:00:00:00:00:00:00:01' |
|
1776 | 1 | mock_switch = get_switch_mock(dpid) |
|
1777 | 1 | mock_interface = get_interface_mock('s1-eth1', 1, mock_switch) |
|
1778 | 1 | mock_interface.remove_tag_ranges = MagicMock() |
|
1779 | 1 | self.napp.controller.get_interface_by_id = MagicMock() |
|
1780 | 1 | self.napp.controller.get_interface_by_id.return_value = None |
|
1781 | 1 | url = f"{self.base_endpoint}/interfaces/{interface_id}/tag_ranges" |
|
1782 | 1 | response = await self.api_client.delete(url) |
|
1783 | assert response.status_code == 404 |
||
1784 | assert mock_interface.remove_tag_ranges.call_count == 0 |
||
1785 | |||
1786 | 1 | async def test_delete_tag_range_type_error(self): |
|
1787 | """Test delete_tag_range TagRangeError""" |
||
1788 | 1 | self.napp.controller.loop = asyncio.get_running_loop() |
|
1789 | 1 | interface_id = '00:00:00:00:00:00:00:01:1' |
|
1790 | 1 | dpid = '00:00:00:00:00:00:00:01' |
|
1791 | 1 | mock_switch = get_switch_mock(dpid) |
|
1792 | 1 | mock_interface = get_interface_mock('s1-eth1', 1, mock_switch) |
|
1793 | 1 | mock_interface.remove_tag_ranges = MagicMock() |
|
1794 | 1 | remove_tag = mock_interface.remove_tag_ranges |
|
1795 | 1 | remove_tag.side_effect = KytosTagtypeNotSupported("") |
|
1796 | 1 | self.napp.controller.get_interface_by_id = MagicMock() |
|
1797 | 1 | self.napp.controller.get_interface_by_id.return_value = mock_interface |
|
1798 | 1 | url = f"{self.base_endpoint}/interfaces/{interface_id}/tag_ranges" |
|
1799 | 1 | response = await self.api_client.delete(url) |
|
1800 | assert response.status_code == 400 |
||
1801 | |||
1802 | 1 | View Code Duplication | async def test_get_all_tag_ranges(self): |
0 ignored issues
–
show
|
|||
1803 | """Test get_all_tag_ranges""" |
||
1804 | 1 | self.napp.controller.loop = asyncio.get_running_loop() |
|
1805 | 1 | dpid = '00:00:00:00:00:00:00:01' |
|
1806 | 1 | switch = get_switch_mock(dpid) |
|
1807 | 1 | interface = get_interface_mock('s1-eth1', 1, switch) |
|
1808 | 1 | tags = {'vlan': [[1, 4095]]} |
|
1809 | 1 | special_tags = {'vlan': ["vlan"]} |
|
1810 | 1 | interface.tag_ranges = tags |
|
1811 | 1 | interface.available_tags = tags |
|
1812 | 1 | interface.special_available_tags = special_tags |
|
1813 | 1 | interface.special_tags = special_tags |
|
1814 | 1 | switch.interfaces = {1: interface} |
|
1815 | 1 | self.napp.controller.switches = {dpid: switch} |
|
1816 | 1 | url = f"{self.base_endpoint}/interfaces/tag_ranges" |
|
1817 | 1 | response = await self.api_client.get(url) |
|
1818 | 1 | expected = {dpid + ":1": { |
|
1819 | 'available_tags': tags, |
||
1820 | 'tag_ranges': tags, |
||
1821 | 'special_available_tags': special_tags, |
||
1822 | 'special_tags': special_tags |
||
1823 | }} |
||
1824 | 1 | assert response.status_code == 200 |
|
1825 | 1 | assert response.json() == expected |
|
1826 | |||
1827 | 1 | View Code Duplication | async def test_get_tag_ranges_by_intf(self): |
0 ignored issues
–
show
|
|||
1828 | """Test get_tag_ranges_by_intf""" |
||
1829 | 1 | self.napp.controller.loop = asyncio.get_running_loop() |
|
1830 | 1 | dpid = '00:00:00:00:00:00:00:01' |
|
1831 | 1 | switch = get_switch_mock(dpid) |
|
1832 | 1 | interface = get_interface_mock('s1-eth1', 1, switch) |
|
1833 | 1 | tags = {'vlan': [[1, 4095]]} |
|
1834 | 1 | special_tags = {'vlan': ["vlan"]} |
|
1835 | 1 | interface.tag_ranges = tags |
|
1836 | 1 | interface.available_tags = tags |
|
1837 | 1 | interface.special_available_tags = special_tags |
|
1838 | 1 | interface.special_tags = special_tags |
|
1839 | 1 | self.napp.controller.get_interface_by_id = MagicMock() |
|
1840 | 1 | self.napp.controller.get_interface_by_id.return_value = interface |
|
1841 | 1 | url = f"{self.base_endpoint}/interfaces/{dpid}:1/tag_ranges" |
|
1842 | 1 | response = await self.api_client.get(url) |
|
1843 | 1 | expected = { |
|
1844 | '00:00:00:00:00:00:00:01:1': { |
||
1845 | "available_tags": tags, |
||
1846 | "tag_ranges": tags, |
||
1847 | 'special_available_tags': special_tags, |
||
1848 | 'special_tags': special_tags |
||
1849 | } |
||
1850 | } |
||
1851 | 1 | assert response.status_code == 200 |
|
1852 | 1 | assert response.json() == expected |
|
1853 | |||
1854 | 1 | async def test_get_tag_ranges_by_intf_error(self): |
|
1855 | """Test get_tag_ranges_by_intf with NotFound""" |
||
1856 | 1 | self.napp.controller.loop = asyncio.get_running_loop() |
|
1857 | 1 | dpid = '00:00:00:00:00:00:00:01' |
|
1858 | 1 | self.napp.controller.get_interface_by_id = MagicMock() |
|
1859 | 1 | self.napp.controller.get_interface_by_id.return_value = None |
|
1860 | 1 | url = f"{self.base_endpoint}/interfaces/{dpid}:1/tag_ranges" |
|
1861 | 1 | response = await self.api_client.get(url) |
|
1862 | assert response.status_code == 404 |
||
1863 | |||
1864 | 1 | async def test_set_special_tags(self): |
|
1865 | """Test set_special_tags""" |
||
1866 | 1 | self.napp.controller.loop = asyncio.get_running_loop() |
|
1867 | 1 | interface_id = '00:00:00:00:00:00:00:01:1' |
|
1868 | 1 | dpid = '00:00:00:00:00:00:00:01' |
|
1869 | 1 | mock_switch = get_switch_mock(dpid) |
|
1870 | 1 | mock_intf = get_interface_mock('s1-eth1', 1, mock_switch) |
|
1871 | 1 | mock_intf.set_special_tags = MagicMock() |
|
1872 | 1 | self.napp.handle_on_interface_tags = MagicMock() |
|
1873 | 1 | self.napp.controller.get_interface_by_id = MagicMock() |
|
1874 | 1 | self.napp.controller.get_interface_by_id.return_value = mock_intf |
|
1875 | 1 | payload = { |
|
1876 | "tag_type": "vlan", |
||
1877 | "special_tags": ["untagged"], |
||
1878 | } |
||
1879 | 1 | url = f"{self.base_endpoint}/interfaces/{interface_id}/"\ |
|
1880 | "special_tags" |
||
1881 | 1 | response = await self.api_client.post(url, json=payload) |
|
1882 | 1 | assert response.status_code == 200 |
|
1883 | |||
1884 | 1 | args = mock_intf.set_special_tags.call_args[0] |
|
1885 | 1 | assert args[0] == payload["tag_type"] |
|
1886 | 1 | assert args[1] == payload['special_tags'] |
|
1887 | 1 | assert self.napp.handle_on_interface_tags.call_count == 1 |
|
1888 | |||
1889 | # KytosTagError |
||
1890 | 1 | mock_intf.set_special_tags.side_effect = KytosTagtypeNotSupported("") |
|
1891 | 1 | url = f"{self.base_endpoint}/interfaces/{interface_id}/"\ |
|
1892 | "special_tags" |
||
1893 | 1 | response = await self.api_client.post(url, json=payload) |
|
1894 | assert response.status_code == 400 |
||
1895 | assert self.napp.handle_on_interface_tags.call_count == 1 |
||
1896 | |||
1897 | # Interface Not Found |
||
1898 | self.napp.controller.get_interface_by_id.return_value = None |
||
1899 | url = f"{self.base_endpoint}/interfaces/{interface_id}/"\ |
||
1900 | "special_tags" |
||
1901 | response = await self.api_client.post(url, json=payload) |
||
1902 | assert response.status_code == 404 |
||
1903 | assert self.napp.handle_on_interface_tags.call_count == 1 |
||
1904 | |||
1905 | 1 | async def test_delete_link(self): |
|
1906 | """Test delete_link""" |
||
1907 | 1 | dpid_a = '00:00:00:00:00:00:00:01' |
|
1908 | 1 | dpid_b = '00:00:00:00:00:00:00:02' |
|
1909 | 1 | link_id = 'mock_link' |
|
1910 | 1 | mock_switch_a = get_switch_mock(dpid_a) |
|
1911 | 1 | mock_switch_b = get_switch_mock(dpid_b) |
|
1912 | 1 | mock_interface_a = get_interface_mock('s1-eth1', 1, mock_switch_a) |
|
1913 | 1 | mock_interface_b = get_interface_mock('s2-eth1', 1, mock_switch_b) |
|
1914 | 1 | mock_link = get_link_mock(mock_interface_a, mock_interface_b) |
|
1915 | 1 | mock_link.id = link_id |
|
1916 | 1 | mock_link.status = EntityStatus.DISABLED |
|
1917 | 1 | mock_interface_a.link = mock_link |
|
1918 | 1 | mock_interface_b.link = mock_link |
|
1919 | 1 | self.napp.controller.links = {link_id: mock_link} |
|
1920 | |||
1921 | 1 | call_count = self.napp.controller.buffers.app.put.call_count |
|
1922 | 1 | endpoint = f"{self.base_endpoint}/links/{link_id}" |
|
1923 | 1 | response = await self.api_client.delete(endpoint) |
|
1924 | 1 | assert response.status_code == 200 |
|
1925 | 1 | assert self.napp.topo_controller.delete_link.call_count == 1 |
|
1926 | 1 | assert len(self.napp.controller.links) == 0 |
|
1927 | 1 | call_count += 2 |
|
1928 | 1 | assert self.napp.controller.buffers.app.put.call_count == call_count |
|
1929 | |||
1930 | # Link is up |
||
1931 | 1 | self.napp.controller.links = {link_id: mock_link} |
|
1932 | 1 | mock_link.status = EntityStatus.UP |
|
1933 | 1 | endpoint = f"{self.base_endpoint}/links/{link_id}" |
|
1934 | 1 | response = await self.api_client.delete(endpoint) |
|
1935 | assert response.status_code == 409 |
||
1936 | assert self.napp.topo_controller.delete_link.call_count == 1 |
||
1937 | assert self.napp.controller.buffers.app.put.call_count == call_count |
||
1938 | |||
1939 | # Link does not exist |
||
1940 | del self.napp.controller.links[link_id] |
||
1941 | endpoint = f"{self.base_endpoint}/links/{link_id}" |
||
1942 | response = await self.api_client.delete(endpoint) |
||
1943 | assert response.status_code == 404 |
||
1944 | assert self.napp.topo_controller.delete_link.call_count == 1 |
||
1945 | assert self.napp.controller.buffers.app.put.call_count == call_count |
||
1946 | |||
1947 | 1 | @patch('napps.kytos.topology.main.Main.get_flows_by_switch') |
|
1948 | 1 | async def test_delete_switch(self, mock_get): |
|
1949 | """Test delete_switch""" |
||
1950 | # Error 404 NotFound |
||
1951 | 1 | dpid = '00:00:00:00:00:00:00:01' |
|
1952 | 1 | endpoint = f"{self.base_endpoint}/switches/{dpid}" |
|
1953 | 1 | response = await self.api_client.delete(endpoint) |
|
1954 | assert response.status_code == 404, response |
||
1955 | |||
1956 | # Error 409 Switch not disabled |
||
1957 | mock_switch = get_switch_mock(dpid) |
||
1958 | mock_switch.status = EntityStatus.UP |
||
1959 | self.napp.controller.switches = {dpid: mock_switch} |
||
1960 | endpoint = f"{self.base_endpoint}/switches/{dpid}" |
||
1961 | response = await self.api_client.delete(endpoint) |
||
1962 | assert response.status_code == 409, response |
||
1963 | |||
1964 | # Error 409 Interface vlan is being used |
||
1965 | 1 | mock_intf = MagicMock(all_tags_available=lambda: False) |
|
1966 | mock_switch.interfaces = {1: mock_intf} |
||
1967 | mock_switch.status = EntityStatus.DISABLED |
||
1968 | endpoint = f"{self.base_endpoint}/switches/{dpid}" |
||
1969 | response = await self.api_client.delete(endpoint) |
||
1970 | assert response.status_code == 409, response |
||
1971 | |||
1972 | # Error 409 Swith have links |
||
1973 | 1 | mock_switch.interfaces[1].all_tags_available = lambda: True |
|
1974 | mock_switch_2 = get_switch_mock("00:00:00:00:00:00:00:02") |
||
1975 | mock_interface_a = get_interface_mock('s1-eth1', 1, mock_switch) |
||
1976 | mock_interface_b = get_interface_mock('s2-eth1', 1, mock_switch_2) |
||
1977 | mock_link = get_link_mock(mock_interface_a, mock_interface_b) |
||
1978 | self.napp.controller.links = {'0e2b5d7bc858b9f38db11b69': mock_link} |
||
1979 | endpoint = f"{self.base_endpoint}/switches/{dpid}" |
||
1980 | response = await self.api_client.delete(endpoint) |
||
1981 | assert response.status_code == 409, response |
||
1982 | |||
1983 | # Error 409 Switch has flows |
||
1984 | mock_get.return_value = {dpid: {}} |
||
1985 | endpoint = f"{self.base_endpoint}/switches/{dpid}" |
||
1986 | response = await self.api_client.delete(endpoint) |
||
1987 | assert response.status_code == 409, response |
||
1988 | |||
1989 | # Success 202 |
||
1990 | mock_get.return_value = {} |
||
1991 | self.napp.controller.links = {} |
||
1992 | endpoint = f"{self.base_endpoint}/switches/{dpid}" |
||
1993 | response = await self.api_client.delete(endpoint) |
||
1994 | 1 | assert response.status_code == 200, response |
|
1995 | |||
1996 | 1 | def test_notify_link_status(self): |
|
1997 | """Test notify_link_enabled_state""" |
||
1998 | 1 | self.napp.controller.buffers.app.put.reset_mock() |
|
1999 | 1 | link = Mock() |
|
2000 | 1 | link.id = 'mock_link' |
|
2001 | 1 | self.napp.notify_link_enabled_state(link, 'enabled') |
|
2002 | 1 | assert self.napp.controller.buffers.app.put.call_count == 1 |
|
2003 | |||
2004 | 1 | self.napp.notify_link_enabled_state(link, 'disabled') |
|
2005 | 1 | assert self.napp.controller.buffers.app.put.call_count == 2 |
|
2006 | |||
2007 | 1 | @patch('napps.kytos.topology.main.tenacity.nap.time') |
|
2008 | 1 | @patch('httpx.get') |
|
2009 | 1 | def test_get_flows_by_switch(self, mock_get, _): |
|
2010 | """Test get_flows_by_switch""" |
||
2011 | 1 | dpid = "00:01" |
|
2012 | 1 | mock_get.return_value.status_code = 400 |
|
2013 | 1 | with pytest.raises(tenacity.RetryError): |
|
2014 | 1 | self.napp.get_flows_by_switch(dpid) |
|
2015 | |||
2016 | 1 | mock_get.return_value.status_code = 200 |
|
2017 | 1 | mock_get.return_value.is_server_error = False |
|
2018 | 1 | expected = {dpid: "mocked_flows"} |
|
2019 | 1 | mock_get.return_value.json.return_value = expected |
|
2020 | 1 | actual = self.napp.get_flows_by_switch(dpid) |
|
2021 | 1 | assert actual == "mocked_flows" |
|
2022 | |||
2023 | 1 | @patch('napps.kytos.topology.main.Main.get_intf_usage') |
|
2024 | 1 | @patch('napps.kytos.topology.main.Main._delete_interface') |
|
2025 | 1 | async def test_delete_interface_api(self, mock_delete, mock_usage): |
|
2026 | """Test delete interface API call""" |
||
2027 | 1 | switch_id = "00:00:00:00:00:00:00:01" |
|
2028 | 1 | intf_id = "00:00:00:00:00:00:00:01:1" |
|
2029 | |||
2030 | # Error 400 Invalid interface id |
||
2031 | 1 | endpoint = f"{self.base_endpoint}/interfaces/{intf_id}x" |
|
2032 | 1 | response = await self.api_client.delete(endpoint) |
|
2033 | assert response.status_code == 400, response |
||
2034 | |||
2035 | # Error 404 Switch not found |
||
2036 | endpoint = f"{self.base_endpoint}/interfaces/{intf_id}" |
||
2037 | response = await self.api_client.delete(endpoint) |
||
2038 | assert response.status_code == 404, response |
||
2039 | |||
2040 | # Error 404 Interface not found |
||
2041 | mock_switch = get_switch_mock(switch_id) |
||
2042 | mock_switch.interfaces = {} |
||
2043 | self.napp.controller.switches = {switch_id: mock_switch} |
||
2044 | response = await self.api_client.delete(endpoint) |
||
2045 | assert response.status_code == 404, response |
||
2046 | |||
2047 | # Error 409 Interface is used |
||
2048 | mock_switch.interfaces = {1: Mock()} |
||
2049 | self.napp.controller.switches = {switch_id: mock_switch} |
||
2050 | mock_usage.return_value = "It is enabled or active." |
||
2051 | response = await self.api_client.delete(endpoint) |
||
2052 | assert response.status_code == 409, response |
||
2053 | |||
2054 | # Success |
||
2055 | mock_usage.return_value = None |
||
2056 | mock_delete.return_value = True |
||
2057 | response = await self.api_client.delete(endpoint) |
||
2058 | 1 | assert response.status_code == 200, response |
|
2059 | |||
2060 | 1 | def test_get_intf_usage(self): |
|
2061 | """Test get_intf_usage""" |
||
2062 | 1 | switch_id = "00:00:00:00:00:00:00:01" |
|
2063 | 1 | mock_switch = get_switch_mock(switch_id) |
|
2064 | 1 | mock_intf = get_interface_mock('s1-eth1', 1, mock_switch) |
|
2065 | |||
2066 | 1 | mock_intf.is_enabled.return_value = False |
|
2067 | 1 | mock_intf.is_active.return_value = True |
|
2068 | 1 | actual_usage = self.napp.get_intf_usage(mock_intf) |
|
2069 | 1 | assert actual_usage == "It is enabled or active." |
|
2070 | |||
2071 | 1 | mock_intf.is_active.return_value = False |
|
2072 | 1 | mock_intf.link = Mock() |
|
2073 | 1 | actual_usage = self.napp.get_intf_usage(mock_intf) |
|
2074 | 1 | assert "It has a link," in actual_usage |
|
2075 | |||
2076 | 1 | mock_intf.link = None |
|
2077 | 1 | self.napp.get_flow_id_by_intf = MagicMock(return_value="mock_flow") |
|
2078 | 1 | actual_usage = self.napp.get_intf_usage(mock_intf) |
|
2079 | 1 | assert "There is a flow installed" in actual_usage |
|
2080 | |||
2081 | 1 | self.napp.get_flow_id_by_intf.return_value = None |
|
2082 | 1 | actual_usage = self.napp.get_intf_usage(mock_intf) |
|
2083 | 1 | assert actual_usage is None |
|
2084 | |||
2085 | 1 | @patch('napps.kytos.topology.main.Main.get_flows_by_switch') |
|
2086 | 1 | def test_get_flow_id_by_intf(self, mock_flows): |
|
2087 | """Test get_flow_id_by_intf""" |
||
2088 | 1 | flows = [ |
|
2089 | { |
||
2090 | "flow": { |
||
2091 | "match": {"in_port": 1, "dl_vlan": 200}, |
||
2092 | }, |
||
2093 | "flow_id": "flow_0", |
||
2094 | }, |
||
2095 | { |
||
2096 | "flow": { |
||
2097 | "actions": [{"action_type": "output", "port": 1}] |
||
2098 | }, |
||
2099 | "flow_id": "flow_1", |
||
2100 | }, |
||
2101 | { |
||
2102 | "flow": { |
||
2103 | "instructions": [{ |
||
2104 | "instruction_type": "apply_actions", |
||
2105 | "actions": [{"action_type": "output", "port": 1}] |
||
2106 | }] |
||
2107 | }, |
||
2108 | "flow_id": "flow_2", |
||
2109 | }, |
||
2110 | { |
||
2111 | "flow": { |
||
2112 | "match": {"dl_src": "ee:ee:ee:ee:ee:02"}, |
||
2113 | }, |
||
2114 | "flow_id": "flow_3", |
||
2115 | } |
||
2116 | ] |
||
2117 | |||
2118 | 1 | switch_id = "00:00:00:00:00:00:00:01" |
|
2119 | 1 | mock_switch = get_switch_mock(switch_id) |
|
2120 | 1 | mock_intf = get_interface_mock('s1-eth1', 1, mock_switch) |
|
2121 | |||
2122 | 1 | mock_flows.return_value = [flows[0]] |
|
2123 | 1 | flow_id = self.napp.get_flow_id_by_intf(mock_intf) |
|
2124 | 1 | assert flow_id == flows[0]["flow_id"] |
|
2125 | |||
2126 | 1 | mock_flows.return_value = [flows[1]] |
|
2127 | 1 | flow_id = self.napp.get_flow_id_by_intf(mock_intf) |
|
2128 | 1 | assert flow_id == flows[1]["flow_id"] |
|
2129 | |||
2130 | 1 | mock_flows.return_value = [flows[2]] |
|
2131 | 1 | flow_id = self.napp.get_flow_id_by_intf(mock_intf) |
|
2132 | 1 | assert flow_id == flows[2]["flow_id"] |
|
2133 | |||
2134 | 1 | mock_flows.return_value = [flows[3]] |
|
2135 | 1 | flow_id = self.napp.get_flow_id_by_intf(mock_intf) |
|
2136 | 1 | assert flow_id is None |
|
2137 | |||
2138 | 1 | def test_delete_interface(self): |
|
2139 | """Test _delete_interface""" |
||
2140 | 1 | switch_id = "00:00:00:00:00:00:00:01" |
|
2141 | 1 | mock_switch = get_switch_mock(switch_id) |
|
2142 | 1 | mock_intf = get_interface_mock('s1-eth1', 1, mock_switch) |
|
2143 | 1 | self.napp._delete_interface(mock_intf) |
|
2144 | 1 | assert mock_switch.remove_interface.call_count == 1 |
|
2145 | 1 | assert self.napp.topo_controller.upsert_switch.call_count == 1 |
|
2146 | 1 | delete = self.napp.topo_controller.delete_interface_from_details |
|
2147 | 1 | assert delete.call_count == 1 |
|
2148 | |||
2149 | 1 | def test_detect_mismatched_link(self): |
|
2150 | """Test detect_mismatched_link""" |
||
2151 | 1 | mock_link_1 = MagicMock(id='link_1') |
|
2152 | 1 | mock_link_1.endpoint_a = MagicMock(link=mock_link_1) |
|
2153 | 1 | mock_link_1.endpoint_b = MagicMock(link=None) |
|
2154 | 1 | assert self.napp.detect_mismatched_link(mock_link_1) |
|
2155 | |||
2156 | 1 | mock_link_1.endpoint_a.link = None |
|
2157 | 1 | mock_link_1.endpoint_b.link = mock_link_1 |
|
2158 | 1 | assert self.napp.detect_mismatched_link(mock_link_1) |
|
2159 | |||
2160 | 1 | mock_link_2 = MagicMock(id='link_2') |
|
2161 | 1 | mock_link_1.endpoint_a.link = mock_link_2 |
|
2162 | 1 | assert self.napp.detect_mismatched_link(mock_link_1) |
|
2163 | |||
2164 | 1 | mock_link_1.endpoint_a.link = mock_link_1 |
|
2165 | 1 | assert not self.napp.detect_mismatched_link(mock_link_1) |
|
2166 | |||
2167 | 1 | @patch('napps.kytos.topology.main.Main.detect_mismatched_link') |
|
2168 | 1 | def test_link_status_mismatched(self, mock_detect_mismatched_link): |
|
2169 | """Test link_status_mismatched""" |
||
2170 | 1 | mock_link_1 = MagicMock() |
|
2171 | 1 | mock_detect_mismatched_link.return_value = True |
|
2172 | 1 | assert (self.napp.link_status_mismatched(mock_link_1) |
|
2173 | == EntityStatus.DOWN) |
||
2174 | |||
2175 | 1 | mock_detect_mismatched_link.return_value = False |
|
2176 | assert self.napp.link_status_mismatched(mock_link_1) is None |
||
2177 |