/
test_vn_service.py
54 lines (41 loc) · 1.94 KB
/
test_vn_service.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
from cvm.constants import VNC_ROOT_DOMAIN, VNC_VCENTER_PROJECT
from tests.utils import assert_vn_model_state
def test_update_vns_no_vns(vn_service, database, vcenter_api_client, vnc_api_client):
database.vmis_to_update = []
vn_service.update_vns()
assert database.get_all_vn_models() == []
vcenter_api_client.get_dpg_by_key.assert_not_called()
vnc_api_client.read_vn.assert_not_called()
def test_update_vns(vn_service, database, vcenter_api_client, vnc_api_client, vmi_model, vnc_vn_1, portgroup):
vmi_model.vcenter_port.portgroup_key = 'dvportgroup-1'
database.vmis_to_update.append(vmi_model)
vcenter_api_client.get_dpg_by_key.return_value = portgroup
vnc_api_client.read_vn.return_value = vnc_vn_1
vn_service.update_vns()
vn_model = database.get_vn_model_by_key('dvportgroup-1')
assert vn_model is not None
assert_vn_model_state(
vn_model,
key='dvportgroup-1',
vnc_vn=vnc_vn_1,
vmware_vn=portgroup,
)
vcenter_api_client.get_dpg_by_key.assert_called_once_with('dvportgroup-1')
vcenter_api_client.enable_vlan_override.assert_called_once_with(portgroup)
vcenter_api_client.set_vlan_trunk.assert_called_once_with(portgroup)
fq_name = [VNC_ROOT_DOMAIN, VNC_VCENTER_PROJECT, 'DPG1']
vnc_api_client.read_vn.assert_called_once_with(fq_name)
def test_sync_vns(vn_service, database, vcenter_api_client, vnc_api_client, vnc_vn_1, portgroup):
vnc_api_client.get_vns_by_project.return_value = [vnc_vn_1]
vcenter_api_client.get_dpg_by_name.return_value = portgroup
vn_service.sync_vns()
vn_model = database.get_vn_model_by_key('dvportgroup-1')
assert vn_model is not None
assert_vn_model_state(
vn_model,
key='dvportgroup-1',
vnc_vn=vnc_vn_1,
vmware_vn=portgroup,
)
vcenter_api_client.enable_vlan_override.assert_called_once_with(portgroup)
vcenter_api_client.set_vlan_trunk.assert_called_once_with(portgroup)