improved duplicate testing code

2026-01-01 04:10:06 -05:00 · 2020-06-17 14:53:00 +02:00
parent 14a5d43dc8
commit 83cc8832cb
2 changed files with 10 additions and 20 deletions
--- a/cookbook/tests/test_setup.py
+++ b/cookbook/tests/test_setup.py
@@ -37,3 +37,9 @@ class TestBase(TestCase):
        user = self.create_login_user('superuser_client', 'admin')
        user.is_superuser = True
        user.save()
+
+    def batch_requests(self, clients, url, method='get', payload={}, content_type=''):
+        for c in clients:
+            if method == 'get':
+                r = c[0].get(url)
+                self.assertEqual(r.status_code, c[1], msg=f'GET request failed for user {auth.get_user(c[0])} when testing url {url}')