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
