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