feat: raise AuthenticationError

Signed-off-by: Charles Larivier <charles@dribbble.com>

Author: Charles Larivier

src/metabase/exceptions.py

@@ -1,2 +1,6 @@
 class NotFoundError(Exception):
     pass
+
+
+class AuthenticationError(Exception):
+    pass

src/metabase/metabase.py

@@ -2,6 +2,8 @@
 
 import requests
 
+from metabase.exceptions import AuthenticationError
+
 
 class Singleton(type):
     _instances = WeakValueDictionary()
@@ -40,6 +42,10 @@ def token(self):
                 self.host + "/api/session",
                 json={"username": self.user, "password": self.password},
             )
+
+            if response.status_code != 200:
+                raise AuthenticationError(response.content.decode())
+
             self._token = response.json()["id"]
 
         return self._token

tests/test_metabase.py

@@ -1,7 +1,7 @@
 from unittest import TestCase
 
 from metabase import Metabase
-
+from metabase.exceptions import AuthenticationError
 from tests.helpers import IntegrationTestCase
 
 
@@ -27,11 +27,18 @@ def test_host(self):
 
     def test_token(self):
         """Ensure Metabase.token returns Metabase._token if not None, else gets a new token."""
-        metabase = Metabase(host="example.co.", user="", password="", token="123")
+        metabase = Metabase(host="example.com", user="", password="", token="123")
         self.assertEqual(metabase.token, "123")
 
         # TODO: add test case when token is None
 
+    def test_token_invalid_auth(self):
+        """Ensure Metabase.token raises AuthenticationException is the user or password is invalid."""
+        metabase = Metabase(host="http://0.0.0.0:3000", user="", password="")
+
+        with self.assertRaises(AuthenticationError):
+            _ = metabase.token
+
     def test_headers(self):
         """Ensure Metabase.headers returns a dictionary with the token."""
         metabase = Metabase(host="example.com", user="", password="", token="123")