testutils.py 890 B

1234567891011121314151617181920
  1. class ApiTestsMixin(object):
  2. def assertApiResultsAreEmpty(self, response):
  3. self.assertEqual(response.status_code, 200)
  4. self.assertEqual(response.json()['results'], [])
  5. def assertApiResultsEqual(self, response, items):
  6. self.assertEqual(response.status_code, 200)
  7. results_ids = [r['id'] for r in response.json()['results']]
  8. items_ids = [i.id for i in items]
  9. self.assertEqual(results_ids, items_ids)
  10. def assertInApiResults(self, response, item):
  11. self.assertEqual(response.status_code, 200)
  12. results_ids = [r['id'] for r in response.json()['results']]
  13. self.assertIn(item.id, results_ids)
  14. def assertNotInApiResults(self, response, item):
  15. self.assertEqual(response.status_code, 200)
  16. results_ids = [r['id'] for r in response.json()['results']]
  17. self.assertNotIn(item.id, results_ids)