# 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
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'}]
+ 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')
+ 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'}
- ]
+ 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)
+ 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')
+ 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')
+ 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': {}}
+ dummy_nspd = {"ns_desc": {}}
_ = mpc.do_placement_computation(dummy_nspd)
mock_create.assert_called_with(dummy_nspd)
- mock_run.assert_called_with('%model', {})
+ mock_run.assert_called_with("%model", {})