def test_add_virtal_server_exists(mock_get_network):
mock_network = objectify.Element('NetworkRecord')
mock_network.attrib['name'] = 'test-network'
mock_network.attrib['href'] = 'test-network-href'
mock_get_network.return_value = mock_network
mock_client = get_mock_client()
driver = EdgeGatewayDriver(mock_client, 'test-name')
driver.load()
mock_service_profile = objectify.Element('ServiceProfile')
# Add virtual server.
driver.add_virtual_server(
'test-vs-one', '0.0.0.0', 'test-pool', 'test-network',
[mock_service_profile])
# Add virtual server again.
with nose.tools.assert_raises(errors.VCloudResourceConflict):
driver.add_virtual_server(
'test-vs-one', '0.0.0.1', 'test-pool', 'test-network',
[mock_service_profile])
# Add virtual server with same IP.
with nose.tools.assert_raises(errors.VCloudResourceConflict):
driver.add_virtual_server(
'test-vs-two', '0.0.0.0', 'test-pool', 'test-network',
[mock_service_profile])
def test_add_virtual_server(mock_get_network):
mock_network = objectify.Element('NetworkRecord')
mock_network.attrib['name'] = 'test-network'
mock_network.attrib['href'] = 'test-network-href'
mock_get_network.return_value = mock_network
mock_client = get_mock_client()
driver = EdgeGatewayDriver(mock_client, 'test-name')
driver.load()
# No existing virtual servers.
mock_service_profile = objectify.Element('ServiceProfile')
driver.add_virtual_server(
'test-vs-one', '0.0.0.0', 'test-pool', 'test-network',
[mock_service_profile])
assert hasattr(driver.config, 'LoadBalancerService')
assert hasattr(driver.config.LoadBalancerService, 'VirtualServer')
# Existing virtual servers.
driver.add_virtual_server(
'test-vs-two', '0.0.0.1', 'test-pool', 'test-network',
[mock_service_profile])
assert len(driver.config.LoadBalancerService.VirtualServer) == 2
