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