osm_pla/test/test_mznModelGenerator.py

# 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 datetime
import logging
# import unittest
from unittest import TestCase
# import random
# from operator import itemgetter
import re

from jinja2 import Template

from osm_pla.placement.mznplacement import MznModelGenerator

test_ns_placement_data_str = {
    'vim_accounts': ['vim' + vim_account.replace('-', '_') for vim_account in ['aaaaaaaa-38f5-438d-b8ee-3f93b3531f87',
                                                                               'bbbbbbbb-38f5-438d-b8ee-3f93b3531f87',
                                                                               'cccccccc-ed84-4e49-b5df-a9d117bd731f',
                                                                               'dddddddd-ed84-4e49-b5df-a9d117bd731f',
                                                                               'eeeeeeee-38f5-438d-b8ee-3f93b3531f87']],
    '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]],
    'trp_link_jitter': [[0, 5, 10, 15, 20], [0, 0, 10, 15, 20], [0, 0, 0, 15, 20], [0, 0, 0, 0, 20],
                        [0, 0, 0, 0, 0]],
    '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]],
    'ns_desc': [
        {'vnf_id': 'one', 'vnf_price_per_vim': [50, 51, 52, 53, 54]},
        {'vnf_id': 'two', 'vnf_price_per_vim': [20, 21, 22, 23, 24]},
        {'vnf_id': 'three', 'vnf_price_per_vim': [70, 71, 72, 73, 74]},
        {'vnf_id': 'four', 'vnf_price_per_vim': [40, 41, 42, 43, 44]}],
    'vld_desc': [{'cp_refs': ['one', 'two'], 'latency': 150, 'jitter': 30},
                 {'cp_refs': ['two', 'three'], 'latency': 140, 'jitter': 30},
                 {'cp_refs': ['three', 'four'], 'latency': 130, 'jitter': 30}],
    'generator_data': {'file': __file__, 'time': datetime.datetime.now()}
}

test_ns_placement_data_str_no_vld_constraints = {
    'vim_accounts': ['vim' + vim_account.replace('-', '_') for vim_account in ['aaaaaaaa-38f5-438d-b8ee-3f93b3531f87',
                                                                               'bbbbbbbb-38f5-438d-b8ee-3f93b3531f87',
                                                                               'cccccccc-ed84-4e49-b5df-a9d117bd731f',
                                                                               'dddddddd-ed84-4e49-b5df-a9d117bd731f',
                                                                               'eeeeeeee-38f5-438d-b8ee-3f93b3531f87']],
    '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]],
    'trp_link_jitter': [[0, 5, 10, 15, 20], [0, 0, 10, 15, 20], [0, 0, 0, 15, 20], [0, 0, 0, 0, 20],
                        [0, 0, 0, 0, 0]],
    '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]],
    'ns_desc': [
        {'vnf_id': 'one', 'vnf_price_per_vim': [50, 51, 52, 53, 54]},
        {'vnf_id': 'two', 'vnf_price_per_vim': [20, 21, 22, 23, 24]},
        {'vnf_id': 'three', 'vnf_price_per_vim': [70, 71, 72, 73, 74]},
        {'vnf_id': 'four', 'vnf_price_per_vim': [40, 41, 42, 43, 44]}],
    'vld_desc': [{'cp_refs': ['one', 'two']},
                 {'cp_refs': ['two', 'three']},
                 {'cp_refs': ['three', 'four']}],
    'generator_data': {'file': __file__, 'time': datetime.datetime.now()}
}

test_ns_placement_data = {
    'vim_accounts': ['vim' + vim_account.replace('-', '_') for vim_account in ['aaaaaaaa-38f5-438d-b8ee-3f93b3531f87',
                                                                               'bbbbbbbb-38f5-438d-b8ee-3f93b3531f87',
                                                                               'cccccccc-ed84-4e49-b5df-a9d117bd731f',
                                                                               'dddddddd-ed84-4e49-b5df-a9d117bd731f',
                                                                               'eeeeeeee-38f5-438d-b8ee-3f93b3531f87']],
    '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]],
    'trp_link_jitter': [[0, 5, 10, 15, 20], [0, 0, 10, 15, 20], [0, 0, 0, 15, 20], [0, 0, 0, 0, 20],
                        [0, 0, 0, 0, 0]],
    '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]],
    'ns_desc': [
        {'vnf_id': '1', 'vnf_price_per_vim': [50, 51, 52, 53, 54]},
        {'vnf_id': '2', 'vnf_price_per_vim': [20, 21, 22, 23, 24]},
        {'vnf_id': '3', 'vnf_price_per_vim': [70, 71, 72, 73, 74]},
        {'vnf_id': '4', 'vnf_price_per_vim': [40, 41, 42, 43, 44]}],
    'vld_desc': [{'cp_refs': ['1', '2'], 'latency': 150, 'jitter': 30},
                 {'cp_refs': ['2', '3'], 'latency': 140, 'jitter': 30},
                 {'cp_refs': ['3', '4'], 'latency': 130, 'jitter': 30}],
    'generator_data': {'file': __file__, 'time': datetime.datetime.now()}
}

test_ns_placement_data_w_pinning = {
    'vim_accounts': ['vim' + vim_account.replace('-', '_') for vim_account in ['aaaaaaaa-38f5-438d-b8ee-3f93b3531f87',
                                                                               'bbbbbbbb-38f5-438d-b8ee-3f93b3531f87',
                                                                               'cccccccc-ed84-4e49-b5df-a9d117bd731f',
                                                                               'dddddddd-ed84-4e49-b5df-a9d117bd731f',
                                                                               'eeeeeeee-38f5-438d-b8ee-3f93b3531f87']],
    '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]],
    'trp_link_jitter': [[0, 5, 10, 15, 20], [0, 0, 10, 15, 20], [0, 0, 0, 15, 20], [0, 0, 0, 0, 20],
                        [0, 0, 0, 0, 0]],
    '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]],
    'ns_desc': [
        {'vnf_id': '1', 'vnf_price_per_vim': [50, 51, 52, 53, 54]},
        {'vnf_id': '2', 'vnf_price_per_vim': [20, 21, 22, 23, 24],
         'vim_account': 'vimeeeeeeee_38f5_438d_b8ee_3f93b3531f87'},
        {'vnf_id': '3', 'vnf_price_per_vim': [70, 71, 72, 73, 74]},
        {'vnf_id': '4', 'vnf_price_per_vim': [40, 41, 42, 43, 44],
         'vim_account': 'vimcccccccc_ed84_4e49_b5df_a9d117bd731f'}],
    'vld_desc': [{'cp_refs': ['1', '2'], 'latency': 150, 'jitter': 30},
                 {'cp_refs': ['2', '3'], 'latency': 140, 'jitter': 30},
                 {'cp_refs': ['3', '4'], 'latency': 130, 'jitter': 30}],
    'generator_data': {'file': __file__, 'time': datetime.datetime.now()}
}

test_ns_placement_data_w_pinning_str = {
    'vim_accounts': ['vim' + vim_account.replace('-', '_') for vim_account in ['aaaaaaaa-38f5-438d-b8ee-3f93b3531f87',
                                                                               'bbbbbbbb-38f5-438d-b8ee-3f93b3531f87',
                                                                               'cccccccc-ed84-4e49-b5df-a9d117bd731f',
                                                                               'dddddddd-ed84-4e49-b5df-a9d117bd731f',
                                                                               'eeeeeeee-38f5-438d-b8ee-3f93b3531f87']],
    '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]],
    'trp_link_jitter': [[0, 5, 10, 15, 20], [0, 0, 10, 15, 20], [0, 0, 0, 15, 20], [0, 0, 0, 0, 20],
                        [0, 0, 0, 0, 0]],
    '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]],
    'ns_desc': [
        {'vnf_id': 'one', 'vnf_price_per_vim': [50, 51, 52, 53, 54]},
        {'vnf_id': 'two', 'vnf_price_per_vim': [20, 21, 22, 23, 24],
         'vim_account': 'vimeeeeeeee_38f5_438d_b8ee_3f93b3531f87'},
        {'vnf_id': 'three', 'vnf_price_per_vim': [70, 71, 72, 73, 74]},
        {'vnf_id': 'four', 'vnf_price_per_vim': [40, 41, 42, 43, 44],
         'vim_account': 'vimcccccccc_ed84_4e49_b5df_a9d117bd731f'}],
    'vld_desc': [{'cp_refs': ['one', 'two'], 'latency': 150, 'jitter': 30},
                 {'cp_refs': ['two', 'three'], 'latency': 140, 'jitter': 30},
                 {'cp_refs': ['three', 'four'], 'latency': 130, 'jitter': 30}],
    'generator_data': {'file': __file__, 'time': datetime.datetime.now()}
}

test_ns_placement_data_str_no_vld = {
    'vim_accounts': ['vim' + vim_account.replace('-', '_') for vim_account in ['aaaaaaaa-38f5-438d-b8ee-3f93b3531f87',
                                                                               'bbbbbbbb-38f5-438d-b8ee-3f93b3531f87',
                                                                               'cccccccc-ed84-4e49-b5df-a9d117bd731f',
                                                                               'dddddddd-ed84-4e49-b5df-a9d117bd731f',
                                                                               'eeeeeeee-38f5-438d-b8ee-3f93b3531f87']],
    '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]],
    'trp_link_jitter': [[0, 5, 10, 15, 20], [0, 0, 10, 15, 20], [0, 0, 0, 15, 20], [0, 0, 0, 0, 20],
                        [0, 0, 0, 0, 0]],
    '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]],
    'ns_desc': [
        {'vnf_id': 'one', 'vnf_price_per_vim': [50, 51, 52, 53, 54]}],
    'vld_desc': [],
    'generator_data': {'file': __file__, 'time': datetime.datetime.now()}
}

expected_model_fragment = """
%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,5,10,15,20,
|0,0,10,15,20,
|0,0,0,15,20,
|0,0,0,0,20,
|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 = [50,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;
"""
expected_model_fragment_str = """
%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,5,10,15,20,
|0,0,10,15,20,
|0,0,0,15,20,
|0,0,0,0,20,
|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_one = [50,51,52,53,54];
array[Vims] of int: vim_price_list_two = [20,21,22,23,24];
array[Vims] of int: vim_price_list_three = [70,71,72,73,74];
array[Vims] of int: vim_price_list_four = [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: VNFone;
var Vims: VNFtwo;
var Vims: VNFthree;
var Vims: VNFfour;


% These are the set of rules for selecting DCs to VNFs
constraint trp_link_latency[VNFone, VNFtwo] <= 150;
constraint trp_link_latency[VNFtwo, VNFthree] <= 140;
constraint trp_link_latency[VNFthree, VNFfour] <= 130;
constraint trp_link_jitter[VNFone, VNFtwo] <= 30;
constraint trp_link_jitter[VNFtwo, VNFthree] <= 30;
constraint trp_link_jitter[VNFthree, VNFfour] <= 30;

% Calculate the cost for VNFs and cost for transport link and total cost
var int: used_transport_cost =trp_link_price_list[VNFone, VNFtwo]+
trp_link_price_list[VNFtwo, VNFthree]+
trp_link_price_list[VNFthree, VNFfour];

var int: used_vim_cost =vim_price_list_one[VNFone]+
vim_price_list_two[VNFtwo]+
vim_price_list_three[VNFthree]+
vim_price_list_four[VNFfour];

var int: total_cost = used_transport_cost + used_vim_cost;

solve minimize total_cost;
"""

expected_model_fragment_str_no_vld_constraints = """
%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,5,10,15,20,
|0,0,10,15,20,
|0,0,0,15,20,
|0,0,0,0,20,
|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_one = [50,51,52,53,54];
array[Vims] of int: vim_price_list_two = [20,21,22,23,24];
array[Vims] of int: vim_price_list_three = [70,71,72,73,74];
array[Vims] of int: vim_price_list_four = [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: VNFone;
var Vims: VNFtwo;
var Vims: VNFthree;
var Vims: VNFfour;


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

% Calculate the cost for VNFs and cost for transport link and total cost
var int: used_transport_cost =trp_link_price_list[VNFone, VNFtwo]+
trp_link_price_list[VNFtwo, VNFthree]+
trp_link_price_list[VNFthree, VNFfour];

var int: used_vim_cost =vim_price_list_one[VNFone]+
vim_price_list_two[VNFtwo]+
vim_price_list_three[VNFthree]+
vim_price_list_four[VNFfour];

var int: total_cost = used_transport_cost + used_vim_cost;

solve minimize total_cost;
"""

expected_model_fragment_w_pinning = """
%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,5,10,15,20,
|0,0,10,15,20,
|0,0,0,15,20,
|0,0,0,0,20,
|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 = [50,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;
Vims: VNF2 = vimeeeeeeee_38f5_438d_b8ee_3f93b3531f87;
var Vims: VNF3;
Vims: VNF4 = vimcccccccc_ed84_4e49_b5df_a9d117bd731f;


% 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;
"""

expected_model_fragment_w_pinning_str = """
%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,5,10,15,20,
|0,0,10,15,20,
|0,0,0,15,20,
|0,0,0,0,20,
|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_one = [50,51,52,53,54];
array[Vims] of int: vim_price_list_two = [20,21,22,23,24];
array[Vims] of int: vim_price_list_three = [70,71,72,73,74];
array[Vims] of int: vim_price_list_four = [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: VNFone;
Vims: VNFtwo = vimeeeeeeee_38f5_438d_b8ee_3f93b3531f87;
var Vims: VNFthree;
Vims: VNFfour = vimcccccccc_ed84_4e49_b5df_a9d117bd731f;


% These are the set of rules for selecting DCs to VNFs
constraint trp_link_latency[VNFone, VNFtwo] <= 150;
constraint trp_link_latency[VNFtwo, VNFthree] <= 140;
constraint trp_link_latency[VNFthree, VNFfour] <= 130;
constraint trp_link_jitter[VNFone, VNFtwo] <= 30;
constraint trp_link_jitter[VNFtwo, VNFthree] <= 30;
constraint trp_link_jitter[VNFthree, VNFfour] <= 30;

% Calculate the cost for VNFs and cost for transport link and total cost
var int: used_transport_cost =trp_link_price_list[VNFone, VNFtwo]+
trp_link_price_list[VNFtwo, VNFthree]+
trp_link_price_list[VNFthree, VNFfour];

var int: used_vim_cost =vim_price_list_one[VNFone]+
vim_price_list_two[VNFtwo]+
vim_price_list_three[VNFthree]+
vim_price_list_four[VNFfour];

var int: total_cost = used_transport_cost + used_vim_cost;

solve minimize total_cost;
"""

expected_model_fragment_str_no_vld = """
%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,5,10,15,20,
|0,0,10,15,20,
|0,0,0,15,20,
|0,0,0,0,20,
|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_one = [50,51,52,53,54];


% 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: VNFone;

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

% Calculate the cost for VNFs and cost for transport link and total cost
var int: used_transport_cost =0;

var int: used_vim_cost =vim_price_list_one[VNFone];

var int: total_cost = used_transport_cost + used_vim_cost;

solve minimize total_cost;
"""


class TestMznModelGenerator(TestCase):
    def test_create_model(self):
        mg = MznModelGenerator(logging.getLogger(__name__))
        mzn_model = mg.create_model(test_ns_placement_data_str)

        # so asserting exact content is difficult due to the datetime.now(), therefore we ignore the first lines
        self.assertTrue(expected_model_fragment_str.replace('\n', '') in
                        mzn_model.replace('\n', ''), "faulty model generated")

    def test_create_model_no_vld_constraints(self):
        """
        instantiate w/o constraints in nsd or order params has a valid model
        :return:
        """
        mg = MznModelGenerator(logging.getLogger(__name__))
        mzn_model = mg.create_model(test_ns_placement_data_str_no_vld_constraints)

        # so asserting exact content is difficult due to the datetime.now(), therefore we ignore the first lines
        self.assertTrue(expected_model_fragment_str_no_vld_constraints.replace('\n', '') in
                        mzn_model.replace('\n', ''), "faulty model generated")

    def test_create_model_w_pinning(self):
        mg = MznModelGenerator(logging.getLogger(__name__))
        mzn_model = mg.create_model(test_ns_placement_data_w_pinning_str)

        # so asserting exact content is difficult due to the datetime.now(), therefore we ignore the first lines
        self.assertTrue(expected_model_fragment_w_pinning_str.replace('\n', '') in
                        mzn_model.replace('\n', ''), "faulty model generated")

    def test_create_model_no_vld(self):
        mg = MznModelGenerator(logging.getLogger(__name__))
        mzn_model = mg.create_model(test_ns_placement_data_str_no_vld)

        # so asserting exact content is difficult due to the datetime.now(), therefore we ignore the first lines
        self.assertTrue(expected_model_fragment_str_no_vld.replace('\n', '') in
                        mzn_model.replace('\n', ''), "faulty model generated")

    def test__load_jinja_template(self):
        """

        add other test to check exception if template not loaded (e.g. invalid template name,
        perhaps also valid name but invalid content (in case jinja2 detects such things))
        """
        mg = MznModelGenerator(logging.getLogger(__name__))
        template = mg._load_jinja_template()  # Note we use the default template
        self.assertTrue(isinstance(template, Template), "failed to load jinja2 template")

    def test_vim_account_replace(self):
        mg = MznModelGenerator(logging.getLogger(__name__))
        nspd = test_ns_placement_data_str
        mzn_model = mg.create_model(nspd)

        expected = '%This is the NETWORK RESOURCE MODEL' + '\n' + 'enum  Vims = {' + '\n'
        for val in test_ns_placement_data_str['vim_accounts']:
            expected = expected + val.replace('-', '_') + ',\n'
        expected = expected[:-2] + '}; % The vim-accounts'
        res = re.findall(expected, mzn_model)
        self.assertEqual(1, len(res), "vim accounts didnt replace from - to _")

    def test_trp_link_price_list(self):
        mg = MznModelGenerator(logging.getLogger(__name__))
        mzn_model = mg.create_model(test_ns_placement_data_str)

        expected = 'array\\[Vims, Vims\\] of int: trp_link_price_list = \\['
        for price_list in test_ns_placement_data_str['trp_link_price_list']:
            expected = expected + '\\|' + (str(price_list)[1:-1]).replace(" ", "") + ',\n'
        expected = expected + '\\|\\]; % Transport link price list'
        res = re.findall(expected, mzn_model)
        self.assertEqual(1, len(res), "price list is not correct")

    def test_link_latency(self):
        mg = MznModelGenerator(logging.getLogger(__name__))
        mzn_model = mg.create_model(test_ns_placement_data_str)

        expected = 'array\\[Vims, Vims\\] of int: trp_link_latency = \\['
        for link_latency in test_ns_placement_data_str['trp_link_latency']:
            expected = expected + '\\|' + (str(link_latency)[1:-1]).replace(" ", "") + ',\n'
        expected = expected + '\\|\\]; % Transport link latency between data centers'
        res = re.findall(expected, mzn_model)
        self.assertEqual(1, len(res), "trp_link_latency values is not correct")

    def test_link_jitter(self):
        mg = MznModelGenerator(logging.getLogger(__name__))
        mzn_model = mg.create_model(test_ns_placement_data_str)

        expected = 'array\\[Vims, Vims\\] of int: trp_link_jitter = \\['
        for link_jitter in test_ns_placement_data_str['trp_link_jitter']:
            expected = expected + '\\|' + (str(link_jitter)[1:-1]).replace(" ", "") + ',\n'
        expected = expected + '\\|\\]; % Transport link jitter between data centers'

        res = re.findall(expected, mzn_model)

        self.assertEqual(1, len(res), "trp_link_jitter values is not correct")

    def test_price_per_vim(self):
        mg = MznModelGenerator(logging.getLogger(__name__))
        mzn_model = mg.create_model(test_ns_placement_data_w_pinning_str)

        expected = ""
        for price_list in test_ns_placement_data_w_pinning_str['ns_desc']:
            expected += 'array\\[Vims\\] of int: vim_price_list_' + price_list.get('vnf_id') + " = "
            temp = str(price_list.get('vnf_price_per_vim'))[1:-1].replace(" ", "")
            expected += "\\[" + temp + "\\];\n"

        res = re.findall(expected, mzn_model)
        self.assertEqual(1, len(res), "mzn_model contains pinning")

    def test_pinning(self):
        mg = MznModelGenerator(logging.getLogger(__name__))
        mzn_model = mg.create_model(test_ns_placement_data_str)

        expected = ""
        for pin_list in test_ns_placement_data_str['ns_desc']:
            if pin_list.get('vim_account'):
                expected += 'Vims: VNF' + pin_list.get('vnf_id') + ' = ' + pin_list.get('vim_account') + ';\n'
            else:
                expected += 'var Vims: VNF' + pin_list.get('vnf_id') + ';\n'

        res = re.findall(expected, mzn_model)
        self.assertEqual(1, len(res), "mzn_model has no pinning")

    def test__without_pinning(self):
        mg = MznModelGenerator(logging.getLogger(__name__))
        mzn_model = mg.create_model(test_ns_placement_data_w_pinning_str)

        expected = ""
        for pin_list in test_ns_placement_data_w_pinning_str['ns_desc']:
            if pin_list.get('vim_account'):
                expected += 'Vims: VNF' + pin_list.get('vnf_id') + ' = ' + pin_list.get('vim_account') + ';\n'
            else:
                expected += 'var Vims: VNF' + pin_list.get('vnf_id') + ';\n'

        res = re.findall(expected, mzn_model)
        self.assertEqual(1, len(res), "mzn_model contains pinning")

    def test__without_constraints_for_jitter_and_latency(self):
        mg = MznModelGenerator(logging.getLogger(__name__))
        mzn_model = mg.create_model(test_ns_placement_data_str_no_vld_constraints)

        expected_latency = "constraint trp_link_latency"
        expected_jitter = "constraint trp_link_jitter"
        latency_or_jitter_was_found = 0
        for l_o_j in test_ns_placement_data_str_no_vld_constraints['vld_desc']:
            if l_o_j.get('latency') or l_o_j.get('jitter'):
                latency_or_jitter_was_found = 1

        res_latency = re.findall(expected_latency, mzn_model)
        res_jitter = re.findall(expected_jitter, mzn_model)
        self.assertEqual(0, latency_or_jitter_was_found, "Jitter or latency was found in the test input")
        self.assertEqual(0, len(res_latency), "constraint trp_link_latency was found in mzn_model")
        self.assertEqual(0, len(res_jitter), "constraint trp_link_latency was found in mzn_model")

    def test__constraints_for_jitter_and_latency(self):
        mg = MznModelGenerator(logging.getLogger(__name__))
        mzn_model = mg.create_model(test_ns_placement_data_str)

        expected_latency = ""
        expected_jitter = ""
        latency_or_jitter_was_found = 0
        for l_o_j in test_ns_placement_data_str['vld_desc']:
            if not (l_o_j.get('latency') or l_o_j.get('jitter')):
                latency_or_jitter_was_found = 1
            expected_latency += "constraint trp_link_latency" + "\\[VNF" + l_o_j.get('cp_refs')[0] + ", VNF" + \
                                l_o_j.get('cp_refs')[1] + "\\] \\<= " + str(l_o_j.get('latency')) + ";\n\n"

            expected_jitter += "constraint trp_link_jitter" + "\\[VNF" + l_o_j.get('cp_refs')[0] + ", VNF" + \
                               l_o_j.get('cp_refs')[1] + "\\] \\<= " + str(l_o_j.get('jitter')) + ";\n\n"

        res = re.findall(expected_latency + expected_jitter, mzn_model)
        self.assertEqual(0, latency_or_jitter_was_found, "Jitter or latency was not found in the test input")
        self.assertEqual(1, len(res), "faulty model generated")