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)