def test_add_pool_exists(): mock_client = get_mock_client() driver = EdgeGatewayDriver(mock_client, 'test-name') driver.load() mock_service_port = objectify.Element('ServicePort') mock_member = objectify.Element('Member') # Add pool. driver.add_pool('test-pool-one', [mock_service_port], [mock_member]) # Add pool again. with nose.tools.assert_raises(errors.VCloudResourceConflict): driver.add_pool('test-pool-one', [mock_service_port], [mock_member])
def test_add_pool(): mock_client = get_mock_client() driver = EdgeGatewayDriver(mock_client, 'test-name') driver.load() # No existing pools. mock_service_port = objectify.Element('ServicePort') mock_member = objectify.Element('Member') driver.add_pool('test-pool-one', [mock_service_port], [mock_member]) assert hasattr(driver.config, 'LoadBalancerService') assert hasattr(driver.config.LoadBalancerService, 'Pool') # Existing rules. driver.add_pool('test-pool-two', [mock_service_port], [mock_member]) assert len(driver.config.LoadBalancerService.Pool) == 2