def test_add_firewall_rule_exists():
mock_client = get_mock_client()
driver = EdgeGatewayDriver(mock_client, 'test-name')
driver.load()
# Add rule.
driver.add_firewall_rule('test-rule-one', 'TCP', 'any', 80, 'any')
# Add rule again.
with nose.tools.assert_raises(errors.VCloudResourceConflict):
driver.add_firewall_rule('test-rule-one', 'TCP', 'any', 80, 'any')
def test_add_firewall_rule():
mock_client = get_mock_client()
driver = EdgeGatewayDriver(mock_client, 'test-name')
driver.load()
# No existing rules.
driver.add_firewall_rule('test-rule-one', 'TCP', 'any', 80, 'any')
assert hasattr(driver.config, 'FirewallService')
assert hasattr(driver.config.FirewallService, 'FirewallRule')
# Existing rules.
driver.add_firewall_rule('test-rule-two', 'TCP', 'any', 80, 'any')
assert len(driver.config.FirewallService.FirewallRule) == 2
