def test_load(self):
data = self.load_from_file('firewalls/single.json')
url = self.base_url + "firewalls/12345"
responses.add(responses.GET,
url,
body=data,
status=200,
content_type='application/json')
firewall = digitalocean.Firewall(id=12345, token=self.token)
f = firewall.load()
self.assert_get_url_equal(responses.calls[0].request.url, url)
self.assertEqual(f.id, 12345)
self.assertEqual(f.name, "firewall")
self.assertEqual(f.status, "succeeded")
self.assertEqual(f.inbound_rules[0].ports, "80")
self.assertEqual(f.inbound_rules[0].protocol, "tcp")
self.assertEqual(f.inbound_rules[0].sources.load_balancer_uids,
["12345"])
self.assertEqual(f.inbound_rules[0].sources.addresses, [])
self.assertEqual(f.inbound_rules[0].sources.tags, [])
self.assertEqual(f.outbound_rules[0].ports, "80")
self.assertEqual(f.outbound_rules[0].protocol, "tcp")
self.assertEqual(f.outbound_rules[0].destinations.load_balancer_uids,
[])
self.assertEqual(f.outbound_rules[0].destinations.addresses,
["0.0.0.0/0", "::/0"])
self.assertEqual(f.outbound_rules[0].destinations.tags, [])
self.assertEqual(f.created_at, "2017-05-23T21:24:00Z")
self.assertEqual(f.droplet_ids, [12345])
self.assertEqual(f.tags, [])
self.assertEqual(f.pending_changes, [])
def writeFirewall(self, dropletName):
self.dropletName = dropletName
self.createInboundRules()
self.createOutboundRules()
self.digitalOceanFirewall = digitalocean.Firewall(
token=self.token,
name=self.dropletName + "Firewall",
inbound_rules=self.inboundRules,
outbound_rules=self.outboundRules)
self.digitalOceanFirewall.create()
def setUp(self):
super(TestFirewall, self).setUp()
data = self.load_from_file('firewalls/single.json')
url = self.base_url + "firewalls/12345"
responses.add(responses.GET,
url,
body=data,
status=200,
content_type='application/json')
self.firewall = digitalocean.Firewall(id=12345,
token=self.token).load()
def test_load(self):
data = self.load_from_file('firewalls/single.json')
url = self.base_url + "firewalls/12345"
responses.add(responses.GET,
url,
body=data,
status=200,
content_type='application/json')
firewall = digitalocean.Firewall(id=12345, token=self.token)
f = firewall.load()
self.assert_get_url_equal(responses.calls[0].request.url, url)
self.assertEqual(f.id, 12345)
self.assertEqual(f.name, "firewall")
self.assertEqual(f.status, "succeeded")
# TODO: Assert the inbound and outbound rules
self.assertEqual(f.created_at, "2017-05-23T21:24:00Z")
self.assertEqual(f.droplet_ids, [12345])
self.assertEqual(f.tags, [])
self.assertEqual(f.pending_changes, [])