def test_get_not_found(self):
"""
Test that getting a non-existent domain raises a StrongarmHttpError.
"""
name = 'non-existent.example.com'
msg = 'The domain does not exist.'
responses.add(responses.GET, strongarm.host + Domain.endpoint + name + '/',
status=404, content_type='application/json',
body=json.dumps({'detail': msg}))
with self.assertRaises(strongarm.StrongarmHttpError) as exp:
Domain.get(name)
self.assertEqual(len(responses.calls), 1)
self.assertEqual(exp.exception.status_code, 404)
self.assertEqual(exp.exception.detail, msg)
def test_get_exists(self):
"""
Test that getting an existing blackholed domain returns an instance of
Domain with the right attributes.
"""
name = self.get_response['name']
responses.add(responses.GET, strongarm.host + Domain.endpoint + name + '/',
body=json.dumps(self.get_response),
content_type='application/json/')
domain = Domain.get(name)
self.assertEqual(len(responses.calls), 1)
self.assertIsInstance(domain, Domain)
self.assertEqual(domain.name, name)
def test_delete_success(self):
"""
Test that deleting a domain succeeds silently.
"""
name = self.get_response['name']
responses.add(responses.GET, strongarm.host + Domain.endpoint + name + '/',
body=json.dumps(self.get_response),
content_type='application/json/')
responses.add(responses.DELETE, strongarm.host + Domain.endpoint + name + '/',
status=204)
# First get the domain to be deleted
domain = Domain.get(name)
self.assertEqual(len(responses.calls), 1)
# Now delete the domain.
result = domain.delete()
self.assertEqual(len(responses.calls), 2)
self.assertIsNone(result)
