1
|
|
|
"""Module to test the KytosGraph in graph.py.""" |
2
|
1 |
|
import pytest |
3
|
1 |
|
from itertools import combinations |
4
|
|
|
|
5
|
|
|
# pylint: disable=import-error |
6
|
1 |
|
from tests.integration.edges_settings import EdgesSettings |
7
|
|
|
|
8
|
|
|
|
9
|
1 |
|
class TestPathsEdges(EdgesSettings): |
10
|
|
|
"""TestPathsEdges.""" |
11
|
|
|
|
12
|
1 |
|
@pytest.mark.parametrize("source,destination", |
13
|
|
|
combinations(["User1", "User2", "User3", "User4"], 2)) |
14
|
1 |
|
def test_k_shortest_paths_among_users(self, source, destination): |
15
|
|
|
"""Tests paths between all users using unconstrained path algorithm.""" |
16
|
1 |
|
self.initializer() |
17
|
1 |
|
paths = self.graph.k_shortest_paths(source, destination) |
18
|
1 |
|
assert paths |
19
|
1 |
|
for path in paths: |
20
|
1 |
|
assert path[0] == source |
21
|
1 |
|
assert path[-1] == destination |
22
|
|
|
|
23
|
1 |
|
@pytest.mark.parametrize("source,destination", |
24
|
|
|
combinations(["User1", "User2", "User3", "User4"], 2)) |
25
|
1 |
|
def test_constrained_k_shortest_paths_among_users(self, source, destination): |
26
|
|
|
"""Tests paths between all users using constrained path algorithm, |
27
|
|
|
with no constraints set. |
28
|
|
|
""" |
29
|
1 |
|
self.initializer() |
30
|
1 |
|
paths = self.graph.constrained_k_shortest_paths( |
31
|
|
|
source, destination |
32
|
|
|
) |
33
|
1 |
|
assert paths |
34
|
1 |
|
for path in paths: |
35
|
1 |
|
assert path["hops"][0] == source |
36
|
1 |
|
assert path["hops"][-1] == destination |
37
|
|
|
|
38
|
1 |
|
def test_cspf_delay_spf_attribute_between_u1_u4(self): |
39
|
|
|
"""Test CSPF delay spf attribute between user1 and user4.""" |
40
|
1 |
|
self.initializer() |
41
|
1 |
|
source = "User1" |
42
|
1 |
|
destination = "User4" |
43
|
1 |
|
spf_attribute = "delay" |
44
|
1 |
|
paths = self.graph.constrained_k_shortest_paths( |
45
|
|
|
source, destination, weight=spf_attribute |
46
|
|
|
) |
47
|
1 |
|
assert paths |
48
|
1 |
|
for path in paths: |
49
|
1 |
|
assert path["hops"][0] == source |
50
|
1 |
|
assert path["hops"][-1] == destination |
51
|
1 |
|
paths = self.graph.path_cost_builder(paths, weight=spf_attribute) |
52
|
1 |
|
assert paths[0]["cost"] == 105 + 1 + 1 |
53
|
|
|
|
54
|
1 |
View Code Duplication |
def test_cspf_reliability_between_u1_u2(self): |
|
|
|
|
55
|
|
|
"""Test CSPF reliability constraint between user1 and user2.""" |
56
|
1 |
|
self.initializer() |
57
|
1 |
|
source = "User1" |
58
|
1 |
|
destination = "User2" |
59
|
1 |
|
paths = self.graph.constrained_k_shortest_paths( |
60
|
|
|
source, destination, mandatory_metrics={"reliability": 10} |
61
|
|
|
) |
62
|
1 |
|
assert not paths |
63
|
|
|
|
64
|
1 |
|
paths = self.graph.constrained_k_shortest_paths( |
65
|
|
|
source, destination, mandatory_metrics={"reliability": 3} |
66
|
|
|
) |
67
|
1 |
|
assert paths |
68
|
|
|
|
69
|
1 |
|
for path in paths: |
70
|
1 |
|
assert path["hops"][0] == source |
71
|
1 |
|
assert path["hops"][-1] == destination |
72
|
1 |
|
assert path["metrics"] == {"reliability": 3} |
73
|
1 |
|
paths = self.graph.path_cost_builder(paths) |
74
|
1 |
|
assert paths[0]["cost"] == 12 |
75
|
|
|
|
76
|
1 |
|
def test_cspf_bandwidth_between_u1_u4(self): |
77
|
|
|
"""Test CSPF bandwidth constraint between user1 and user4.""" |
78
|
1 |
|
self.initializer() |
79
|
1 |
|
source = "User1" |
80
|
1 |
|
destination = "User4" |
81
|
1 |
|
spf_attribute = "delay" |
82
|
1 |
|
paths = self.graph.constrained_k_shortest_paths( |
83
|
|
|
source, |
84
|
|
|
destination, |
85
|
|
|
weight=spf_attribute, |
86
|
|
|
mandatory_metrics={"bandwidth": 200}, |
87
|
|
|
) |
88
|
1 |
|
assert not paths |
89
|
|
|
|
90
|
1 |
|
paths = self.graph.constrained_k_shortest_paths( |
91
|
|
|
source, |
92
|
|
|
destination, |
93
|
|
|
weight=spf_attribute, |
94
|
|
|
mandatory_metrics={"bandwidth": 100}, |
95
|
|
|
) |
96
|
1 |
|
assert paths |
97
|
|
|
|
98
|
1 |
|
for path in paths: |
99
|
1 |
|
assert path["hops"][0] == source |
100
|
1 |
|
assert path["hops"][-1] == destination |
101
|
1 |
|
assert path["metrics"] == {"bandwidth": 100} |
102
|
1 |
|
paths = self.graph.path_cost_builder(paths, weight=spf_attribute) |
103
|
1 |
|
assert paths[0]["cost"] == 122 |
104
|
|
|
|
105
|
1 |
View Code Duplication |
def test_cspf_delay_between_u2_u3(self): |
|
|
|
|
106
|
|
|
"""Test CSPF delay constraint between user2 and user3.""" |
107
|
1 |
|
self.initializer() |
108
|
1 |
|
source = "User2" |
109
|
1 |
|
destination = "User3" |
110
|
|
|
|
111
|
1 |
|
paths = self.graph.constrained_k_shortest_paths( |
112
|
|
|
source, destination, mandatory_metrics={"delay": 1} |
113
|
|
|
) |
114
|
1 |
|
assert not paths |
115
|
|
|
|
116
|
1 |
|
paths = self.graph.constrained_k_shortest_paths( |
117
|
|
|
source, destination, mandatory_metrics={"delay": 50} |
118
|
|
|
) |
119
|
1 |
|
assert paths |
120
|
1 |
|
for path in paths: |
121
|
1 |
|
assert path["hops"][0] == source |
122
|
1 |
|
assert path["hops"][-1] == destination |
123
|
1 |
|
paths = self.graph.path_cost_builder(paths) |
124
|
1 |
|
assert paths[0]["cost"] >= 3 |
125
|
|
|
|
126
|
1 |
View Code Duplication |
def test_cspf_ownership_between_s4_s6(self): |
|
|
|
|
127
|
|
|
"""Test CSPF ownership constraint between switch4 and switch6.""" |
128
|
1 |
|
self.initializer() |
129
|
1 |
|
source = "S6:2" |
130
|
1 |
|
destination = "S4:2" |
131
|
|
|
|
132
|
1 |
|
paths = self.graph.constrained_k_shortest_paths( |
133
|
|
|
source, destination, mandatory_metrics={"ownership": "B"} |
134
|
|
|
) |
135
|
1 |
|
assert not paths |
136
|
|
|
|
137
|
1 |
|
paths = self.graph.constrained_k_shortest_paths( |
138
|
|
|
source, destination, mandatory_metrics={"ownership": "A"} |
139
|
|
|
) |
140
|
1 |
|
assert paths |
141
|
1 |
|
for path in paths: |
142
|
1 |
|
assert path["hops"][0] == source |
143
|
1 |
|
assert path["hops"][-1] == destination |
144
|
1 |
|
paths = self.graph.path_cost_builder(paths) |
145
|
1 |
|
assert paths[0]["cost"] >= 3 |
146
|
|
|
|
147
|
1 |
|
def test_cspf_flexible_between_s4_s6(self): |
148
|
|
|
"""Test CSPF flexible constraint between switch4 and switch6.""" |
149
|
1 |
|
self.initializer() |
150
|
1 |
|
source = "S6:2" |
151
|
1 |
|
destination = "S4:2" |
152
|
|
|
|
153
|
1 |
|
paths = self.graph.constrained_k_shortest_paths( |
154
|
|
|
source, |
155
|
|
|
destination, |
156
|
|
|
mandatory_metrics={"reliability": 2}, |
157
|
|
|
flexible_metrics={"bandwidth": 60}, |
158
|
|
|
minimium_hits=1, |
159
|
|
|
) |
160
|
1 |
|
assert paths |
161
|
1 |
|
for path in paths: |
162
|
1 |
|
assert path["hops"][0] == source |
163
|
1 |
|
assert path["hops"][-1] == destination |
164
|
1 |
|
paths = self.graph.path_cost_builder(paths) |
165
|
1 |
|
assert paths[0]["cost"] >= 3 |
166
|
|
|
|
167
|
1 |
|
def test_cspf_paths_mandatory_with_flexible(self): |
168
|
|
|
"""Tests paths between all users using constrained path algorithm, |
169
|
|
|
with the delay constraint set to 50, the bandwidth constraint |
170
|
|
|
set to 100, the reliability constraint set to 3, and the ownership |
171
|
|
|
constraint set to 'B' |
172
|
|
|
|
173
|
|
|
Tests conducted with all but ownership flexible |
174
|
|
|
""" |
175
|
1 |
|
combos = combinations(["User1", "User2", "User3", "User4"], 2) |
176
|
1 |
|
self.initializer() |
177
|
|
|
|
178
|
1 |
|
for source, destination in combos: |
179
|
1 |
|
paths = self.graph.constrained_k_shortest_paths( |
180
|
|
|
source, |
181
|
|
|
destination, |
182
|
|
|
mandatory_metrics={"ownership": "B"}, |
183
|
|
|
flexible_metrics={ |
184
|
|
|
"delay": 50, |
185
|
|
|
"bandwidth": 100, |
186
|
|
|
"reliability": 3, |
187
|
|
|
}, |
188
|
|
|
) |
189
|
1 |
|
for path in paths: |
190
|
1 |
|
hops_set = set(path["hops"]) |
191
|
|
|
|
192
|
|
|
# delay = 50 checks |
193
|
1 |
|
if "delay" in path["metrics"]: |
194
|
1 |
|
nodes = set([ |
195
|
|
|
"S1:1", "S2:1", "S3:1", "S5:1", "S4:2", "User1:2", |
196
|
|
|
"S5:5", "S8:2", "S5:6", "User1:3", "S6:3", "S9:1", |
197
|
|
|
"S6:4", "S9:2", "S6:5", "S10:1", "S8:5", "S9:4", |
198
|
|
|
"User1:4", "User4:3" |
199
|
|
|
]) |
200
|
1 |
|
assert not nodes & hops_set |
201
|
|
|
|
202
|
|
|
# bandwidth = 100 checks |
203
|
1 |
|
if "bandwidth" in path["metrics"]: |
204
|
1 |
|
nodes = set(["S3:1", "S5:1", "User1:4", "User4:3"]) |
205
|
1 |
|
assert not nodes & hops_set |
206
|
|
|
|
207
|
|
|
# reliability = 3 checks |
208
|
1 |
|
if "reliability" in path["metrics"]: |
209
|
1 |
|
nodes = set(["S4:1", "S5:2", "S5:3", "S6:1"]) |
210
|
1 |
|
assert not nodes & hops_set |
211
|
|
|
|
212
|
|
|
# ownership = "B" checks |
213
|
1 |
|
assert "ownership" in path["metrics"] |
214
|
1 |
|
nodes = set([ |
215
|
|
|
"S4:1", "S5:2", "User1:2", "S5:4", |
216
|
|
|
"S6:2", "S6:5", "S10:1", "S8:6", "S10:2", |
217
|
|
|
"S10:3", "User2:1" |
218
|
|
|
]) |
219
|
1 |
|
assert not nodes & hops_set |
220
|
|
|
|
221
|
1 |
|
def test_ownership_type_error(self): |
222
|
|
|
"""Tests that TypeError.""" |
223
|
1 |
|
self.initializer() |
224
|
|
|
|
225
|
1 |
|
with pytest.raises(TypeError): |
226
|
1 |
|
self.graph.constrained_k_shortest_paths( |
227
|
|
|
"User1", "User2", mandatory_metrics={"ownership": 1} |
228
|
|
|
) |
229
|
|
|
|