Total Complexity | 95 |
Total Lines | 2084 |
Duplicated Lines | 7.58 % |
Coverage | 100% |
Changes | 0 |
Duplicate code is one of the most pungent code smells. A rule that is often used is to re-structure code once it is duplicated in three or more places.
Common duplication problems, and corresponding solutions are:
Complex classes like build.tests.unit.models.test_evc_deploy often do a lot of different things. To break such a class down, we need to identify a cohesive component within that class. A common approach to find such a component is to look for fields/methods that share the same prefixes, or suffixes.
Once you have determined the fields that belong together, you can apply the Extract Class refactoring. If the component makes sense as a sub-class, Extract Subclass is also a candidate, and is often faster.
1 | """Method to thest EVCDeploy class.""" |
||
2 | 1 | import sys |
|
3 | 1 | from unittest.mock import MagicMock, Mock, call, patch |
|
4 | 1 | import operator |
|
5 | 1 | import pytest |
|
6 | 1 | from kytos.lib.helpers import get_controller_mock |
|
7 | |||
8 | 1 | from kytos.core.common import EntityStatus |
|
9 | 1 | from kytos.core.exceptions import KytosNoTagAvailableError |
|
10 | 1 | from kytos.core.interface import Interface |
|
11 | 1 | from kytos.core.switch import Switch |
|
12 | 1 | from httpx import TimeoutException |
|
13 | # pylint: disable=wrong-import-position |
||
14 | 1 | sys.path.insert(0, "/var/lib/kytos/napps/..") |
|
15 | # pylint: enable=wrong-import-position |
||
16 | |||
17 | 1 | from napps.kytos.mef_eline.exceptions import (FlowModException, # NOQA |
|
18 | EVCPathNotInstalled) |
||
19 | 1 | from napps.kytos.mef_eline.models import EVC, EVCDeploy, Path # NOQA |
|
20 | 1 | from napps.kytos.mef_eline.settings import (ANY_SB_PRIORITY, # NOQA |
|
21 | EPL_SB_PRIORITY, EVPL_SB_PRIORITY, |
||
22 | MANAGER_URL, |
||
23 | SDN_TRACE_CP_URL, |
||
24 | UNTAGGED_SB_PRIORITY) |
||
25 | 1 | from napps.kytos.mef_eline.tests.helpers import (get_link_mocked, # NOQA |
|
26 | get_uni_mocked) |
||
27 | |||
28 | |||
29 | # pylint: disable=too-many-public-methods, too-many-lines |
||
30 | 1 | class TestEVC(): |
|
31 | """Tests to verify EVC class.""" |
||
32 | |||
33 | 1 | def setup_method(self): |
|
34 | """Setup method""" |
||
35 | 1 | attributes = { |
|
36 | "controller": get_controller_mock(), |
||
37 | "name": "circuit_for_tests", |
||
38 | "uni_a": get_uni_mocked(is_valid=True), |
||
39 | "uni_z": get_uni_mocked(is_valid=True), |
||
40 | } |
||
41 | 1 | self.evc_deploy = EVCDeploy(**attributes) |
|
42 | |||
43 | 1 | def test_primary_links_zipped_empty(self): |
|
44 | """Test primary links zipped method.""" |
||
45 | 1 | assert not self.evc_deploy.links_zipped(None) |
|
46 | |||
47 | 1 | @staticmethod |
|
48 | 1 | @patch("napps.kytos.mef_eline.models.evc.log") |
|
49 | 1 | def test_should_deploy_case1(log_mock): |
|
50 | """Test should deploy method without primary links.""" |
||
51 | 1 | log_mock.debug.return_value = True |
|
52 | 1 | attributes = { |
|
53 | "controller": get_controller_mock(), |
||
54 | "name": "custom_name", |
||
55 | "uni_a": get_uni_mocked(is_valid=True), |
||
56 | "uni_z": get_uni_mocked(is_valid=True), |
||
57 | } |
||
58 | |||
59 | 1 | evc = EVC(**attributes) |
|
60 | 1 | evc.should_deploy() |
|
61 | 1 | log_mock.debug.assert_called_with("Path is empty.") |
|
62 | |||
63 | 1 | View Code Duplication | @patch("napps.kytos.mef_eline.models.evc.log") |
|
|||
64 | 1 | def test_should_deploy_case2(self, log_mock): |
|
65 | """Test should deploy method with disable circuit.""" |
||
66 | 1 | log_mock.debug.return_value = True |
|
67 | 1 | attributes = { |
|
68 | "controller": get_controller_mock(), |
||
69 | "name": "custom_name", |
||
70 | "uni_a": get_uni_mocked(is_valid=True), |
||
71 | "uni_z": get_uni_mocked(is_valid=True), |
||
72 | "primary_links": [get_link_mocked(), get_link_mocked()], |
||
73 | } |
||
74 | 1 | evc = EVC(**attributes) |
|
75 | |||
76 | 1 | assert evc.should_deploy(attributes["primary_links"]) is False |
|
77 | 1 | log_mock.debug.assert_called_with(f"{evc} is disabled.") |
|
78 | |||
79 | 1 | View Code Duplication | @patch("napps.kytos.mef_eline.models.evc.log") |
80 | 1 | def test_should_deploy_case3(self, log_mock): |
|
81 | """Test should deploy method with enabled and not active circuit.""" |
||
82 | 1 | log_mock.debug.return_value = True |
|
83 | 1 | attributes = { |
|
84 | "controller": get_controller_mock(), |
||
85 | "name": "custom_name", |
||
86 | "uni_a": get_uni_mocked(is_valid=True), |
||
87 | "uni_z": get_uni_mocked(is_valid=True), |
||
88 | "primary_links": [get_link_mocked(), get_link_mocked()], |
||
89 | "enabled": True, |
||
90 | } |
||
91 | 1 | evc = EVC(**attributes) |
|
92 | 1 | assert evc.should_deploy(attributes["primary_links"]) is True |
|
93 | 1 | log_mock.debug.assert_called_with(f"{evc} will be deployed.") |
|
94 | |||
95 | 1 | @patch("napps.kytos.mef_eline.models.evc.log") |
|
96 | 1 | def test_should_deploy_case4(self, log_mock): |
|
97 | """Test should deploy method with enabled and active circuit.""" |
||
98 | 1 | log_mock.debug.return_value = True |
|
99 | 1 | attributes = { |
|
100 | "controller": get_controller_mock(), |
||
101 | "name": "custom_name", |
||
102 | "uni_a": get_uni_mocked(is_valid=True), |
||
103 | "uni_z": get_uni_mocked(is_valid=True), |
||
104 | "primary_links": [get_link_mocked(), get_link_mocked()], |
||
105 | "enabled": True, |
||
106 | "active": True, |
||
107 | } |
||
108 | 1 | evc = EVC(**attributes) |
|
109 | 1 | assert evc.should_deploy(attributes["primary_links"]) is False |
|
110 | |||
111 | 1 | @patch("napps.kytos.mef_eline.models.evc.httpx") |
|
112 | 1 | def test_send_flow_mods_case1(self, httpx_mock): |
|
113 | """Test if _send_flow_mods is sending flow_mods to be installed.""" |
||
114 | 1 | flow_mods = {"00:01": {"flows": [20]}} |
|
115 | |||
116 | 1 | response = MagicMock() |
|
117 | 1 | response.status_code = 201 |
|
118 | 1 | response.is_server_error = False |
|
119 | 1 | httpx_mock.post.return_value = response |
|
120 | |||
121 | # pylint: disable=protected-access |
||
122 | 1 | EVC._send_flow_mods(flow_mods, "install") |
|
123 | |||
124 | 1 | expected_endpoint = f"{MANAGER_URL}/flows" |
|
125 | 1 | expected_data = flow_mods |
|
126 | 1 | expected_data["force"] = False |
|
127 | 1 | assert httpx_mock.post.call_count == 1 |
|
128 | 1 | httpx_mock.post.assert_called_once_with( |
|
129 | expected_endpoint, json=expected_data, timeout=30 |
||
130 | ) |
||
131 | |||
132 | 1 | @patch("napps.kytos.mef_eline.models.evc.httpx") |
|
133 | 1 | def test_send_flow_mods_case2(self, httpx_mock): |
|
134 | """Test if _send_flow_mods are sending flow_mods to be deleted |
||
135 | and by_switch.""" |
||
136 | 1 | flow_mods = {"00:01": {"flows": [20]}} |
|
137 | 1 | response = MagicMock() |
|
138 | 1 | response.status_code = 201 |
|
139 | 1 | response.is_server_error = False |
|
140 | 1 | httpx_mock.request.return_value = response |
|
141 | |||
142 | # pylint: disable=protected-access |
||
143 | 1 | EVC._send_flow_mods( |
|
144 | flow_mods, command='delete', force=True, by_switch=True |
||
145 | ) |
||
146 | |||
147 | 1 | expected_endpoint = f"{MANAGER_URL}/flows_by_switch/?force={True}" |
|
148 | 1 | assert httpx_mock.request.call_count == 1 |
|
149 | 1 | httpx_mock.request.assert_called_once_with( |
|
150 | "DELETE", expected_endpoint, json=flow_mods, timeout=30 |
||
151 | ) |
||
152 | |||
153 | 1 | @patch("time.sleep") |
|
154 | 1 | @patch("napps.kytos.mef_eline.models.evc.httpx") |
|
155 | 1 | def test_send_flow_mods_error(self, httpx_mock, _): |
|
156 | """Test flow_manager call fails.""" |
||
157 | 1 | flow_mods = {"00:01": {"flows": [20]}} |
|
158 | 1 | response = MagicMock() |
|
159 | 1 | response.status_code = 415 |
|
160 | 1 | httpx_mock.post.return_value = response |
|
161 | |||
162 | # pylint: disable=protected-access |
||
163 | 1 | with pytest.raises(FlowModException): |
|
164 | 1 | EVC._send_flow_mods( |
|
165 | flow_mods, |
||
166 | command='delete', |
||
167 | force=True |
||
168 | ) |
||
169 | 1 | assert httpx_mock.request.call_count == 3 |
|
170 | |||
171 | 1 | def test_prepare_flow_mod(self): |
|
172 | """Test prepare flow_mod method.""" |
||
173 | 1 | interface_a = Interface("eth0", 1, Mock(spec=Switch)) |
|
174 | 1 | interface_z = Interface("eth1", 3, Mock(spec=Switch)) |
|
175 | 1 | attributes = { |
|
176 | "table_group": {"epl": 0, "evpl": 0}, |
||
177 | "controller": get_controller_mock(), |
||
178 | "name": "custom_name", |
||
179 | "uni_a": get_uni_mocked(is_valid=True), |
||
180 | "uni_z": get_uni_mocked(is_valid=True), |
||
181 | "primary_links": [get_link_mocked(), get_link_mocked()], |
||
182 | "enabled": True, |
||
183 | "active": True, |
||
184 | } |
||
185 | 1 | evc = EVC(**attributes) |
|
186 | |||
187 | # pylint: disable=protected-access |
||
188 | 1 | flow_mod = evc._prepare_flow_mod(interface_a, interface_z) |
|
189 | 1 | expected_flow_mod = { |
|
190 | "match": {"in_port": interface_a.port_number}, |
||
191 | "cookie": evc.get_cookie(), |
||
192 | "owner": "mef_eline", |
||
193 | "actions": [ |
||
194 | {"action_type": "output", "port": interface_z.port_number} |
||
195 | ], |
||
196 | "priority": EVPL_SB_PRIORITY, |
||
197 | "table_group": "evpl", |
||
198 | "table_id": 0, |
||
199 | } |
||
200 | 1 | assert expected_flow_mod == flow_mod |
|
201 | |||
202 | 1 | evc.sb_priority = 1234 |
|
203 | 1 | flow_mod = evc._prepare_flow_mod(interface_a, interface_z, 3) |
|
204 | 1 | assert flow_mod["priority"] == 1234 |
|
205 | 1 | assert flow_mod["actions"][1]["action_type"] == "set_queue" |
|
206 | 1 | assert flow_mod["actions"][1]["queue_id"] == 3 |
|
207 | |||
208 | 1 | def test_prepare_pop_flow(self): |
|
209 | """Test prepare pop flow method.""" |
||
210 | 1 | attributes = { |
|
211 | "table_group": {"epl": 0, "evpl": 0}, |
||
212 | "controller": get_controller_mock(), |
||
213 | "name": "custom_name", |
||
214 | "uni_a": get_uni_mocked(interface_port=1, is_valid=True), |
||
215 | "uni_z": get_uni_mocked(interface_port=2, is_valid=True), |
||
216 | } |
||
217 | 1 | evc = EVC(**attributes) |
|
218 | 1 | interface_a = evc.uni_a.interface |
|
219 | 1 | interface_z = evc.uni_z.interface |
|
220 | 1 | in_vlan = 10 |
|
221 | |||
222 | # pylint: disable=protected-access |
||
223 | 1 | flow_mod = evc._prepare_pop_flow( |
|
224 | interface_a, interface_z, in_vlan |
||
225 | ) |
||
226 | |||
227 | 1 | expected_flow_mod = { |
|
228 | "match": {"in_port": interface_a.port_number, |
||
229 | "dl_vlan": in_vlan}, |
||
230 | "cookie": evc.get_cookie(), |
||
231 | "owner": "mef_eline", |
||
232 | "actions": [ |
||
233 | {"action_type": "pop_vlan"}, |
||
234 | {"action_type": "output", "port": interface_z.port_number}, |
||
235 | ], |
||
236 | "priority": EVPL_SB_PRIORITY, |
||
237 | "table_group": "evpl", |
||
238 | "table_id": 0, |
||
239 | } |
||
240 | 1 | assert expected_flow_mod == flow_mod |
|
241 | |||
242 | # pylint: disable=too-many-branches |
||
243 | 1 | @pytest.mark.parametrize( |
|
244 | "in_vlan_a,in_vlan_z", |
||
245 | [ |
||
246 | (100, 50), |
||
247 | (100, 100), |
||
248 | (100, "4096/4096"), |
||
249 | (100, 0), |
||
250 | (100, None), |
||
251 | ("4096/4096", 50), |
||
252 | ("4096/4096", "4096/4096"), |
||
253 | ("4096/4096", 0), |
||
254 | ("4096/4096", None), |
||
255 | (0, 50), |
||
256 | (0, "4096/4096"), |
||
257 | (0, 0), |
||
258 | (0, None), |
||
259 | (None, 50), |
||
260 | (None, "4096/4096"), |
||
261 | (None, 0), |
||
262 | (None, None), |
||
263 | ] |
||
264 | ) |
||
265 | 1 | def test_prepare_push_flow(self, in_vlan_a, in_vlan_z): |
|
266 | """Test prepare push flow method.""" |
||
267 | 1 | attributes = { |
|
268 | "table_group": {"evpl": 3, "epl": 4}, |
||
269 | "controller": get_controller_mock(), |
||
270 | "name": "custom_name", |
||
271 | "uni_a": get_uni_mocked(interface_port=1, is_valid=True), |
||
272 | "uni_z": get_uni_mocked(interface_port=2, is_valid=True), |
||
273 | } |
||
274 | 1 | evc = EVC(**attributes) |
|
275 | 1 | interface_a = evc.uni_a.interface |
|
276 | 1 | interface_z = evc.uni_z.interface |
|
277 | 1 | out_vlan_a = 20 |
|
278 | |||
279 | # pylint: disable=protected-access |
||
280 | 1 | flow_mod = evc._prepare_push_flow(interface_a, interface_z, |
|
281 | in_vlan_a, out_vlan_a, |
||
282 | in_vlan_z) |
||
283 | 1 | expected_flow_mod = { |
|
284 | 'match': {'in_port': interface_a.port_number}, |
||
285 | 'cookie': evc.get_cookie(), |
||
286 | 'owner': 'mef_eline', |
||
287 | 'table_id': 3, |
||
288 | 'table_group': 'evpl', |
||
289 | 'actions': [ |
||
290 | {'action_type': 'push_vlan', 'tag_type': 's'}, |
||
291 | {'action_type': 'set_vlan', 'vlan_id': out_vlan_a}, |
||
292 | { |
||
293 | 'action_type': 'output', |
||
294 | 'port': interface_z.port_number |
||
295 | } |
||
296 | ], |
||
297 | "priority": EVPL_SB_PRIORITY, |
||
298 | } |
||
299 | 1 | expected_flow_mod["priority"] = evc.get_priority(in_vlan_a) |
|
300 | 1 | if in_vlan_a is not None: |
|
301 | 1 | expected_flow_mod['match']['dl_vlan'] = in_vlan_a |
|
302 | 1 | if in_vlan_z not in evc.special_cases and in_vlan_a != in_vlan_z: |
|
303 | 1 | new_action = {"action_type": "set_vlan", |
|
304 | "vlan_id": in_vlan_z} |
||
305 | 1 | expected_flow_mod["actions"].insert(0, new_action) |
|
306 | 1 | if in_vlan_a not in evc.special_cases: |
|
307 | 1 | if in_vlan_z == 0: |
|
308 | 1 | new_action = {"action_type": "pop_vlan"} |
|
309 | 1 | expected_flow_mod["actions"].insert(0, new_action) |
|
310 | 1 | elif in_vlan_a == "4096/4096": |
|
311 | 1 | if in_vlan_z == 0: |
|
312 | 1 | new_action = {"action_type": "pop_vlan"} |
|
313 | 1 | expected_flow_mod["actions"].insert(0, new_action) |
|
314 | 1 | elif not in_vlan_a: |
|
315 | 1 | if in_vlan_a is None: |
|
316 | 1 | expected_flow_mod["table_group"] = "epl" |
|
317 | 1 | expected_flow_mod["table_id"] = 4 |
|
318 | 1 | if in_vlan_z not in evc.special_cases: |
|
319 | 1 | new_action = {"action_type": "push_vlan", |
|
320 | "tag_type": "c"} |
||
321 | 1 | expected_flow_mod["actions"].insert(0, new_action) |
|
322 | 1 | assert expected_flow_mod == flow_mod |
|
323 | |||
324 | 1 | @staticmethod |
|
325 | 1 | def create_evc_inter_switch(tag_value_a=82, tag_value_z=83): |
|
326 | """Create inter-switch EVC with two links in the path""" |
||
327 | 1 | uni_a = get_uni_mocked( |
|
328 | interface_port=2, |
||
329 | tag_value=tag_value_a, |
||
330 | switch_id=1, |
||
331 | switch_dpid=1, |
||
332 | is_valid=True, |
||
333 | ) |
||
334 | 1 | uni_z = get_uni_mocked( |
|
335 | interface_port=3, |
||
336 | tag_value=tag_value_z, |
||
337 | switch_id=3, |
||
338 | switch_dpid=3, |
||
339 | is_valid=True, |
||
340 | ) |
||
341 | |||
342 | 1 | attributes = { |
|
343 | "controller": get_controller_mock(), |
||
344 | "name": "custom_name", |
||
345 | "id": "1", |
||
346 | "uni_a": uni_a, |
||
347 | "uni_z": uni_z, |
||
348 | "primary_links": [ |
||
349 | get_link_mocked( |
||
350 | switch_a=Switch(1), |
||
351 | switch_b=Switch(2), |
||
352 | endpoint_a_port=9, |
||
353 | endpoint_b_port=10, |
||
354 | metadata={"s_vlan": 5}, |
||
355 | ), |
||
356 | get_link_mocked( |
||
357 | switch_a=Switch(2), |
||
358 | switch_b=Switch(3), |
||
359 | endpoint_a_port=11, |
||
360 | endpoint_b_port=12, |
||
361 | metadata={"s_vlan": 6}, |
||
362 | ), |
||
363 | ], |
||
364 | "table_group": {"epl": 0, "evpl": 0} |
||
365 | } |
||
366 | 1 | return EVC(**attributes) |
|
367 | |||
368 | 1 | @patch("httpx.post") |
|
369 | 1 | @patch("napps.kytos.mef_eline.controllers.ELineController.upsert_evc") |
|
370 | 1 | @patch("napps.kytos.mef_eline.models.evc.log") |
|
371 | 1 | @patch("napps.kytos.mef_eline.models.path.Path.choose_vlans") |
|
372 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVC._install_flows") |
|
373 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVC._install_direct_uni_flows") |
|
374 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVC.activate") |
|
375 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVC.should_deploy") |
|
376 | 1 | def test_deploy_successfully(self, *args): |
|
377 | """Test if all methods to deploy are called.""" |
||
378 | # pylint: disable=too-many-locals |
||
379 | 1 | ( |
|
380 | should_deploy_mock, |
||
381 | activate_mock, |
||
382 | install_direct_uni_flows_mock, |
||
383 | install_unni_flows_mock, |
||
384 | chose_vlans_mock, |
||
385 | log_mock, |
||
386 | _, |
||
387 | httpx_mock, |
||
388 | ) = args |
||
389 | |||
390 | 1 | response = MagicMock() |
|
391 | 1 | response.status_code = 201 |
|
392 | 1 | response.is_server_error = False |
|
393 | 1 | httpx_mock.return_value = response |
|
394 | |||
395 | 1 | should_deploy_mock.return_value = True |
|
396 | 1 | evc = self.create_evc_inter_switch() |
|
397 | 1 | deployed = evc.deploy_to_path(evc.primary_links) |
|
398 | |||
399 | 1 | assert should_deploy_mock.call_count == 1 |
|
400 | 1 | assert activate_mock.call_count == 1 |
|
401 | 1 | assert install_unni_flows_mock.call_count == 1 |
|
402 | 1 | assert chose_vlans_mock.call_count == 1 |
|
403 | 1 | log_mock.info.assert_called_with(f"{evc} was deployed.") |
|
404 | 1 | assert deployed is True |
|
405 | |||
406 | # intra switch EVC |
||
407 | 1 | evc = self.create_evc_intra_switch() |
|
408 | 1 | assert evc.deploy_to_path(evc.primary_links) is True |
|
409 | 1 | assert install_direct_uni_flows_mock.call_count == 1 |
|
410 | 1 | assert activate_mock.call_count == 2 |
|
411 | 1 | assert log_mock.info.call_count == 2 |
|
412 | 1 | log_mock.info.assert_called_with(f"{evc} was deployed.") |
|
413 | |||
414 | 1 | @patch("httpx.post") |
|
415 | 1 | @patch("napps.kytos.mef_eline.models.evc.log") |
|
416 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVC.discover_new_paths") |
|
417 | 1 | @patch("napps.kytos.mef_eline.models.path.Path.choose_vlans") |
|
418 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVC._install_flows") |
|
419 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVC.activate") |
|
420 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVC.should_deploy") |
|
421 | 1 | @patch("napps.kytos.mef_eline.models.EVC.sync") |
|
422 | 1 | def test_deploy_fail(self, *args): |
|
423 | """Test if all methods is ignored when the should_deploy is false.""" |
||
424 | # pylint: disable=too-many-locals |
||
425 | 1 | ( |
|
426 | sync_mock, |
||
427 | should_deploy_mock, |
||
428 | activate_mock, |
||
429 | install_unni_flows_mock, |
||
430 | choose_vlans_mock, |
||
431 | discover_new_paths_mock, |
||
432 | log_mock, |
||
433 | httpx_mock, |
||
434 | ) = args |
||
435 | |||
436 | 1 | response = MagicMock() |
|
437 | 1 | response.status_code = 201 |
|
438 | 1 | httpx_mock.return_value = response |
|
439 | |||
440 | 1 | evc = self.create_evc_inter_switch() |
|
441 | 1 | should_deploy_mock.return_value = False |
|
442 | 1 | discover_new_paths_mock.return_value = [] |
|
443 | 1 | deployed = evc.deploy_to_path() |
|
444 | |||
445 | 1 | assert discover_new_paths_mock.call_count == 1 |
|
446 | 1 | assert should_deploy_mock.call_count == 1 |
|
447 | 1 | assert activate_mock.call_count == 0 |
|
448 | 1 | assert install_unni_flows_mock.call_count == 0 |
|
449 | 1 | assert choose_vlans_mock.call_count == 0 |
|
450 | 1 | assert log_mock.info.call_count == 0 |
|
451 | 1 | assert sync_mock.call_count == 0 |
|
452 | 1 | assert deployed is False |
|
453 | |||
454 | # NoTagAvailable on static path |
||
455 | 1 | should_deploy_mock.return_value = True |
|
456 | 1 | choose_vlans_mock.side_effect = KytosNoTagAvailableError(MagicMock()) |
|
457 | 1 | assert evc.deploy_to_path(evc.primary_links) is False |
|
458 | |||
459 | # NoTagAvailable on dynamic path |
||
460 | 1 | should_deploy_mock.return_value = False |
|
461 | 1 | discover_new_paths_mock.return_value = [Path(['a', 'b'])] |
|
462 | 1 | choose_vlans_mock.side_effect = KytosNoTagAvailableError(MagicMock()) |
|
463 | 1 | assert evc.deploy_to_path(evc.primary_links) is False |
|
464 | |||
465 | 1 | @patch("napps.kytos.mef_eline.models.evc.log") |
|
466 | 1 | @patch( |
|
467 | "napps.kytos.mef_eline.models.evc.EVC.discover_new_paths", |
||
468 | return_value=[], |
||
469 | ) |
||
470 | 1 | @patch("napps.kytos.mef_eline.models.path.Path.choose_vlans") |
|
471 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVC._install_flows") |
|
472 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVC.should_deploy") |
|
473 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVC.remove_current_flows") |
|
474 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVC.sync") |
|
475 | 1 | def test_deploy_error(self, *args): |
|
476 | """Test if all methods is ignored when the should_deploy is false.""" |
||
477 | # pylint: disable=too-many-locals |
||
478 | 1 | ( |
|
479 | sync_mock, |
||
480 | remove_current_flows, |
||
481 | should_deploy_mock, |
||
482 | install_unni_flows, |
||
483 | choose_vlans_mock, |
||
484 | discover_new_paths, |
||
485 | log_mock, |
||
486 | ) = args |
||
487 | |||
488 | 1 | install_unni_flows.side_effect = EVCPathNotInstalled |
|
489 | 1 | should_deploy_mock.return_value = True |
|
490 | 1 | uni_a = get_uni_mocked( |
|
491 | interface_port=2, |
||
492 | tag_value=82, |
||
493 | switch_id="switch_uni_a", |
||
494 | is_valid=True, |
||
495 | ) |
||
496 | 1 | uni_z = get_uni_mocked( |
|
497 | interface_port=3, |
||
498 | tag_value=83, |
||
499 | switch_id="switch_uni_z", |
||
500 | is_valid=True, |
||
501 | ) |
||
502 | |||
503 | 1 | primary_links = [ |
|
504 | get_link_mocked( |
||
505 | endpoint_a_port=9, endpoint_b_port=10, metadata={"s_vlan": 5} |
||
506 | ), |
||
507 | get_link_mocked( |
||
508 | endpoint_a_port=11, endpoint_b_port=12, metadata={"s_vlan": 6} |
||
509 | ), |
||
510 | ] |
||
511 | |||
512 | 1 | attributes = { |
|
513 | "controller": get_controller_mock(), |
||
514 | "name": "custom_name", |
||
515 | "uni_a": uni_a, |
||
516 | "uni_z": uni_z, |
||
517 | "primary_links": primary_links, |
||
518 | "queue_id": 5, |
||
519 | } |
||
520 | # Setup path to deploy |
||
521 | 1 | path = Path() |
|
522 | 1 | path.append(primary_links[0]) |
|
523 | 1 | path.append(primary_links[1]) |
|
524 | |||
525 | 1 | evc = EVC(**attributes) |
|
526 | |||
527 | 1 | deployed = evc.deploy_to_path(path) |
|
528 | |||
529 | 1 | assert discover_new_paths.call_count == 0 |
|
530 | 1 | assert should_deploy_mock.call_count == 1 |
|
531 | 1 | assert install_unni_flows.call_count == 1 |
|
532 | 1 | assert choose_vlans_mock.call_count == 1 |
|
533 | 1 | assert log_mock.error.call_count == 1 |
|
534 | 1 | assert sync_mock.call_count == 0 |
|
535 | 1 | assert remove_current_flows.call_count == 2 |
|
536 | 1 | assert deployed is False |
|
537 | |||
538 | 1 | @patch("napps.kytos.mef_eline.models.evc.emit_event") |
|
539 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVC.get_failover_path_candidates") |
|
540 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVC._install_flows") |
|
541 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVC.remove_path_flows") |
|
542 | 1 | @patch("napps.kytos.mef_eline.models.EVC.sync") |
|
543 | 1 | def test_setup_failover_path(self, *args): |
|
544 | """Test setup_failover_path method.""" |
||
545 | 1 | ( |
|
546 | sync_mock, |
||
547 | remove_path_flows_mock, |
||
548 | install_unni_flows_mock, |
||
549 | get_failover_path_candidates_mock, |
||
550 | emit_event_mock, |
||
551 | ) = args |
||
552 | |||
553 | # case1: early return intra switch |
||
554 | 1 | evc1 = self.create_evc_intra_switch() |
|
555 | |||
556 | 1 | assert evc1.setup_failover_path() is False |
|
557 | 1 | assert sync_mock.call_count == 0 |
|
558 | |||
559 | # case2: early return not eligible for path failover |
||
560 | 1 | evc2 = self.create_evc_inter_switch() |
|
561 | 1 | evc2.is_eligible_for_failover_path = MagicMock(return_value=False) |
|
562 | |||
563 | 1 | assert evc2.setup_failover_path() is False |
|
564 | 1 | assert sync_mock.call_count == 0 |
|
565 | |||
566 | # case3: success failover_path setup |
||
567 | 1 | evc2.is_eligible_for_failover_path = MagicMock(return_value=True) |
|
568 | 1 | evc2.failover_path = ["link1", "link2"] |
|
569 | 1 | path_mock = MagicMock() |
|
570 | 1 | path_mock.__iter__.return_value = ["link3"] |
|
571 | 1 | get_failover_path_candidates_mock.return_value = [None, path_mock] |
|
572 | 1 | mock_choose = path_mock.choose_vlans |
|
573 | |||
574 | 1 | assert evc2.setup_failover_path() is True |
|
575 | 1 | remove_path_flows_mock.assert_called_with(["link1", "link2"]) |
|
576 | 1 | mock_choose.assert_called() |
|
577 | 1 | install_unni_flows_mock.assert_called_with(path_mock, skip_in=True) |
|
578 | 1 | assert evc2.failover_path == path_mock |
|
579 | 1 | assert sync_mock.call_count == 1 |
|
580 | 1 | assert emit_event_mock.call_count == 1 |
|
581 | 1 | assert emit_event_mock.call_args[0][1] == "failover_deployed" |
|
582 | |||
583 | # case 4: failed to setup failover_path - No Tag available |
||
584 | 1 | evc2.failover_path = [] |
|
585 | 1 | mock_choose.side_effect = KytosNoTagAvailableError(MagicMock()) |
|
586 | 1 | sync_mock.call_count = 0 |
|
587 | |||
588 | 1 | assert evc2.setup_failover_path() is False |
|
589 | 1 | assert len(list(evc2.failover_path)) == 0 |
|
590 | 1 | assert sync_mock.call_count == 1 |
|
591 | |||
592 | # case 5: failed to setup failover_path - FlowMod exception |
||
593 | 1 | evc2.failover_path = [] |
|
594 | 1 | mock_choose.side_effect = None |
|
595 | 1 | install_unni_flows_mock.side_effect = EVCPathNotInstalled("error") |
|
596 | 1 | sync_mock.call_count = 0 |
|
597 | |||
598 | 1 | assert evc2.setup_failover_path() is False |
|
599 | 1 | assert len(list(evc2.failover_path)) == 0 |
|
600 | 1 | assert sync_mock.call_count == 1 |
|
601 | 1 | remove_path_flows_mock.assert_called_with(path_mock) |
|
602 | |||
603 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVC.deploy_to_path") |
|
604 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVC.discover_new_paths") |
|
605 | 1 | def test_deploy_to_backup_path1( |
|
606 | self, discover_new_paths_mocked, deploy_to_path_mocked |
||
607 | ): |
||
608 | """Test deployment when dynamic_backup_path is False in same switch""" |
||
609 | 1 | uni_a = get_uni_mocked(interface_port=2, tag_value=82, is_valid=True) |
|
610 | 1 | uni_z = get_uni_mocked(interface_port=3, tag_value=83, is_valid=True) |
|
611 | |||
612 | 1 | switch = Mock(spec=Switch) |
|
613 | 1 | uni_a.interface.switch = switch |
|
614 | 1 | uni_z.interface.switch = switch |
|
615 | |||
616 | 1 | attributes = { |
|
617 | "controller": get_controller_mock(), |
||
618 | "name": "custom_name", |
||
619 | "uni_a": uni_a, |
||
620 | "uni_z": uni_z, |
||
621 | "enabled": True, |
||
622 | "dynamic_backup_path": False, |
||
623 | } |
||
624 | |||
625 | 1 | evc = EVC(**attributes) |
|
626 | 1 | discover_new_paths_mocked.return_value = [] |
|
627 | 1 | deploy_to_path_mocked.return_value = True |
|
628 | |||
629 | 1 | deployed = evc.deploy_to_backup_path() |
|
630 | |||
631 | 1 | deploy_to_path_mocked.assert_called_once_with() |
|
632 | 1 | assert deployed is True |
|
633 | |||
634 | 1 | @patch("httpx.post") |
|
635 | 1 | @patch("napps.kytos.mef_eline.controllers.ELineController.upsert_evc") |
|
636 | 1 | @patch("napps.kytos.mef_eline.models.evc.log") |
|
637 | 1 | @patch("napps.kytos.mef_eline.models.path.Path.choose_vlans") |
|
638 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVC._install_flows") |
|
639 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVC.activate") |
|
640 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVC.should_deploy") |
|
641 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVC.discover_new_paths") |
|
642 | 1 | def test_deploy_without_path_case1(self, *args): |
|
643 | """Test if not path is found a dynamic path is used.""" |
||
644 | # pylint: disable=too-many-locals |
||
645 | 1 | ( |
|
646 | discover_new_paths_mocked, |
||
647 | should_deploy_mock, |
||
648 | activate_mock, |
||
649 | install_unni_flows_mock, |
||
650 | chose_vlans_mock, |
||
651 | log_mock, |
||
652 | _, |
||
653 | httpx_mock, |
||
654 | ) = args |
||
655 | |||
656 | 1 | response = MagicMock() |
|
657 | 1 | response.status_code = 201 |
|
658 | 1 | httpx_mock.return_value = response |
|
659 | |||
660 | 1 | should_deploy_mock.return_value = False |
|
661 | 1 | uni_a = get_uni_mocked( |
|
662 | interface_port=2, |
||
663 | tag_value=82, |
||
664 | switch_id="switch_uni_a", |
||
665 | is_valid=True, |
||
666 | ) |
||
667 | 1 | uni_z = get_uni_mocked( |
|
668 | interface_port=3, |
||
669 | tag_value=83, |
||
670 | switch_id="switch_uni_z", |
||
671 | is_valid=True, |
||
672 | ) |
||
673 | |||
674 | 1 | attributes = { |
|
675 | "controller": get_controller_mock(), |
||
676 | "name": "custom_name", |
||
677 | "uni_a": uni_a, |
||
678 | "uni_z": uni_z, |
||
679 | "enabled": True, |
||
680 | "dynamic_backup_path": False, |
||
681 | } |
||
682 | |||
683 | 1 | dynamic_backup_path = Path( |
|
684 | [ |
||
685 | get_link_mocked( |
||
686 | endpoint_a_port=9, |
||
687 | endpoint_b_port=10, |
||
688 | metadata={"s_vlan": 5}, |
||
689 | ), |
||
690 | get_link_mocked( |
||
691 | endpoint_a_port=11, |
||
692 | endpoint_b_port=12, |
||
693 | metadata={"s_vlan": 6}, |
||
694 | ), |
||
695 | ] |
||
696 | ) |
||
697 | |||
698 | 1 | evc = EVC(**attributes) |
|
699 | 1 | discover_new_paths_mocked.return_value = [dynamic_backup_path] |
|
700 | |||
701 | 1 | deployed = evc.deploy_to_path() |
|
702 | |||
703 | 1 | assert should_deploy_mock.call_count == 1 |
|
704 | 1 | assert discover_new_paths_mocked.call_count == 1 |
|
705 | 1 | assert activate_mock.call_count == 1 |
|
706 | 1 | assert install_unni_flows_mock.call_count == 1 |
|
707 | 1 | assert chose_vlans_mock.call_count == 1 |
|
708 | 1 | log_mock.info.assert_called_with(f"{evc} was deployed.") |
|
709 | 1 | assert deployed is True |
|
710 | |||
711 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVCDeploy.deploy_to_primary_path") |
|
712 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVCDeploy.deploy_to_backup_path") |
|
713 | 1 | @patch("napps.kytos.mef_eline.models.evc.emit_event") |
|
714 | 1 | def test_deploy(self, *args): |
|
715 | """Test method deploy""" |
||
716 | 1 | (emit_event_mock, deploy_primary_mock, deploy_backup_mock) = args |
|
717 | |||
718 | # case 1: deploy to primary |
||
719 | 1 | self.evc_deploy.archived = False |
|
720 | 1 | deploy_primary_mock.return_value = True |
|
721 | 1 | assert self.evc_deploy.deploy() |
|
722 | 1 | assert emit_event_mock.call_count == 1 |
|
723 | |||
724 | # case 2: deploy to backup |
||
725 | 1 | deploy_primary_mock.return_value = False |
|
726 | 1 | deploy_backup_mock.return_value = True |
|
727 | 1 | assert self.evc_deploy.deploy() |
|
728 | 1 | assert emit_event_mock.call_count == 2 |
|
729 | |||
730 | # case 3: fail to deploy to primary and backup |
||
731 | 1 | deploy_backup_mock.return_value = False |
|
732 | 1 | assert self.evc_deploy.deploy() is False |
|
733 | 1 | assert emit_event_mock.call_count == 2 |
|
734 | |||
735 | # case 4: archived |
||
736 | 1 | self.evc_deploy.archived = True |
|
737 | 1 | assert self.evc_deploy.deploy() is False |
|
738 | 1 | assert emit_event_mock.call_count == 2 |
|
739 | |||
740 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVCDeploy.remove_current_flows") |
|
741 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVCDeploy.sync") |
|
742 | 1 | @patch("napps.kytos.mef_eline.models.evc.emit_event") |
|
743 | 1 | def test_remove(self, *args): |
|
744 | """Test method remove""" |
||
745 | 1 | (emit_event_mock, sync_mock, remove_flows_mock) = args |
|
746 | 1 | self.evc_deploy.remove() |
|
747 | 1 | remove_flows_mock.assert_called() |
|
748 | 1 | sync_mock.assert_called() |
|
749 | 1 | emit_event_mock.assert_called() |
|
750 | 1 | assert self.evc_deploy.is_enabled() is False |
|
751 | |||
752 | 1 | @patch("napps.kytos.mef_eline.controllers.ELineController.upsert_evc") |
|
753 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVC._send_flow_mods") |
|
754 | 1 | @patch("napps.kytos.mef_eline.models.evc.log.error") |
|
755 | 1 | def test_remove_current_flows(self, *args): |
|
756 | """Test remove current flows.""" |
||
757 | # pylint: disable=too-many-locals |
||
758 | 1 | (log_error_mock, send_flow_mods_mocked, _) = args |
|
759 | 1 | uni_a = get_uni_mocked( |
|
760 | interface_port=2, |
||
761 | tag_value=82, |
||
762 | switch_id="switch_uni_a", |
||
763 | is_valid=True, |
||
764 | ) |
||
765 | 1 | uni_z = get_uni_mocked( |
|
766 | interface_port=3, |
||
767 | tag_value=83, |
||
768 | switch_id="switch_uni_z", |
||
769 | is_valid=True, |
||
770 | ) |
||
771 | |||
772 | 1 | switch_a = Switch("00:00:00:00:00:01") |
|
773 | 1 | switch_b = Switch("00:00:00:00:00:02") |
|
774 | 1 | switch_c = Switch("00:00:00:00:00:03") |
|
775 | |||
776 | 1 | attributes = { |
|
777 | "controller": get_controller_mock(), |
||
778 | "name": "custom_name", |
||
779 | "uni_a": uni_a, |
||
780 | "uni_z": uni_z, |
||
781 | "active": True, |
||
782 | "enabled": True, |
||
783 | "primary_links": [ |
||
784 | get_link_mocked( |
||
785 | switch_a=switch_a, |
||
786 | switch_b=switch_b, |
||
787 | endpoint_a_port=9, |
||
788 | endpoint_b_port=10, |
||
789 | metadata={"s_vlan": 5}, |
||
790 | ), |
||
791 | get_link_mocked( |
||
792 | switch_a=switch_b, |
||
793 | switch_b=switch_c, |
||
794 | endpoint_a_port=11, |
||
795 | endpoint_b_port=12, |
||
796 | metadata={"s_vlan": 6}, |
||
797 | ), |
||
798 | ], |
||
799 | } |
||
800 | |||
801 | 1 | evc = EVC(**attributes) |
|
802 | |||
803 | 1 | evc.current_path = evc.primary_links |
|
804 | 1 | evc.remove_current_flows() |
|
805 | |||
806 | 1 | assert send_flow_mods_mocked.call_count == 1 |
|
807 | 1 | assert evc.is_active() is False |
|
808 | 1 | flows = [ |
|
809 | {"cookie": evc.get_cookie(), "cookie_mask": 18446744073709551615, |
||
810 | "owner": "mef_eline"} |
||
811 | ] |
||
812 | 1 | expected_flows = { |
|
813 | "switches": [ |
||
814 | switch_a.id, switch_b.id, switch_c.id, |
||
815 | evc.uni_a.interface.switch.id, |
||
816 | evc.uni_z.interface.switch.id, |
||
817 | ], |
||
818 | "flows": flows |
||
819 | } |
||
820 | 1 | args = send_flow_mods_mocked.call_args[0] |
|
821 | 1 | assert set(expected_flows["switches"]) == set(args[0]["switches"]) |
|
822 | 1 | assert expected_flows["flows"] == args[0]["flows"] |
|
823 | 1 | assert 'delete' == args[1] |
|
824 | |||
825 | 1 | send_flow_mods_mocked.side_effect = FlowModException("error") |
|
826 | 1 | evc.remove_current_flows() |
|
827 | 1 | log_error_mock.assert_called() |
|
828 | |||
829 | 1 | @patch("napps.kytos.mef_eline.controllers.ELineController.upsert_evc") |
|
830 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVC._send_flow_mods") |
|
831 | 1 | @patch("napps.kytos.mef_eline.models.evc.log.error") |
|
832 | 1 | def test_remove_failover_flows_exclude_uni_switches(self, *args): |
|
833 | """Test remove failover flows excluding UNI switches.""" |
||
834 | # pylint: disable=too-many-locals |
||
835 | 1 | (log_error_mock, send_flow_mods_mocked, mock_upsert) = args |
|
836 | 1 | uni_a = get_uni_mocked( |
|
837 | interface_port=2, |
||
838 | tag_value=82, |
||
839 | switch_id="00:00:00:00:00:00:00:01", |
||
840 | is_valid=True, |
||
841 | ) |
||
842 | 1 | uni_z = get_uni_mocked( |
|
843 | interface_port=3, |
||
844 | tag_value=83, |
||
845 | switch_id="00:00:00:00:00:00:00:03", |
||
846 | is_valid=True, |
||
847 | ) |
||
848 | |||
849 | 1 | switch_a = Switch("00:00:00:00:00:00:00:01") |
|
850 | 1 | switch_b = Switch("00:00:00:00:00:00:00:02") |
|
851 | 1 | switch_c = Switch("00:00:00:00:00:00:00:03") |
|
852 | |||
853 | 1 | attributes = { |
|
854 | "controller": get_controller_mock(), |
||
855 | "name": "custom_name", |
||
856 | "uni_a": uni_a, |
||
857 | "uni_z": uni_z, |
||
858 | "active": True, |
||
859 | "enabled": True, |
||
860 | "failover_path": [ |
||
861 | get_link_mocked( |
||
862 | switch_a=switch_a, |
||
863 | switch_b=switch_b, |
||
864 | endpoint_a_port=9, |
||
865 | endpoint_b_port=10, |
||
866 | metadata={"s_vlan": 5}, |
||
867 | ), |
||
868 | get_link_mocked( |
||
869 | switch_a=switch_b, |
||
870 | switch_b=switch_c, |
||
871 | endpoint_a_port=11, |
||
872 | endpoint_b_port=12, |
||
873 | metadata={"s_vlan": 6}, |
||
874 | ), |
||
875 | ], |
||
876 | } |
||
877 | |||
878 | 1 | evc = EVC(**attributes) |
|
879 | 1 | evc.remove_failover_flows(exclude_uni_switches=True, sync=True) |
|
880 | |||
881 | 1 | assert send_flow_mods_mocked.call_count == 1 |
|
882 | 1 | flows = [ |
|
883 | {"cookie": evc.get_cookie(), |
||
884 | "cookie_mask": int(0xffffffffffffffff), |
||
885 | "owner": "mef_eline"} |
||
886 | ] |
||
887 | 1 | expected_flows = { |
|
888 | "switches": [switch_b.id], |
||
889 | "flows": flows |
||
890 | } |
||
891 | 1 | send_flow_mods_mocked.assert_any_call(expected_flows, 'delete', |
|
892 | force=True) |
||
893 | 1 | assert mock_upsert.call_count == 1 |
|
894 | |||
895 | 1 | send_flow_mods_mocked.side_effect = FlowModException("error") |
|
896 | 1 | evc.remove_current_flows() |
|
897 | 1 | log_error_mock.assert_called() |
|
898 | |||
899 | 1 | @patch("napps.kytos.mef_eline.controllers.ELineController.upsert_evc") |
|
900 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVC._send_flow_mods") |
|
901 | 1 | def test_remove_failover_flows_include_all(self, *args): |
|
902 | """Test remove failover flows including UNI switches.""" |
||
903 | # pylint: disable=too-many-locals |
||
904 | 1 | (send_flow_mods_mocked, mock_upsert) = args |
|
905 | 1 | uni_a = get_uni_mocked( |
|
906 | interface_port=2, |
||
907 | tag_value=82, |
||
908 | switch_id="00:00:00:00:00:00:00:01", |
||
909 | is_valid=True, |
||
910 | ) |
||
911 | 1 | uni_z = get_uni_mocked( |
|
912 | interface_port=3, |
||
913 | tag_value=83, |
||
914 | switch_id="00:00:00:00:00:00:00:03", |
||
915 | is_valid=True, |
||
916 | ) |
||
917 | |||
918 | 1 | switch_a = Switch("00:00:00:00:00:00:00:01") |
|
919 | 1 | switch_b = Switch("00:00:00:00:00:00:00:02") |
|
920 | 1 | switch_c = Switch("00:00:00:00:00:00:00:03") |
|
921 | |||
922 | 1 | attributes = { |
|
923 | "controller": get_controller_mock(), |
||
924 | "name": "custom_name", |
||
925 | "uni_a": uni_a, |
||
926 | "uni_z": uni_z, |
||
927 | "active": True, |
||
928 | "enabled": True, |
||
929 | "failover_path": [ |
||
930 | get_link_mocked( |
||
931 | switch_a=switch_a, |
||
932 | switch_b=switch_b, |
||
933 | endpoint_a_port=9, |
||
934 | endpoint_b_port=10, |
||
935 | metadata={"s_vlan": 5}, |
||
936 | ), |
||
937 | get_link_mocked( |
||
938 | switch_a=switch_b, |
||
939 | switch_b=switch_c, |
||
940 | endpoint_a_port=11, |
||
941 | endpoint_b_port=12, |
||
942 | metadata={"s_vlan": 6}, |
||
943 | ), |
||
944 | ], |
||
945 | } |
||
946 | |||
947 | 1 | evc = EVC(**attributes) |
|
948 | 1 | evc.remove_failover_flows(exclude_uni_switches=False, sync=True) |
|
949 | |||
950 | 1 | assert send_flow_mods_mocked.call_count == 1 |
|
951 | 1 | flows = [ |
|
952 | {"cookie": evc.get_cookie(), |
||
953 | "cookie_mask": int(0xffffffffffffffff), |
||
954 | "owner": "mef_eline"} |
||
955 | ] |
||
956 | 1 | expected_flows = { |
|
957 | "switches": [switch_a.id, switch_b.id, switch_c.id], |
||
958 | "flows": flows |
||
959 | } |
||
960 | 1 | args = send_flow_mods_mocked.call_args[0] |
|
961 | 1 | assert set(args[0]["switches"]) == set(expected_flows["switches"]) |
|
962 | 1 | assert args[0]["flows"] == expected_flows["flows"] |
|
963 | 1 | assert args[1] == 'delete' |
|
964 | 1 | assert mock_upsert.call_count == 1 |
|
965 | |||
966 | 1 | @staticmethod |
|
967 | 1 | def create_evc_intra_switch(): |
|
968 | """Create intra-switch EVC.""" |
||
969 | 1 | switch = Mock(spec=Switch) |
|
970 | 1 | switch.dpid = 2 |
|
971 | 1 | switch.id = switch.dpid |
|
972 | 1 | interface_a = Interface("eth0", 1, switch) |
|
973 | 1 | interface_z = Interface("eth1", 3, switch) |
|
974 | 1 | uni_a = get_uni_mocked( |
|
975 | tag_value=82, |
||
976 | is_valid=True, |
||
977 | ) |
||
978 | 1 | uni_z = get_uni_mocked( |
|
979 | tag_value=84, |
||
980 | is_valid=True, |
||
981 | ) |
||
982 | 1 | uni_a.interface = interface_a |
|
983 | 1 | uni_z.interface = interface_z |
|
984 | 1 | attributes = { |
|
985 | "table_group": {"epl": 0, "evpl": 0}, |
||
986 | "controller": get_controller_mock(), |
||
987 | "name": "custom_name", |
||
988 | "id": "1", |
||
989 | "uni_a": uni_a, |
||
990 | "uni_z": uni_z, |
||
991 | "enabled": True, |
||
992 | } |
||
993 | 1 | return EVC(**attributes) |
|
994 | |||
995 | # pylint: disable=too-many-branches |
||
996 | 1 | @pytest.mark.parametrize( |
|
997 | "uni_a,uni_z", |
||
998 | [ |
||
999 | (100, 50), |
||
1000 | (100, "4096/4096"), |
||
1001 | (100, 0), |
||
1002 | (100, None), |
||
1003 | ("4096/4096", 50), |
||
1004 | ("4096/4096", "4096/4096"), |
||
1005 | ("4096/4096", 0), |
||
1006 | ("4096/4096", None), |
||
1007 | (0, 50), |
||
1008 | (0, "4096/4096"), |
||
1009 | (0, 0), |
||
1010 | (0, None), |
||
1011 | (None, 50), |
||
1012 | (None, "4096/4096"), |
||
1013 | (None, 0), |
||
1014 | (None, None), |
||
1015 | ] |
||
1016 | ) |
||
1017 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVC._send_flow_mods") |
|
1018 | 1 | def test_deploy_direct_uni_flows(self, send_flow_mods_mock, uni_a, uni_z): |
|
1019 | """Test _install_direct_uni_flows""" |
||
1020 | 1 | evc = TestEVC.create_evc_intra_switch() |
|
1021 | 1 | expected_dpid = evc.uni_a.interface.switch.id |
|
1022 | |||
1023 | 1 | expected_flows = [ |
|
1024 | { |
||
1025 | "match": {"in_port": 1}, |
||
1026 | "cookie": evc.get_cookie(), |
||
1027 | "owner": "mef_eline", |
||
1028 | "table_id": 0, |
||
1029 | "table_group": "epl", |
||
1030 | "actions": [ |
||
1031 | {"action_type": "output", "port": 3}, |
||
1032 | ], |
||
1033 | "priority": EPL_SB_PRIORITY |
||
1034 | }, |
||
1035 | { |
||
1036 | "match": {"in_port": 3}, |
||
1037 | "cookie": evc.get_cookie(), |
||
1038 | "owner": "mef_eline", |
||
1039 | "table_id": 0, |
||
1040 | "table_group": "epl", |
||
1041 | "actions": [ |
||
1042 | {"action_type": "output", "port": 1}, |
||
1043 | ], |
||
1044 | "priority": EPL_SB_PRIORITY |
||
1045 | } |
||
1046 | ] |
||
1047 | 1 | evc.uni_a = get_uni_mocked(tag_value=uni_a, is_valid=True) |
|
1048 | 1 | evc.uni_a.interface.port_number = 1 |
|
1049 | 1 | evc.uni_z = get_uni_mocked(tag_value=uni_z, is_valid=True) |
|
1050 | 1 | evc.uni_z.interface.port_number = 3 |
|
1051 | 1 | expected_dpid = evc.uni_a.interface.switch.id |
|
1052 | 1 | evc._install_direct_uni_flows() |
|
1053 | 1 | if uni_a is not None: |
|
1054 | 1 | expected_flows[0]["match"]["dl_vlan"] = uni_a |
|
1055 | 1 | expected_flows[0]["table_group"] = "evpl" |
|
1056 | 1 | if uni_z is not None: |
|
1057 | 1 | expected_flows[1]["match"]["dl_vlan"] = uni_z |
|
1058 | 1 | expected_flows[1]["table_group"] = "evpl" |
|
1059 | 1 | expected_flows[0]["priority"] = EVC.get_priority(uni_a) |
|
1060 | 1 | expected_flows[1]["priority"] = EVC.get_priority(uni_z) |
|
1061 | 1 | if uni_z not in evc.special_cases: |
|
1062 | 1 | expected_flows[0]["actions"].insert( |
|
1063 | 0, {"action_type": "set_vlan", "vlan_id": uni_z} |
||
1064 | ) |
||
1065 | 1 | if uni_a not in evc.special_cases: |
|
1066 | 1 | expected_flows[1]["actions"].insert( |
|
1067 | 0, {"action_type": "set_vlan", |
||
1068 | "vlan_id": uni_a} |
||
1069 | ) |
||
1070 | 1 | if not uni_z: |
|
1071 | 1 | expected_flows[1]["actions"].insert( |
|
1072 | 0, {"action_type": "push_vlan", |
||
1073 | "tag_type": "c"} |
||
1074 | ) |
||
1075 | 1 | if uni_z == 0: |
|
1076 | 1 | new_action = {"action_type": "pop_vlan"} |
|
1077 | 1 | expected_flows[0]["actions"].insert(0, new_action) |
|
1078 | 1 | elif uni_a == "4096/4096": |
|
1079 | 1 | if uni_z == 0: |
|
1080 | 1 | new_action = {"action_type": "pop_vlan"} |
|
1081 | 1 | expected_flows[0]["actions"].insert(0, new_action) |
|
1082 | 1 | elif uni_a == 0: |
|
1083 | 1 | if uni_z not in evc.special_cases: |
|
1084 | 1 | expected_flows[0]["actions"].insert( |
|
1085 | 0, {"action_type": "push_vlan", |
||
1086 | "tag_type": "c"} |
||
1087 | ) |
||
1088 | 1 | if uni_z: |
|
1089 | 1 | new_action = {"action_type": "pop_vlan"} |
|
1090 | 1 | expected_flows[1]["actions"].insert(0, new_action) |
|
1091 | 1 | elif uni_a is None: |
|
1092 | 1 | if uni_z not in evc.special_cases: |
|
1093 | 1 | expected_flows[0]["actions"].insert( |
|
1094 | 0, {"action_type": "push_vlan", |
||
1095 | "tag_type": "c"} |
||
1096 | ) |
||
1097 | 1 | flow_mods = {"switches": [expected_dpid], "flows": expected_flows} |
|
1098 | 1 | send_flow_mods_mock.assert_called_with(flow_mods, "install") |
|
1099 | |||
1100 | 1 | def test_is_affected_by_link(self): |
|
1101 | """Test is_affected_by_link method""" |
||
1102 | 1 | self.evc_deploy.current_path = Path(['a', 'b', 'c']) |
|
1103 | 1 | assert self.evc_deploy.is_affected_by_link('b') is True |
|
1104 | |||
1105 | 1 | def test_is_backup_path_affected_by_link(self): |
|
1106 | """Test is_backup_path_affected_by_link method""" |
||
1107 | 1 | self.evc_deploy.backup_path = Path(['a', 'b', 'c']) |
|
1108 | 1 | assert self.evc_deploy.is_backup_path_affected_by_link('d') is False |
|
1109 | |||
1110 | 1 | def test_is_primary_path_affected_by_link(self): |
|
1111 | """Test is_primary_path_affected_by_link method""" |
||
1112 | 1 | self.evc_deploy.primary_path = Path(['a', 'b', 'c']) |
|
1113 | 1 | assert self.evc_deploy.is_primary_path_affected_by_link('c') is True |
|
1114 | |||
1115 | 1 | def test_is_using_primary_path(self): |
|
1116 | """Test is_using_primary_path method""" |
||
1117 | 1 | self.evc_deploy.primary_path = Path(['a', 'b', 'c']) |
|
1118 | 1 | self.evc_deploy.current_path = Path(['e', 'f', 'g']) |
|
1119 | 1 | assert self.evc_deploy.is_using_primary_path() is False |
|
1120 | |||
1121 | 1 | def test_is_using_backup_path(self): |
|
1122 | """Test is_using_backup_path method""" |
||
1123 | 1 | self.evc_deploy.backup_path = Path(['a', 'b', 'c']) |
|
1124 | 1 | self.evc_deploy.current_path = Path(['e', 'f', 'g']) |
|
1125 | 1 | assert self.evc_deploy.is_using_backup_path() is False |
|
1126 | |||
1127 | 1 | @patch('napps.kytos.mef_eline.models.path.Path.status') |
|
1128 | 1 | def test_is_using_dynamic_path(self, mock_status): |
|
1129 | """Test is_using_dynamic_path method""" |
||
1130 | 1 | mock_status.return_value = False |
|
1131 | 1 | self.evc_deploy.backup_path = Path([]) |
|
1132 | 1 | self.evc_deploy.primary_path = Path([]) |
|
1133 | 1 | assert self.evc_deploy.is_using_dynamic_path() is False |
|
1134 | |||
1135 | 1 | def test_get_path_status(self): |
|
1136 | """Test get_path_status method""" |
||
1137 | 1 | path = Path([]) |
|
1138 | 1 | assert self.evc_deploy.get_path_status(path) == EntityStatus.DISABLED |
|
1139 | 1 | path = Path([ |
|
1140 | get_link_mocked(status=EntityStatus.UP), |
||
1141 | get_link_mocked(status=EntityStatus.DOWN) |
||
1142 | ]) |
||
1143 | 1 | assert self.evc_deploy.get_path_status(path) == EntityStatus.DOWN |
|
1144 | 1 | path = Path([ |
|
1145 | get_link_mocked(status=EntityStatus.UP), |
||
1146 | get_link_mocked(status=EntityStatus.UP) |
||
1147 | ]) |
||
1148 | 1 | assert self.evc_deploy.get_path_status(path) == EntityStatus.UP |
|
1149 | |||
1150 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVC._prepare_uni_flows") |
|
1151 | 1 | def test_get_failover_flows(self, prepare_uni_flows_mock): |
|
1152 | """Test get_failover_flows method.""" |
||
1153 | 1 | evc = self.create_evc_inter_switch() |
|
1154 | 1 | evc.failover_path = Path([]) |
|
1155 | 1 | assert len(evc.get_failover_flows()) == 0 |
|
1156 | |||
1157 | 1 | path = MagicMock() |
|
1158 | 1 | evc.failover_path = path |
|
1159 | 1 | evc.get_failover_flows() |
|
1160 | 1 | prepare_uni_flows_mock.assert_called_with(path, skip_out=True) |
|
1161 | |||
1162 | 1 | @patch("napps.kytos.mef_eline.models.evc.log") |
|
1163 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVC._send_flow_mods") |
|
1164 | 1 | @patch("napps.kytos.mef_eline.models.path.Path.make_vlans_available") |
|
1165 | 1 | def test_remove_path_flows(self, *args): |
|
1166 | """Test remove path flows.""" |
||
1167 | 1 | ( |
|
1168 | make_vlans_available_mock, |
||
1169 | send_flow_mods_mock, |
||
1170 | log_mock |
||
1171 | ) = args |
||
1172 | |||
1173 | 1 | evc = self.create_evc_inter_switch() |
|
1174 | |||
1175 | 1 | evc.remove_path_flows() |
|
1176 | 1 | make_vlans_available_mock.assert_not_called() |
|
1177 | |||
1178 | 1 | expected_flows_1 = [ |
|
1179 | { |
||
1180 | 'cookie': 12249790986447749121, |
||
1181 | 'cookie_mask': 18446744073709551615, |
||
1182 | "owner": "mef_eline", |
||
1183 | 'match': {'in_port': 9, 'dl_vlan': 5} |
||
1184 | }, |
||
1185 | ] |
||
1186 | 1 | expected_flows_2 = [ |
|
1187 | { |
||
1188 | 'cookie': 12249790986447749121, |
||
1189 | 'cookie_mask': 18446744073709551615, |
||
1190 | "owner": "mef_eline", |
||
1191 | 'match': {'in_port': 10, 'dl_vlan': 5} |
||
1192 | }, |
||
1193 | { |
||
1194 | 'cookie': 12249790986447749121, |
||
1195 | 'cookie_mask': 18446744073709551615, |
||
1196 | "owner": "mef_eline", |
||
1197 | 'match': {'in_port': 11, 'dl_vlan': 6} |
||
1198 | }, |
||
1199 | ] |
||
1200 | 1 | expected_flows_3 = [ |
|
1201 | { |
||
1202 | 'cookie': 12249790986447749121, |
||
1203 | 'cookie_mask': 18446744073709551615, |
||
1204 | "owner": "mef_eline", |
||
1205 | 'match': {'in_port': 12, 'dl_vlan': 6} |
||
1206 | }, |
||
1207 | ] |
||
1208 | |||
1209 | 1 | dpid_flows = evc.remove_path_flows(evc.primary_links) |
|
1210 | 1 | assert dpid_flows |
|
1211 | 1 | assert len(dpid_flows) == 3 |
|
1212 | 1 | assert sum(len(flows) for flows in dpid_flows.values()) == len( |
|
1213 | expected_flows_1 |
||
1214 | ) + len(expected_flows_2) + len(expected_flows_3) |
||
1215 | 1 | send_flow_mods_mock.assert_called_once() |
|
1216 | 1 | expected_flows = { |
|
1217 | 1: {"flows": expected_flows_1}, |
||
1218 | 2: {"flows": expected_flows_2}, |
||
1219 | 3: {"flows": expected_flows_3} |
||
1220 | } |
||
1221 | 1 | send_flow_mods_mock.assert_called_with( |
|
1222 | expected_flows, "delete", force=True, by_switch=True |
||
1223 | ) |
||
1224 | |||
1225 | 1 | send_flow_mods_mock.side_effect = FlowModException("err") |
|
1226 | 1 | evc.remove_path_flows(evc.primary_links) |
|
1227 | 1 | log_mock.error.assert_called() |
|
1228 | |||
1229 | 1 | @patch("httpx.put") |
|
1230 | 1 | def test_run_bulk_sdntraces(self, put_mock): |
|
1231 | """Test run_bulk_sdntraces method for bulk request.""" |
||
1232 | 1 | evc = self.create_evc_inter_switch() |
|
1233 | 1 | response = MagicMock() |
|
1234 | 1 | response.status_code = 200 |
|
1235 | 1 | response.json.return_value = {"result": "ok"} |
|
1236 | 1 | put_mock.return_value = response |
|
1237 | |||
1238 | 1 | expected_endpoint = f"{SDN_TRACE_CP_URL}/traces" |
|
1239 | 1 | expected_payload = [ |
|
1240 | { |
||
1241 | 'trace': { |
||
1242 | 'switch': {'dpid': 1, 'in_port': 2}, |
||
1243 | 'eth': {'dl_type': 0x8100, 'dl_vlan': 82} |
||
1244 | } |
||
1245 | } |
||
1246 | ] |
||
1247 | 1 | arg_tuple = [(evc.uni_a.interface, evc.uni_a.user_tag.value)] |
|
1248 | 1 | result = EVCDeploy.run_bulk_sdntraces(arg_tuple) |
|
1249 | 1 | put_mock.assert_called_with( |
|
1250 | expected_endpoint, |
||
1251 | json=expected_payload, |
||
1252 | timeout=30 |
||
1253 | ) |
||
1254 | 1 | assert result['result'] == "ok" |
|
1255 | |||
1256 | 1 | response.status_code = 400 |
|
1257 | 1 | result = EVCDeploy.run_bulk_sdntraces(arg_tuple) |
|
1258 | 1 | assert result == {"result": []} |
|
1259 | |||
1260 | 1 | put_mock.side_effect = TimeoutException('Timeout') |
|
1261 | 1 | response.status_code = 200 |
|
1262 | 1 | result = EVCDeploy.run_bulk_sdntraces(arg_tuple) |
|
1263 | 1 | assert result == {"result": []} |
|
1264 | |||
1265 | 1 | @patch("httpx.put") |
|
1266 | 1 | def test_run_bulk_sdntraces_special_vlan(self, put_mock): |
|
1267 | """Test run_bulk_sdntraces method for bulk request.""" |
||
1268 | 1 | evc = self.create_evc_inter_switch() |
|
1269 | 1 | response = MagicMock() |
|
1270 | 1 | response.status_code = 200 |
|
1271 | 1 | put_mock.return_value = response |
|
1272 | |||
1273 | 1 | expected_endpoint = f"{SDN_TRACE_CP_URL}/traces" |
|
1274 | 1 | expected_payload = [ |
|
1275 | { |
||
1276 | 'trace': { |
||
1277 | 'switch': {'dpid': 1, 'in_port': 2} |
||
1278 | } |
||
1279 | } |
||
1280 | ] |
||
1281 | 1 | evc.uni_a.user_tag.value = 'untagged' |
|
1282 | 1 | EVCDeploy.run_bulk_sdntraces( |
|
1283 | [(evc.uni_a.interface, evc.uni_a.user_tag.value)] |
||
1284 | ) |
||
1285 | 1 | put_mock.assert_called_with( |
|
1286 | expected_endpoint, |
||
1287 | json=expected_payload, |
||
1288 | timeout=30 |
||
1289 | ) |
||
1290 | 1 | args = put_mock.call_args[1]['json'][0] |
|
1291 | 1 | assert 'eth' not in args |
|
1292 | |||
1293 | 1 | evc.uni_a.user_tag.value = 0 |
|
1294 | 1 | EVCDeploy.run_bulk_sdntraces( |
|
1295 | [(evc.uni_a.interface, evc.uni_a.user_tag.value)] |
||
1296 | ) |
||
1297 | 1 | put_mock.assert_called_with( |
|
1298 | expected_endpoint, |
||
1299 | json=expected_payload, |
||
1300 | timeout=30 |
||
1301 | ) |
||
1302 | 1 | args = put_mock.call_args[1]['json'][0]['trace'] |
|
1303 | 1 | assert 'eth' not in args |
|
1304 | |||
1305 | 1 | evc.uni_a.user_tag.value = '5/2' |
|
1306 | 1 | EVCDeploy.run_bulk_sdntraces( |
|
1307 | [(evc.uni_a.interface, evc.uni_a.user_tag.value)] |
||
1308 | ) |
||
1309 | 1 | put_mock.assert_called_with( |
|
1310 | expected_endpoint, |
||
1311 | json=expected_payload, |
||
1312 | timeout=30 |
||
1313 | ) |
||
1314 | 1 | args = put_mock.call_args[1]['json'][0]['trace'] |
|
1315 | 1 | assert 'eth' not in args |
|
1316 | |||
1317 | 1 | expected_payload[0]['trace']['eth'] = {'dl_type': 0x8100, 'dl_vlan': 1} |
|
1318 | 1 | evc.uni_a.user_tag.value = 'any' |
|
1319 | 1 | EVCDeploy.run_bulk_sdntraces( |
|
1320 | [(evc.uni_a.interface, evc.uni_a.user_tag.value)] |
||
1321 | ) |
||
1322 | 1 | put_mock.assert_called_with( |
|
1323 | expected_endpoint, |
||
1324 | json=expected_payload, |
||
1325 | timeout=30 |
||
1326 | ) |
||
1327 | 1 | args = put_mock.call_args[1]['json'][0]['trace'] |
|
1328 | 1 | assert args['eth'] == {'dl_type': 33024, 'dl_vlan': 1} |
|
1329 | |||
1330 | 1 | evc.uni_a.user_tag.value = '4096/4096' |
|
1331 | 1 | EVCDeploy.run_bulk_sdntraces( |
|
1332 | [(evc.uni_a.interface, evc.uni_a.user_tag.value)] |
||
1333 | ) |
||
1334 | 1 | put_mock.assert_called_with( |
|
1335 | expected_endpoint, |
||
1336 | json=expected_payload, |
||
1337 | timeout=30 |
||
1338 | ) |
||
1339 | 1 | args = put_mock.call_args[1]['json'][0]['trace'] |
|
1340 | 1 | assert args['eth'] == {'dl_type': 33024, 'dl_vlan': 1} |
|
1341 | |||
1342 | 1 | expected_payload[0]['trace']['eth'] = { |
|
1343 | 'dl_type': 0x8100, |
||
1344 | 'dl_vlan': 10 |
||
1345 | } |
||
1346 | 1 | evc.uni_a.user_tag.value = '10/10' |
|
1347 | 1 | EVCDeploy.run_bulk_sdntraces( |
|
1348 | [(evc.uni_a.interface, evc.uni_a.user_tag.value)] |
||
1349 | ) |
||
1350 | 1 | put_mock.assert_called_with( |
|
1351 | expected_endpoint, |
||
1352 | json=expected_payload, |
||
1353 | timeout=30 |
||
1354 | ) |
||
1355 | 1 | args = put_mock.call_args[1]['json'][0]['trace'] |
|
1356 | 1 | assert args['eth'] == {'dl_type': 33024, 'dl_vlan': 10} |
|
1357 | |||
1358 | 1 | expected_payload[0]['trace']['eth'] = { |
|
1359 | 'dl_type': 0x8100, |
||
1360 | 'dl_vlan': 1 |
||
1361 | } |
||
1362 | 1 | evc.uni_a.user_tag.value = '5/3' |
|
1363 | 1 | EVCDeploy.run_bulk_sdntraces( |
|
1364 | [(evc.uni_a.interface, evc.uni_a.user_tag.value)] |
||
1365 | ) |
||
1366 | 1 | put_mock.assert_called_with( |
|
1367 | expected_endpoint, |
||
1368 | json=expected_payload, |
||
1369 | timeout=30 |
||
1370 | ) |
||
1371 | 1 | args = put_mock.call_args[1]['json'][0]['trace'] |
|
1372 | 1 | assert args['eth'] == {'dl_type': 33024, 'dl_vlan': 1} |
|
1373 | |||
1374 | 1 | expected_payload[0]['trace']['eth'] = { |
|
1375 | 'dl_type': 0x8100, |
||
1376 | 'dl_vlan': 10 |
||
1377 | } |
||
1378 | 1 | evc.uni_a.user_tag.value = 10 |
|
1379 | 1 | EVCDeploy.run_bulk_sdntraces( |
|
1380 | [(evc.uni_a.interface, evc.uni_a.user_tag.value)] |
||
1381 | ) |
||
1382 | 1 | put_mock.assert_called_with( |
|
1383 | expected_endpoint, |
||
1384 | json=expected_payload, |
||
1385 | timeout=30 |
||
1386 | ) |
||
1387 | |||
1388 | 1 | @patch("napps.kytos.mef_eline.models.evc.log") |
|
1389 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVCDeploy.run_bulk_sdntraces") |
|
1390 | 1 | def test_check_list_traces_ordered_unordered(self, run_bulk_mock, _): |
|
1391 | """Test check_list_traces with UNIs ordered and unordered.""" |
||
1392 | 1 | evc = self.create_evc_inter_switch() |
|
1393 | |||
1394 | 1 | for link in evc.primary_links: |
|
1395 | 1 | link.metadata['s_vlan'] = MagicMock(value=link.metadata['s_vlan']) |
|
1396 | 1 | evc.current_path = evc.primary_links |
|
1397 | |||
1398 | 1 | trace_a = [ |
|
1399 | { |
||
1400 | "dpid": 1, |
||
1401 | "port": 2, |
||
1402 | "time": "t1", |
||
1403 | "type": "starting", |
||
1404 | "vlan": 82 |
||
1405 | }, |
||
1406 | { |
||
1407 | "dpid": 2, |
||
1408 | "port": 10, |
||
1409 | "time": "t2", |
||
1410 | "type": "intermediary", |
||
1411 | "vlan": 5 |
||
1412 | }, |
||
1413 | {"dpid": 3, "port": 12, "time": "t3", "type": "last", "vlan": 6}, |
||
1414 | ] |
||
1415 | 1 | trace_z = [ |
|
1416 | { |
||
1417 | "dpid": 3, |
||
1418 | "port": 3, |
||
1419 | "time": "t1", |
||
1420 | "type": "starting", |
||
1421 | "vlan": 83 |
||
1422 | }, |
||
1423 | { |
||
1424 | "dpid": 2, |
||
1425 | "port": 11, |
||
1426 | "time": "t2", |
||
1427 | "type": "intermediary", |
||
1428 | "vlan": 6 |
||
1429 | }, |
||
1430 | {"dpid": 1, "port": 9, "time": "t3", "type": "last", "vlan": 5}, |
||
1431 | ] |
||
1432 | |||
1433 | 1 | run_bulk_mock.return_value = {"result": [trace_a, trace_z]} |
|
1434 | 1 | result = EVCDeploy.check_list_traces([evc]) |
|
1435 | 1 | assert result[evc.id] |
|
1436 | |||
1437 | # swapped UNIs since uni_a and uni_z might not be ordered with cur path |
||
1438 | 1 | run_bulk_mock.return_value = {"result": [trace_z, trace_a]} |
|
1439 | 1 | evc.uni_a, evc.uni_z = evc.uni_z, evc.uni_a |
|
1440 | 1 | result = EVCDeploy.check_list_traces([evc]) |
|
1441 | 1 | assert result[evc.id] |
|
1442 | |||
1443 | 1 | @patch("napps.kytos.mef_eline.models.evc.log") |
|
1444 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVCDeploy.run_bulk_sdntraces") |
|
1445 | 1 | def test_check_list_traces(self, run_bulk_sdntraces_mock, _): |
|
1446 | """Test check_list_traces method.""" |
||
1447 | 1 | evc = self.create_evc_inter_switch() |
|
1448 | |||
1449 | 1 | for link in evc.primary_links: |
|
1450 | 1 | link.metadata['s_vlan'] = MagicMock(value=link.metadata['s_vlan']) |
|
1451 | 1 | evc.current_path = evc.primary_links |
|
1452 | |||
1453 | 1 | trace_a = [ |
|
1454 | { |
||
1455 | "dpid": 1, |
||
1456 | "port": 2, |
||
1457 | "time": "t1", |
||
1458 | "type": "starting", |
||
1459 | "vlan": 82 |
||
1460 | }, |
||
1461 | { |
||
1462 | "dpid": 2, |
||
1463 | "port": 10, |
||
1464 | "time": "t2", |
||
1465 | "type": "intermediary", |
||
1466 | "vlan": 5 |
||
1467 | }, |
||
1468 | {"dpid": 3, "port": 12, "time": "t3", "type": "last", "vlan": 6}, |
||
1469 | ] |
||
1470 | 1 | trace_z = [ |
|
1471 | { |
||
1472 | "dpid": 3, |
||
1473 | "port": 3, |
||
1474 | "time": "t1", |
||
1475 | "type": "starting", |
||
1476 | "vlan": 83 |
||
1477 | }, |
||
1478 | { |
||
1479 | "dpid": 2, |
||
1480 | "port": 11, |
||
1481 | "time": "t2", |
||
1482 | "type": "intermediary", |
||
1483 | "vlan": 6 |
||
1484 | }, |
||
1485 | {"dpid": 1, "port": 9, "time": "t3", "type": "last", "vlan": 5}, |
||
1486 | ] |
||
1487 | |||
1488 | 1 | run_bulk_sdntraces_mock.return_value = { |
|
1489 | "result": [trace_a, trace_z] |
||
1490 | } |
||
1491 | 1 | result = EVCDeploy.check_list_traces([evc]) |
|
1492 | 1 | assert result[evc.id] is True |
|
1493 | |||
1494 | # case2: fail incomplete trace from uni_a |
||
1495 | 1 | run_bulk_sdntraces_mock.return_value = { |
|
1496 | "result": [ |
||
1497 | trace_a[:2], |
||
1498 | trace_z |
||
1499 | ] |
||
1500 | } |
||
1501 | 1 | result = EVCDeploy.check_list_traces([evc]) |
|
1502 | 1 | assert result[evc.id] is False |
|
1503 | |||
1504 | # case3: fail incomplete trace from uni_z |
||
1505 | 1 | run_bulk_sdntraces_mock.return_value = { |
|
1506 | "result": [ |
||
1507 | trace_a, |
||
1508 | trace_z[:2] |
||
1509 | ] |
||
1510 | } |
||
1511 | 1 | result = EVCDeploy.check_list_traces([evc]) |
|
1512 | 1 | assert result[evc.id] is False |
|
1513 | |||
1514 | # case4: fail wrong vlan id in trace from uni_a |
||
1515 | 1 | trace_a[1]["vlan"] = 5 |
|
1516 | 1 | trace_z[1]["vlan"] = 99 |
|
1517 | 1 | run_bulk_sdntraces_mock.return_value = { |
|
1518 | "result": [trace_a, trace_z] |
||
1519 | } |
||
1520 | 1 | result = EVCDeploy.check_list_traces([evc]) |
|
1521 | 1 | assert result[evc.id] is False |
|
1522 | |||
1523 | # case5: fail wrong vlan id in trace from uni_z |
||
1524 | 1 | trace_a[1]["vlan"] = 99 |
|
1525 | 1 | run_bulk_sdntraces_mock.return_value = { |
|
1526 | "result": [trace_a, trace_z] |
||
1527 | } |
||
1528 | 1 | result = EVCDeploy.check_list_traces([evc]) |
|
1529 | 1 | assert result[evc.id] is False |
|
1530 | |||
1531 | # case6: success when no output in traces |
||
1532 | 1 | trace_a[1]["vlan"] = 5 |
|
1533 | 1 | trace_z[1]["vlan"] = 6 |
|
1534 | 1 | result = EVCDeploy.check_list_traces([evc]) |
|
1535 | 1 | assert result[evc.id] is True |
|
1536 | |||
1537 | # case7: fail when output is None in trace_a or trace_b |
||
1538 | 1 | trace_a[-1]["out"] = None |
|
1539 | 1 | result = EVCDeploy.check_list_traces([evc]) |
|
1540 | 1 | assert result[evc.id] is False |
|
1541 | 1 | trace_a[-1].pop("out", None) |
|
1542 | 1 | trace_z[-1]["out"] = None |
|
1543 | 1 | result = EVCDeploy.check_list_traces([evc]) |
|
1544 | 1 | assert result[evc.id] is False |
|
1545 | |||
1546 | # case8: success when the output is correct on both uni |
||
1547 | 1 | trace_a[-1]["out"] = {"port": 3, "vlan": 83} |
|
1548 | 1 | trace_z[-1]["out"] = {"port": 2, "vlan": 82} |
|
1549 | 1 | result = EVCDeploy.check_list_traces([evc]) |
|
1550 | 1 | assert result[evc.id] is True |
|
1551 | |||
1552 | # case9: fail if any output is incorrect |
||
1553 | 1 | trace_a[-1]["out"] = {"port": 3, "vlan": 99} |
|
1554 | 1 | trace_z[-1]["out"] = {"port": 2, "vlan": 82} |
|
1555 | 1 | result = EVCDeploy.check_list_traces([evc]) |
|
1556 | 1 | assert result[evc.id] is False |
|
1557 | 1 | trace_a[-1]["out"] = {"port": 3, "vlan": 83} |
|
1558 | 1 | trace_z[-1]["out"] = {"port": 2, "vlan": 99} |
|
1559 | 1 | result = EVCDeploy.check_list_traces([evc]) |
|
1560 | 1 | assert result[evc.id] is False |
|
1561 | |||
1562 | 1 | View Code Duplication | @patch("napps.kytos.mef_eline.models.evc.log") |
1563 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVCDeploy.run_bulk_sdntraces") |
|
1564 | 1 | def test_check_list_traces_any_cases(self, run_bulk_sdntraces_mock, _): |
|
1565 | """Test check_list_traces method.""" |
||
1566 | 1 | evc = self.create_evc_inter_switch("any", "any") |
|
1567 | |||
1568 | 1 | for link in evc.primary_links: |
|
1569 | 1 | link.metadata['s_vlan'] = MagicMock(value=link.metadata['s_vlan']) |
|
1570 | 1 | evc.current_path = evc.primary_links |
|
1571 | |||
1572 | 1 | trace_a = [ |
|
1573 | { |
||
1574 | "dpid": 1, |
||
1575 | "port": 2, |
||
1576 | "time": "t1", |
||
1577 | "type": "starting", |
||
1578 | "vlan": 1 |
||
1579 | }, |
||
1580 | { |
||
1581 | "dpid": 2, |
||
1582 | "port": 10, |
||
1583 | "time": "t2", |
||
1584 | "type": "intermediary", |
||
1585 | "vlan": 5 |
||
1586 | }, |
||
1587 | { |
||
1588 | "dpid": 3, |
||
1589 | "port": 12, |
||
1590 | 'out': {'port': 3, 'vlan': 1}, |
||
1591 | "time": "t3", |
||
1592 | "type": "last", |
||
1593 | "vlan": 6 |
||
1594 | }, |
||
1595 | ] |
||
1596 | 1 | trace_z = [ |
|
1597 | { |
||
1598 | "dpid": 3, |
||
1599 | "port": 3, |
||
1600 | "time": "t1", |
||
1601 | "type": "starting", |
||
1602 | "vlan": 1 |
||
1603 | }, |
||
1604 | { |
||
1605 | "dpid": 2, |
||
1606 | "port": 11, |
||
1607 | "time": "t2", |
||
1608 | "type": "intermediary", |
||
1609 | "vlan": 6 |
||
1610 | }, |
||
1611 | { |
||
1612 | "dpid": 1, |
||
1613 | "port": 9, |
||
1614 | 'out': {'port': 2, 'vlan': 1}, |
||
1615 | "time": "t3", |
||
1616 | "type": "last", |
||
1617 | "vlan": 5 |
||
1618 | }, |
||
1619 | ] |
||
1620 | |||
1621 | 1 | run_bulk_sdntraces_mock.return_value = { |
|
1622 | "result": [trace_a, trace_z] |
||
1623 | } |
||
1624 | 1 | result = EVCDeploy.check_list_traces([evc]) |
|
1625 | 1 | assert result[evc.id] is True |
|
1626 | |||
1627 | 1 | View Code Duplication | @patch("napps.kytos.mef_eline.models.evc.log") |
1628 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVCDeploy.run_bulk_sdntraces") |
|
1629 | 1 | def test_check_list_traces_untagged_cases(self, bulk_sdntraces_mock, _): |
|
1630 | """Test check_list_traces method.""" |
||
1631 | 1 | evc = self.create_evc_inter_switch("untagged", "untagged") |
|
1632 | |||
1633 | 1 | for link in evc.primary_links: |
|
1634 | 1 | link.metadata['s_vlan'] = MagicMock(value=link.metadata['s_vlan']) |
|
1635 | 1 | evc.current_path = evc.primary_links |
|
1636 | |||
1637 | 1 | trace_a = [ |
|
1638 | { |
||
1639 | "dpid": 1, |
||
1640 | "port": 2, |
||
1641 | "time": "t1", |
||
1642 | "type": "starting", |
||
1643 | "vlan": 0 |
||
1644 | }, |
||
1645 | { |
||
1646 | "dpid": 2, |
||
1647 | "port": 10, |
||
1648 | "time": "t2", |
||
1649 | "type": "intermediary", |
||
1650 | "vlan": 5 |
||
1651 | }, |
||
1652 | { |
||
1653 | "dpid": 3, |
||
1654 | "port": 12, |
||
1655 | 'out': {'port': 3}, |
||
1656 | "time": "t3", "type": |
||
1657 | "last", |
||
1658 | "vlan": 6 |
||
1659 | }, |
||
1660 | ] |
||
1661 | 1 | trace_z = [ |
|
1662 | { |
||
1663 | "dpid": 3, |
||
1664 | "port": 3, |
||
1665 | "time": "t1", |
||
1666 | "type": "starting", |
||
1667 | "vlan": 0 |
||
1668 | }, |
||
1669 | { |
||
1670 | "dpid": 2, |
||
1671 | "port": 11, |
||
1672 | "time": "t2", |
||
1673 | "type": "intermediary", |
||
1674 | "vlan": 6 |
||
1675 | }, |
||
1676 | { |
||
1677 | "dpid": 1, |
||
1678 | "port": 9, |
||
1679 | 'out': {'port': 2}, |
||
1680 | "time": "t3", |
||
1681 | "type": "last", |
||
1682 | "vlan": 5 |
||
1683 | }, |
||
1684 | ] |
||
1685 | |||
1686 | 1 | bulk_sdntraces_mock.return_value = { |
|
1687 | "result": [trace_a, trace_z] |
||
1688 | } |
||
1689 | 1 | result = EVCDeploy.check_list_traces([evc]) |
|
1690 | 1 | assert result[evc.id] is True |
|
1691 | |||
1692 | 1 | @patch("napps.kytos.mef_eline.models.evc.log") |
|
1693 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVCDeploy.run_bulk_sdntraces") |
|
1694 | 1 | def test_check_list_traces_invalid_types(self, run_bulk_sdntraces_mock, _): |
|
1695 | """Test check_list_traces method for invalid traces by trace type.""" |
||
1696 | 1 | evc = self.create_evc_inter_switch() |
|
1697 | |||
1698 | 1 | for link in evc.primary_links: |
|
1699 | 1 | link.metadata['s_vlan'] = MagicMock(value=link.metadata['s_vlan']) |
|
1700 | 1 | evc.current_path = evc.primary_links |
|
1701 | |||
1702 | 1 | trace_a = [ |
|
1703 | { |
||
1704 | "dpid": 1, |
||
1705 | "port": 2, |
||
1706 | "time": "t1", |
||
1707 | "type": "starting", |
||
1708 | "vlan": 82 |
||
1709 | }, |
||
1710 | { |
||
1711 | "dpid": 2, |
||
1712 | "port": 10, |
||
1713 | "time": "t2", |
||
1714 | "type": "intermediary", |
||
1715 | "vlan": 5 |
||
1716 | }, |
||
1717 | {"dpid": 3, "port": 12, "time": "t3", "type": "last", "vlan": 6}, |
||
1718 | ] |
||
1719 | 1 | trace_z = [ |
|
1720 | { |
||
1721 | "dpid": 3, |
||
1722 | "port": 3, |
||
1723 | "time": "t1", |
||
1724 | "type": "starting", |
||
1725 | "vlan": 83 |
||
1726 | }, |
||
1727 | { |
||
1728 | "dpid": 2, |
||
1729 | "port": 11, |
||
1730 | "time": "t2", |
||
1731 | "type": "intermediary", |
||
1732 | "vlan": 6 |
||
1733 | }, |
||
1734 | { |
||
1735 | "dpid": 1, |
||
1736 | "port": 9, |
||
1737 | "time": "t3", |
||
1738 | "type": "last", |
||
1739 | "vlan": 5 |
||
1740 | }, |
||
1741 | ] |
||
1742 | |||
1743 | 1 | run_bulk_sdntraces_mock.return_value = { |
|
1744 | "result": [trace_a, trace_z] |
||
1745 | } |
||
1746 | 1 | result = EVCDeploy.check_list_traces([evc]) |
|
1747 | |||
1748 | 1 | assert result[evc.id] is True |
|
1749 | |||
1750 | 1 | trace_z = [ |
|
1751 | { |
||
1752 | "dpid": 3, |
||
1753 | "port": 3, |
||
1754 | "time": "t1", |
||
1755 | "type": "starting", |
||
1756 | "vlan": 83 |
||
1757 | }, |
||
1758 | { |
||
1759 | "dpid": 2, |
||
1760 | "port": 11, |
||
1761 | "time": "t2", |
||
1762 | "type": "loop", |
||
1763 | "vlan": 6 |
||
1764 | }, |
||
1765 | ] |
||
1766 | |||
1767 | 1 | run_bulk_sdntraces_mock.return_value = { |
|
1768 | "result": [trace_a, trace_z] |
||
1769 | } |
||
1770 | 1 | result = EVCDeploy.check_list_traces([evc]) |
|
1771 | # type loop |
||
1772 | 1 | assert result[evc.id] is False |
|
1773 | |||
1774 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVCDeploy.check_trace") |
|
1775 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVCDeploy.check_range") |
|
1776 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVCDeploy.run_bulk_sdntraces") |
|
1777 | 1 | def test_check_list_traces_vlan_list(self, *args): |
|
1778 | """Test check_list_traces with vlan list""" |
||
1779 | 1 | mock_bulk, mock_range, mock_trace = args |
|
1780 | 1 | mask_list = [1, '2/4094', '4/4094'] |
|
1781 | 1 | evc = self.create_evc_inter_switch([[1, 5]], [[1, 5]]) |
|
1782 | 1 | evc.uni_a.user_tag.mask_list = mask_list |
|
1783 | 1 | evc.uni_z.user_tag.mask_list = mask_list |
|
1784 | 1 | mock_bulk.return_value = {"result": ["mock"] * 6} |
|
1785 | 1 | mock_range.return_value = True |
|
1786 | 1 | actual_return = EVC.check_list_traces([evc]) |
|
1787 | 1 | assert actual_return == {evc._id: True} |
|
1788 | 1 | assert mock_trace.call_count == 0 |
|
1789 | 1 | assert mock_range.call_count == 1 |
|
1790 | 1 | args = mock_range.call_args[0] |
|
1791 | 1 | assert args[0] == evc |
|
1792 | 1 | assert args[1] == ["mock"] * 6 |
|
1793 | |||
1794 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVCDeploy.check_trace") |
|
1795 | 1 | @patch("napps.kytos.mef_eline.models.evc.log") |
|
1796 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVCDeploy.run_bulk_sdntraces") |
|
1797 | 1 | def test_check_list_traces_empty(self, mock_bulk, mock_log, mock_trace): |
|
1798 | """Test check_list_traces with empty return""" |
||
1799 | 1 | evc = self.create_evc_inter_switch(1, 1) |
|
1800 | 1 | actual_return = EVC.check_list_traces([]) |
|
1801 | 1 | assert not actual_return |
|
1802 | |||
1803 | 1 | mock_bulk.return_value = {"result": []} |
|
1804 | 1 | actual_return = EVC.check_list_traces([evc]) |
|
1805 | 1 | assert not actual_return |
|
1806 | |||
1807 | 1 | mock_bulk.return_value = {"result": ["mock"]} |
|
1808 | 1 | mock_trace.return_value = True |
|
1809 | 1 | actual_return = EVC.check_list_traces([evc]) |
|
1810 | 1 | assert mock_log.error.call_count == 1 |
|
1811 | 1 | assert not actual_return |
|
1812 | |||
1813 | 1 | @patch( |
|
1814 | "napps.kytos.mef_eline.models.path.DynamicPathManager" |
||
1815 | ".get_disjoint_paths" |
||
1816 | ) |
||
1817 | 1 | def test_get_failover_path_vandidates(self, get_disjoint_paths_mock): |
|
1818 | """Test get_failover_path_candidates method""" |
||
1819 | 1 | self.evc_deploy.get_failover_path_candidates() |
|
1820 | 1 | get_disjoint_paths_mock.assert_called_once() |
|
1821 | |||
1822 | 1 | def test_is_failover_path_affected_by_link(self): |
|
1823 | """Test is_failover_path_affected_by_link method""" |
||
1824 | 1 | link1 = get_link_mocked(endpoint_a_port=1, endpoint_b_port=2) |
|
1825 | 1 | link2 = get_link_mocked(endpoint_a_port=3, endpoint_b_port=4) |
|
1826 | 1 | link3 = get_link_mocked(endpoint_a_port=5, endpoint_b_port=6) |
|
1827 | 1 | self.evc_deploy.failover_path = Path([link1, link2]) |
|
1828 | 1 | assert self.evc_deploy.is_failover_path_affected_by_link(link1) is True |
|
1829 | 1 | assert self.evc_deploy.is_failover_path_affected_by_link(link3) \ |
|
1830 | is False |
||
1831 | |||
1832 | 1 | def test_is_eligible_for_failover_path(self): |
|
1833 | """Test is_eligible_for_failover_path method""" |
||
1834 | 1 | assert self.evc_deploy.is_eligible_for_failover_path() is False |
|
1835 | 1 | self.evc_deploy.dynamic_backup_path = True |
|
1836 | 1 | self.evc_deploy.primary_path = Path([]) |
|
1837 | 1 | self.evc_deploy.backup_path = Path([]) |
|
1838 | 1 | assert self.evc_deploy.is_eligible_for_failover_path() is True |
|
1839 | |||
1840 | 1 | def test_get_value_from_uni_tag(self): |
|
1841 | """Test _get_value_from_uni_tag""" |
||
1842 | 1 | uni = get_uni_mocked(tag_value="any") |
|
1843 | 1 | value = EVC._get_value_from_uni_tag(uni) |
|
1844 | 1 | assert value == "4096/4096" |
|
1845 | |||
1846 | 1 | uni.user_tag.value = "untagged" |
|
1847 | 1 | value = EVC._get_value_from_uni_tag(uni) |
|
1848 | 1 | assert value == 0 |
|
1849 | |||
1850 | 1 | uni.user_tag.value = 100 |
|
1851 | 1 | value = EVC._get_value_from_uni_tag(uni) |
|
1852 | 1 | assert value == 100 |
|
1853 | |||
1854 | 1 | uni.user_tag = None |
|
1855 | 1 | value = EVC._get_value_from_uni_tag(uni) |
|
1856 | 1 | assert value is None |
|
1857 | |||
1858 | 1 | uni = get_uni_mocked(tag_value=[[12, 20]]) |
|
1859 | 1 | uni.user_tag.mask_list = ['12/4092', '16/4092', '20/4094'] |
|
1860 | |||
1861 | 1 | value = EVC._get_value_from_uni_tag(uni) |
|
1862 | 1 | assert value == ['12/4092', '16/4092', '20/4094'] |
|
1863 | |||
1864 | 1 | def test_get_priority(self): |
|
1865 | """Test get_priority_from_vlan""" |
||
1866 | 1 | evpl_value = EVC.get_priority(100) |
|
1867 | 1 | assert evpl_value == EVPL_SB_PRIORITY |
|
1868 | |||
1869 | 1 | untagged_value = EVC.get_priority(0) |
|
1870 | 1 | assert untagged_value == UNTAGGED_SB_PRIORITY |
|
1871 | |||
1872 | 1 | any_value = EVC.get_priority("4096/4096") |
|
1873 | 1 | assert any_value == ANY_SB_PRIORITY |
|
1874 | |||
1875 | 1 | epl_value = EVC.get_priority(None) |
|
1876 | 1 | assert epl_value == EPL_SB_PRIORITY |
|
1877 | |||
1878 | 1 | epl_value = EVC.get_priority([[1, 5]]) |
|
1879 | 1 | assert epl_value == EVPL_SB_PRIORITY |
|
1880 | |||
1881 | 1 | def test_set_flow_table_group_id(self): |
|
1882 | """Test set_flow_table_group_id""" |
||
1883 | 1 | self.evc_deploy.table_group = {"epl": 3, "evpl": 4} |
|
1884 | 1 | flow_mod = {} |
|
1885 | 1 | self.evc_deploy.set_flow_table_group_id(flow_mod, 100) |
|
1886 | 1 | assert flow_mod["table_group"] == "evpl" |
|
1887 | 1 | assert flow_mod["table_id"] == 4 |
|
1888 | 1 | self.evc_deploy.set_flow_table_group_id(flow_mod, None) |
|
1889 | 1 | assert flow_mod["table_group"] == "epl" |
|
1890 | 1 | assert flow_mod["table_id"] == 3 |
|
1891 | |||
1892 | 1 | def test_get_endpoint_by_id(self): |
|
1893 | """Test get_endpoint_by_id""" |
||
1894 | 1 | link = MagicMock() |
|
1895 | 1 | link.endpoint_a.switch.id = "01" |
|
1896 | 1 | link.endpoint_b.switch.id = "02" |
|
1897 | 1 | result = self.evc_deploy.get_endpoint_by_id(link, "01", operator.eq) |
|
1898 | 1 | assert result == link.endpoint_a |
|
1899 | 1 | result = self.evc_deploy.get_endpoint_by_id(link, "01", operator.ne) |
|
1900 | 1 | assert result == link.endpoint_b |
|
1901 | |||
1902 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVC._prepare_pop_flow") |
|
1903 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVC.get_endpoint_by_id") |
|
1904 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVC._prepare_push_flow") |
|
1905 | 1 | def test_prepare_uni_flows(self, mock_push, mock_endpoint, _): |
|
1906 | """Test _prepare_uni_flows""" |
||
1907 | 1 | mask_list = [1, '2/4094', '4/4094'] |
|
1908 | 1 | uni_a = get_uni_mocked(interface_port=1, tag_value=[[1, 5]]) |
|
1909 | 1 | uni_a.user_tag.mask_list = mask_list |
|
1910 | 1 | uni_z = get_uni_mocked(interface_port=2, tag_value=[[1, 5]]) |
|
1911 | 1 | uni_z.user_tag.mask_list = mask_list |
|
1912 | 1 | mock_endpoint.return_value = "mock_endpoint" |
|
1913 | 1 | attributes = { |
|
1914 | "table_group": {"evpl": 3, "epl": 4}, |
||
1915 | "controller": get_controller_mock(), |
||
1916 | "name": "custom_name", |
||
1917 | "uni_a": uni_a, |
||
1918 | "uni_z": uni_z, |
||
1919 | } |
||
1920 | 1 | evc = EVC(**attributes) |
|
1921 | 1 | link = get_link_mocked() |
|
1922 | 1 | evc._prepare_uni_flows(Path([link])) |
|
1923 | 1 | call_list = [] |
|
1924 | 1 | for i in range(0, 3): |
|
1925 | 1 | call_list.append(call( |
|
1926 | uni_a.interface, |
||
1927 | "mock_endpoint", |
||
1928 | mask_list[i], |
||
1929 | None, |
||
1930 | mask_list, |
||
1931 | queue_id=-1 |
||
1932 | )) |
||
1933 | 1 | for i in range(0, 3): |
|
1934 | 1 | call_list.append(call( |
|
1935 | uni_z.interface, |
||
1936 | "mock_endpoint", |
||
1937 | mask_list[i], |
||
1938 | None, |
||
1939 | mask_list, |
||
1940 | queue_id=-1 |
||
1941 | )) |
||
1942 | 1 | mock_push.assert_has_calls(call_list) |
|
1943 | |||
1944 | 1 | def test_prepare_direct_uni_flows(self): |
|
1945 | """Test _prepare_direct_uni_flows""" |
||
1946 | 1 | mask_list = [1, '2/4094', '4/4094'] |
|
1947 | 1 | uni_a = get_uni_mocked(interface_port=1, tag_value=[[1, 5]]) |
|
1948 | 1 | uni_a.user_tag.mask_list = mask_list |
|
1949 | 1 | uni_z = get_uni_mocked(interface_port=2, tag_value=[[1, 5]]) |
|
1950 | 1 | uni_z.user_tag.mask_list = mask_list |
|
1951 | 1 | attributes = { |
|
1952 | "table_group": {"evpl": 3, "epl": 4}, |
||
1953 | "controller": get_controller_mock(), |
||
1954 | "name": "custom_name", |
||
1955 | "uni_a": uni_a, |
||
1956 | "uni_z": uni_z, |
||
1957 | } |
||
1958 | 1 | evc = EVC(**attributes) |
|
1959 | 1 | flows = evc._prepare_direct_uni_flows()[1] |
|
1960 | 1 | assert len(flows) == 6 |
|
1961 | 1 | for i in range(0, 3): |
|
1962 | 1 | assert flows[i]["match"]["in_port"] == 1 |
|
1963 | 1 | assert flows[i]["match"]["dl_vlan"] == mask_list[i] |
|
1964 | 1 | assert flows[i]["priority"] == EVPL_SB_PRIORITY |
|
1965 | 1 | for i in range(3, 6): |
|
1966 | 1 | assert flows[i]["match"]["in_port"] == 2 |
|
1967 | 1 | assert flows[i]["match"]["dl_vlan"] == mask_list[i-3] |
|
1968 | 1 | assert flows[i]["priority"] == EVPL_SB_PRIORITY |
|
1969 | |||
1970 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVCDeploy.check_trace") |
|
1971 | 1 | def test_check_range(self, mock_check_range): |
|
1972 | """Test check_range""" |
||
1973 | 1 | mask_list = [1, '2/4094', '4/4094'] |
|
1974 | 1 | uni_a = get_uni_mocked(interface_port=1, tag_value=[[1, 5]]) |
|
1975 | 1 | uni_a.user_tag.mask_list = mask_list |
|
1976 | 1 | uni_z = get_uni_mocked(interface_port=2, tag_value=[[1, 5]]) |
|
1977 | 1 | uni_z.user_tag.mask_list = mask_list |
|
1978 | 1 | attributes = { |
|
1979 | "table_group": {"evpl": 3, "epl": 4}, |
||
1980 | "controller": get_controller_mock(), |
||
1981 | "name": "custom_name", |
||
1982 | "uni_a": uni_a, |
||
1983 | "uni_z": uni_z, |
||
1984 | } |
||
1985 | 1 | circuit = EVC(**attributes) |
|
1986 | 1 | traces = list(range(0, 6)) |
|
1987 | 1 | mock_check_range.return_value = True |
|
1988 | 1 | check = EVC.check_range(circuit, traces) |
|
1989 | 1 | call_list = [] |
|
1990 | 1 | for i in range(0, 3): |
|
1991 | 1 | call_list.append(call( |
|
1992 | circuit.id, circuit.name, |
||
1993 | mask_list[i], mask_list[i], |
||
1994 | uni_a.interface, |
||
1995 | uni_z.interface, |
||
1996 | circuit.current_path, |
||
1997 | i*2, i*2+1 |
||
1998 | )) |
||
1999 | 1 | mock_check_range.assert_has_calls(call_list) |
|
2000 | 1 | assert check |
|
2001 | |||
2002 | 1 | mock_check_range.side_effect = [True, False, True] |
|
2003 | 1 | check = EVC.check_range(circuit, traces) |
|
2004 | 1 | assert check is False |
|
2005 | |||
2006 | 1 | def test_add_tag_errors(self): |
|
2007 | """Test add_tag_errors""" |
||
2008 | 1 | msg = "No available path was found." |
|
2009 | 1 | tag_errors = [] |
|
2010 | 1 | tag_errors.append('Mocked error 1') |
|
2011 | 1 | actual = self.evc_deploy.add_tag_errors(msg, tag_errors) |
|
2012 | 1 | assert actual == ('No available path was found. 1 path was rejected' |
|
2013 | f' with message: {tag_errors}') |
||
2014 | |||
2015 | 1 | tag_errors.append('Mocked error 2') |
|
2016 | 1 | actual = self.evc_deploy.add_tag_errors(msg, tag_errors) |
|
2017 | 1 | assert actual == ('No available path was found. 2 paths were' |
|
2018 | f' rejected with messages: {tag_errors}') |
||
2019 | |||
2020 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVCDeploy.setup_failover_path") |
|
2021 | 1 | def test_try_setup_failover_path(self, setup_failover_mock): |
|
2022 | """Test try_setup_failover_path""" |
||
2023 | 1 | self.evc_deploy.failover_path = True |
|
2024 | 1 | self.evc_deploy.current_path = False |
|
2025 | 1 | self.evc_deploy.is_active = MagicMock(return_value=False) |
|
2026 | 1 | self.evc_deploy.try_setup_failover_path() |
|
2027 | 1 | assert setup_failover_mock.call_count == 0 |
|
2028 | |||
2029 | 1 | self.evc_deploy.failover_path = False |
|
2030 | 1 | self.evc_deploy.current_path = True |
|
2031 | 1 | self.evc_deploy.is_active = MagicMock(return_value=True) |
|
2032 | 1 | self.evc_deploy.try_setup_failover_path() |
|
2033 | 1 | assert setup_failover_mock.call_count == 1 |
|
2034 | |||
2035 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVCDeploy._send_flow_mods") |
|
2036 | 1 | @patch( |
|
2037 | "napps.kytos.mef_eline.models.evc.EVCDeploy._prepare_direct_uni_flows" |
||
2038 | ) |
||
2039 | 1 | def test_install_direct_uni_flows_error(self, prepare_mock, send_mock): |
|
2040 | """Test _install_direct_uni_flows with errors""" |
||
2041 | 1 | prepare_mock.return_value = True, True |
|
2042 | 1 | send_mock.side_effect = FlowModException |
|
2043 | 1 | with pytest.raises(EVCPathNotInstalled): |
|
2044 | 1 | self.evc_deploy._install_direct_uni_flows() |
|
2045 | |||
2046 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVCDeploy._send_flow_mods") |
|
2047 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVCDeploy._prepare_nni_flows") |
|
2048 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVCDeploy._prepare_uni_flows") |
|
2049 | 1 | def test_install_flows_error( |
|
2050 | self, prepare_uni_mock, prepare_nni_mock, send_flow_mock |
||
2051 | ): |
||
2052 | """Test _install_flows with error""" |
||
2053 | 1 | prepare_nni_mock.return_value = {'1': [1]} |
|
2054 | 1 | prepare_uni_mock.return_value = {'2': [2]} |
|
2055 | 1 | send_flow_mock.side_effect = FlowModException('err') |
|
2056 | 1 | with pytest.raises(EVCPathNotInstalled): |
|
2057 | 1 | self.evc_deploy._install_flows() |
|
2058 | |||
2059 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVCDeploy._send_flow_mods") |
|
2060 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVCDeploy._prepare_nni_flows") |
|
2061 | 1 | @patch("napps.kytos.mef_eline.models.evc.EVCDeploy._prepare_uni_flows") |
|
2062 | 1 | def test_install_flows( |
|
2063 | self, prepare_uni_mock, prepare_nni_mock, send_flow_mock |
||
2064 | ): |
||
2065 | """Test that the dictionary contains uni and nni flows sent |
||
2066 | to be installed from _install_uni_flows.""" |
||
2067 | 1 | prepare_nni_mock.return_value = {'00:01': ["flow1"]} |
|
2068 | 1 | prepare_uni_mock.return_value = { |
|
2069 | '00:02': ["flow1"], '00:01': ["flow2"] |
||
2070 | } |
||
2071 | 1 | out_flows = self.evc_deploy._install_flows() |
|
2072 | 1 | expected_out_fows = { |
|
2073 | '00:01': ["flow1", "flow2"], |
||
2074 | '00:02': ["flow1"] |
||
2075 | } |
||
2076 | 1 | assert out_flows == expected_out_fows |
|
2077 | |||
2078 | 1 | expected_installed_flows = { |
|
2079 | '00:01': {"flows": ["flow1", "flow2"]}, |
||
2080 | '00:02': {"flows": ["flow1"]} |
||
2081 | } |
||
2082 | 1 | send_flow_mock.assert_called_with( |
|
2083 | expected_installed_flows, "install", by_switch=True |
||
2084 | ) |
||
2085 |