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 |
|
# from collections import Counter |
17 |
1 |
from unittest import TestCase, mock |
18 |
|
|
19 |
|
# import osm_pla |
20 |
1 |
from osm_pla.placement.mznplacement import MznPlacementConductor, MznModelGenerator |
21 |
|
|
22 |
1 |
test_mzn_model = """ |
23 |
|
% This minizinc model is generated using |
24 |
|
% C:/Users/LG/PycharmProjects/dynamic_jijna2_mzn/osm_pla/placement/mznplacement.py |
25 |
|
% at 2019-10-24 11:12:02.058905. |
26 |
|
|
27 |
|
%This is the NETWORK RESOURCE MODEL |
28 |
|
enum Vims = { |
29 |
|
vimaaaaaaaa_38f5_438d_b8ee_3f93b3531f87, |
30 |
|
vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87, |
31 |
|
vimcccccccc_ed84_4e49_b5df_a9d117bd731f, |
32 |
|
vimdddddddd_ed84_4e49_b5df_a9d117bd731f, |
33 |
|
vimeeeeeeee_38f5_438d_b8ee_3f93b3531f87}; % The vim-accounts |
34 |
|
array[Vims, Vims] of int: trp_link_latency = [|0,50,100,150,200, |
35 |
|
|0,0,100,150,200, |
36 |
|
|0,0,0,150,200, |
37 |
|
|0,0,0,0,200, |
38 |
|
|0,0,0,0,0, |
39 |
|
|]; % Transport link latency between data centers |
40 |
|
array[Vims, Vims] of int: trp_link_jitter = [|0,50,100,150,200, |
41 |
|
|0,0,100,150,200, |
42 |
|
|0,0,0,150,200, |
43 |
|
|0,0,0,0,200, |
44 |
|
|0,0,0,0,0, |
45 |
|
|]; % Transport link jitter between data centers |
46 |
|
array[Vims, Vims] of int: trp_link_price_list = [|0,5,6,6,7, |
47 |
|
|0,0,6,6,7, |
48 |
|
|0,0,0,6,7, |
49 |
|
|0,0,0,0,7, |
50 |
|
|0,0,0,0,0, |
51 |
|
|]; % Transport link price list |
52 |
|
array[Vims] of int: vim_price_list_1 = [500,51,52,53,54]; |
53 |
|
array[Vims] of int: vim_price_list_2 = [20,21,22,23,24]; |
54 |
|
array[Vims] of int: vim_price_list_3 = [70,71,72,73,74]; |
55 |
|
array[Vims] of int: vim_price_list_4 = [40,41,42,43,44]; |
56 |
|
|
57 |
|
|
58 |
|
% This is the NETWORK BASIC LOAD MODEL (CONSUMED) |
59 |
|
% NOTE. This is not applicable in OSM Release 7 |
60 |
|
|
61 |
|
% This is the SERVICE CONSUMPTION MODEL |
62 |
|
% These are the variables, i.e. which DC to select for each VNF |
63 |
|
var Vims: VNF1; |
64 |
|
var Vims: VNF2; |
65 |
|
var Vims: VNF3; |
66 |
|
var Vims: VNF4; |
67 |
|
|
68 |
|
|
69 |
|
% These are the set of rules for selecting DCs to VNFs |
70 |
|
constraint trp_link_latency[VNF1, VNF2] <= 150; |
71 |
|
constraint trp_link_latency[VNF2, VNF3] <= 140; |
72 |
|
constraint trp_link_latency[VNF3, VNF4] <= 130; |
73 |
|
constraint trp_link_jitter[VNF1, VNF2] <= 30; |
74 |
|
constraint trp_link_jitter[VNF2, VNF3] <= 30; |
75 |
|
constraint trp_link_jitter[VNF3, VNF4] <= 30; |
76 |
|
|
77 |
|
% Calculate the cost for VNFs and cost for transport link and total cost |
78 |
|
var int: used_transport_cost =trp_link_price_list[VNF1, VNF2]+ |
79 |
|
trp_link_price_list[VNF2, VNF3]+ |
80 |
|
trp_link_price_list[VNF3, VNF4]; |
81 |
|
|
82 |
|
var int: used_vim_cost =vim_price_list_1[VNF1]+ |
83 |
|
vim_price_list_2[VNF2]+ |
84 |
|
vim_price_list_3[VNF3]+ |
85 |
|
vim_price_list_4[VNF4]; |
86 |
|
|
87 |
|
var int: total_cost = used_transport_cost + used_vim_cost; |
88 |
|
|
89 |
|
solve minimize total_cost; |
90 |
|
|
91 |
|
""" |
92 |
|
|
93 |
1 |
test_mzn_model_w_pinning = """ |
94 |
|
% This minizinc model is generated using |
95 |
|
% C:/Users/LG/PycharmProjects/dynamic_jijna2_mzn/osm_pla/placement/mznplacement.py |
96 |
|
% at 2019-10-24 11:12:02.058905. |
97 |
|
|
98 |
|
%This is the NETWORK RESOURCE MODEL |
99 |
|
enum Vims = { |
100 |
|
vimaaaaaaaa_38f5_438d_b8ee_3f93b3531f87, |
101 |
|
vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87, |
102 |
|
vimcccccccc_ed84_4e49_b5df_a9d117bd731f, |
103 |
|
vimdddddddd_ed84_4e49_b5df_a9d117bd731f, |
104 |
|
vimeeeeeeee_38f5_438d_b8ee_3f93b3531f87}; % The vim-accounts |
105 |
|
array[Vims, Vims] of int: trp_link_latency = [|0,50,100,150,200, |
106 |
|
|0,0,100,150,200, |
107 |
|
|0,0,0,150,200, |
108 |
|
|0,0,0,0,200, |
109 |
|
|0,0,0,0,0, |
110 |
|
|]; % Transport link latency between data centers |
111 |
|
array[Vims, Vims] of int: trp_link_jitter = [|0,50,100,150,200, |
112 |
|
|0,0,100,150,200, |
113 |
|
|0,0,0,150,200, |
114 |
|
|0,0,0,0,200, |
115 |
|
|0,0,0,0,0, |
116 |
|
|]; % Transport link jitter between data centers |
117 |
|
array[Vims, Vims] of int: trp_link_price_list = [|0,5,6,6,7, |
118 |
|
|0,0,6,6,7, |
119 |
|
|0,0,0,6,7, |
120 |
|
|0,0,0,0,7, |
121 |
|
|0,0,0,0,0, |
122 |
|
|]; % Transport link price list |
123 |
|
array[Vims] of int: vim_price_list_1 = [500,51,52,53,54]; |
124 |
|
array[Vims] of int: vim_price_list_2 = [20,21,22,23,24]; |
125 |
|
array[Vims] of int: vim_price_list_3 = [70,71,72,73,74]; |
126 |
|
array[Vims] of int: vim_price_list_4 = [40,41,42,43,44]; |
127 |
|
|
128 |
|
|
129 |
|
% This is the NETWORK BASIC LOAD MODEL (CONSUMED) |
130 |
|
% NOTE. This is not applicable in OSM Release 7 |
131 |
|
|
132 |
|
% This is the SERVICE CONSUMPTION MODEL |
133 |
|
% These are the variables, i.e. which DC to select for each VNF |
134 |
|
Vims: VNF1 = vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87; |
135 |
|
var Vims: VNF2; |
136 |
|
Vims: VNF3 = vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87; |
137 |
|
var Vims: VNF4; |
138 |
|
|
139 |
|
|
140 |
|
% These are the set of rules for selecting DCs to VNFs |
141 |
|
constraint trp_link_latency[VNF1, VNF2] <= 150; |
142 |
|
constraint trp_link_latency[VNF2, VNF3] <= 140; |
143 |
|
constraint trp_link_latency[VNF3, VNF4] <= 130; |
144 |
|
constraint trp_link_jitter[VNF1, VNF2] <= 30; |
145 |
|
constraint trp_link_jitter[VNF2, VNF3] <= 30; |
146 |
|
constraint trp_link_jitter[VNF3, VNF4] <= 30; |
147 |
|
|
148 |
|
% Calculate the cost for VNFs and cost for transport link and total cost |
149 |
|
var int: used_transport_cost =trp_link_price_list[VNF1, VNF2]+ |
150 |
|
trp_link_price_list[VNF2, VNF3]+ |
151 |
|
trp_link_price_list[VNF3, VNF4]; |
152 |
|
|
153 |
|
var int: used_vim_cost =vim_price_list_1[VNF1]+ |
154 |
|
vim_price_list_2[VNF2]+ |
155 |
|
vim_price_list_3[VNF3]+ |
156 |
|
vim_price_list_4[VNF4]; |
157 |
|
|
158 |
|
var int: total_cost = used_transport_cost + used_vim_cost; |
159 |
|
|
160 |
|
solve minimize total_cost; |
161 |
|
|
162 |
|
""" |
163 |
|
|
164 |
1 |
test_mzn_unsatisfiable_model = """ |
165 |
|
var 1..2: item1; |
166 |
|
var 1..2: item2; |
167 |
|
constraint item1 + item2 == 5; |
168 |
|
|
169 |
|
solve satisfy; |
170 |
|
""" |
171 |
|
|
172 |
|
|
173 |
1 |
class TestMznPlacementConductor(TestCase): |
174 |
1 |
def test__run_placement_model(self): |
175 |
1 |
expected_result = [{'vimAccountId': 'bbbbbbbb-38f5-438d-b8ee-3f93b3531f87', 'member-vnf-index': '1'}, |
176 |
|
{'vimAccountId': 'aaaaaaaa-38f5-438d-b8ee-3f93b3531f87', 'member-vnf-index': '2'}, |
177 |
|
{'vimAccountId': 'aaaaaaaa-38f5-438d-b8ee-3f93b3531f87', 'member-vnf-index': '3'}, |
178 |
|
{'vimAccountId': 'aaaaaaaa-38f5-438d-b8ee-3f93b3531f87', 'member-vnf-index': '4'}] |
179 |
|
|
180 |
1 |
mpc = MznPlacementConductor(logging.getLogger(__name__)) |
181 |
1 |
placement = mpc._run_placement_model(mzn_model=test_mzn_model, ns_desc={}) |
182 |
|
# sort the result to ease assert with expected result |
183 |
1 |
sorted_placement = sorted(placement, key=lambda k: k['member-vnf-index']) |
184 |
1 |
self.assertEqual(expected_result, sorted_placement, 'Faulty syntax or content') |
185 |
|
|
186 |
1 |
def test__run_placement_model_w_pinning(self): |
187 |
1 |
expected_result = [{'vimAccountId': 'bbbbbbbb-38f5-438d-b8ee-3f93b3531f87', 'member-vnf-index': '1'}, |
188 |
|
{'vimAccountId': 'bbbbbbbb-38f5-438d-b8ee-3f93b3531f87', 'member-vnf-index': '2'}, |
189 |
|
{'vimAccountId': 'bbbbbbbb-38f5-438d-b8ee-3f93b3531f87', 'member-vnf-index': '3'}, |
190 |
|
{'vimAccountId': 'aaaaaaaa-38f5-438d-b8ee-3f93b3531f87', 'member-vnf-index': '4'}] |
191 |
|
|
192 |
1 |
ns_desc = [{'vnf_price_per_vim': [10, 9, 7, 8], 'vnf_id': '2'}, |
193 |
|
{'vim_account': 'vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87', 'vnf_price_per_vim': [10, 9, 7, 8], |
194 |
|
'vnf_id': '1'}, |
195 |
|
{'vnf_price_per_vim': [10, 9, 7, 8], 'vnf_id': '4'}, |
196 |
|
{'vim_account': 'vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87', 'vnf_price_per_vim': [10, 9, 7, 8], |
197 |
|
'vnf_id': '3'} |
198 |
|
] |
199 |
|
|
200 |
1 |
mpc = MznPlacementConductor(logging.getLogger(__name__)) |
201 |
1 |
placement = mpc._run_placement_model(mzn_model=test_mzn_model_w_pinning, ns_desc=ns_desc) |
202 |
|
# sort the result to ease assert with expected result |
203 |
1 |
sorted_placement = sorted(placement, key=lambda k: k['member-vnf-index']) |
204 |
1 |
self.assertEqual(expected_result, sorted_placement, 'Faulty syntax or content') |
205 |
|
|
206 |
1 |
def test__run_placement_model_unsatisfiable(self): |
207 |
1 |
mpc = MznPlacementConductor(logging.getLogger(__name__)) |
208 |
1 |
self.assertEqual([{}], mpc._run_placement_model(mzn_model=test_mzn_unsatisfiable_model, ns_desc={}), |
209 |
|
"Faulty syntax or content for unsatisfiable model") |
210 |
|
|
211 |
1 |
@mock.patch.object(MznModelGenerator, 'create_model', side_effect=['%model']) |
212 |
1 |
@mock.patch.object(MznPlacementConductor, '_run_placement_model') |
213 |
|
def test_do_placement_computation(self, mock_run, mock_create): |
214 |
1 |
mpc = MznPlacementConductor(logging.getLogger(__name__)) |
215 |
1 |
dummy_nspd = {'ns_desc': {}} |
216 |
1 |
_ = mpc.do_placement_computation(dummy_nspd) |
217 |
1 |
mock_create.assert_called_with(dummy_nspd) |
218 |
1 |
mock_run.assert_called_with('%model', {}) |