changes for my endpoints

2021-03-12 10:22:23 +00:00
parent 6862d62bf5
commit 37f222e6c9
2 changed files with 13 additions and 1 deletions
--- a/products/tests.py
+++ b/products/tests.py
@@ -1004,6 +1004,18 @@ class MyProductsViewTest(APITestCase):
        # check response
        self.assertEqual(response.status_code, status.HTTP_401_UNAUTHORIZED)

+    def test_auth_user_without_company(self):
+        # Authenticate
+        token = get_tokens_for_user(self.user)
+        self.client.credentials(HTTP_AUTHORIZATION=f"Bearer {token['access']}")
+
+        # Query endpoint
+        response = self.client.get(self.endpoint)
+        payload = response.json()
+        # Assert forbidden code
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.assertEquals([], payload)
+

 class AdminProductViewSetTest(APITestCase):