osm_pla/test/test_mznPlacementConductor.py |
---|
1 |
|
# Copyright 2020 ArctosLabs Scandinavia AB |
2 |
|
# |
3 |
|
# Licensed under the Apache License, Version 2.0 (the "License"); |
4 |
|
# you may not use this file except in compliance with the License. |
5 |
|
# You may obtain a copy of the License at |
6 |
|
# |
7 |
|
# http://www.apache.org/licenses/LICENSE-2.0 |
8 |
|
# |
9 |
|
# Unless required by applicable law or agreed to in writing, software |
10 |
|
# distributed under the License is distributed on an "AS IS" BASIS, |
11 |
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or |
12 |
|
# implied. |
13 |
|
# See the License for the specific language governing permissions and |
14 |
|
# limitations under the License. |
15 |
1 |
import logging |
16 |
|
|
17 |
|
# from collections import Counter |
18 |
1 |
from unittest import TestCase, mock |
19 |
|
|
20 |
|
# import osm_pla |
21 |
1 |
from osm_pla.placement.mznplacement import MznPlacementConductor, MznModelGenerator |
22 |
|
|
23 |
1 |
test_mzn_model = """ |
24 |
|
% This minizinc model is generated using |
25 |
|
% C:/Users/LG/PycharmProjects/dynamic_jijna2_mzn/osm_pla/placement/mznplacement.py |
26 |
|
% at 2019-10-24 11:12:02.058905. |
27 |
|
|
28 |
|
%This is the NETWORK RESOURCE MODEL |
29 |
|
enum Vims = { |
30 |
|
vimaaaaaaaa_38f5_438d_b8ee_3f93b3531f87, |
31 |
|
vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87, |
32 |
|
vimcccccccc_ed84_4e49_b5df_a9d117bd731f, |
33 |
|
vimdddddddd_ed84_4e49_b5df_a9d117bd731f, |
34 |
|
vimeeeeeeee_38f5_438d_b8ee_3f93b3531f87}; % The vim-accounts |
35 |
|
array[Vims, Vims] of int: trp_link_latency = [|0,50,100,150,200, |
36 |
|
|0,0,100,150,200, |
37 |
|
|0,0,0,150,200, |
38 |
|
|0,0,0,0,200, |
39 |
|
|0,0,0,0,0, |
40 |
|
|]; % Transport link latency between data centers |
41 |
|
array[Vims, Vims] of int: trp_link_jitter = [|0,50,100,150,200, |
42 |
|
|0,0,100,150,200, |
43 |
|
|0,0,0,150,200, |
44 |
|
|0,0,0,0,200, |
45 |
|
|0,0,0,0,0, |
46 |
|
|]; % Transport link jitter between data centers |
47 |
|
array[Vims, Vims] of int: trp_link_price_list = [|0,5,6,6,7, |
48 |
|
|0,0,6,6,7, |
49 |
|
|0,0,0,6,7, |
50 |
|
|0,0,0,0,7, |
51 |
|
|0,0,0,0,0, |
52 |
|
|]; % Transport link price list |
53 |
|
array[Vims] of int: vim_price_list_1 = [500,51,52,53,54]; |
54 |
|
array[Vims] of int: vim_price_list_2 = [20,21,22,23,24]; |
55 |
|
array[Vims] of int: vim_price_list_3 = [70,71,72,73,74]; |
56 |
|
array[Vims] of int: vim_price_list_4 = [40,41,42,43,44]; |
57 |
|
|
58 |
|
|
59 |
|
% This is the NETWORK BASIC LOAD MODEL (CONSUMED) |
60 |
|
% NOTE. This is not applicable in OSM Release 7 |
61 |
|
|
62 |
|
% This is the SERVICE CONSUMPTION MODEL |
63 |
|
% These are the variables, i.e. which DC to select for each VNF |
64 |
|
var Vims: VNF1; |
65 |
|
var Vims: VNF2; |
66 |
|
var Vims: VNF3; |
67 |
|
var Vims: VNF4; |
68 |
|
|
69 |
|
|
70 |
|
% These are the set of rules for selecting DCs to VNFs |
71 |
|
constraint trp_link_latency[VNF1, VNF2] <= 150; |
72 |
|
constraint trp_link_latency[VNF2, VNF3] <= 140; |
73 |
|
constraint trp_link_latency[VNF3, VNF4] <= 130; |
74 |
|
constraint trp_link_jitter[VNF1, VNF2] <= 30; |
75 |
|
constraint trp_link_jitter[VNF2, VNF3] <= 30; |
76 |
|
constraint trp_link_jitter[VNF3, VNF4] <= 30; |
77 |
|
|
78 |
|
% Calculate the cost for VNFs and cost for transport link and total cost |
79 |
|
var int: used_transport_cost =trp_link_price_list[VNF1, VNF2]+ |
80 |
|
trp_link_price_list[VNF2, VNF3]+ |
81 |
|
trp_link_price_list[VNF3, VNF4]; |
82 |
|
|
83 |
|
var int: used_vim_cost =vim_price_list_1[VNF1]+ |
84 |
|
vim_price_list_2[VNF2]+ |
85 |
|
vim_price_list_3[VNF3]+ |
86 |
|
vim_price_list_4[VNF4]; |
87 |
|
|
88 |
|
var int: total_cost = used_transport_cost + used_vim_cost; |
89 |
|
|
90 |
|
solve minimize total_cost; |
91 |
|
|
92 |
|
""" |
93 |
|
|
94 |
1 |
test_mzn_model_w_pinning = """ |
95 |
|
% This minizinc model is generated using |
96 |
|
% C:/Users/LG/PycharmProjects/dynamic_jijna2_mzn/osm_pla/placement/mznplacement.py |
97 |
|
% at 2019-10-24 11:12:02.058905. |
98 |
|
|
99 |
|
%This is the NETWORK RESOURCE MODEL |
100 |
|
enum Vims = { |
101 |
|
vimaaaaaaaa_38f5_438d_b8ee_3f93b3531f87, |
102 |
|
vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87, |
103 |
|
vimcccccccc_ed84_4e49_b5df_a9d117bd731f, |
104 |
|
vimdddddddd_ed84_4e49_b5df_a9d117bd731f, |
105 |
|
vimeeeeeeee_38f5_438d_b8ee_3f93b3531f87}; % The vim-accounts |
106 |
|
array[Vims, Vims] of int: trp_link_latency = [|0,50,100,150,200, |
107 |
|
|0,0,100,150,200, |
108 |
|
|0,0,0,150,200, |
109 |
|
|0,0,0,0,200, |
110 |
|
|0,0,0,0,0, |
111 |
|
|]; % Transport link latency between data centers |
112 |
|
array[Vims, Vims] of int: trp_link_jitter = [|0,50,100,150,200, |
113 |
|
|0,0,100,150,200, |
114 |
|
|0,0,0,150,200, |
115 |
|
|0,0,0,0,200, |
116 |
|
|0,0,0,0,0, |
117 |
|
|]; % Transport link jitter between data centers |
118 |
|
array[Vims, Vims] of int: trp_link_price_list = [|0,5,6,6,7, |
119 |
|
|0,0,6,6,7, |
120 |
|
|0,0,0,6,7, |
121 |
|
|0,0,0,0,7, |
122 |
|
|0,0,0,0,0, |
123 |
|
|]; % Transport link price list |
124 |
|
array[Vims] of int: vim_price_list_1 = [500,51,52,53,54]; |
125 |
|
array[Vims] of int: vim_price_list_2 = [20,21,22,23,24]; |
126 |
|
array[Vims] of int: vim_price_list_3 = [70,71,72,73,74]; |
127 |
|
array[Vims] of int: vim_price_list_4 = [40,41,42,43,44]; |
128 |
|
|
129 |
|
|
130 |
|
% This is the NETWORK BASIC LOAD MODEL (CONSUMED) |
131 |
|
% NOTE. This is not applicable in OSM Release 7 |
132 |
|
|
133 |
|
% This is the SERVICE CONSUMPTION MODEL |
134 |
|
% These are the variables, i.e. which DC to select for each VNF |
135 |
|
Vims: VNF1 = vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87; |
136 |
|
var Vims: VNF2; |
137 |
|
Vims: VNF3 = vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87; |
138 |
|
var Vims: VNF4; |
139 |
|
|
140 |
|
|
141 |
|
% These are the set of rules for selecting DCs to VNFs |
142 |
|
constraint trp_link_latency[VNF1, VNF2] <= 150; |
143 |
|
constraint trp_link_latency[VNF2, VNF3] <= 140; |
144 |
|
constraint trp_link_latency[VNF3, VNF4] <= 130; |
145 |
|
constraint trp_link_jitter[VNF1, VNF2] <= 30; |
146 |
|
constraint trp_link_jitter[VNF2, VNF3] <= 30; |
147 |
|
constraint trp_link_jitter[VNF3, VNF4] <= 30; |
148 |
|
|
149 |
|
% Calculate the cost for VNFs and cost for transport link and total cost |
150 |
|
var int: used_transport_cost =trp_link_price_list[VNF1, VNF2]+ |
151 |
|
trp_link_price_list[VNF2, VNF3]+ |
152 |
|
trp_link_price_list[VNF3, VNF4]; |
153 |
|
|
154 |
|
var int: used_vim_cost =vim_price_list_1[VNF1]+ |
155 |
|
vim_price_list_2[VNF2]+ |
156 |
|
vim_price_list_3[VNF3]+ |
157 |
|
vim_price_list_4[VNF4]; |
158 |
|
|
159 |
|
var int: total_cost = used_transport_cost + used_vim_cost; |
160 |
|
|
161 |
|
solve minimize total_cost; |
162 |
|
|
163 |
|
""" |
164 |
|
|
165 |
1 |
test_mzn_unsatisfiable_model = """ |
166 |
|
var 1..2: item1; |
167 |
|
var 1..2: item2; |
168 |
|
constraint item1 + item2 == 5; |
169 |
|
|
170 |
|
solve satisfy; |
171 |
|
""" |
172 |
|
|
173 |
|
|
174 |
1 |
class TestMznPlacementConductor(TestCase): |
175 |
1 |
def test__run_placement_model(self): |
176 |
1 |
expected_result = [ |
177 |
|
{ |
178 |
|
"vimAccountId": "bbbbbbbb-38f5-438d-b8ee-3f93b3531f87", |
179 |
|
"member-vnf-index": "1", |
180 |
|
}, |
181 |
|
{ |
182 |
|
"vimAccountId": "aaaaaaaa-38f5-438d-b8ee-3f93b3531f87", |
183 |
|
"member-vnf-index": "2", |
184 |
|
}, |
185 |
|
{ |
186 |
|
"vimAccountId": "aaaaaaaa-38f5-438d-b8ee-3f93b3531f87", |
187 |
|
"member-vnf-index": "3", |
188 |
|
}, |
189 |
|
{ |
190 |
|
"vimAccountId": "aaaaaaaa-38f5-438d-b8ee-3f93b3531f87", |
191 |
|
"member-vnf-index": "4", |
192 |
|
}, |
193 |
|
] |
194 |
|
|
195 |
1 |
mpc = MznPlacementConductor(logging.getLogger(__name__)) |
196 |
1 |
placement = mpc._run_placement_model(mzn_model=test_mzn_model, ns_desc={}) |
197 |
|
# sort the result to ease assert with expected result |
198 |
1 |
sorted_placement = sorted(placement, key=lambda k: k["member-vnf-index"]) |
199 |
1 |
self.assertEqual(expected_result, sorted_placement, "Faulty syntax or content") |
200 |
|
|
201 |
1 |
def test__run_placement_model_w_pinning(self): |
202 |
1 |
expected_result = [ |
203 |
|
{ |
204 |
|
"vimAccountId": "bbbbbbbb-38f5-438d-b8ee-3f93b3531f87", |
205 |
|
"member-vnf-index": "1", |
206 |
|
}, |
207 |
|
{ |
208 |
|
"vimAccountId": "bbbbbbbb-38f5-438d-b8ee-3f93b3531f87", |
209 |
|
"member-vnf-index": "2", |
210 |
|
}, |
211 |
|
{ |
212 |
|
"vimAccountId": "bbbbbbbb-38f5-438d-b8ee-3f93b3531f87", |
213 |
|
"member-vnf-index": "3", |
214 |
|
}, |
215 |
|
{ |
216 |
|
"vimAccountId": "aaaaaaaa-38f5-438d-b8ee-3f93b3531f87", |
217 |
|
"member-vnf-index": "4", |
218 |
|
}, |
219 |
|
] |
220 |
|
|
221 |
1 |
ns_desc = [ |
222 |
|
{"vnf_price_per_vim": [10, 9, 7, 8], "vnf_id": "2"}, |
223 |
|
{ |
224 |
|
"vim_account": "vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87", |
225 |
|
"vnf_price_per_vim": [10, 9, 7, 8], |
226 |
|
"vnf_id": "1", |
227 |
|
}, |
228 |
|
{"vnf_price_per_vim": [10, 9, 7, 8], "vnf_id": "4"}, |
229 |
|
{ |
230 |
|
"vim_account": "vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87", |
231 |
|
"vnf_price_per_vim": [10, 9, 7, 8], |
232 |
|
"vnf_id": "3", |
233 |
|
}, |
234 |
|
] |
235 |
|
|
236 |
1 |
mpc = MznPlacementConductor(logging.getLogger(__name__)) |
237 |
1 |
placement = mpc._run_placement_model( |
238 |
|
mzn_model=test_mzn_model_w_pinning, ns_desc=ns_desc |
239 |
|
) |
240 |
|
# sort the result to ease assert with expected result |
241 |
1 |
sorted_placement = sorted(placement, key=lambda k: k["member-vnf-index"]) |
242 |
1 |
self.assertEqual(expected_result, sorted_placement, "Faulty syntax or content") |
243 |
|
|
244 |
1 |
def test__run_placement_model_unsatisfiable(self): |
245 |
1 |
mpc = MznPlacementConductor(logging.getLogger(__name__)) |
246 |
1 |
self.assertEqual( |
247 |
|
[{}], |
248 |
|
mpc._run_placement_model( |
249 |
|
mzn_model=test_mzn_unsatisfiable_model, ns_desc={} |
250 |
|
), |
251 |
|
"Faulty syntax or content for unsatisfiable model", |
252 |
|
) |
253 |
|
|
254 |
1 |
@mock.patch.object(MznModelGenerator, "create_model", side_effect=["%model"]) |
255 |
1 |
@mock.patch.object(MznPlacementConductor, "_run_placement_model") |
256 |
1 |
def test_do_placement_computation(self, mock_run, mock_create): |
257 |
1 |
mpc = MznPlacementConductor(logging.getLogger(__name__)) |
258 |
1 |
dummy_nspd = {"ns_desc": {}} |
259 |
1 |
_ = mpc.do_placement_computation(dummy_nspd) |
260 |
1 |
mock_create.assert_called_with(dummy_nspd) |
261 |
1 |
mock_run.assert_called_with("%model", {}) |