def test_do_not_copy(vmware_fixture):
vmware_handler, client, si_mock = vmware_fixture
vnic1 = Mock(portgroup="p1")
vnic2 = Mock(portgroup="p2")
console_vnic1 = Mock(portgroup="p3")
console_vnic2 = Mock(portgroup="p4")
view1 = Mock(vim.View)
view1.name = "host"
view1.config = Mock(network=Mock(
vnic=[vnic1, vnic2], consoleVnic=[console_vnic1, console_vnic2]))
container = Mock(view=[view1])
view_manager = Mock()
view_manager.CreateContainerView = Mock(return_value=container)
si_mock.RetrieveContent.return_value.viewManager = view_manager
assert client.get_do_not_copy_list("host") == [
vnic1,
vnic2,
console_vnic1,
console_vnic2,
]
assert client.is_in_do_not_copy_list("host", "p1") is True
assert client.is_in_do_not_copy_list("host", "p2") is True
assert client.is_in_do_not_copy_list("host", "p9009") is False
def test_get_host_virtual_switches(vmware_fixture):
vmware_handler, client, si_mock = vmware_fixture
vswitch = Mock()
view1 = Mock(vim.View)
view1.name = "host"
view1.config = Mock(network=Mock(vswitch=vswitch))
container = Mock(view=[view1])
view_manager = Mock()
view_manager.CreateContainerView = Mock(return_value=container)
si_mock.RetrieveContent.return_value.viewManager = view_manager
assert client.get_host_virtual_switches("host") == vswitch
def test_get_host_portgroups(vmware_fixture):
vmware_handler, client, si_mock = vmware_fixture
portgroup = Mock()
view1 = Mock(vim.View)
view1.name = "host"
view1.config = Mock(network=Mock(portgroup=portgroup))
container = Mock(view=[view1])
view_manager = Mock()
view_manager.CreateContainerView = Mock(return_value=container)
si_mock.RetrieveContent.return_value.viewManager = view_manager
assert client.get_host_portgroups("host") == portgroup
def test_has_vswitch(vmware_fixture):
vmware_handler, client, si_mock = vmware_fixture
switch = Mock()
switch.name = "s1"
view1 = Mock(vim.View)
view1.name = "host"
view1.config = Mock(network=Mock(vswitch=[switch]))
container = Mock(view=[view1])
view_manager = Mock()
view_manager.CreateContainerView = Mock(return_value=container)
si_mock.RetrieveContent.return_value.viewManager = view_manager
assert client.has_virtual_switch("host", "s1") is True
assert client.has_virtual_switch("host", "s3") is False
def test_get_portgroup_by_vswitch(vmware_fixture):
vmware_handler, client, si_mock = vmware_fixture
portgroup1 = Mock(vswitch="ab-cd-switch1")
portgroup2 = Mock(vswitch="ab-cd-switch0")
portgroup3 = Mock(vswitch="ab-cd-switch2")
view1 = Mock(vim.View)
view1.name = "host"
view1.config = Mock(network=Mock(
portgroup=[portgroup1, portgroup2, portgroup3]))
container = Mock(view=[view1])
view_manager = Mock()
view_manager.CreateContainerView = Mock(return_value=container)
si_mock.RetrieveContent.return_value.viewManager = view_manager
assert client.get_portgroup_by_vswitch("host", "switch1") == [portgroup1]
def test_get_host_system(vmware_fixture):
vmware_handler, client, si_mock = vmware_fixture
view1 = Mock(vim.View)
view1.name = "name1"
view2 = Mock(vim.View)
view2.name = "host2"
view3 = Mock(vim.View)
view3.name = "name3"
container = Mock(view=[view1, view2, view3])
view_manager = Mock()
view_manager.CreateContainerView = Mock(return_value=container)
si_mock.RetrieveContent.return_value.viewManager = view_manager
actual = client.get_host_system("host2")
si_mock.RetrieveContent.assert_called()
assert actual == view2
def test_has_portgroup(vmware_fixture):
vmware_handler, client, si_mock = vmware_fixture
vnic = Mock()
spec = Mock()
spec.name = "p1"
vnic.spec = spec
view1 = Mock(vim.View)
view1.name = "host"
view1.config = Mock(network=Mock(portgroup=[vnic]))
container = Mock(view=[view1])
view_manager = Mock()
view_manager.CreateContainerView = Mock(return_value=container)
si_mock.RetrieveContent.return_value.viewManager = view_manager
assert client.has_portgroup("host", "p1") is True
assert client.has_portgroup("host", "p187") is False
def test_add_host_portgroup(vmware_fixture):
vmware_handler, client, si_mock = vmware_fixture
view1 = Mock(vim.View)
view1.name = "host"
config_manager = Mock()
network_system = Mock()
network_system.AddPortGroup = Mock(return_value=Mock())
config_manager.networkSystem = network_system
view1.configManager = config_manager
container = Mock(view=[view1])
view_manager = Mock()
view_manager.CreateContainerView = Mock(return_value=container)
si_mock.RetrieveContent.return_value.viewManager = view_manager
portgroup_spec = Mock()
# when
client.add_host_portgroup("host", portgroup_spec)
# then
network_system.AddPortGroup.assert_called_with(portgroup_spec)
