osm_pla/test/test_mznPlacementConductor.py - osm/PLA - Gitiles

 # Copyright 2020 ArctosLabs Scandinavia AB
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
 #
 #    http://www.apache.org/licenses/LICENSE-2.0
 #
 # Unless required by applicable law or agreed to in writing, software
 # distributed under the License is distributed on an "AS IS" BASIS,
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
 # implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
 import logging

 # from collections import Counter
 from unittest import TestCase, mock

 # import osm_pla
 from osm_pla.placement.mznplacement import MznPlacementConductor, MznModelGenerator

 test_mzn_model = """
 % This minizinc model is generated using
 % C:/Users/LG/PycharmProjects/dynamic_jijna2_mzn/osm_pla/placement/mznplacement.py
 % at 2019-10-24 11:12:02.058905.

 %This is the NETWORK RESOURCE MODEL
 enum  Vims = {
 vimaaaaaaaa_38f5_438d_b8ee_3f93b3531f87,
 vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87,
 vimcccccccc_ed84_4e49_b5df_a9d117bd731f,
 vimdddddddd_ed84_4e49_b5df_a9d117bd731f,
 vimeeeeeeee_38f5_438d_b8ee_3f93b3531f87}; % The vim-accounts
 array[Vims, Vims] of int: trp_link_latency = [|0,50,100,150,200,
 |0,0,100,150,200,
 |0,0,0,150,200,
 |0,0,0,0,200,
 |0,0,0,0,0,
 |]; % Transport link latency between data centers
 array[Vims, Vims] of int: trp_link_jitter = [|0,50,100,150,200,
 |0,0,100,150,200,
 |0,0,0,150,200,
 |0,0,0,0,200,
 |0,0,0,0,0,
 |]; % Transport link jitter between data centers
 array[Vims, Vims] of int: trp_link_price_list = [|0,5,6,6,7,
 |0,0,6,6,7,
 |0,0,0,6,7,
 |0,0,0,0,7,
 |0,0,0,0,0,
 |]; % Transport link price list
 array[Vims] of int: vim_price_list_1 = [500,51,52,53,54];
 array[Vims] of int: vim_price_list_2 = [20,21,22,23,24];
 array[Vims] of int: vim_price_list_3 = [70,71,72,73,74];
 array[Vims] of int: vim_price_list_4 = [40,41,42,43,44];


 % This is the NETWORK BASIC LOAD MODEL (CONSUMED)
 % NOTE. This is not applicable in OSM Release 7

 % This is the SERVICE CONSUMPTION MODEL
 % These are the variables, i.e. which DC to select for each VNF
 var Vims: VNF1;
 var Vims: VNF2;
 var Vims: VNF3;
 var Vims: VNF4;


 % These are the set of rules for selecting DCs to VNFs
 constraint trp_link_latency[VNF1, VNF2] <= 150;
 constraint trp_link_latency[VNF2, VNF3] <= 140;
 constraint trp_link_latency[VNF3, VNF4] <= 130;
 constraint trp_link_jitter[VNF1, VNF2] <= 30;
 constraint trp_link_jitter[VNF2, VNF3] <= 30;
 constraint trp_link_jitter[VNF3, VNF4] <= 30;

 % Calculate the cost for VNFs and cost for transport link and total cost
 var int: used_transport_cost =trp_link_price_list[VNF1, VNF2]+
 trp_link_price_list[VNF2, VNF3]+
 trp_link_price_list[VNF3, VNF4];

 var int: used_vim_cost =vim_price_list_1[VNF1]+
 vim_price_list_2[VNF2]+
 vim_price_list_3[VNF3]+
 vim_price_list_4[VNF4];

 var int: total_cost = used_transport_cost + used_vim_cost;

 solve minimize total_cost;

 """

 test_mzn_model_w_pinning = """
 % This minizinc model is generated using
 % C:/Users/LG/PycharmProjects/dynamic_jijna2_mzn/osm_pla/placement/mznplacement.py
 % at 2019-10-24 11:12:02.058905.

 %This is the NETWORK RESOURCE MODEL
 enum  Vims = {
 vimaaaaaaaa_38f5_438d_b8ee_3f93b3531f87,
 vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87,
 vimcccccccc_ed84_4e49_b5df_a9d117bd731f,
 vimdddddddd_ed84_4e49_b5df_a9d117bd731f,
 vimeeeeeeee_38f5_438d_b8ee_3f93b3531f87}; % The vim-accounts
 array[Vims, Vims] of int: trp_link_latency = [|0,50,100,150,200,
 |0,0,100,150,200,
 |0,0,0,150,200,
 |0,0,0,0,200,
 |0,0,0,0,0,
 |]; % Transport link latency between data centers
 array[Vims, Vims] of int: trp_link_jitter = [|0,50,100,150,200,
 |0,0,100,150,200,
 |0,0,0,150,200,
 |0,0,0,0,200,
 |0,0,0,0,0,
 |]; % Transport link jitter between data centers
 array[Vims, Vims] of int: trp_link_price_list = [|0,5,6,6,7,
 |0,0,6,6,7,
 |0,0,0,6,7,
 |0,0,0,0,7,
 |0,0,0,0,0,
 |]; % Transport link price list
 array[Vims] of int: vim_price_list_1 = [500,51,52,53,54];
 array[Vims] of int: vim_price_list_2 = [20,21,22,23,24];
 array[Vims] of int: vim_price_list_3 = [70,71,72,73,74];
 array[Vims] of int: vim_price_list_4 = [40,41,42,43,44];


 % This is the NETWORK BASIC LOAD MODEL (CONSUMED)
 % NOTE. This is not applicable in OSM Release 7

 % This is the SERVICE CONSUMPTION MODEL
 % These are the variables, i.e. which DC to select for each VNF
 Vims: VNF1 = vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87;
 var Vims: VNF2;
 Vims: VNF3 = vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87;
 var Vims: VNF4;


 % These are the set of rules for selecting DCs to VNFs
 constraint trp_link_latency[VNF1, VNF2] <= 150;
 constraint trp_link_latency[VNF2, VNF3] <= 140;
 constraint trp_link_latency[VNF3, VNF4] <= 130;
 constraint trp_link_jitter[VNF1, VNF2] <= 30;
 constraint trp_link_jitter[VNF2, VNF3] <= 30;
 constraint trp_link_jitter[VNF3, VNF4] <= 30;

 % Calculate the cost for VNFs and cost for transport link and total cost
 var int: used_transport_cost =trp_link_price_list[VNF1, VNF2]+
 trp_link_price_list[VNF2, VNF3]+
 trp_link_price_list[VNF3, VNF4];

 var int: used_vim_cost =vim_price_list_1[VNF1]+
 vim_price_list_2[VNF2]+
 vim_price_list_3[VNF3]+
 vim_price_list_4[VNF4];

 var int: total_cost = used_transport_cost + used_vim_cost;

 solve minimize total_cost;

 """

 test_mzn_unsatisfiable_model = """
 var 1..2: item1;
 var 1..2: item2;
 constraint item1 + item2 == 5;

 solve satisfy;
 """


 class TestMznPlacementConductor(TestCase):
     def test__run_placement_model(self):
         expected_result = [
             {
                 "vimAccountId": "bbbbbbbb-38f5-438d-b8ee-3f93b3531f87",
                 "member-vnf-index": "1",
             },
             {
                 "vimAccountId": "aaaaaaaa-38f5-438d-b8ee-3f93b3531f87",
                 "member-vnf-index": "2",
             },
             {
                 "vimAccountId": "aaaaaaaa-38f5-438d-b8ee-3f93b3531f87",
                 "member-vnf-index": "3",
             },
             {
                 "vimAccountId": "aaaaaaaa-38f5-438d-b8ee-3f93b3531f87",
                 "member-vnf-index": "4",
             },
         ]

         mpc = MznPlacementConductor(logging.getLogger(__name__))
         placement = mpc._run_placement_model(mzn_model=test_mzn_model, ns_desc={})
         # sort the result to ease assert with expected result
         sorted_placement = sorted(placement, key=lambda k: k["member-vnf-index"])
         self.assertEqual(expected_result, sorted_placement, "Faulty syntax or content")

     def test__run_placement_model_w_pinning(self):
         expected_result = [
             {
                 "vimAccountId": "bbbbbbbb-38f5-438d-b8ee-3f93b3531f87",
                 "member-vnf-index": "1",
             },
             {
                 "vimAccountId": "bbbbbbbb-38f5-438d-b8ee-3f93b3531f87",
                 "member-vnf-index": "2",
             },
             {
                 "vimAccountId": "bbbbbbbb-38f5-438d-b8ee-3f93b3531f87",
                 "member-vnf-index": "3",
             },
             {
                 "vimAccountId": "aaaaaaaa-38f5-438d-b8ee-3f93b3531f87",
                 "member-vnf-index": "4",
             },
         ]

         ns_desc = [
             {"vnf_price_per_vim": [10, 9, 7, 8], "vnf_id": "2"},
             {
                 "vim_account": "vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87",
                 "vnf_price_per_vim": [10, 9, 7, 8],
                 "vnf_id": "1",
             },
             {"vnf_price_per_vim": [10, 9, 7, 8], "vnf_id": "4"},
             {
                 "vim_account": "vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87",
                 "vnf_price_per_vim": [10, 9, 7, 8],
                 "vnf_id": "3",
             },
         ]

         mpc = MznPlacementConductor(logging.getLogger(__name__))
         placement = mpc._run_placement_model(
             mzn_model=test_mzn_model_w_pinning, ns_desc=ns_desc
         )
         # sort the result to ease assert with expected result
         sorted_placement = sorted(placement, key=lambda k: k["member-vnf-index"])
         self.assertEqual(expected_result, sorted_placement, "Faulty syntax or content")

     def test__run_placement_model_unsatisfiable(self):
         mpc = MznPlacementConductor(logging.getLogger(__name__))
         self.assertEqual(
             [{}],
             mpc._run_placement_model(
                 mzn_model=test_mzn_unsatisfiable_model, ns_desc={}
             ),
             "Faulty syntax or content for unsatisfiable model",
         )

     @mock.patch.object(MznModelGenerator, "create_model", side_effect=["%model"])
     @mock.patch.object(MznPlacementConductor, "_run_placement_model")
     def test_do_placement_computation(self, mock_run, mock_create):
         mpc = MznPlacementConductor(logging.getLogger(__name__))
         dummy_nspd = {"ns_desc": {}}
         _ = mpc.do_placement_computation(dummy_nspd)
         mock_create.assert_called_with(dummy_nspd)
         mock_run.assert_called_with("%model", {})
	# Copyright 2020 ArctosLabs Scandinavia AB
	#
	# Licensed under the Apache License, Version 2.0 (the "License");
	# you may not use this file except in compliance with the License.
	# You may obtain a copy of the License at
	#
	# http://www.apache.org/licenses/LICENSE-2.0
	#
	# Unless required by applicable law or agreed to in writing, software
	# distributed under the License is distributed on an "AS IS" BASIS,
	# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
	# implied.
	# See the License for the specific language governing permissions and
	# limitations under the License.
	import logging

	# from collections import Counter
	from unittest import TestCase, mock

	# import osm_pla
	from osm_pla.placement.mznplacement import MznPlacementConductor, MznModelGenerator

	test_mzn_model = """
	% This minizinc model is generated using
	% C:/Users/LG/PycharmProjects/dynamic_jijna2_mzn/osm_pla/placement/mznplacement.py
	% at 2019-10-24 11:12:02.058905.

	%This is the NETWORK RESOURCE MODEL
	enum Vims = {
	vimaaaaaaaa_38f5_438d_b8ee_3f93b3531f87,
	vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87,
	vimcccccccc_ed84_4e49_b5df_a9d117bd731f,
	vimdddddddd_ed84_4e49_b5df_a9d117bd731f,
	vimeeeeeeee_38f5_438d_b8ee_3f93b3531f87}; % The vim-accounts
	array[Vims, Vims] of int: trp_link_latency = [\|0,50,100,150,200,
	\|0,0,100,150,200,
	\|0,0,0,150,200,
	\|0,0,0,0,200,
	\|0,0,0,0,0,
	\|]; % Transport link latency between data centers
	array[Vims, Vims] of int: trp_link_jitter = [\|0,50,100,150,200,
	\|0,0,100,150,200,
	\|0,0,0,150,200,
	\|0,0,0,0,200,
	\|0,0,0,0,0,
	\|]; % Transport link jitter between data centers
	array[Vims, Vims] of int: trp_link_price_list = [\|0,5,6,6,7,
	\|0,0,6,6,7,
	\|0,0,0,6,7,
	\|0,0,0,0,7,
	\|0,0,0,0,0,
	\|]; % Transport link price list
	array[Vims] of int: vim_price_list_1 = [500,51,52,53,54];
	array[Vims] of int: vim_price_list_2 = [20,21,22,23,24];
	array[Vims] of int: vim_price_list_3 = [70,71,72,73,74];
	array[Vims] of int: vim_price_list_4 = [40,41,42,43,44];


	% This is the NETWORK BASIC LOAD MODEL (CONSUMED)
	% NOTE. This is not applicable in OSM Release 7

	% This is the SERVICE CONSUMPTION MODEL
	% These are the variables, i.e. which DC to select for each VNF
	var Vims: VNF1;
	var Vims: VNF2;
	var Vims: VNF3;
	var Vims: VNF4;


	% These are the set of rules for selecting DCs to VNFs
	constraint trp_link_latency[VNF1, VNF2] <= 150;
	constraint trp_link_latency[VNF2, VNF3] <= 140;
	constraint trp_link_latency[VNF3, VNF4] <= 130;
	constraint trp_link_jitter[VNF1, VNF2] <= 30;
	constraint trp_link_jitter[VNF2, VNF3] <= 30;
	constraint trp_link_jitter[VNF3, VNF4] <= 30;

	% Calculate the cost for VNFs and cost for transport link and total cost
	var int: used_transport_cost =trp_link_price_list[VNF1, VNF2]+
	trp_link_price_list[VNF2, VNF3]+
	trp_link_price_list[VNF3, VNF4];

	var int: used_vim_cost =vim_price_list_1[VNF1]+
	vim_price_list_2[VNF2]+
	vim_price_list_3[VNF3]+
	vim_price_list_4[VNF4];

	var int: total_cost = used_transport_cost + used_vim_cost;

	solve minimize total_cost;

	"""

	test_mzn_model_w_pinning = """
	% This minizinc model is generated using
	% C:/Users/LG/PycharmProjects/dynamic_jijna2_mzn/osm_pla/placement/mznplacement.py
	% at 2019-10-24 11:12:02.058905.

	%This is the NETWORK RESOURCE MODEL
	enum Vims = {
	vimaaaaaaaa_38f5_438d_b8ee_3f93b3531f87,
	vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87,
	vimcccccccc_ed84_4e49_b5df_a9d117bd731f,
	vimdddddddd_ed84_4e49_b5df_a9d117bd731f,
	vimeeeeeeee_38f5_438d_b8ee_3f93b3531f87}; % The vim-accounts
	array[Vims, Vims] of int: trp_link_latency = [\|0,50,100,150,200,
	\|0,0,100,150,200,
	\|0,0,0,150,200,
	\|0,0,0,0,200,
	\|0,0,0,0,0,
	\|]; % Transport link latency between data centers
	array[Vims, Vims] of int: trp_link_jitter = [\|0,50,100,150,200,
	\|0,0,100,150,200,
	\|0,0,0,150,200,
	\|0,0,0,0,200,
	\|0,0,0,0,0,
	\|]; % Transport link jitter between data centers
	array[Vims, Vims] of int: trp_link_price_list = [\|0,5,6,6,7,
	\|0,0,6,6,7,
	\|0,0,0,6,7,
	\|0,0,0,0,7,
	\|0,0,0,0,0,
	\|]; % Transport link price list
	array[Vims] of int: vim_price_list_1 = [500,51,52,53,54];
	array[Vims] of int: vim_price_list_2 = [20,21,22,23,24];
	array[Vims] of int: vim_price_list_3 = [70,71,72,73,74];
	array[Vims] of int: vim_price_list_4 = [40,41,42,43,44];


	% This is the NETWORK BASIC LOAD MODEL (CONSUMED)
	% NOTE. This is not applicable in OSM Release 7

	% This is the SERVICE CONSUMPTION MODEL
	% These are the variables, i.e. which DC to select for each VNF
	Vims: VNF1 = vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87;
	var Vims: VNF2;
	Vims: VNF3 = vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87;
	var Vims: VNF4;


	% These are the set of rules for selecting DCs to VNFs
	constraint trp_link_latency[VNF1, VNF2] <= 150;
	constraint trp_link_latency[VNF2, VNF3] <= 140;
	constraint trp_link_latency[VNF3, VNF4] <= 130;
	constraint trp_link_jitter[VNF1, VNF2] <= 30;
	constraint trp_link_jitter[VNF2, VNF3] <= 30;
	constraint trp_link_jitter[VNF3, VNF4] <= 30;

	% Calculate the cost for VNFs and cost for transport link and total cost
	var int: used_transport_cost =trp_link_price_list[VNF1, VNF2]+
	trp_link_price_list[VNF2, VNF3]+
	trp_link_price_list[VNF3, VNF4];

	var int: used_vim_cost =vim_price_list_1[VNF1]+
	vim_price_list_2[VNF2]+
	vim_price_list_3[VNF3]+
	vim_price_list_4[VNF4];

	var int: total_cost = used_transport_cost + used_vim_cost;

	solve minimize total_cost;

	"""

	test_mzn_unsatisfiable_model = """
	var 1..2: item1;
	var 1..2: item2;
	constraint item1 + item2 == 5;

	solve satisfy;
	"""


	class TestMznPlacementConductor(TestCase):
	def test__run_placement_model(self):
	expected_result = [
	{
	"vimAccountId": "bbbbbbbb-38f5-438d-b8ee-3f93b3531f87",
	"member-vnf-index": "1",
	},
	{
	"vimAccountId": "aaaaaaaa-38f5-438d-b8ee-3f93b3531f87",
	"member-vnf-index": "2",
	},
	{
	"vimAccountId": "aaaaaaaa-38f5-438d-b8ee-3f93b3531f87",
	"member-vnf-index": "3",
	},
	{
	"vimAccountId": "aaaaaaaa-38f5-438d-b8ee-3f93b3531f87",
	"member-vnf-index": "4",
	},
	]

	mpc = MznPlacementConductor(logging.getLogger(__name__))
	placement = mpc._run_placement_model(mzn_model=test_mzn_model, ns_desc={})
	# sort the result to ease assert with expected result
	sorted_placement = sorted(placement, key=lambda k: k["member-vnf-index"])
	self.assertEqual(expected_result, sorted_placement, "Faulty syntax or content")

	def test__run_placement_model_w_pinning(self):
	expected_result = [
	{
	"vimAccountId": "bbbbbbbb-38f5-438d-b8ee-3f93b3531f87",
	"member-vnf-index": "1",
	},
	{
	"vimAccountId": "bbbbbbbb-38f5-438d-b8ee-3f93b3531f87",
	"member-vnf-index": "2",
	},
	{
	"vimAccountId": "bbbbbbbb-38f5-438d-b8ee-3f93b3531f87",
	"member-vnf-index": "3",
	},
	{
	"vimAccountId": "aaaaaaaa-38f5-438d-b8ee-3f93b3531f87",
	"member-vnf-index": "4",
	},
	]

	ns_desc = [
	{"vnf_price_per_vim": [10, 9, 7, 8], "vnf_id": "2"},
	{
	"vim_account": "vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87",
	"vnf_price_per_vim": [10, 9, 7, 8],
	"vnf_id": "1",
	},
	{"vnf_price_per_vim": [10, 9, 7, 8], "vnf_id": "4"},
	{
	"vim_account": "vimbbbbbbbb_38f5_438d_b8ee_3f93b3531f87",
	"vnf_price_per_vim": [10, 9, 7, 8],
	"vnf_id": "3",
	},
	]

	mpc = MznPlacementConductor(logging.getLogger(__name__))
	placement = mpc._run_placement_model(
	mzn_model=test_mzn_model_w_pinning, ns_desc=ns_desc
	)
	# sort the result to ease assert with expected result
	sorted_placement = sorted(placement, key=lambda k: k["member-vnf-index"])
	self.assertEqual(expected_result, sorted_placement, "Faulty syntax or content")

	def test__run_placement_model_unsatisfiable(self):
	mpc = MznPlacementConductor(logging.getLogger(__name__))
	self.assertEqual(
	[{}],
	mpc._run_placement_model(
	mzn_model=test_mzn_unsatisfiable_model, ns_desc={}
	),
	"Faulty syntax or content for unsatisfiable model",
	)

	@mock.patch.object(MznModelGenerator, "create_model", side_effect=["%model"])
	@mock.patch.object(MznPlacementConductor, "_run_placement_model")
	def test_do_placement_computation(self, mock_run, mock_create):
	mpc = MznPlacementConductor(logging.getLogger(__name__))
	dummy_nspd = {"ns_desc": {}}
	_ = mpc.do_placement_computation(dummy_nspd)
	mock_create.assert_called_with(dummy_nspd)
	mock_run.assert_called_with("%model", {})