Passed
Pull Request — master (#259)
by Aldo
04:39
created

)   A

Complexity

Conditions 1

Size

Total Lines 16
Code Lines 14

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 8
CRAP Score 1

Importance

Changes 0
Metric Value
cc 1
eloc 14
nop 1
dl 0
loc 16
ccs 8
cts 8
cp 1
crap 1
rs 9.7
c 0
b 0
f 0
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_b = get_interface_mock('s2-eth1', 1, mock_switch_b)
441 1
        mock_interface_b.id = dpid_b + ':1'
442 1
        mock_interface_b.available_tags = [1, 2, 3]
443 1
        mock_switch_a.interfaces = {1: mock_interface_a}
444 1
        mock_switch_b.interfaces = {1: mock_interface_b}
445 1
        self.napp.controller.switches[dpid_a] = mock_switch_a
446 1
        self.napp.controller.switches[dpid_b] = mock_switch_b
447 1
        link_attrs = {
448
            'enabled': True,
449
            'id': link_id,
450
            'metadata': {},
451
            'endpoint_a': {
452
                'id': mock_interface_a.id
453
            },
454
            'endpoint_b': {
455
                'id': mock_interface_b.id
456
            }
457
        }
458
459 1
        self.napp._load_link(link_attrs)
460
461 1
        assert len(self.napp.controller.links) == 1
462 1
        link = list(self.napp.controller.links.values())[0]
463
464 1
        assert link.endpoint_a.id == mock_interface_a.id
465 1
        assert link.endpoint_b.id == mock_interface_b.id
466 1
        assert mock_interface_a.nni
467 1
        assert mock_interface_b.nni
468 1
        assert mock_interface_a.update_link.call_count == 1
469 1
        assert mock_interface_b.update_link.call_count == 1
470
471
        # test enable/disable
472 1
        link_id = '4d42dc08522'
473 1
        mock_interface_a = get_interface_mock('s1-eth1', 1, mock_switch_a)
474 1
        mock_interface_b = get_interface_mock('s2-eth1', 1, mock_switch_b)
475 1
        mock_link = get_link_mock(mock_interface_a, mock_interface_b)
476 1
        mock_link.id = link_id
477
478 1
        self.napp.controller.get_link_or_create = MagicMock()
479 1
        mock_get_link_or_create = self.napp.controller.get_link_or_create
480 1
        mock_get_link_or_create.return_value = (mock_link, True)
481
482
        # enable link
483 1
        link_attrs['enabled'] = True
484 1
        self.napp.controller.links = {link_id: mock_link}
485 1
        self.napp._load_link(link_attrs)
486 1
        assert mock_link.enable.call_count == 1
487
        # disable link
488 1
        link_attrs['enabled'] = False
489 1
        self.napp.controller.links = {link_id: mock_link}
490 1
        self.napp._load_link(link_attrs)
491 1
        assert mock_link.disable.call_count == 1
492
493 1
    def test_fail_load_link(self):
494
        """Test fail load_link."""
495 1
        self.napp.controller.get_link_or_create = MagicMock()
496 1
        mock_get_link_or_create = self.napp.controller.get_link_or_create
497 1
        dpid_a = '00:00:00:00:00:00:00:01'
498 1
        dpid_b = '00:00:00:00:00:00:00:02'
499 1
        link_id = '4d42dc08522'
500 1
        mock_switch_a = get_switch_mock(dpid_a)
501 1
        mock_switch_b = get_switch_mock(dpid_b)
502 1
        mock_interface_a_1 = get_interface_mock('s1-eth1', 1, mock_switch_a)
503 1
        mock_interface_b_1 = get_interface_mock('s2-eth1', 1, mock_switch_b)
504 1
        mock_link = get_link_mock(mock_interface_a_1, mock_interface_b_1)
505 1
        mock_link.id = link_id
506 1
        self.napp.controller.links = {link_id: mock_link}
507 1
        mock_get_link_or_create.return_value = mock_link
508
509 1
        link_attrs_fail = {
510
            'enabled': True,
511
            'id': link_id,
512
            'metadata': {},
513
            'endpoint_a': {
514
                'id': f"{dpid_a}:999",
515
            },
516
            'endpoint_b': {
517
                'id': f"{dpid_b}:999",
518
            }
519
        }
520 1
        with pytest.raises(RestoreError):
521 1
            self.napp._load_link(link_attrs_fail)
522
523 1
        link_attrs_fail = {
524
            'enabled': True,
525
            'id': link_id,
526
            'endpoint_a': {
527
                'id': f"{dpid_a}:1",
528
            },
529
            'endpoint_b': {
530
                'id': f"{dpid_b}:1",
531
            }
532
        }
533 1
        with pytest.raises(RestoreError):
534 1
            self.napp._load_link(link_attrs_fail)
535
536 1
    @patch('napps.kytos.topology.main.Main.notify_switch_links_status')
537 1
    @patch('napps.kytos.topology.main.Main.notify_topology_update')
538 1
    async def test_enable_switch(self, mock_notify_topo, mock_sw_l_status):
539
        """Test enable_switch."""
540 1
        dpid = "00:00:00:00:00:00:00:01"
541 1
        mock_switch = get_switch_mock(dpid)
542 1
        self.napp.controller.switches = {dpid: mock_switch}
543
544 1
        endpoint = f"{self.base_endpoint}/switches/{dpid}/enable"
545 1
        response = await self.api_client.post(endpoint)
546 1
        assert response.status_code == 201
547 1
        assert mock_switch.enable.call_count == 1
548 1
        self.napp.topo_controller.enable_switch.assert_called_once_with(dpid)
549 1
        mock_notify_topo.assert_called()
550 1
        mock_sw_l_status.assert_called()
551
552
        # fail case
553 1
        mock_switch.enable.call_count = 0
554 1
        dpid = "00:00:00:00:00:00:00:02"
555 1
        endpoint = f"{self.base_endpoint}/switches/{dpid}/enable"
556 1
        response = await self.api_client.post(endpoint)
557
        assert response.status_code == 404
558
        assert mock_switch.enable.call_count == 0
559
560 1
    @patch('napps.kytos.topology.main.Main.notify_link_enabled_state')
561 1
    @patch('napps.kytos.topology.main.Main.notify_switch_links_status')
562 1
    @patch('napps.kytos.topology.main.Main.notify_topology_update')
563 1
    async def test_disable_switch(self, *args):
564
        """Test disable_switch."""
565 1
        mock_notify_topo, mock_sw_l_status, mock_noti_link = args
566 1
        dpid = "00:00:00:00:00:00:00:01"
567 1
        mock_switch = get_switch_mock(dpid)
568 1
        interface = Mock()
569 1
        interface.link.is_enabled = lambda: True
570 1
        interface.link.link_lock = MagicMock()
571 1
        mock_switch.interfaces = {1: interface}
572 1
        self.napp.controller.switches = {dpid: mock_switch}
573
574 1
        endpoint = f"{self.base_endpoint}/switches/{dpid}/disable"
575 1
        response = await self.api_client.post(endpoint)
576 1
        assert response.status_code == 201
577 1
        assert mock_switch.disable.call_count == 1
578 1
        assert mock_noti_link.call_count == 1
579 1
        assert mock_noti_link.call_args[0][0] == interface.link
580 1
        assert mock_noti_link.call_args[0][1] == "disabled"
581 1
        assert interface.link.disable.call_count == 1
582 1
        assert self.napp.topo_controller.bulk_disable_links.call_count == 1
583 1
        self.napp.topo_controller.disable_switch.assert_called_once_with(dpid)
584 1
        mock_notify_topo.assert_called()
585 1
        mock_sw_l_status.assert_called()
586
587
        # fail case
588 1
        mock_switch.disable.call_count = 0
589 1
        dpid = "00:00:00:00:00:00:00:02"
590 1
        endpoint = f"{self.base_endpoint}/switches/{dpid}/disable"
591 1
        response = await self.api_client.post(endpoint)
592
        assert response.status_code == 404
593
        assert mock_switch.disable.call_count == 0
594
595 1
    async def test_get_switch_metadata(self):
596
        """Test get_switch_metadata."""
597 1
        dpid = "00:00:00:00:00:00:00:01"
598 1
        mock_switch = get_switch_mock(dpid)
599 1
        mock_switch.metadata = "A"
600 1
        self.napp.controller.switches = {dpid: mock_switch}
601
602 1
        endpoint = f"{self.base_endpoint}/switches/{dpid}/metadata"
603 1
        response = await self.api_client.get(endpoint)
604 1
        assert response.status_code == 200
605 1
        assert response.json() == {"metadata": mock_switch.metadata}
606
607
        # fail case
608 1
        dpid = "00:00:00:00:00:00:00:02"
609 1
        endpoint = f"{self.base_endpoint}/switches/{dpid}/metadata"
610 1
        response = await self.api_client.get(endpoint)
611
        assert response.status_code == 404
612
613 1
    @patch('napps.kytos.topology.main.Main.notify_metadata_changes')
614 1
    async def test_add_switch_metadata(
615
        self, mock_metadata_changes
616
    ):
617
        """Test add_switch_metadata."""
618 1
        self.napp.controller.loop = asyncio.get_running_loop()
619 1
        dpid = "00:00:00:00:00:00:00:01"
620 1
        mock_switch = get_switch_mock(dpid)
621 1
        self.napp.controller.switches = {dpid: mock_switch}
622 1
        payload = {"data": "A"}
623
624 1
        endpoint = f"{self.base_endpoint}/switches/{dpid}/metadata"
625 1
        response = await self.api_client.post(endpoint, json=payload)
626 1
        assert response.status_code == 201
627
628 1
        mock_metadata_changes.assert_called()
629 1
        self.napp.topo_controller.add_switch_metadata.assert_called_once_with(
630
            dpid, payload
631
        )
632
633
        # fail case
634 1
        dpid = "00:00:00:00:00:00:00:02"
635 1
        endpoint = f"{self.base_endpoint}/switches/{dpid}/metadata"
636 1
        response = await self.api_client.post(endpoint, json=payload)
637
        assert response.status_code == 404
638
639 1
    async def test_add_switch_metadata_wrong_format(self):
640
        """Test add_switch_metadata_wrong_format."""
641 1
        self.napp.controller.loop = asyncio.get_running_loop()
642 1
        dpid = "00:00:00:00:00:00:00:01"
643 1
        payload = 'A'
644
645 1
        endpoint = f"{self.base_endpoint}/switches/{dpid}/metadata"
646 1
        response = await self.api_client.post(endpoint, json=payload)
647
        assert response.status_code == 400
648
649
        payload = None
650
        response = await self.api_client.post(endpoint, json=payload)
651
        assert response.status_code == 415
652
653 1
    @patch('napps.kytos.topology.main.Main.notify_metadata_changes')
654 1
    async def test_delete_switch_metadata(
655
        self, mock_metadata_changes
656
    ):
657
        """Test delete_switch_metadata."""
658 1
        self.napp.controller.loop = asyncio.get_running_loop()
659 1
        dpid = "00:00:00:00:00:00:00:01"
660 1
        mock_switch = get_switch_mock(dpid)
661 1
        mock_switch.metadata = {"A": "A"}
662 1
        self.napp.controller.switches = {dpid: mock_switch}
663
664 1
        key = "A"
665 1
        endpoint = f"{self.base_endpoint}/switches/{dpid}/metadata/{key}"
666 1
        response = await self.api_client.delete(endpoint)
667
668 1
        assert response.status_code == 200
669 1
        assert mock_metadata_changes.call_count == 1
670 1
        del_key_mock = self.napp.topo_controller.delete_switch_metadata_key
671 1
        del_key_mock.assert_called_with(
672
            dpid, key
673
        )
674
675
        # 404, Metadata not found
676 1
        mock_switch.metadata = {}
677 1
        endpoint = f"{self.base_endpoint}/switches/{dpid}/metadata/{key}"
678 1
        response = await self.api_client.delete(endpoint)
679
        assert response.status_code == 404
680
681
        # 404, Switch not found
682
        key = "A"
683
        dpid = "00:00:00:00:00:00:00:02"
684
        endpoint = f"{self.base_endpoint}/switches/{dpid}/metadata/{key}"
685
        response = await self.api_client.delete(endpoint)
686
        assert mock_metadata_changes.call_count == 1
687
        assert response.status_code == 404
688
689
    # pylint: disable=too-many-statements
690 1
    @patch('napps.kytos.topology.main.Main.notify_topology_update')
691 1
    async def test_enable_interfaces(self, mock_notify_topo):
692
        """Test enable_interfaces."""
693 1
        dpid = '00:00:00:00:00:00:00:01'
694 1
        mock_switch = get_switch_mock(dpid)
695 1
        mock_interface_1 = get_interface_mock('s1-eth1', 1, mock_switch)
696 1
        mock_interface_1.link = Mock()
697 1
        mock_interface_1.link.link_lock = MagicMock()
698 1
        mock_interface_1.link._enabled = True
699 1
        mock_interface_2 = get_interface_mock('s1-eth2', 2, mock_switch)
700 1
        mock_interface_2.link = Mock()
701 1
        mock_interface_2.link.link_lock = MagicMock()
702 1
        mock_interface_2.link._enabled = False
703 1
        mock_switch.interfaces = {1: mock_interface_1, 2: mock_interface_2}
704
705
        # Switch not found
706 1
        interface_id = '00:00:00:00:00:00:00:01:1'
707 1
        endpoint = f"{self.base_endpoint}/interfaces/{interface_id}/enable"
708 1
        response = await self.api_client.post(endpoint)
709
        assert response.status_code == 404
710
711
        # Switch not enabled
712 1
        mock_switch.is_enabled = lambda: False
713
        self.napp.controller.switches = {dpid: mock_switch}
714
        endpoint = f"{self.base_endpoint}/interfaces/{interface_id}/enable"
715
        response = await self.api_client.post(endpoint)
716
        assert response.status_code == 409
717
718
        # Success
719 1
        mock_switch.is_enabled = lambda: True
720
        endpoint = f"{self.base_endpoint}/interfaces/{interface_id}/enable"
721
        response = await self.api_client.post(endpoint)
722 1
        assert response.status_code == 200
723 1
        assert mock_interface_1.enable.call_count == 1
724 1
        assert mock_interface_2.enable.call_count == 0
725 1
        self.napp.topo_controller.enable_interface.assert_called_with(
726
            interface_id
727
        )
728 1
        mock_notify_topo.assert_called()
729
730 1
        mock_interface_1.enable.call_count = 0
731 1
        mock_interface_2.enable.call_count = 0
732 1
        endpoint = f"{self.base_endpoint}/interfaces/switch/{dpid}/enable"
733 1
        response = await self.api_client.post(endpoint)
734 1
        assert response.status_code == 200
735 1
        self.napp.topo_controller.upsert_switch.assert_called_with(
736
            mock_switch.id, mock_switch.as_dict()
737
        )
738 1
        assert mock_interface_1.enable.call_count == 1
739 1
        assert mock_interface_2.enable.call_count == 1
740
741
        # test interface not found
742 1
        interface_id = '00:00:00:00:00:00:00:01:3'
743 1
        mock_interface_1.enable.call_count = 0
744 1
        mock_interface_2.enable.call_count = 0
745 1
        endpoint = f"{self.base_endpoint}/interfaces/{interface_id}/enable"
746 1
        response = await self.api_client.post(endpoint)
747
        assert response.status_code == 404
748
        assert mock_interface_1.enable.call_count == 0
749
        assert mock_interface_2.enable.call_count == 0
750
751
        # test switch not found
752
        dpid = '00:00:00:00:00:00:00:02'
753
        endpoint = f"{self.base_endpoint}/interfaces/{interface_id}/enable"
754
        response = await self.api_client.post(endpoint)
755
        assert response.status_code == 404
756
        assert mock_interface_1.enable.call_count == 0
757
        assert mock_interface_2.enable.call_count == 0
758
759 1
    @patch('napps.kytos.topology.main.Main.notify_link_enabled_state')
760 1
    @patch('napps.kytos.topology.main.Main.notify_topology_update')
761 1
    async def test_disable_interfaces(self, mock_notify_topo, mock_noti_link):
762
        """Test disable_interfaces."""
763 1
        interface_id = '00:00:00:00:00:00:00:01:1'
764 1
        dpid = '00:00:00:00:00:00:00:01'
765 1
        mock_switch = get_switch_mock(dpid)
766 1
        mock_interface_1 = get_interface_mock('s1-eth1', 1, mock_switch)
767 1
        mock_interface_1.link = Mock()
768 1
        mock_interface_1.link.is_enabled = lambda: True
769 1
        mock_interface_1.link.link_lock = MagicMock()
770 1
        mock_interface_2 = get_interface_mock('s1-eth2', 2, mock_switch)
771 1
        mock_interface_2.link = Mock()
772 1
        mock_interface_2.link.is_enabled = lambda: False
773 1
        mock_interface_2.link.link_lock = MagicMock()
774 1
        mock_switch.interfaces = {1: mock_interface_1, 2: mock_interface_2}
775 1
        self.napp.controller.switches = {dpid: mock_switch}
776
777 1
        endpoint = f"{self.base_endpoint}/interfaces/{interface_id}/disable"
778 1
        response = await self.api_client.post(endpoint)
779 1
        assert response.status_code == 200
780
781 1
        self.napp.topo_controller.disable_interface.assert_called_with(
782
            interface_id
783
        )
784 1
        assert mock_interface_1.disable.call_count == 1
785 1
        assert mock_interface_2.disable.call_count == 0
786 1
        assert mock_interface_1.link.disable.call_count == 1
787 1
        assert mock_noti_link.call_count == 1
788 1
        assert mock_noti_link.call_args[0][0] == mock_interface_1.link
789 1
        assert mock_noti_link.call_args[0][1] == "disabled"
790 1
        assert self.napp.topo_controller.disable_interface.call_count == 1
791 1
        mock_notify_topo.assert_called()
792
793 1
        mock_interface_1.disable.call_count = 0
794 1
        mock_interface_2.disable.call_count = 0
795
796 1
        endpoint = f"{self.base_endpoint}/interfaces/switch/{dpid}/disable"
797 1
        response = await self.api_client.post(endpoint)
798 1
        assert response.status_code == 200
799
800 1
        self.napp.topo_controller.upsert_switch.assert_called_with(
801
            mock_switch.id, mock_switch.as_dict()
802
        )
803 1
        assert mock_interface_1.disable.call_count == 1
804 1
        assert mock_interface_1.link.disable.call_count == 2
805 1
        assert mock_interface_2.disable.call_count == 1
806 1
        assert mock_noti_link.call_count == 2
807 1
        assert mock_noti_link.call_args[0][0] == mock_interface_1.link
808 1
        assert mock_noti_link.call_args[0][1] == "disabled"
809 1
        bulk_controller = self.napp.topo_controller.bulk_disable_links
810 1
        assert bulk_controller.call_count == 2
811 1
        assert len(bulk_controller.call_args[0][0]) == 1
812
813
        # test interface not found
814 1
        interface_id = '00:00:00:00:00:00:00:01:3'
815 1
        mock_interface_1.disable.call_count = 0
816 1
        mock_interface_2.disable.call_count = 0
817 1
        endpoint = f"{self.base_endpoint}/interfaces/{interface_id}/disable"
818 1
        response = await self.api_client.post(endpoint)
819
820
        assert response.status_code == 404
821
        assert mock_interface_1.disable.call_count == 0
822
        assert mock_interface_2.disable.call_count == 0
823
824
        # test switch not found
825
        dpid = '00:00:00:00:00:00:00:02'
826
        endpoint = f"{self.base_endpoint}/interfaces/switch/{dpid}/disable"
827
        response = await self.api_client.post(endpoint)
828
        assert response.status_code == 404
829
        assert mock_interface_1.disable.call_count == 0
830
        assert mock_interface_2.disable.call_count == 0
831
832 1
    async def test_get_interface_metadata(self):
833
        """Test get_interface_metada."""
834 1
        interface_id = '00:00:00:00:00:00:00:01:1'
835 1
        dpid = '00:00:00:00:00:00:00:01'
836 1
        mock_switch = get_switch_mock(dpid)
837 1
        mock_interface = get_interface_mock('s1-eth1', 1, mock_switch)
838 1
        mock_interface.metadata = {"A": "B"}
839 1
        mock_switch.interfaces = {1: mock_interface}
840 1
        self.napp.controller.switches = {dpid: mock_switch}
841
842 1
        endpoint = f"{self.base_endpoint}/interfaces/{interface_id}/metadata"
843 1
        response = await self.api_client.get(endpoint)
844 1
        assert response.status_code == 200
845 1
        assert response.json() == {"metadata": mock_interface.metadata}
846
847
        # fail case switch not found
848 1
        interface_id = '00:00:00:00:00:00:00:02:1'
849 1
        endpoint = f"{self.base_endpoint}/interfaces/{interface_id}/metadata"
850 1
        response = await self.api_client.get(endpoint)
851
        assert response.status_code == 404
852
853
        # fail case interface not found
854
        interface_id = '00:00:00:00:00:00:00:01:2'
855
        endpoint = f"{self.base_endpoint}/interfaces/{interface_id}/metadata"
856
        response = await self.api_client.get(endpoint)
857
        assert response.status_code == 404
858
859 1
    @patch('napps.kytos.topology.main.Main.notify_metadata_changes')
860 1
    async def test_add_interface_metadata(
861
        self, mock_metadata_changes
862
    ):
863
        """Test add_interface_metadata."""
864 1
        self.napp.controller.loop = asyncio.get_running_loop()
865 1
        interface_id = '00:00:00:00:00:00:00:01:1'
866 1
        dpid = '00:00:00:00:00:00:00:01'
867 1
        mock_switch = get_switch_mock(dpid)
868 1
        mock_interface = get_interface_mock('s1-eth1', 1, mock_switch)
869 1
        mock_interface.metadata = {"metada": "A"}
870 1
        mock_switch.interfaces = {1: mock_interface}
871 1
        self.napp.controller.switches = {dpid: mock_switch}
872 1
        payload = {"metada": "A"}
873 1
        endpoint = f"{self.base_endpoint}/interfaces/{interface_id}/metadata"
874 1
        response = await self.api_client.post(endpoint, json=payload)
875 1
        assert response.status_code == 201
876 1
        mock_metadata_changes.assert_called()
877
878
        # fail case switch not found
879 1
        interface_id = '00:00:00:00:00:00:00:02:1'
880 1
        endpoint = f"{self.base_endpoint}/interfaces/{interface_id}/metadata"
881 1
        response = await self.api_client.post(endpoint, json=payload)
882
        assert response.status_code == 404
883
884
        # fail case interface not found
885
        interface_id = '00:00:00:00:00:00:00:01:2'
886
        endpoint = f"{self.base_endpoint}/interfaces/{interface_id}/metadata"
887
        response = await self.api_client.post(endpoint, json=payload)
888
        assert response.status_code == 404
889
890 1
    async def test_add_interface_metadata_wrong_format(self):
891
        """Test add_interface_metadata_wrong_format."""
892 1
        self.napp.controller.loop = asyncio.get_running_loop()
893 1
        interface_id = "00:00:00:00:00:00:00:01:1"
894 1
        endpoint = f"{self.base_endpoint}/interfaces/{interface_id}/metadata"
895 1
        response = await self.api_client.post(endpoint, json='A')
896
        assert response.status_code == 400
897
        response = await self.api_client.post(endpoint, json=None)
898
        assert response.status_code == 415
899
900 1
    async def test_delete_interface_metadata(self):
901
        """Test delete_interface_metadata."""
902 1
        self.napp.controller.loop = asyncio.get_running_loop()
903 1
        interface_id = '00:00:00:00:00:00:00:01:1'
904 1
        dpid = '00:00:00:00:00:00:00:01'
905 1
        mock_switch = get_switch_mock(dpid)
906 1
        mock_interface = get_interface_mock('s1-eth1', 1, mock_switch)
907 1
        mock_interface.remove_metadata.side_effect = [True, False]
908 1
        mock_interface.metadata = {"A": "A"}
909 1
        mock_switch.interfaces = {1: mock_interface}
910 1
        self.napp.controller.switches = {'00:00:00:00:00:00:00:01':
911
                                         mock_switch}
912
913 1
        key = 'A'
914 1
        url = f"{self.base_endpoint}/interfaces/{interface_id}/metadata/{key}"
915 1
        response = await self.api_client.delete(url)
916 1
        assert response.status_code == 200
917
918 1
        del_key_mock = self.napp.topo_controller.delete_interface_metadata_key
919 1
        del_key_mock.assert_called_once_with(interface_id, key)
920
921
        # fail case switch not found
922 1
        key = 'A'
923 1
        interface_id = '00:00:00:00:00:00:00:02:1'
924 1
        url = f"{self.base_endpoint}/interfaces/{interface_id}/metadata/{key}"
925 1
        response = await self.api_client.delete(url)
926
        assert response.status_code == 404
927
928
        # fail case interface not found
929
        key = 'A'
930
        interface_id = '00:00:00:00:00:00:00:01:2'
931
        url = f"{self.base_endpoint}/interfaces/{interface_id}/metadata/{key}"
932
        response = await self.api_client.delete(url)
933
        assert response.status_code == 404
934
935
        # fail case metadata not found
936
        key = 'B'
937
        interface_id = '00:00:00:00:00:00:00:01:1'
938
        url = f"{self.base_endpoint}/interfaces/{interface_id}/metadata/{key}"
939
        response = await self.api_client.delete(url)
940
        assert response.status_code == 404
941
942 1
    @patch('napps.kytos.topology.main.Main.notify_link_enabled_state')
943 1
    @patch('napps.kytos.topology.main.Main.notify_topology_update')
944 1
    async def test_enable_link(self, mock_notify_topo, mock_noti_link):
945
        """Test enable_link."""
946 1
        mock_link = MagicMock(Link)
947 1
        mock_link.link_lock = MagicMock()
948 1
        link_id = "1"
949 1
        mock_link.id = link_id
950 1
        mock_link.is_enabled = lambda: False
951 1
        mock_link.endpoint_a = MagicMock(is_enabled=lambda: True)
952 1
        mock_link.endpoint_b = MagicMock(is_enabled=lambda: True)
953 1
        self.napp.controller.links = {'1': mock_link}
954
955
        # 409, endpoint is/are disabled
956 1
        mock_link.endpoint_a.is_enabled = lambda: False
957 1
        mock_link.endpoint_b.is_enabled = lambda: False
958 1
        endpoint = f"{self.base_endpoint}/links/{link_id}/enable"
959 1
        response = await self.api_client.post(endpoint)
960
        assert response.status_code == 409
961
962 1
        mock_link.endpoint_a.is_enabled = lambda: True
963
        endpoint = f"{self.base_endpoint}/links/{link_id}/enable"
964
        response = await self.api_client.post(endpoint)
965
        assert response.status_code == 409
966
967
        # Success
968 1
        mock_link.endpoint_b.is_enabled = lambda: True
969
        endpoint = f"{self.base_endpoint}/links/{link_id}/enable"
970
        response = await self.api_client.post(endpoint)
971 1
        assert response.status_code == 201
972 1
        assert mock_noti_link.call_count == 1
973 1
        assert mock_noti_link.call_args[0][0] == mock_link
974 1
        assert mock_noti_link.call_args[0][1] == "enabled"
975 1
        mock_notify_topo.assert_called()
976
977
        # 404, link not found
978 1
        link_id = "2"
979 1
        endpoint = f"{self.base_endpoint}/links/{link_id}/enable"
980 1
        response = await self.api_client.post(endpoint)
981
        assert response.status_code == 404
982
        assert mock_noti_link.call_count == 1
983
984 1
    @patch('napps.kytos.topology.main.Main.notify_link_enabled_state')
985 1
    @patch('napps.kytos.topology.main.Main.notify_topology_update')
986 1
    async def test_disable_link(self, mock_notify_topo, mock_notify):
987
        """Test disable_link."""
988 1
        mock_link = MagicMock(Link)
989 1
        mock_link.link_lock = MagicMock()
990 1
        mock_link.is_enabled = lambda: True
991 1
        self.napp.controller.links = {'1': mock_link}
992
993 1
        link_id = "1"
994 1
        endpoint = f"{self.base_endpoint}/links/{link_id}/disable"
995 1
        response = await self.api_client.post(endpoint)
996 1
        assert response.status_code == 201
997 1
        assert mock_notify_topo.call_count == 1
998 1
        assert mock_notify.call_count == 1
999 1
        assert mock_notify.call_args[0][0] == mock_link
1000 1
        assert mock_notify.call_args[0][1] == "disabled"
1001 1
        assert mock_link.disable.call_count == 1
1002 1
        assert self.napp.topo_controller.disable_link.call_count == 1
1003
1004
        # fail case
1005 1
        link_id = "2"
1006 1
        endpoint = f"{self.base_endpoint}/links/{link_id}/disable"
1007 1
        response = await self.api_client.post(endpoint)
1008
        assert response.status_code == 404
1009
1010 1
    def test_handle_lldp_status_updated(self):
1011
        """Test handle_lldp_status_updated."""
1012 1
        event = MagicMock()
1013 1
        self.napp.controller.buffers.app.put = MagicMock()
1014
1015 1
        dpid_a = "00:00:00:00:00:00:00:01"
1016 1
        dpid_b = "00:00:00:00:00:00:00:02"
1017 1
        dpids = [dpid_a, dpid_b]
1018 1
        interface_ids = [f"{dpid}:1" for dpid in dpids]
1019
1020 1
        mock_switch_a = get_switch_mock(dpid_a, 0x04)
1021 1
        mock_switch_b = get_switch_mock(dpid_b, 0x04)
1022 1
        self.napp.controller.switches = {dpid_a: mock_switch_a,
1023
                                         dpid_b: mock_switch_b}
1024
1025 1
        event.content = {"interface_ids": interface_ids, "state": "disabled"}
1026 1
        self.napp.handle_lldp_status_updated(event)
1027
1028 1
        mock_put = self.napp.controller.buffers.app.put
1029 1
        assert mock_put.call_count == len(interface_ids)
1030
1031 1
    def test_handle_topo_controller_upsert_switch(self):
1032
        """Test handle_topo_controller_upsert_switch."""
1033 1
        event = MagicMock()
1034 1
        self.napp.handle_topo_controller_upsert_switch(event)
1035 1
        mock = self.napp.topo_controller.upsert_switch
1036 1
        mock.assert_called_with(event.id, event.as_dict())
1037
1038 1
    async def test_get_link_metadata(self):
1039
        """Test get_link_metadata."""
1040 1
        mock_link = MagicMock(Link)
1041 1
        mock_link.link_lock = MagicMock()
1042 1
        mock_link.metadata = "A"
1043 1
        self.napp.controller.links = {'1': mock_link}
1044 1
        msg_success = {"metadata": "A"}
1045
1046 1
        link_id = "1"
1047 1
        endpoint = f"{self.base_endpoint}/links/{link_id}/metadata"
1048 1
        response = await self.api_client.get(endpoint)
1049 1
        assert response.status_code == 200
1050 1
        assert msg_success == response.json()
1051
1052
        # fail case
1053 1
        link_id = "2"
1054 1
        endpoint = f"{self.base_endpoint}/links/{link_id}/metadata"
1055 1
        response = await self.api_client.get(endpoint)
1056
        assert response.status_code == 404
1057
1058 1
    @patch('napps.kytos.topology.main.Main.notify_topology_update')
1059 1
    @patch('napps.kytos.topology.main.Main.notify_metadata_changes')
1060 1
    async def test_add_link_metadata(
1061
        self,
1062
        mock_metadata_changes,
1063
        mock_topology_update
1064
    ):
1065
        """Test add_link_metadata."""
1066 1
        self.napp.controller.loop = asyncio.get_running_loop()
1067 1
        mock_link = MagicMock(Link)
1068 1
        mock_link.link_lock = MagicMock()
1069 1
        mock_link.metadata = "A"
1070 1
        self.napp.controller.links = {'1': mock_link}
1071 1
        payload = {"metadata": "A"}
1072 1
        link_id = 1
1073
1074 1
        endpoint = f"{self.base_endpoint}/links/{link_id}/metadata"
1075 1
        response = await self.api_client.post(endpoint, json=payload)
1076 1
        assert response.status_code == 201
1077 1
        mock_metadata_changes.assert_called()
1078 1
        mock_topology_update.assert_called()
1079
1080
        # fail case
1081 1
        link_id = 2
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 == 404
1085
1086 1
    async def test_add_link_metadata_wrong_format(self):
1087
        """Test add_link_metadata_wrong_format."""
1088 1
        self.napp.controller.loop = asyncio.get_running_loop()
1089 1
        link_id = 'cf0f4071be426b3f745027f5d22'
1090 1
        payload = "A"
1091 1
        endpoint = f"{self.base_endpoint}/links/{link_id}/metadata"
1092 1
        response = await self.api_client.post(endpoint, json=payload)
1093
        assert response.status_code == 400
1094
1095
        payload = None
1096
        response = await self.api_client.post(endpoint, json=payload)
1097
        assert response.status_code == 415
1098
1099 1
    @patch('napps.kytos.topology.main.Main.notify_topology_update')
1100 1
    @patch('napps.kytos.topology.main.Main.notify_metadata_changes')
1101 1
    async def test_delete_link_metadata(
1102
        self,
1103
        mock_metadata_changes,
1104
        mock_topology_update
1105
    ):
1106
        """Test delete_link_metadata."""
1107 1
        mock_link = MagicMock(Link)
1108 1
        mock_link.link_lock = MagicMock()
1109 1
        mock_link.metadata = {"A": "A"}
1110 1
        mock_link.remove_metadata.side_effect = [True, False]
1111 1
        self.napp.controller.links = {'1': mock_link}
1112
1113 1
        link_id = 1
1114 1
        key = 'A'
1115 1
        endpoint = f"{self.base_endpoint}/links/{link_id}/metadata/{key}"
1116 1
        response = await self.api_client.delete(endpoint)
1117 1
        assert response.status_code == 200
1118 1
        del_mock = self.napp.topo_controller.delete_link_metadata_key
1119 1
        del_mock.assert_called_once_with(mock_link.id, key)
1120 1
        mock_metadata_changes.assert_called()
1121 1
        mock_topology_update.assert_called()
1122
1123
        # fail case link not found
1124 1
        link_id = 2
1125 1
        key = 'A'
1126 1
        endpoint = f"{self.base_endpoint}/links/{link_id}/metadata/{key}"
1127 1
        response = await self.api_client.delete(endpoint)
1128
        assert response.status_code == 404
1129
1130
        # fail case metadata not found
1131
        link_id = 1
1132
        key = 'B'
1133
        endpoint = f"{self.base_endpoint}/links/{link_id}/metadata/{key}"
1134
        response = await self.api_client.delete(endpoint)
1135
        assert response.status_code == 404
1136
1137 1
    @patch('napps.kytos.topology.main.Main.notify_topology_update')
1138 1
    def test_handle_new_switch(self, mock_notify_topology_update):
1139
        """Test handle_new_switch."""
1140 1
        mock_event = MagicMock()
1141 1
        mock_switch = create_autospec(Switch)
1142 1
        mock_event.content['switch'] = mock_switch
1143 1
        self.napp.handle_new_switch(mock_event)
1144 1
        mock = self.napp.topo_controller.upsert_switch
1145 1
        mock.assert_called_once_with(mock_event.content['switch'].id,
1146
                                     mock_event.content['switch'].as_dict())
1147 1
        mock_notify_topology_update.assert_called()
1148
1149 1
    @patch('napps.kytos.topology.main.Main.notify_topology_update')
1150 1
    def test_handle_connection_lost(self, mock_notify_topology_update):
1151
        """Test handle connection_lost."""
1152 1
        mock_event = MagicMock()
1153 1
        mock_switch = create_autospec(Switch)
1154 1
        mock_switch.return_value = True
1155 1
        mock_event.content['source'] = mock_switch
1156 1
        self.napp.handle_connection_lost(mock_event)
1157 1
        mock_notify_topology_update.assert_called()
1158
1159 1
    @patch('napps.kytos.topology.main.Main.handle_interface_link_down')
1160 1
    @patch('napps.kytos.topology.main.Main.handle_interface_link_up')
1161 1
    def test_handle_interface_created(self, mock_link_up, mock_link_down):
1162
        """Test handle_interface_created."""
1163 1
        mock_event = MagicMock()
1164 1
        mock_interface = create_autospec(Interface)
1165 1
        mock_interface.id = "1"
1166 1
        mock_event.content = {'interface': mock_interface}
1167 1
        self.napp.handle_interface_created(mock_event)
1168 1
        mock_link_up.assert_called()
1169 1
        mock_link_down.assert_not_called()
1170
1171 1
    @patch('napps.kytos.topology.main.Main.handle_interface_link_down')
1172 1
    @patch('napps.kytos.topology.main.Main.handle_interface_link_up')
1173 1
    def test_handle_interface_created_inactive(self, mock_link_up,
1174
                                               mock_link_down):
1175
        """Test handle_interface_created inactive."""
1176 1
        mock_event = MagicMock()
1177 1
        mock_interface = create_autospec(Interface)
1178 1
        mock_interface.id = "1"
1179 1
        mock_event.content = {'interface': mock_interface}
1180 1
        mock_interface.is_active.return_value = False
1181 1
        self.napp.handle_interface_created(mock_event)
1182 1
        mock_link_up.assert_not_called()
1183 1
        mock_link_down.assert_called()
1184
1185 1
    def test_handle_interfaces_created(self):
1186
        """Test handle_interfaces_created."""
1187 1
        buffers_app_mock = MagicMock()
1188 1
        self.napp.controller.buffers.app = buffers_app_mock
1189 1
        mock_switch = create_autospec(Switch)
1190 1
        mock_event = MagicMock()
1191 1
        mock_interface = create_autospec(Interface)
1192 1
        mock_interface.id = "1"
1193 1
        mock_interface.switch = mock_switch
1194 1
        mock_interface_two = create_autospec(Interface)
1195 1
        mock_interface_two.id = "2"
1196 1
        mock_event.content = {'interfaces': [mock_interface,
1197
                              mock_interface_two]}
1198 1
        self.napp.handle_interfaces_created(mock_event)
1199 1
        upsert_mock = self.napp.topo_controller.upsert_switch
1200 1
        upsert_mock.assert_called_with(mock_switch.id, mock_switch.as_dict())
1201 1
        assert self.napp.controller.buffers.app.put.call_count == 2
1202
1203 1
    @patch('napps.kytos.topology.main.Main.handle_interface_link_down')
1204 1
    def test_handle_interface_down(self, mock_handle_interface_link_down):
1205
        """Test handle interface down."""
1206 1
        mock_event = MagicMock()
1207 1
        mock_interface = create_autospec(Interface)
1208 1
        mock_event.content['interface'] = mock_interface
1209 1
        self.napp.handle_interface_down(mock_event)
1210 1
        mock_handle_interface_link_down.assert_called()
1211
1212 1
    @patch('napps.kytos.topology.main.Main.handle_interface_down')
1213 1
    def test_interface_deleted(self, mock_handle_interface_link_down):
1214
        """Test interface deleted."""
1215 1
        mock_event = MagicMock()
1216 1
        self.napp.handle_interface_deleted(mock_event)
1217 1
        mock_handle_interface_link_down.assert_called()
1218
1219 1
    @patch('napps.kytos.topology.main.Main.notify_topology_update')
1220 1
    def test_interface_link_up(self, mock_notify_topology_update):
1221
        """Test interface link_up."""
1222 1
        self.napp.controller.buffers.app.put = MagicMock()
1223
1224 1
        tnow = time.time()
1225 1
        mock_switch_a = create_autospec(Switch)
1226 1
        mock_switch_a.is_active.return_value = True
1227 1
        mock_switch_b = create_autospec(Switch)
1228 1
        mock_switch_b.is_active.return_value = True
1229 1
        mock_interface_a = create_autospec(Interface)
1230 1
        mock_interface_a.switch = mock_switch_a
1231 1
        mock_interface_a.is_active.return_value = False
1232 1
        mock_interface_b = create_autospec(Interface)
1233 1
        mock_interface_b.switch = mock_switch_b
1234 1
        mock_interface_b.is_active.return_value = True
1235 1
        mock_link = create_autospec(Link)
1236 1
        mock_link.link_lock = MagicMock()
1237 1
        mock_link.get_metadata.return_value = tnow
1238 1
        mock_link.is_active.return_value = False
1239 1
        mock_link.endpoint_a = mock_interface_a
1240 1
        mock_link.endpoint_b = mock_interface_b
1241 1
        mock_link.status = EntityStatus.UP
1242 1
        mock_interface_a.link = mock_link
1243 1
        mock_interface_b.link = mock_link
1244 1
        event = KytosEvent("kytos.of_core.switch.interface.down")
1245 1
        self.napp.handle_interface_link_up(mock_interface_a, event)
1246 1
        mock_notify_topology_update.assert_called()
1247 1
        assert mock_link.id in self.napp.link_status_change
1248 1
        mock_link.activate.assert_called()
1249 1
        self.napp.controller.buffers.app.put.assert_not_called()
1250
1251 1
        mock_interface_a.is_active.return_value = True
1252 1
        event = KytosEvent("kytos.of_core.switch.interface.down")
1253 1
        self.napp.handle_interface_link_up(mock_interface_a, event)
1254
1255 1
        assert mock_link.id in self.napp.link_status_change
1256 1
        link_status_info = self.napp.link_status_change[mock_link.id]
1257 1
        mock_link.activate.assert_called()
1258 1
        assert self.napp.controller.buffers.app.put.call_count == 1
1259 1
        ev = "kytos/topology.notify_link_up_if_status"
1260 1
        assert self.napp.controller.buffers.app.put.call_args[0][0].name == ev
1261
1262 1
        mock_link.is_active.return_value = True
1263 1
        orig_change_time = link_status_info["last_status_change"]
1264
1265 1
        self.napp.handle_interface_link_up(mock_interface_a, event)
1266
1267 1
        link_status_info = self.napp.link_status_change[mock_link.id]
1268 1
        new_change_time = link_status_info["last_status_change"]
1269 1
        assert orig_change_time == new_change_time
1270
1271 1
    @patch('napps.kytos.topology.main.Main.notify_topology_update')
1272 1
    @patch('napps.kytos.topology.main.Main.notify_link_status_change')
1273 1
    def test_interface_link_down(self, *args):
1274
        """Test interface link down."""
1275 1
        mock_status_change, mock_topology_update = args
1276
1277 1
        mock_interface = create_autospec(Interface)
1278 1
        mock_link = create_autospec(Link)
1279 1
        mock_link.link_lock = MagicMock()
1280 1
        mock_link.is_active.return_value = True
1281 1
        mock_interface.link = mock_link
1282 1
        event = KytosEvent("kytos.of_core.switch.interface.link_up")
1283 1
        self.napp.handle_interface_link_down(mock_interface, event)
1284 1
        mock_topology_update.assert_called()
1285 1
        mock_status_change.assert_called()
1286
1287 1
    @patch('napps.kytos.topology.main.Main.notify_topology_update')
1288 1
    @patch('napps.kytos.topology.main.Main.notify_link_status_change')
1289 1
    def test_interface_link_down_unordered_event(self, *args):
1290
        """Test interface link down unordered event."""
1291 1
        (mock_status_change, mock_topology_update) = args
1292
1293 1
        mock_interface = create_autospec(Interface)
1294 1
        mock_interface.id = "1"
1295 1
        event_2 = KytosEvent("kytos.of_core.switch.interface.down")
1296 1
        event_1 = KytosEvent("kytos.of_core.switch.interface.up")
1297 1
        assert event_1.timestamp > event_2.timestamp
1298 1
        self.napp._intfs_updated_at[mock_interface.id] = event_1.timestamp
1299 1
        self.napp.handle_interface_link_down(mock_interface, event_2)
1300 1
        mock_topology_update.assert_not_called()
1301 1
        mock_status_change.assert_not_called()
1302
1303 1
    @patch('napps.kytos.topology.main.Main.notify_topology_update')
1304 1
    @patch('napps.kytos.topology.main.Main.notify_link_status_change')
1305 1
    def test_interface_link_up_unordered_event(self, *args):
1306
        """Test interface link up unordered event."""
1307 1
        (mock_status_change, mock_topology_update) = args
1308
1309 1
        mock_interface = create_autospec(Interface)
1310 1
        mock_interface.id = "1"
1311 1
        event_2 = KytosEvent("kytos.of_core.switch.interface.up")
1312 1
        event_1 = KytosEvent("kytos.of_core.switch.interface.down")
1313 1
        assert event_1.timestamp > event_2.timestamp
1314 1
        self.napp._intfs_updated_at[mock_interface.id] = event_1.timestamp
1315 1
        self.napp.handle_interface_link_up(mock_interface, event_2)
1316 1
        mock_topology_update.assert_not_called()
1317 1
        mock_status_change.assert_not_called()
1318
1319 1
    @patch('napps.kytos.topology.main.Main.notify_topology_update')
1320 1
    @patch('napps.kytos.topology.main.Main.notify_link_status_change')
1321 1
    def test_handle_link_down(self, *args):
1322
        """Test interface link down."""
1323 1
        (mock_status_change, mock_topology_update) = args
1324
1325 1
        mock_interface = create_autospec(Interface)
1326 1
        mock_link = create_autospec(Link)
1327 1
        mock_link.link_lock = MagicMock()
1328 1
        mock_link.is_active.return_value = True
1329 1
        mock_interface.link = mock_link
1330 1
        self.napp.handle_link_down(mock_interface)
1331 1
        mock_interface.deactivate.assert_not_called()
1332 1
        mock_link.deactivate.assert_called()
1333 1
        assert mock_topology_update.call_count == 1
1334 1
        mock_status_change.assert_called()
1335
1336 1
    @patch('napps.kytos.topology.main.Main.notify_topology_update')
1337 1
    def test_handle_link_down_not_active(self, args):
1338
        """Test interface link down with link not active."""
1339 1
        mock_topology_update = args
1340 1
        self.napp.controller.buffers.app.put = MagicMock()
1341
1342 1
        mock_interface = create_autospec(Interface)
1343 1
        mock_link = create_autospec(Link)
1344 1
        mock_link.link_lock = MagicMock()
1345 1
        mock_link.is_active.return_value = False
1346 1
        mock_link.get_metadata.return_value = False
1347 1
        mock_interface.link = mock_link
1348 1
        self.napp.link_up = set()
1349 1
        self.napp.link_status_change[mock_link.id] = {}
1350 1
        self.napp.handle_link_down(mock_interface)
1351 1
        mock_topology_update.assert_called()
1352 1
        self.napp.controller.buffers.app.put.assert_not_called()
1353
1354 1
    @patch('napps.kytos.topology.main.Main.notify_topology_update')
1355 1
    @patch('napps.kytos.topology.main.Main.notify_link_status_change')
1356 1
    def test_handle_link_down_not_active_last_status(self, *args):
1357
        """Test interface link down with link not active."""
1358 1
        (mock_status_change, mock_topology_update) = args
1359
1360 1
        mock_interface = create_autospec(Interface)
1361 1
        mock_link = create_autospec(Link)
1362 1
        mock_link.link_lock = MagicMock()
1363 1
        mock_link.is_active.return_value = False
1364 1
        mock_link.get_metadata.return_value = True
1365 1
        mock_interface.link = mock_link
1366 1
        self.napp.handle_link_down(mock_interface)
1367 1
        mock_topology_update.assert_called()
1368 1
        mock_status_change.assert_called()
1369
1370 1
    @patch('napps.kytos.topology.main.Main.notify_topology_update')
1371 1
    def test_handle_link_up(self, mock_notify_topology_update):
1372
        """Test handle link up."""
1373 1
        mock_switch_a = create_autospec(Switch)
1374 1
        mock_switch_a.is_active.return_value = True
1375 1
        mock_interface = create_autospec(Interface)
1376 1
        mock_interface.switch = mock_switch_a
1377 1
        mock_interface.is_active.return_value = True
1378 1
        mock_link = MagicMock(status=EntityStatus.UP)
1379 1
        mock_link.is_active.return_value = True
1380 1
        mock_interface.link = mock_link
1381 1
        self.napp.handle_link_up(mock_interface)
1382 1
        mock_interface.activate.assert_not_called()
1383 1
        mock_notify_topology_update.assert_called()
1384 1
        assert self.napp.controller.buffers.app.put.call_count == 2
1385 1
        ev = "kytos/topology.notify_link_up_if_status"
1386 1
        assert self.napp.controller.buffers.app.put.call_args[0][0].name == ev
1387
1388 1
    @patch('time.sleep')
1389 1
    @patch('napps.kytos.topology.main.Main.notify_topology_update')
1390 1
    @patch('napps.kytos.topology.main.Main.notify_link_status_change')
1391 1
    def test_handle_link_up_intf_down(self, *args):
1392
        """Test handle link up but one intf down."""
1393 1
        (mock_status_change, mock_topology_update, _) = args
1394
1395 1
        mock_switch = create_autospec(Switch)
1396 1
        mock_interface = create_autospec(Interface)
1397 1
        mock_interface.switch = mock_switch
1398 1
        mock_link = MagicMock()
1399 1
        mock_link.endpoint_a.is_active.return_value = False
1400 1
        mock_link.is_active.return_value = False
1401 1
        mock_interface.link = mock_link
1402 1
        self.napp.handle_link_up(mock_interface)
1403 1
        mock_interface.activate.assert_not_called()
1404 1
        assert mock_topology_update.call_count == 1
1405 1
        mock_status_change.assert_not_called()
1406
1407 1
    @patch('napps.kytos.topology.main.Main.notify_link_up_if_status')
1408 1
    def test_add_links(self, mock_notify_link_up_if_status):
1409
        """Test add_links."""
1410 1
        mock_link = MagicMock()
1411 1
        self.napp.controller.get_link_or_create = MagicMock()
1412 1
        mock_get_link_or_create = self.napp.controller.get_link_or_create
1413 1
        mock_get_link_or_create.return_value = (mock_link, True)
1414 1
        mock_event = MagicMock()
1415 1
        mock_intf_a = MagicMock()
1416 1
        mock_intf_b = MagicMock()
1417 1
        mock_event.content = {
1418
            "interface_a": mock_intf_a,
1419
            "interface_b": mock_intf_b
1420
        }
1421 1
        self.napp.add_links(mock_event)
1422 1
        assert mock_link.id in self.napp.link_status_change
1423 1
        mock_get_link_or_create.assert_called()
1424 1
        mock_notify_link_up_if_status.assert_called()
1425 1
        mock_intf_a.update_link.assert_called()
1426 1
        mock_intf_b.update_link.assert_called()
1427 1
        mock_link.endpoint_a = mock_intf_a
1428 1
        mock_link.endpoint_b = mock_intf_b
1429
1430 1
    def test_notify_switch_enabled(self):
1431
        """Test notify switch enabled."""
1432 1
        dpid = "00:00:00:00:00:00:00:01"
1433 1
        mock_buffers_put = MagicMock()
1434 1
        self.napp.controller.buffers.app.put = mock_buffers_put
1435 1
        self.napp.notify_switch_enabled(dpid)
1436 1
        mock_buffers_put.assert_called()
1437
1438 1
    def test_notify_switch_disabled(self):
1439
        """Test notify switch disabled."""
1440 1
        dpid = "00:00:00:00:00:00:00:01"
1441 1
        mock_buffers_put = MagicMock()
1442 1
        self.napp.controller.buffers.app.put = mock_buffers_put
1443 1
        self.napp.notify_switch_disabled(dpid)
1444 1
        mock_buffers_put.assert_called()
1445
1446 1
    def test_notify_topology_update(self):
1447
        """Test notify_topology_update."""
1448 1
        mock_buffers_put = MagicMock()
1449 1
        self.napp.controller.buffers.app.put = mock_buffers_put
1450 1
        self.napp.notify_topology_update()
1451 1
        mock_buffers_put.assert_called()
1452
1453 1
    def test_notify_link_status_change(self):
1454
        """Test notify link status change."""
1455 1
        mock_buffers_put = MagicMock()
1456 1
        self.napp.controller.buffers.app.put = mock_buffers_put
1457 1
        mock_link = create_autospec(Link)
1458 1
        mock_link.id = 'test_link'
1459 1
        mock_link.status_reason = frozenset()
1460 1
        mock_link.status = EntityStatus.UP
1461
1462
        # Check when switching to up
1463 1
        self.napp.notify_link_status_change(mock_link, 'test')
1464 1
        assert mock_buffers_put.call_count == 1
1465 1
        args, _ = mock_buffers_put.call_args
1466 1
        event = args[0]
1467 1
        assert event.content['link'] is mock_link
1468 1
        assert event.content['reason'] == 'test'
1469 1
        assert event.name == 'kytos/topology.link_up'
1470
1471
        # Check result when no change
1472 1
        self.napp.notify_link_status_change(mock_link, 'test2')
1473 1
        assert mock_buffers_put.call_count == 1
1474
1475
        # Check when switching to down
1476 1
        mock_link.status_reason = frozenset({'disabled'})
1477 1
        mock_link.status = EntityStatus.DOWN
1478 1
        self.napp.notify_link_status_change(mock_link, 'test3')
1479 1
        assert mock_buffers_put.call_count == 2
1480 1
        args, _ = mock_buffers_put.call_args
1481 1
        event = args[0]
1482 1
        assert event.content['link'] is mock_link
1483 1
        assert event.content['reason'] == 'test3'
1484 1
        assert event.name == 'kytos/topology.link_down'
1485
1486 1
    def test_notify_metadata_changes(self):
1487
        """Test notify metadata changes."""
1488 1
        mock_buffers_put = MagicMock()
1489 1
        self.napp.controller.buffers.app.put = mock_buffers_put
1490 1
        count = 0
1491 1
        for spec in [Switch, Interface, Link]:
1492 1
            mock_obj = create_autospec(spec)
1493 1
            mock_obj.metadata = {"some_key": "some_value"}
1494 1
            self.napp.notify_metadata_changes(mock_obj, 'added')
1495 1
            assert mock_buffers_put.call_count == count+1
1496 1
            count += 1
1497 1
        with pytest.raises(ValueError):
1498 1
            self.napp.notify_metadata_changes(MagicMock(), 'added')
1499
1500 1
    def test_notify_port_created(self):
1501
        """Test notify port created."""
1502 1
        mock_buffers_put = MagicMock()
1503 1
        self.napp.controller.buffers.app.put = mock_buffers_put
1504 1
        event = KytosEvent("some_event")
1505 1
        expected_name = "kytos/topology.port.created"
1506 1
        self.napp.notify_port_created(event)
1507 1
        assert mock_buffers_put.call_count == 1
1508 1
        assert mock_buffers_put.call_args_list[0][0][0].name == expected_name
1509
1510 1
    def test_handle_link_liveness_disabled(self) -> None:
1511
        """Test handle_link_liveness_disabled."""
1512 1
        interfaces = [MagicMock(id=f"intf{n}") for n in range(4)]
1513 1
        links = {
1514
            "link1": MagicMock(id="link1",
1515
                               endpoint_a=interfaces[0],
1516
                               endpoint_b=interfaces[1]),
1517
            "link2": MagicMock(id="link2",
1518
                               endpoint_a=interfaces[2],
1519
                               endpoint_b=interfaces[3]),
1520
        }
1521 1
        self.napp.controller.links = links
1522 1
        self.napp.notify_topology_update = MagicMock()
1523 1
        self.napp.notify_link_status_change = MagicMock()
1524
1525 1
        self.napp.handle_link_liveness_disabled(interfaces)
1526
1527 1
        assert self.napp.notify_topology_update.call_count == 1
1528 1
        assert self.napp.notify_link_status_change.call_count == len(links)
1529
1530 1
    def test_link_status_hook_link_up_timer(self) -> None:
1531
        """Test status hook link up timer."""
1532 1
        last_change = time.time() - self.napp.link_up_timer + 5
1533 1
        link = MagicMock(metadata={"last_status_change": last_change})
1534 1
        self.napp.link_status_change[link.id] = {
1535
            "last_status_change": last_change,
1536
        }
1537 1
        link.is_active.return_value = True
1538 1
        link.is_enabled.return_value = True
1539 1
        res = self.napp.link_status_hook_link_up_timer(link)
1540 1
        assert res == EntityStatus.DOWN
1541
1542 1
        last_change = time.time() - self.napp.link_up_timer
1543 1
        self.napp.link_status_change[link.id] = {
1544
            "last_status_change": last_change,
1545
        }
1546 1
        res = self.napp.link_status_hook_link_up_timer(link)
1547 1
        assert res is None
1548
1549 1
    @patch('napps.kytos.topology.main.Main.notify_link_status_change')
1550 1
    @patch('napps.kytos.topology.main.Main.notify_topology_update')
1551 1
    @patch('time.sleep')
1552 1
    def test_notify_link_up_if_status(
1553
        self,
1554
        mock_sleep,
1555
        mock_notify_topo,
1556
        mock_notify_link,
1557
    ) -> None:
1558
        """Test notify link up if status."""
1559
1560 1
        link = MagicMock(status=EntityStatus.UP)
1561 1
        self.napp.link_status_change[link.id] = {
1562
            "notified_up_at": now(),
1563
        }
1564 1
        assert not self.napp.notify_link_up_if_status(link, "link up")
1565 1
        link.update_metadata.assert_not_called()
1566 1
        mock_notify_topo.assert_not_called()
1567 1
        mock_notify_link.assert_not_called()
1568
1569 1
        link = MagicMock(status=EntityStatus.UP)
1570 1
        orig_time = now() - timedelta(seconds=60)
1571 1
        self.napp.link_status_change[link.id] = {
1572
            "notified_up_at": orig_time,
1573
        }
1574 1
        assert not self.napp.notify_link_up_if_status(link, "link up")
1575 1
        link_status_info = self.napp.link_status_change[link.id]
1576 1
        new_time = link_status_info["notified_up_at"]
1577 1
        assert new_time != orig_time
1578 1
        mock_notify_topo.assert_called()
1579 1
        mock_notify_link.assert_called()
1580
1581 1
        assert mock_sleep.call_count == 2
1582
1583 1
    @patch('napps.kytos.topology.main.Main.notify_link_status_change')
1584 1
    def test_notify_switch_links_status(self, mock_notify_link_status_change):
1585
        """Test switch links notification when switch status change"""
1586 1
        buffers_app_mock = MagicMock()
1587 1
        self.napp.controller.buffers.app = buffers_app_mock
1588 1
        dpid = "00:00:00:00:00:00:00:01"
1589 1
        mock_switch = get_switch_mock(dpid)
1590 1
        link1 = MagicMock()
1591 1
        link1.endpoint_a.switch = mock_switch
1592 1
        self.napp.controller.links = {1: link1}
1593
1594 1
        self.napp.notify_switch_links_status(mock_switch, "link enabled")
1595 1
        assert self.napp.controller.buffers.app.put.call_count == 1
1596
1597 1
        self.napp.notify_switch_links_status(mock_switch, "link disabled")
1598 1
        assert self.napp.controller.buffers.app.put.call_count == 1
1599 1
        assert mock_notify_link_status_change.call_count == 1
1600
1601
        # Without notification
1602 1
        link1.endpoint_a.switch = None
1603 1
        self.napp.notify_switch_links_status(mock_switch, "link enabled")
1604 1
        assert self.napp.controller.buffers.app.put.call_count == 1
1605
1606 1
    @patch('napps.kytos.topology.main.Main.notify_link_status_change')
1607 1
    def test_notify_interface_link_status(self, args):
1608
        """Test interface links notification when enable"""
1609 1
        mock_notify_link_status_change = args
1610 1
        buffers_app_mock = MagicMock()
1611 1
        self.napp.controller.buffers.app = buffers_app_mock
1612 1
        mock_intf = MagicMock(link=MagicMock())
1613 1
        self.napp._notify_interface_link_status([mock_intf], "link enabled")
1614 1
        assert self.napp.controller.buffers.app.put.call_count == 1
1615
1616 1
        self.napp._notify_interface_link_status([mock_intf], "link disabled")
1617 1
        assert mock_notify_link_status_change.call_count == 1
1618 1
        assert self.napp.controller.buffers.app.put.call_count == 1
1619
1620
        # Without notification
1621 1
        mock_intf.link = None
1622 1
        self.napp._notify_interface_link_status([mock_intf], "link enabled")
1623 1
        assert self.napp.controller.buffers.app.put.call_count == 1
1624
1625 1 View Code Duplication
    @patch('napps.kytos.topology.main.Main.notify_topology_update')
0 ignored issues
show
Duplication introduced by
This code seems to be duplicated in your project.
Loading history...
1626 1
    @patch('napps.kytos.topology.main.Main.notify_link_status_change')
1627 1
    def test_interruption_start(
1628
        self,
1629
        mock_notify_link_status_change,
1630
        mock_notify_topology_update
1631
    ):
1632
        """Tests processing of received interruption start events."""
1633 1
        link_a = MagicMock()
1634 1
        link_b = MagicMock()
1635 1
        link_c = MagicMock()
1636 1
        self.napp.controller.links = {
1637
            'link_a': link_a,
1638
            'link_b': link_b,
1639
            'link_c': link_c,
1640
        }
1641 1
        event = KytosEvent(
1642
            "topology.interruption.start",
1643
            {
1644
                'type': 'test_interruption',
1645
                'switches': [
1646
                ],
1647
                'interfaces': [
1648
                ],
1649
                'links': [
1650
                    'link_a',
1651
                    'link_c',
1652
                ],
1653
            }
1654
        )
1655 1
        self.napp.handle_interruption_start(event)
1656 1
        mock_notify_link_status_change.assert_has_calls(
1657
            [
1658
                call(link_a, 'test_interruption'),
1659
                call(link_c, 'test_interruption'),
1660
            ]
1661
        )
1662 1
        assert mock_notify_link_status_change.call_count == 2
1663 1
        mock_notify_topology_update.assert_called_once()
1664
1665 1 View Code Duplication
    @patch('napps.kytos.topology.main.Main.notify_topology_update')
0 ignored issues
show
Duplication introduced by
This code seems to be duplicated in your project.
Loading history...
1666 1
    @patch('napps.kytos.topology.main.Main.notify_link_status_change')
1667 1
    def test_interruption_end(
1668
        self,
1669
        mock_notify_link_status_change,
1670
        mock_notify_topology_update
1671
    ):
1672
        """Tests processing of received interruption end events."""
1673 1
        link_a = MagicMock()
1674 1
        link_b = MagicMock()
1675 1
        link_c = MagicMock()
1676 1
        self.napp.controller.links = {
1677
            'link_a': link_a,
1678
            'link_b': link_b,
1679
            'link_c': link_c,
1680
        }
1681 1
        event = KytosEvent(
1682
            "topology.interruption.start",
1683
            {
1684
                'type': 'test_interruption',
1685
                'switches': [
1686
                ],
1687
                'interfaces': [
1688
                ],
1689
                'links': [
1690
                    'link_a',
1691
                    'link_c',
1692
                ],
1693
            }
1694
        )
1695 1
        self.napp.handle_interruption_end(event)
1696 1
        mock_notify_link_status_change.assert_has_calls(
1697
            [
1698
                call(link_a, 'test_interruption'),
1699
                call(link_c, 'test_interruption'),
1700
            ]
1701
        )
1702 1
        assert mock_notify_link_status_change.call_count == 2
1703 1
        mock_notify_topology_update.assert_called_once()
1704
1705 1
    async def test_set_tag_range(self):
1706
        """Test set_tag_range"""
1707 1
        self.napp.controller.loop = asyncio.get_running_loop()
1708 1
        interface_id = '00:00:00:00:00:00:00:01:1'
1709 1
        dpid = '00:00:00:00:00:00:00:01'
1710 1
        mock_switch = get_switch_mock(dpid)
1711 1
        mock_interface = get_interface_mock('s1-eth1', 1, mock_switch)
1712 1
        mock_interface.set_tag_ranges = MagicMock()
1713 1
        self.napp.handle_on_interface_tags = MagicMock()
1714 1
        self.napp.controller.get_interface_by_id = MagicMock()
1715 1
        self.napp.controller.get_interface_by_id.return_value = mock_interface
1716 1
        payload = {
1717
            "tag_type": "vlan",
1718
            "tag_ranges": [[20, 20], [200, 3000]]
1719
        }
1720 1
        url = f"{self.base_endpoint}/interfaces/{interface_id}/tag_ranges"
1721 1
        response = await self.api_client.post(url, json=payload)
1722 1
        assert response.status_code == 200
1723
1724 1
        args = mock_interface.set_tag_ranges.call_args[0]
1725 1
        assert args[0] == payload['tag_ranges']
1726 1
        assert args[1] == payload['tag_type']
1727 1
        assert self.napp.handle_on_interface_tags.call_count == 1
1728
1729 1
    async def test_set_tag_range_not_found(self):
1730
        """Test set_tag_range. Not found"""
1731 1
        self.napp.controller.loop = asyncio.get_running_loop()
1732 1
        interface_id = '00:00:00:00:00:00:00:01:1'
1733 1
        self.napp.controller.get_interface_by_id = MagicMock()
1734 1
        self.napp.controller.get_interface_by_id.return_value = None
1735 1
        payload = {
1736
            "tag_type": "vlan",
1737
            "tag_ranges": [[20, 20], [200, 3000]]
1738
        }
1739 1
        url = f"{self.base_endpoint}/interfaces/{interface_id}/tag_ranges"
1740 1
        response = await self.api_client.post(url, json=payload)
1741
        assert response.status_code == 404
1742
1743 1 View Code Duplication
    async def test_set_tag_range_tag_error(self):
0 ignored issues
show
Duplication introduced by
This code seems to be duplicated in your project.
Loading history...
1744
        """Test set_tag_range TagRangeError"""
1745 1
        self.napp.controller.loop = asyncio.get_running_loop()
1746 1
        interface_id = '00:00:00:00:00:00:00:01:1'
1747 1
        dpid = '00:00:00:00:00:00:00:01'
1748 1
        mock_switch = get_switch_mock(dpid)
1749 1
        mock_interface = get_interface_mock('s1-eth1', 1, mock_switch)
1750 1
        mock_interface.set_tag_ranges = MagicMock()
1751 1
        mock_interface.set_tag_ranges.side_effect = KytosSetTagRangeError("")
1752 1
        mock_interface.notify_interface_tags = MagicMock()
1753 1
        self.napp.controller.get_interface_by_id = MagicMock()
1754 1
        self.napp.controller.get_interface_by_id.return_value = mock_interface
1755 1
        payload = {
1756
            "tag_type": "vlan",
1757
            "tag_ranges": [[20, 20], [200, 3000]]
1758
        }
1759 1
        url = f"{self.base_endpoint}/interfaces/{interface_id}/tag_ranges"
1760 1
        response = await self.api_client.post(url, json=payload)
1761
        assert response.status_code == 400
1762
        assert mock_interface.notify_interface_tags.call_count == 0
1763
1764 1 View Code Duplication
    async def test_set_tag_range_type_error(self):
0 ignored issues
show
Duplication introduced by
This code seems to be duplicated in your project.
Loading history...
1765
        """Test set_tag_range TagRangeError"""
1766 1
        self.napp.controller.loop = asyncio.get_running_loop()
1767 1
        interface_id = '00:00:00:00:00:00:00:01:1'
1768 1
        dpid = '00:00:00:00:00:00:00:01'
1769 1
        mock_switch = get_switch_mock(dpid)
1770 1
        mock_interface = get_interface_mock('s1-eth1', 1, mock_switch)
1771 1
        mock_interface.set_tag_ranges = MagicMock()
1772 1
        mock_interface.set_tag_ranges.side_effect = KytosTagtypeNotSupported(
1773
            ""
1774
        )
1775 1
        self.napp.handle_on_interface_tags = MagicMock()
1776 1
        self.napp.controller.get_interface_by_id = MagicMock()
1777 1
        self.napp.controller.get_interface_by_id.return_value = mock_interface
1778 1
        payload = {
1779
            "tag_type": "wrong_tag_type",
1780
            "tag_ranges": [[20, 20], [200, 3000]]
1781
        }
1782 1
        url = f"{self.base_endpoint}/interfaces/{interface_id}/tag_ranges"
1783 1
        response = await self.api_client.post(url, json=payload)
1784
        assert response.status_code == 400
1785
        assert self.napp.handle_on_interface_tags.call_count == 0
1786
1787 1 View Code Duplication
    async def test_delete_tag_range(self):
0 ignored issues
show
Duplication introduced by
This code seems to be duplicated in your project.
Loading history...
1788
        """Test delete_tag_range"""
1789 1
        self.napp.controller.loop = asyncio.get_running_loop()
1790 1
        interface_id = '00:00:00:00:00:00:00:01:1'
1791 1
        dpid = '00:00:00:00:00:00:00:01'
1792 1
        mock_switch = get_switch_mock(dpid)
1793 1
        mock_interface = get_interface_mock('s1-eth1', 1, mock_switch)
1794 1
        mock_interface.remove_tag_ranges = MagicMock()
1795 1
        self.napp.handle_on_interface_tags = MagicMock()
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 1
        assert response.status_code == 200
1801 1
        assert mock_interface.remove_tag_ranges.call_count == 1
1802
1803 1 View Code Duplication
    async def test_delete_tag_range_not_found(self):
0 ignored issues
show
Duplication introduced by
This code seems to be duplicated in your project.
Loading history...
1804
        """Test delete_tag_range. Not found"""
1805 1
        self.napp.controller.loop = asyncio.get_running_loop()
1806 1
        interface_id = '00:00:00:00:00:00:00:01:1'
1807 1
        dpid = '00:00:00:00:00:00:00:01'
1808 1
        mock_switch = get_switch_mock(dpid)
1809 1
        mock_interface = get_interface_mock('s1-eth1', 1, mock_switch)
1810 1
        mock_interface.remove_tag_ranges = MagicMock()
1811 1
        self.napp.controller.get_interface_by_id = MagicMock()
1812 1
        self.napp.controller.get_interface_by_id.return_value = None
1813 1
        url = f"{self.base_endpoint}/interfaces/{interface_id}/tag_ranges"
1814 1
        response = await self.api_client.delete(url)
1815
        assert response.status_code == 404
1816
        assert mock_interface.remove_tag_ranges.call_count == 0
1817
1818 1
    async def test_delete_tag_range_type_error(self):
1819
        """Test delete_tag_range TagRangeError"""
1820 1
        self.napp.controller.loop = asyncio.get_running_loop()
1821 1
        interface_id = '00:00:00:00:00:00:00:01:1'
1822 1
        dpid = '00:00:00:00:00:00:00:01'
1823 1
        mock_switch = get_switch_mock(dpid)
1824 1
        mock_interface = get_interface_mock('s1-eth1', 1, mock_switch)
1825 1
        mock_interface.remove_tag_ranges = MagicMock()
1826 1
        remove_tag = mock_interface.remove_tag_ranges
1827 1
        remove_tag.side_effect = KytosTagtypeNotSupported("")
1828 1
        self.napp.controller.get_interface_by_id = MagicMock()
1829 1
        self.napp.controller.get_interface_by_id.return_value = mock_interface
1830 1
        url = f"{self.base_endpoint}/interfaces/{interface_id}/tag_ranges"
1831 1
        response = await self.api_client.delete(url)
1832
        assert response.status_code == 400
1833
1834 1 View Code Duplication
    async def test_get_all_tag_ranges(self):
0 ignored issues
show
Duplication introduced by
This code seems to be duplicated in your project.
Loading history...
1835
        """Test get_all_tag_ranges"""
1836 1
        self.napp.controller.loop = asyncio.get_running_loop()
1837 1
        dpid = '00:00:00:00:00:00:00:01'
1838 1
        switch = get_switch_mock(dpid)
1839 1
        interface = get_interface_mock('s1-eth1', 1, switch)
1840 1
        tags = {'vlan': [[1, 4095]]}
1841 1
        special_tags = {'vlan': ["vlan"]}
1842 1
        interface.tag_ranges = tags
1843 1
        interface.available_tags = tags
1844 1
        interface.special_available_tags = special_tags
1845 1
        interface.special_tags = special_tags
1846 1
        switch.interfaces = {1: interface}
1847 1
        self.napp.controller.switches = {dpid: switch}
1848 1
        url = f"{self.base_endpoint}/interfaces/tag_ranges"
1849 1
        response = await self.api_client.get(url)
1850 1
        expected = {dpid + ":1": {
1851
            'available_tags': tags,
1852
            'tag_ranges': tags,
1853
            'special_available_tags': special_tags,
1854
            'special_tags': special_tags
1855
        }}
1856 1
        assert response.status_code == 200
1857 1
        assert response.json() == expected
1858
1859 1 View Code Duplication
    async def test_get_tag_ranges_by_intf(self):
0 ignored issues
show
Duplication introduced by
This code seems to be duplicated in your project.
Loading history...
1860
        """Test get_tag_ranges_by_intf"""
1861 1
        self.napp.controller.loop = asyncio.get_running_loop()
1862 1
        dpid = '00:00:00:00:00:00:00:01'
1863 1
        switch = get_switch_mock(dpid)
1864 1
        interface = get_interface_mock('s1-eth1', 1, switch)
1865 1
        tags = {'vlan': [[1, 4095]]}
1866 1
        special_tags = {'vlan': ["vlan"]}
1867 1
        interface.tag_ranges = tags
1868 1
        interface.available_tags = tags
1869 1
        interface.special_available_tags = special_tags
1870 1
        interface.special_tags = special_tags
1871 1
        self.napp.controller.get_interface_by_id = MagicMock()
1872 1
        self.napp.controller.get_interface_by_id.return_value = interface
1873 1
        url = f"{self.base_endpoint}/interfaces/{dpid}:1/tag_ranges"
1874 1
        response = await self.api_client.get(url)
1875 1
        expected = {
1876
            '00:00:00:00:00:00:00:01:1': {
1877
                "available_tags": tags,
1878
                "tag_ranges": tags,
1879
                'special_available_tags': special_tags,
1880
                'special_tags': special_tags
1881
            }
1882
        }
1883 1
        assert response.status_code == 200
1884 1
        assert response.json() == expected
1885
1886 1
    async def test_get_tag_ranges_by_intf_error(self):
1887
        """Test get_tag_ranges_by_intf with NotFound"""
1888 1
        self.napp.controller.loop = asyncio.get_running_loop()
1889 1
        dpid = '00:00:00:00:00:00:00:01'
1890 1
        self.napp.controller.get_interface_by_id = MagicMock()
1891 1
        self.napp.controller.get_interface_by_id.return_value = None
1892 1
        url = f"{self.base_endpoint}/interfaces/{dpid}:1/tag_ranges"
1893 1
        response = await self.api_client.get(url)
1894
        assert response.status_code == 404
1895
1896 1
    async def test_set_special_tags(self):
1897
        """Test set_special_tags"""
1898 1
        self.napp.controller.loop = asyncio.get_running_loop()
1899 1
        interface_id = '00:00:00:00:00:00:00:01:1'
1900 1
        dpid = '00:00:00:00:00:00:00:01'
1901 1
        mock_switch = get_switch_mock(dpid)
1902 1
        mock_intf = get_interface_mock('s1-eth1', 1, mock_switch)
1903 1
        mock_intf.set_special_tags = MagicMock()
1904 1
        self.napp.handle_on_interface_tags = MagicMock()
1905 1
        self.napp.controller.get_interface_by_id = MagicMock()
1906 1
        self.napp.controller.get_interface_by_id.return_value = mock_intf
1907 1
        payload = {
1908
            "tag_type": "vlan",
1909
            "special_tags": ["untagged"],
1910
        }
1911 1
        url = f"{self.base_endpoint}/interfaces/{interface_id}/"\
1912
              "special_tags"
1913 1
        response = await self.api_client.post(url, json=payload)
1914 1
        assert response.status_code == 200
1915
1916 1
        args = mock_intf.set_special_tags.call_args[0]
1917 1
        assert args[0] == payload["tag_type"]
1918 1
        assert args[1] == payload['special_tags']
1919 1
        assert self.napp.handle_on_interface_tags.call_count == 1
1920
1921
        # KytosTagError
1922 1
        mock_intf.set_special_tags.side_effect = KytosTagtypeNotSupported("")
1923 1
        url = f"{self.base_endpoint}/interfaces/{interface_id}/"\
1924
              "special_tags"
1925 1
        response = await self.api_client.post(url, json=payload)
1926
        assert response.status_code == 400
1927
        assert self.napp.handle_on_interface_tags.call_count == 1
1928
1929
        # Interface Not Found
1930
        self.napp.controller.get_interface_by_id.return_value = None
1931
        url = f"{self.base_endpoint}/interfaces/{interface_id}/"\
1932
              "special_tags"
1933
        response = await self.api_client.post(url, json=payload)
1934
        assert response.status_code == 404
1935
        assert self.napp.handle_on_interface_tags.call_count == 1
1936
1937 1
    async def test_delete_link(self):
1938
        """Test delete_link"""
1939 1
        dpid_a = '00:00:00:00:00:00:00:01'
1940 1
        dpid_b = '00:00:00:00:00:00:00:02'
1941 1
        link_id = 'mock_link'
1942 1
        mock_switch_a = get_switch_mock(dpid_a)
1943 1
        mock_switch_b = get_switch_mock(dpid_b)
1944 1
        mock_interface_a = get_interface_mock('s1-eth1', 1, mock_switch_a)
1945 1
        mock_interface_b = get_interface_mock('s2-eth1', 1, mock_switch_b)
1946 1
        mock_link = get_link_mock(mock_interface_a, mock_interface_b)
1947 1
        mock_link.id = link_id
1948 1
        mock_link.status = EntityStatus.DISABLED
1949 1
        mock_interface_a.link = mock_link
1950 1
        mock_interface_b.link = mock_link
1951 1
        self.napp.controller.links = {link_id: mock_link}
1952
1953 1
        call_count = self.napp.controller.buffers.app.put.call_count
1954 1
        endpoint = f"{self.base_endpoint}/links/{link_id}"
1955 1
        response = await self.api_client.delete(endpoint)
1956 1
        assert response.status_code == 200
1957 1
        assert self.napp.topo_controller.delete_link.call_count == 1
1958 1
        assert len(self.napp.controller.links) == 0
1959 1
        call_count += 2
1960 1
        assert self.napp.controller.buffers.app.put.call_count == call_count
1961
1962
        # Link is up
1963 1
        self.napp.controller.links = {link_id: mock_link}
1964 1
        mock_link.status = EntityStatus.UP
1965 1
        endpoint = f"{self.base_endpoint}/links/{link_id}"
1966 1
        response = await self.api_client.delete(endpoint)
1967
        assert response.status_code == 409
1968
        assert self.napp.topo_controller.delete_link.call_count == 1
1969
        assert self.napp.controller.buffers.app.put.call_count == call_count
1970
1971
        # Link does not exist
1972
        del self.napp.controller.links[link_id]
1973
        endpoint = f"{self.base_endpoint}/links/{link_id}"
1974
        response = await self.api_client.delete(endpoint)
1975
        assert response.status_code == 404
1976
        assert self.napp.topo_controller.delete_link.call_count == 1
1977
        assert self.napp.controller.buffers.app.put.call_count == call_count
1978
1979 1
    @patch('napps.kytos.topology.main.Main.get_flows_by_switch')
1980 1
    async def test_delete_switch(self, mock_get):
1981
        """Test delete_switch"""
1982
        # Error 404 NotFound
1983 1
        dpid = '00:00:00:00:00:00:00:01'
1984 1
        endpoint = f"{self.base_endpoint}/switches/{dpid}"
1985 1
        response = await self.api_client.delete(endpoint)
1986
        assert response.status_code == 404, response
1987
1988
        # Error 409 Switch not disabled
1989
        mock_switch = get_switch_mock(dpid)
1990
        mock_switch.status = EntityStatus.UP
1991
        self.napp.controller.switches = {dpid: mock_switch}
1992
        endpoint = f"{self.base_endpoint}/switches/{dpid}"
1993
        response = await self.api_client.delete(endpoint)
1994
        assert response.status_code == 409, response
1995
1996
        # Error 409 Interface vlan is being used
1997 1
        mock_intf = MagicMock(all_tags_available=lambda: False)
1998
        mock_switch.interfaces = {1: mock_intf}
1999
        mock_switch.status = EntityStatus.DISABLED
2000
        endpoint = f"{self.base_endpoint}/switches/{dpid}"
2001
        response = await self.api_client.delete(endpoint)
2002
        assert response.status_code == 409, response
2003
2004
        # Error 409 Swith have links
2005 1
        mock_switch.interfaces[1].all_tags_available = lambda: True
2006
        mock_switch_2 = get_switch_mock("00:00:00:00:00:00:00:02")
2007
        mock_interface_a = get_interface_mock('s1-eth1', 1, mock_switch)
2008
        mock_interface_b = get_interface_mock('s2-eth1', 1, mock_switch_2)
2009
        mock_link = get_link_mock(mock_interface_a, mock_interface_b)
2010
        self.napp.controller.links = {'0e2b5d7bc858b9f38db11b69': mock_link}
2011
        endpoint = f"{self.base_endpoint}/switches/{dpid}"
2012
        response = await self.api_client.delete(endpoint)
2013
        assert response.status_code == 409, response
2014
2015
        # Error 409 Switch has flows
2016
        mock_get.return_value = {dpid: {}}
2017
        endpoint = f"{self.base_endpoint}/switches/{dpid}"
2018
        response = await self.api_client.delete(endpoint)
2019
        assert response.status_code == 409, response
2020
2021
        # Success 202
2022
        mock_get.return_value = {}
2023
        self.napp.controller.links = {}
2024
        endpoint = f"{self.base_endpoint}/switches/{dpid}"
2025
        response = await self.api_client.delete(endpoint)
2026 1
        assert response.status_code == 200, response
2027
2028 1
    def test_notify_link_status(self):
2029
        """Test notify_link_enabled_state"""
2030 1
        self.napp.controller.buffers.app.put.reset_mock()
2031 1
        link = Mock()
2032 1
        link.id = 'mock_link'
2033 1
        self.napp.notify_link_enabled_state(link, 'enabled')
2034 1
        assert self.napp.controller.buffers.app.put.call_count == 1
2035
2036 1
        self.napp.notify_link_enabled_state(link, 'disabled')
2037 1
        assert self.napp.controller.buffers.app.put.call_count == 2
2038
2039 1
    @patch('napps.kytos.topology.main.tenacity.nap.time')
2040 1
    @patch('httpx.get')
2041 1
    def test_get_flows_by_switch(self, mock_get, _):
2042
        """Test get_flows_by_switch"""
2043 1
        dpid = "00:01"
2044 1
        mock_get.return_value.status_code = 400
2045 1
        with pytest.raises(tenacity.RetryError):
2046 1
            self.napp.get_flows_by_switch(dpid)
2047
2048 1
        mock_get.return_value.status_code = 200
2049 1
        mock_get.return_value.is_server_error = False
2050 1
        expected = {dpid: "mocked_flows"}
2051 1
        mock_get.return_value.json.return_value = expected
2052 1
        actual = self.napp.get_flows_by_switch(dpid)
2053 1
        assert actual == "mocked_flows"
2054
2055 1
    @patch('napps.kytos.topology.main.Main.get_intf_usage')
2056 1
    @patch('napps.kytos.topology.main.Main._delete_interface')
2057 1
    async def test_delete_interface_api(self, mock_delete, mock_usage):
2058
        """Test delete interface API call"""
2059 1
        switch_id = "00:00:00:00:00:00:00:01"
2060 1
        intf_id = "00:00:00:00:00:00:00:01:1"
2061
2062
        # Error 400 Invalid interface id
2063 1
        endpoint = f"{self.base_endpoint}/interfaces/{intf_id}x"
2064 1
        response = await self.api_client.delete(endpoint)
2065
        assert response.status_code == 400, response
2066
2067
        # Error 404 Switch not found
2068
        endpoint = f"{self.base_endpoint}/interfaces/{intf_id}"
2069
        response = await self.api_client.delete(endpoint)
2070
        assert response.status_code == 404, response
2071
2072
        # Error 404 Interface not found
2073
        mock_switch = get_switch_mock(switch_id)
2074
        mock_switch.interfaces = {}
2075
        self.napp.controller.switches = {switch_id: mock_switch}
2076
        response = await self.api_client.delete(endpoint)
2077
        assert response.status_code == 404, response
2078
2079
        # Error 409 Interface is used
2080
        mock_switch.interfaces = {1: Mock()}
2081
        self.napp.controller.switches = {switch_id: mock_switch}
2082
        mock_usage.return_value = "It is enabled or active."
2083
        response = await self.api_client.delete(endpoint)
2084
        assert response.status_code == 409, response
2085
2086
        # Success
2087
        mock_usage.return_value = None
2088
        mock_delete.return_value = True
2089
        response = await self.api_client.delete(endpoint)
2090 1
        assert response.status_code == 200, response
2091
2092 1
    def test_get_intf_usage(self):
2093
        """Test get_intf_usage"""
2094 1
        switch_id = "00:00:00:00:00:00:00:01"
2095 1
        mock_switch = get_switch_mock(switch_id)
2096 1
        mock_intf = get_interface_mock('s1-eth1', 1, mock_switch)
2097
2098 1
        mock_intf.is_enabled.return_value = False
2099 1
        mock_intf.is_active.return_value = True
2100 1
        actual_usage = self.napp.get_intf_usage(mock_intf)
2101 1
        assert actual_usage == "It is enabled or active."
2102
2103 1
        mock_intf.is_active.return_value = False
2104 1
        mock_intf.link = Mock()
2105 1
        actual_usage = self.napp.get_intf_usage(mock_intf)
2106 1
        assert "It has a link," in actual_usage
2107
2108 1
        mock_intf.link = None
2109 1
        self.napp.get_flow_id_by_intf = MagicMock(return_value="mock_flow")
2110 1
        actual_usage = self.napp.get_intf_usage(mock_intf)
2111 1
        assert "There is a flow installed" in actual_usage
2112
2113 1
        self.napp.get_flow_id_by_intf.return_value = None
2114 1
        actual_usage = self.napp.get_intf_usage(mock_intf)
2115 1
        assert actual_usage is None
2116
2117 1
    @patch('napps.kytos.topology.main.Main.get_flows_by_switch')
2118 1
    def test_get_flow_id_by_intf(self, mock_flows):
2119
        """Test get_flow_id_by_intf"""
2120 1
        flows = [
2121
            {
2122
                "flow": {
2123
                    "match": {"in_port": 1, "dl_vlan": 200},
2124
                },
2125
                "flow_id": "flow_0",
2126
            },
2127
            {
2128
                "flow": {
2129
                    "actions": [{"action_type": "output", "port": 1}]
2130
                },
2131
                "flow_id": "flow_1",
2132
            },
2133
            {
2134
                "flow": {
2135
                    "instructions": [{
2136
                        "instruction_type": "apply_actions",
2137
                        "actions": [{"action_type": "output", "port": 1}]
2138
                    }]
2139
                },
2140
                "flow_id": "flow_2",
2141
            },
2142
            {
2143
                "flow": {
2144
                    "match": {"dl_src": "ee:ee:ee:ee:ee:02"},
2145
                },
2146
                "flow_id": "flow_3",
2147
            }
2148
        ]
2149
2150 1
        switch_id = "00:00:00:00:00:00:00:01"
2151 1
        mock_switch = get_switch_mock(switch_id)
2152 1
        mock_intf = get_interface_mock('s1-eth1', 1, mock_switch)
2153
2154 1
        mock_flows.return_value = [flows[0]]
2155 1
        flow_id = self.napp.get_flow_id_by_intf(mock_intf)
2156 1
        assert flow_id == flows[0]["flow_id"]
2157
2158 1
        mock_flows.return_value = [flows[1]]
2159 1
        flow_id = self.napp.get_flow_id_by_intf(mock_intf)
2160 1
        assert flow_id == flows[1]["flow_id"]
2161
2162 1
        mock_flows.return_value = [flows[2]]
2163 1
        flow_id = self.napp.get_flow_id_by_intf(mock_intf)
2164 1
        assert flow_id == flows[2]["flow_id"]
2165
2166 1
        mock_flows.return_value = [flows[3]]
2167 1
        flow_id = self.napp.get_flow_id_by_intf(mock_intf)
2168 1
        assert flow_id is None
2169
2170 1
    def test_delete_interface(self):
2171
        """Test _delete_interface"""
2172 1
        switch_id = "00:00:00:00:00:00:00:01"
2173 1
        mock_switch = get_switch_mock(switch_id)
2174 1
        mock_intf = get_interface_mock('s1-eth1', 1, mock_switch)
2175 1
        self.napp._delete_interface(mock_intf)
2176 1
        assert mock_switch.remove_interface.call_count == 1
2177 1
        assert self.napp.topo_controller.upsert_switch.call_count == 1
2178 1
        delete = self.napp.topo_controller.delete_interface_from_details
2179
        assert delete.call_count == 1
2180