django-pfx 1.4.dev172__tar.gz → 1.4.dev174__tar.gz

{django_pfx-1.4.dev172 → django_pfx-1.4.dev174}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: django-pfx
-Version: 1.4.dev172
+Version: 1.4.dev174
 Summary: Django PFX is a toolkit designed to streamline the development of RESTful APIs using the Django framework.
 Author: Hervé Martinet
 Author-email: herve.martinet@gmail.com

{django_pfx-1.4.dev172 → django_pfx-1.4.dev174}/django_pfx.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: django-pfx
-Version: 1.4.dev172
+Version: 1.4.dev174
 Summary: Django PFX is a toolkit designed to streamline the development of RESTful APIs using the Django framework.
 Author: Hervé Martinet
 Author-email: herve.martinet@gmail.com

{django_pfx-1.4.dev172 → django_pfx-1.4.dev174}/pfx/pfxcore/test.py

@@ -73,21 +73,22 @@ def get_url(path, **params):
 
 
 class APIClient(DjangoClient):
-
-    token = None
-    default_locale = None
-    auth_cookie = None
+    class AuthFailedException(Exception):
+        pass
 
     def __init__(self, enforce_csrf_checks=False, raise_request_exception=True,
                  default_locale=None, with_cookie=False, **defaults):
         super().__init__(enforce_csrf_checks, raise_request_exception,
                          **defaults)
+        self.token = None
+        self.auth_cookie = None
         self.default_locale = default_locale
         self.with_cookie = with_cookie
 
     def login(self, username, password='test',
-              path='/api/auth/login', locale=None, remember_me=False):
-        self.token = None
+              path='/api/auth/login', locale=None, remember_me=False,
+              print_response=False):
+        self.logout()  # Ensure cleaning token and cookies
         response = self.post(
             get_url(path, mode=(self.with_cookie and 'cookie' or 'jwt')),
             dict(
@@ -95,6 +96,8 @@ class APIClient(DjangoClient):
                 password=password,
                 remember_me=remember_me),
             locale=locale)
+        if print_response:
+            response.print()
         if response.status_code == 200:
             if self.with_cookie:
                 self.auth_cookie = get_auth_cookie(response)
@@ -103,10 +106,13 @@ class APIClient(DjangoClient):
             else:
                 self.token = response.json_content['token']
             return True
-        return False
+        raise APIClient.AuthFailedException(
+            f"Authentication failed for user {username} on {path}")
 
     def logout(self):
         self.token = None
+        self.auth_cookie = None
+        self.cookies.clear()
 
     def generic(self, method, path, data='',
                 content_type='application/octet-stream', secure=False,