Merge pull request #851 from adamtheturtle/auth-failure

Add AuthFailure exception

Author: adamtheturtle

dev-requirements.txt

@@ -1,6 +1,6 @@
 PyYAML==3.13
 Pygments==2.2.0
-VWS-Python-Mock==2018.10.01.1
+VWS-Python-Mock==2018.10.02.0
 autoflake==1.2
 check-manifest==0.37
 codecov==2.0.15 # Upload coverage data

src/vws/exceptions.py

@@ -50,6 +50,21 @@ def __init__(self, response: Response) -> None:
         self.response = response
 
 
+class AuthenticationFailure(Exception):
+    """
+    Exception raised when Vuforia returns a response with a result code
+    'AuthenticationFailure'.
+    """
+
+    def __init__(self, response: Response) -> None:
+        """
+        Args:
+            response: The response to a request to Vuforia.
+        """
+        super().__init__()
+        self.response = response
+
+
 class TargetStatusProcessing(Exception):
     """
     Exception raised when Vuforia returns a response with a result code

src/vws/vws.py

@@ -15,6 +15,7 @@
 
 from vws._authorization import authorization_header, rfc_1123_date
 from vws.exceptions import (
+    AuthenticationFailure,
     BadImage,
     Fail,
     ImageTooLarge,
@@ -102,6 +103,7 @@ def _raise_for_result_code(
         return
 
     exception = {
+        'AuthenticationFailure': AuthenticationFailure,
         'BadImage': BadImage,
         'Fail': Fail,
         'ImageTooLarge': ImageTooLarge,

tests/test_exceptions.py

@@ -14,6 +14,7 @@
 
 from vws import VWS
 from vws.exceptions import (
+    AuthenticationFailure,
     BadImage,
     Fail,
     ImageTooLarge,
@@ -97,7 +98,7 @@ def test_request_quota_reached() -> None:
 
 def test_fail(high_quality_image: io.BytesIO) -> None:
     """
-    A ``Fail`` exception is raised when there are authentication issues.
+    A ``Fail`` exception is raised when the server access key does not exist.
     """
     with MockVWS():
         client = VWS(
@@ -201,3 +202,26 @@ def test_metadata_too_large(
         )
 
     assert exc.value.response.status_code == codes.UNPROCESSABLE_ENTITY
+
+
+def test_authentication_failure(high_quality_image: io.BytesIO) -> None:
+    """
+    An ``AuthenticationFailure`` exception is raised when the server access key
+    does not exist.
+    """
+    database = VuforiaDatabase()
+    with MockVWS() as mock:
+        mock.add_database(database=database)
+        client = VWS(
+            server_access_key=database.server_access_key.decode(),
+            server_secret_key='a',
+        )
+
+        with pytest.raises(AuthenticationFailure) as exc:
+            client.add_target(
+                name='x',
+                width=1,
+                image=high_quality_image,
+            )
+
+        assert exc.value.response.status_code == codes.UNAUTHORIZED