# Add rules to a firewall
from oneandone.client import OneAndOneService
from oneandone.client import FirewallPolicyRule
client = OneAndOneService('675fbe491b27896b57e76867604f8255')
new_rule1 = FirewallPolicyRule(protocol='TCP', port_from=90, port_to=90,
source='0.0.0.0')
new_rule2 = FirewallPolicyRule(protocol='TCP', port_from=333, port_to=333,
source='0.0.0.0')
new_rules = [new_rule1, new_rule2]
response = client.add_firewall_policy_rule(firewall_id='',
firewall_policy_rules=new_rules)
# Attach servers to a firewall
from oneandone.client import OneAndOneService
from oneandone.client import AttachServer
client = OneAndOneService('675fbe491b27896b57e76867604f8255')
server1 = AttachServer(server_ip_id='')
server2 = AttachServer(server_ip_id='')
servers = [server1, server2]
response = client.attach_server_firewall_policy(firewall_id='',
class TestFirewallPolicy(unittest.TestCase):
def setUp(self):
self.client = OneAndOneService('USER-API-KEY')
# 'GET' Methods
@responses.activate
def test_list_policies(self):
with open('mock-api/list-firewalls.json') as f:
data = json.load(f)
test_id = data[0]['id']
responses.add(responses.GET,
'https://cloudpanel-api.1and1.com/v1/firewall_policies',
body=json.dumps(data),
status=200,
content_type="application/json")
r = self.client.list_firewall_policies()
self.assertEqual(r[0]['id'], test_id)
@responses.activate
def test_get_policy(self):
with open('mock-api/get-firewall.json') as f:
data = json.load(f)
firewall_id = data['id']
responses.add(
responses.GET,
'https://cloudpanel-api.1and1.com/v1/firewall_policies/%s' %
firewall_id,
body=json.dumps(data),
status=200,
content_type="application/json")
r = self.client.get_firewall(firewall_id=firewall_id)
self.assertEqual(r['id'], firewall_id)
@responses.activate
def test_list_firewall_servers(self):
with open('mock-api/list-server-ips-fp.json') as f:
data = json.load(f)
firewall_id = 'FIREWALL_ID'
server_ip_id = data[0]['id']
responses.add(
responses.GET,
'https://cloudpanel-api.1and1.com/v1/firewall_policies/%s/server_ips'
% firewall_id,
body=json.dumps(data),
status=200,
content_type="application/json")
r = self.client.list_firewall_servers(firewall_id=firewall_id)
self.assertEqual(r[0]['id'], server_ip_id)
@responses.activate
def test_get_firewall_server(self):
with open('mock-api/get-server-ip-fp.json') as f:
data = json.load(f)
firewall_id = 'FIREWALL_ID'
server_ip_id = data['id']
responses.add(
responses.GET,
'https://cloudpanel-api.1and1.com/v1/firewall_policies/%s/server_ips/%s'
% (firewall_id, server_ip_id),
body=json.dumps(data),
status=200,
content_type="application/json")
r = self.client.get_firewall_server(firewall_id=firewall_id,
server_ip_id=server_ip_id)
self.assertEqual(r['id'], server_ip_id)
@responses.activate
def test_list_policy_rules(self):
with open('mock-api/list-fp-rules.json') as f:
data = json.load(f)
firewall_id = 'FIREWALL_ID'
rule_id = data[0]['id']
responses.add(
responses.GET,
'https://cloudpanel-api.1and1.com/v1/firewall_policies/%s/rules' %
firewall_id,
body=json.dumps(data),
status=200,
content_type="application/json")
r = self.client.list_firewall_policy_rules(firewall_id=firewall_id)
self.assertEqual(r[0]['id'], rule_id)
@responses.activate
def test_get_policy_rule(self):
with open('mock-api/get-fp-rule.json') as f:
data = json.load(f)
firewall_id = 'FIREWALL_ID'
rule_id = data['id']
responses.add(
responses.GET,
'https://cloudpanel-api.1and1.com/v1/firewall_policies/%s/rules/%s'
% (firewall_id, rule_id),
body=json.dumps(data),
status=200,
content_type="application/json")
r = self.client.get_firewall_policy_rule(firewall_id=firewall_id,
rule_id=rule_id)
self.assertEqual(r['id'], rule_id)
# 'PUT' Methods
@responses.activate
def test_modify_firewall(self):
with open('mock-api/modify-fp.json') as f:
data = json.load(f)
firewall_id = data['id']
name = data['name']
description = data['description']
responses.add(
responses.PUT,
'https://cloudpanel-api.1and1.com/v1/firewall_policies/%s' %
firewall_id,
body=json.dumps(data),
status=200,
content_type="application/json")
r = self.client.modify_firewall(firewall_id=firewall_id,
description=description)
self.assertEqual(r['name'], name)
self.assertEqual(r['description'], description)
# 'POST' Methods
@responses.activate
def test_create_firewall_policy(self):
with open('mock-api/create-fp.json') as f:
data = json.load(f)
fp1 = FirewallPolicy(name=data['name'],
description=data['description'])
rule1 = FirewallPolicyRule(protocol=data['rules'][0]['protocol'],
port_from=data['rules'][0]['port_from'],
port_to=data['rules'][0]['port_to'],
source=data['rules'][0]['source'])
rule2 = FirewallPolicyRule(protocol=data['rules'][1]['protocol'],
port_from=data['rules'][1]['port_from'],
port_to=data['rules'][1]['port_to'],
source=data['rules'][1]['source'])
rules = [rule1, rule2]
responses.add(responses.POST,
'https://cloudpanel-api.1and1.com/v1/firewall_policies',
body=json.dumps(data),
status=202,
content_type="application/json")
r = self.client.create_firewall_policy(firewall_policy=fp1,
firewall_policy_rules=rules)
self.assertEqual(r['name'], fp1.specs['name'])
self.assertEqual(r['description'], fp1.specs['description'])
self.assertEqual(r['rules'][0]['protocol'], rule1.rule_set['protocol'])
self.assertEqual(r['rules'][1]['protocol'], rule2.rule_set['protocol'])
@responses.activate
def test_assign_servers(self):
with open('mock-api/assign-ip-fp.json') as f:
data = json.load(f)
firewall_id = data['id']
server1 = AttachServer(server_ip_id=data['server_ips'][0]['id'])
servers = [server1]
responses.add(
responses.POST,
'https://cloudpanel-api.1and1.com/v1/firewall_policies/%s/server_ips'
% firewall_id,
body=json.dumps(data),
status=202,
content_type="application/json")
r = self.client.attach_server_firewall_policy(firewall_id=firewall_id,
server_ips=servers)
self.assertEqual(r['id'], firewall_id)
self.assertEqual(r['server_ips'][0]['id'], server1.server_ip_id)
@responses.activate
def test_add_rule(self):
with open('mock-api/add-rule-fp.json') as f:
data = json.load(f)
firewall_id = data['id']
rule1 = FirewallPolicyRule(protocol=data['rules'][2]['protocol'],
port_from=data['rules'][2]['port_from'],
port_to=data['rules'][2]['port_to'],
source=data['rules'][2]['source'])
rules = [rule1]
responses.add(
responses.POST,
'https://cloudpanel-api.1and1.com/v1/firewall_policies/%s/rules' %
firewall_id,
body=json.dumps(data),
status=202,
content_type="application/json")
r = self.client.add_firewall_policy_rule(firewall_id=firewall_id,
firewall_policy_rules=rules)
self.assertEqual(r['rules'][2]['protocol'], rule1.rule_set['protocol'])
self.assertEqual(r['rules'][2]['port_from'],
rule1.rule_set['port_from'])
self.assertEqual(r['rules'][2]['port_to'], rule1.rule_set['port_to'])
self.assertEqual(r['rules'][2]['source'], rule1.rule_set['source'])
# 'DELETE' Methods
@responses.activate
def test_remove_server_firewall(self):
with open('mock-api/remove-ip-fp.json') as f:
data = json.load(f)
firewall_id = 'FIREWALL_ID'
server_ip_id = 'SERVER_IP_ID'
responses.add(
responses.DELETE,
'https://cloudpanel-api.1and1.com/v1/firewall_policies/%s/server_ips/%s'
% (firewall_id, server_ip_id),
body=json.dumps(data),
status=202,
content_type="application/json")
r = self.client.remove_firewall_server(firewall_id=firewall_id,
server_ip_id=server_ip_id)
self.assertEqual(r['state'], 'CONFIGURING')
self.assertEqual(r['server_ips'], [])
@responses.activate
def test_remove_firewall_rule(self):
with open('mock-api/delete-rule-fp.json') as f:
data = json.load(f)
firewall_id = 'FIREWALL_ID'
rule_id = 'RULE_ID'
responses.add(
responses.DELETE,
'https://cloudpanel-api.1and1.com/v1/firewall_policies/%s/rules/%s'
% (firewall_id, rule_id),
body=json.dumps(data),
status=202,
content_type="application/json")
r = self.client.remove_firewall_rule(firewall_id=firewall_id,
rule_id=rule_id)
self.assertNotEqual(len(r['rules']), 3)
@responses.activate
def test_delete_firewall(self):
with open('mock-api/delete-fp.json') as f:
data = json.load(f)
firewall_id = 'FIREWALL_ID'
responses.add(
responses.DELETE,
'https://cloudpanel-api.1and1.com/v1/firewall_policies/%s' %
firewall_id,
body=json.dumps(data),
status=202,
content_type="application/json")
r = self.client.delete_firewall(firewall_id=firewall_id)
self.assertEqual(r['state'], 'REMOVING')
# List all servers attached to a firewall
servers = client.list_firewall_servers(firewall_id=new_firewall['id'])
# Retrieve a server attached to a firewall
server = client.get_firewall_server(firewall_id=new_firewall['id'],
server_ip_id=serverResp['ips'][0]['id'])
# Add rules to a firewall
new_rule1 = FirewallPolicyRule(protocol='TCP', port=90, source='0.0.0.0')
new_rule2 = FirewallPolicyRule(protocol='TCP', port=333, source='0.0.0.0')
new_rules = [new_rule1, new_rule2]
new_rule = client.add_firewall_policy_rule(firewall_id=new_firewall['id'],
firewall_policy_rules=new_rules)
# List all rules of a firewall
rules = client.list_firewall_policy_rules(firewall_id=new_firewall['id'])
# Retrieve a firewall rule
rule = client.get_firewall_policy_rule(firewall_id=new_firewall['id'],
rule_id=new_rule['rules'][0]['id'])
# Modify a firewall
modified_firewall = client.modify_firewall(firewall_id=new_firewall['id'],
name='New Name',
description='New Description')
# Remove a firewall rule
response = client.remove_firewall_rule(firewall_id=new_firewall['id'],
client = OneAndOneService('675fbe491b27896b57e76867604f8255')
new_rule1 = FirewallPolicyRule(protocol='TCP',
port_from=90,
port_to=90,
source='0.0.0.0')
new_rule2 = FirewallPolicyRule(protocol='TCP',
port_from=333,
port_to=333,
source='0.0.0.0')
new_rules = [new_rule1, new_rule2]
response = client.add_firewall_policy_rule(firewall_id='',
firewall_policy_rules=new_rules)
# Attach servers to a firewall
from oneandone.client import OneAndOneService
from oneandone.client import AttachServer
client = OneAndOneService('675fbe491b27896b57e76867604f8255')
server1 = AttachServer(server_ip_id='')
server2 = AttachServer(server_ip_id='')
servers = [server1, server2]
response = client.attach_server_firewall_policy(firewall_id='',
server_ips=servers)
class TestFirewallPolicy(unittest.TestCase):
def setUp(self):
self.client = OneAndOneService('USER-API-KEY')
# 'GET' Methods
@responses.activate
def test_list_policies(self):
with open('mock-api/list-firewalls.json') as f:
data = json.load(f)
test_id = data[0]['id']
responses.add(responses.GET, 'https://cloudpanel-api.1and1.com/v1/firewall_policies',
body=json.dumps(data), status=200,
content_type="application/json")
r = self.client.list_firewall_policies()
self.assertEqual(r[0]['id'], test_id)
@responses.activate
def test_get_policy(self):
with open('mock-api/get-firewall.json') as f:
data = json.load(f)
firewall_id = data['id']
responses.add(responses.GET, 'https://cloudpanel-api.1and1.com/v1/firewall_policies/%s' % firewall_id,
body=json.dumps(data), status=200,
content_type="application/json")
r = self.client.get_firewall(firewall_id=firewall_id)
self.assertEqual(r['id'], firewall_id)
@responses.activate
def test_list_firewall_servers(self):
with open('mock-api/list-server-ips-fp.json') as f:
data = json.load(f)
firewall_id = 'FIREWALL_ID'
server_ip_id = data[0]['id']
responses.add(responses.GET, 'https://cloudpanel-api.1and1.com/v1/firewall_policies/%s/server_ips' % firewall_id,
body=json.dumps(data), status=200,
content_type="application/json")
r = self.client.list_firewall_servers(firewall_id=firewall_id)
self.assertEqual(r[0]['id'], server_ip_id)
@responses.activate
def test_get_firewall_server(self):
with open('mock-api/get-server-ip-fp.json') as f:
data = json.load(f)
firewall_id = 'FIREWALL_ID'
server_ip_id = data['id']
responses.add(responses.GET, 'https://cloudpanel-api.1and1.com/v1/firewall_policies/%s/server_ips/%s' % (firewall_id, server_ip_id),
body=json.dumps(data), status=200,
content_type="application/json")
r = self.client.get_firewall_server(firewall_id=firewall_id, server_ip_id=server_ip_id)
self.assertEqual(r['id'], server_ip_id)
@responses.activate
def test_list_policy_rules(self):
with open('mock-api/list-fp-rules.json') as f:
data = json.load(f)
firewall_id = 'FIREWALL_ID'
rule_id = data[0]['id']
responses.add(responses.GET, 'https://cloudpanel-api.1and1.com/v1/firewall_policies/%s/rules' % firewall_id,
body=json.dumps(data), status=200,
content_type="application/json")
r = self.client.list_firewall_policy_rules(firewall_id=firewall_id)
self.assertEqual(r[0]['id'], rule_id)
@responses.activate
def test_get_policy_rule(self):
with open('mock-api/get-fp-rule.json') as f:
data = json.load(f)
firewall_id = 'FIREWALL_ID'
rule_id = data['id']
responses.add(responses.GET, 'https://cloudpanel-api.1and1.com/v1/firewall_policies/%s/rules/%s' % (firewall_id, rule_id),
body=json.dumps(data), status=200,
content_type="application/json")
r = self.client.get_firewall_policy_rule(firewall_id=firewall_id, rule_id=rule_id)
self.assertEqual(r['id'], rule_id)
# 'PUT' Methods
@responses.activate
def test_modify_firewall(self):
with open('mock-api/modify-fp.json') as f:
data = json.load(f)
firewall_id = data['id']
name = data['name']
description = data['description']
responses.add(responses.PUT, 'https://cloudpanel-api.1and1.com/v1/firewall_policies/%s' % firewall_id,
body=json.dumps(data), status=200,
content_type="application/json")
r = self.client.modify_firewall(firewall_id=firewall_id, description=description)
self.assertEqual(r['name'], name)
self.assertEqual(r['description'], description)
# 'POST' Methods
@responses.activate
def test_create_firewall_policy(self):
with open('mock-api/create-fp.json') as f:
data = json.load(f)
fp1 = FirewallPolicy(name=data['name'], description=data['description'])
rule1 = FirewallPolicyRule(protocol=data['rules'][0]['protocol'], port_from=data['rules'][0]['port_from'], port_to=data['rules'][0]['port_to'],
source=data['rules'][0]['source'])
rule2 = FirewallPolicyRule(protocol=data['rules'][1]['protocol'], port_from=data['rules'][1]['port_from'], port_to=data['rules'][1]['port_to'],
source=data['rules'][1]['source'])
rules = [rule1, rule2]
responses.add(responses.POST, 'https://cloudpanel-api.1and1.com/v1/firewall_policies',
body=json.dumps(data), status=202,
content_type="application/json")
r = self.client.create_firewall_policy(firewall_policy=fp1, firewall_policy_rules=rules)
self.assertEqual(r['name'], fp1.specs['name'])
self.assertEqual(r['description'], fp1.specs['description'])
self.assertEqual(r['rules'][0]['protocol'], rule1.rule_set['protocol'])
self.assertEqual(r['rules'][1]['protocol'], rule2.rule_set['protocol'])
@responses.activate
def test_assign_servers(self):
with open('mock-api/assign-ip-fp.json') as f:
data = json.load(f)
firewall_id = data['id']
server1 = AttachServer(server_ip_id=data['server_ips'][0]['id'])
servers = [server1]
responses.add(responses.POST, 'https://cloudpanel-api.1and1.com/v1/firewall_policies/%s/server_ips' % firewall_id,
body=json.dumps(data), status=202,
content_type="application/json")
r = self.client.attach_server_firewall_policy(firewall_id=firewall_id, server_ips=servers)
self.assertEqual(r['id'], firewall_id)
self.assertEqual(r['server_ips'][0]['id'], server1.server_ip_id)
@responses.activate
def test_add_rule(self):
with open('mock-api/add-rule-fp.json') as f:
data = json.load(f)
firewall_id = data['id']
rule1 = FirewallPolicyRule(protocol=data['rules'][2]['protocol'], port_from=data['rules'][2]['port_from'], port_to=data['rules'][2]['port_to'],
source=data['rules'][2]['source'])
rules = [rule1]
responses.add(responses.POST, 'https://cloudpanel-api.1and1.com/v1/firewall_policies/%s/rules' % firewall_id,
body=json.dumps(data), status=202,
content_type="application/json")
r = self.client.add_firewall_policy_rule(firewall_id=firewall_id, firewall_policy_rules=rules)
self.assertEqual(r['rules'][2]['protocol'], rule1.rule_set['protocol'])
self.assertEqual(r['rules'][2]['port_from'], rule1.rule_set['port_from'])
self.assertEqual(r['rules'][2]['port_to'], rule1.rule_set['port_to'])
self.assertEqual(r['rules'][2]['source'], rule1.rule_set['source'])
# 'DELETE' Methods
@responses.activate
def test_remove_server_firewall(self):
with open('mock-api/remove-ip-fp.json') as f:
data = json.load(f)
firewall_id = 'FIREWALL_ID'
server_ip_id = 'SERVER_IP_ID'
responses.add(responses.DELETE, 'https://cloudpanel-api.1and1.com/v1/firewall_policies/%s/server_ips/%s' % (firewall_id, server_ip_id),
body=json.dumps(data), status=202,
content_type="application/json")
r = self.client.remove_firewall_server(firewall_id=firewall_id, server_ip_id=server_ip_id)
self.assertEqual(r['state'], 'CONFIGURING')
self.assertEqual(r['server_ips'], [])
@responses.activate
def test_remove_firewall_rule(self):
with open('mock-api/delete-rule-fp.json') as f:
data = json.load(f)
firewall_id = 'FIREWALL_ID'
rule_id = 'RULE_ID'
responses.add(responses.DELETE, 'https://cloudpanel-api.1and1.com/v1/firewall_policies/%s/rules/%s' % (firewall_id, rule_id),
body=json.dumps(data), status=202,
content_type="application/json")
r = self.client.remove_firewall_rule(firewall_id=firewall_id, rule_id=rule_id)
self.assertNotEqual(len(r['rules']), 3)
@responses.activate
def test_delete_firewall(self):
with open('mock-api/delete-fp.json') as f:
data = json.load(f)
firewall_id = 'FIREWALL_ID'
responses.add(responses.DELETE, 'https://cloudpanel-api.1and1.com/v1/firewall_policies/%s' % firewall_id,
body=json.dumps(data), status=202,
content_type="application/json")
r = self.client.delete_firewall(firewall_id=firewall_id)
self.assertEqual(r['state'], 'REMOVING')