Avoids unnecessary calls when pytest API fixture fails

Geotab · Oct 10, 2024 · 9262637 · 9262637
1 parent 10d5d55
commit 9262637
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tests/test_api_async.py b/tests/test_api_async.py
@@ -38,8 +38,8 @@ def async_populated_api():
         session = API(USERNAME, password=PASSWORD, database=DATABASE, server=SERVER, cert=cert)
         try:
             session.authenticate()
-        except MyGeotabException as exception:
-            pytest.fail(exception)
+        except (MyGeotabException, AuthenticationException) as exception:
+            pytest.fail(str(exception))
             return
         yield session
     else:

diff --git a/tests/test_api_call.py b/tests/test_api_call.py
@@ -48,8 +48,8 @@ def populated_api():
         session = api.API(USERNAME, password=PASSWORD, database=DATABASE, server=SERVER, cert=cert)
         try:
             session.authenticate()
-        except api.MyGeotabException as exception:
-            pytest.fail(exception)
+        except (api.MyGeotabException, api.AuthenticationException) as exception:
+            pytest.fail(str(exception))
             return
         yield session
     else: