Coverage for osm_pla/test/test_mznPlacementConductor.py: 100%

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. 

15import logging 

16# from collections import Counter 

17from unittest import TestCase, mock 

18 

19# import osm_pla 

20from osm_pla.placement.mznplacement import MznPlacementConductor, MznModelGenerator 

21 

22test_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 

28enum Vims = { 

29vimaaaaaaaa_38f5_438d_b8ee_3f93b3531f87, 

30vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87, 

31vimcccccccc_ed84_4e49_b5df_a9d117bd731f, 

32vimdddddddd_ed84_4e49_b5df_a9d117bd731f, 

33vimeeeeeeee_38f5_438d_b8ee_3f93b3531f87}; % The vim-accounts 

34array[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 

40array[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 

46array[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 

52array[Vims] of int: vim_price_list_1 = [500,51,52,53,54]; 

53array[Vims] of int: vim_price_list_2 = [20,21,22,23,24]; 

54array[Vims] of int: vim_price_list_3 = [70,71,72,73,74]; 

55array[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 

63var Vims: VNF1; 

64var Vims: VNF2; 

65var Vims: VNF3; 

66var Vims: VNF4; 

67 

68 

69% These are the set of rules for selecting DCs to VNFs 

70constraint trp_link_latency[VNF1, VNF2] <= 150; 

71constraint trp_link_latency[VNF2, VNF3] <= 140; 

72constraint trp_link_latency[VNF3, VNF4] <= 130; 

73constraint trp_link_jitter[VNF1, VNF2] <= 30; 

74constraint trp_link_jitter[VNF2, VNF3] <= 30; 

75constraint trp_link_jitter[VNF3, VNF4] <= 30; 

76 

77% Calculate the cost for VNFs and cost for transport link and total cost 

78var int: used_transport_cost =trp_link_price_list[VNF1, VNF2]+ 

79trp_link_price_list[VNF2, VNF3]+ 

80trp_link_price_list[VNF3, VNF4]; 

81 

82var int: used_vim_cost =vim_price_list_1[VNF1]+ 

83vim_price_list_2[VNF2]+ 

84vim_price_list_3[VNF3]+ 

85vim_price_list_4[VNF4]; 

86 

87var int: total_cost = used_transport_cost + used_vim_cost; 

88 

89solve minimize total_cost; 

90 

91""" 

92 

93test_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 

99enum Vims = { 

100vimaaaaaaaa_38f5_438d_b8ee_3f93b3531f87, 

101vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87, 

102vimcccccccc_ed84_4e49_b5df_a9d117bd731f, 

103vimdddddddd_ed84_4e49_b5df_a9d117bd731f, 

104vimeeeeeeee_38f5_438d_b8ee_3f93b3531f87}; % The vim-accounts 

105array[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 

111array[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 

117array[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 

123array[Vims] of int: vim_price_list_1 = [500,51,52,53,54]; 

124array[Vims] of int: vim_price_list_2 = [20,21,22,23,24]; 

125array[Vims] of int: vim_price_list_3 = [70,71,72,73,74]; 

126array[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 

134Vims: VNF1 = vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87; 

135var Vims: VNF2; 

136Vims: VNF3 = vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87; 

137var Vims: VNF4; 

138 

139 

140% These are the set of rules for selecting DCs to VNFs 

141constraint trp_link_latency[VNF1, VNF2] <= 150; 

142constraint trp_link_latency[VNF2, VNF3] <= 140; 

143constraint trp_link_latency[VNF3, VNF4] <= 130; 

144constraint trp_link_jitter[VNF1, VNF2] <= 30; 

145constraint trp_link_jitter[VNF2, VNF3] <= 30; 

146constraint trp_link_jitter[VNF3, VNF4] <= 30; 

147 

148% Calculate the cost for VNFs and cost for transport link and total cost 

149var int: used_transport_cost =trp_link_price_list[VNF1, VNF2]+ 

150trp_link_price_list[VNF2, VNF3]+ 

151trp_link_price_list[VNF3, VNF4]; 

152 

153var int: used_vim_cost =vim_price_list_1[VNF1]+ 

154vim_price_list_2[VNF2]+ 

155vim_price_list_3[VNF3]+ 

156vim_price_list_4[VNF4]; 

157 

158var int: total_cost = used_transport_cost + used_vim_cost; 

159 

160solve minimize total_cost; 

161 

162""" 

163 

164test_mzn_unsatisfiable_model = """ 

165var 1..2: item1; 

166var 1..2: item2; 

167constraint item1 + item2 == 5; 

168 

169solve satisfy; 

170""" 

171 

172 

173class TestMznPlacementConductor(TestCase): 

174 def test__run_placement_model(self): 

175 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 mpc = MznPlacementConductor(logging.getLogger(__name__)) 

181 placement = mpc._run_placement_model(mzn_model=test_mzn_model, ns_desc={}) 

182 # sort the result to ease assert with expected result 

183 sorted_placement = sorted(placement, key=lambda k: k['member-vnf-index']) 

184 self.assertEqual(expected_result, sorted_placement, 'Faulty syntax or content') 

185 

186 def test__run_placement_model_w_pinning(self): 

187 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 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 mpc = MznPlacementConductor(logging.getLogger(__name__)) 

201 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 sorted_placement = sorted(placement, key=lambda k: k['member-vnf-index']) 

204 self.assertEqual(expected_result, sorted_placement, 'Faulty syntax or content') 

205 

206 def test__run_placement_model_unsatisfiable(self): 

207 mpc = MznPlacementConductor(logging.getLogger(__name__)) 

208 self.assertEqual([{}], mpc._run_placement_model(mzn_model=test_mzn_unsatisfiable_model, ns_desc={}), 

209 "Faulty syntax or content for unsatisfiable model") 

210 

211 @mock.patch.object(MznModelGenerator, 'create_model', side_effect=['%model']) 

212 @mock.patch.object(MznPlacementConductor, '_run_placement_model') 

213 def test_do_placement_computation(self, mock_run, mock_create): 

214 mpc = MznPlacementConductor(logging.getLogger(__name__)) 

215 dummy_nspd = {'ns_desc': {}} 

216 _ = mpc.do_placement_computation(dummy_nspd) 

217 mock_create.assert_called_with(dummy_nspd) 

218 mock_run.assert_called_with('%model', {})