mock google auth

Author: rog555

abnosql/plugins/table/cosmos.py

@@ -119,6 +119,7 @@ def strip_cosmos_attrs(item):
 
 class Table(TableBase):
 
+    @cosmos_ex_handler()
     def __init__(
         self, pm: PM, name: str, config: t.Optional[dict] = None
     ) -> None:

abnosql/plugins/table/dynamodb.py

@@ -151,6 +151,7 @@ def get_dynamodb_kwargs(
 
 class Table(TableBase):
 
+    @dynamodb_ex_handler()
     def __init__(
         self, pm: PM, name: str, config: t.Optional[dict] = None
     ) -> None:

abnosql/plugins/table/firestore.py

@@ -5,6 +5,7 @@
 import logging
 import os
 from os.path import expanduser
+import sys
 
 import typing as t
 
@@ -32,7 +33,8 @@
 hookimpl = pluggy.HookimplMarker('abnosql.table')
 
 try:
-    from google.api_core import exceptions as gex  # type: ignore
+    from google.api_core.exceptions import ClientError  # type: ignore
+    from google.auth.exceptions import GoogleAuthError  # type: ignore
     from google.cloud import firestore  # type: ignore
 except ImportError:
     MISSING_DEPS = True
@@ -53,20 +55,23 @@ def decorator(func):
         def wrapper(*args, **kwargs):
             try:
                 return func(*args, **kwargs)
-            except gex.ClientError as e:
+            except ClientError as e:
                 if raise_not_found and e.code in [404]:
                     raise ex.NotFoundException() from None
                 raise ex.ValidationException(detail=e) from None
+            except GoogleAuthError as e:
+                raise ex.ConfigException(detail=e) from None
             except ex.NoSQLException:
                 raise
             except Exception as e:
-                raise ex.PluginException(detail=e)
+                raise ex.PluginException(detail=e) from None
         return wrapper
     return decorator
 
 
 class Table(TableBase):
 
+    @firestore_ex_handler()
     def __init__(
         self, pm: PM, name: str, config: t.Optional[dict] = None
     ) -> None:
@@ -97,14 +102,19 @@ def _get_client(self):
                 if val is not None:
                     kwargs[attr] = val
                 break
-        cred_file = os.path.join(
-            expanduser('~'),
-            '.config',
-            'gcloud',
-            'application_default_credentials.json'
-        )
-        if os.path.isfile(cred_file):
-            os.environ['GOOGLE_APPLICATION_CREDENTIALS'] = cred_file
+        cred_file = os.path.join(*(
+            (
+                tuple(os.environ.get('APPDATA', ''))
+                if sys.platform == 'win32'
+                else (expanduser('~'), '.config')
+            ) + (
+                'gcloud',
+                'application_default_credentials.json'
+            )
+        ))
+        gac = 'GOOGLE_APPLICATION_CREDENTIALS'
+        if gac not in os.environ and os.path.isfile(cred_file):
+            os.environ[gac] = cred_file
         return firestore.Client(**kwargs)
 
     def _docid(self, **kwargs):

abnosql/plugins/table/memory.py

@@ -159,11 +159,13 @@ def wrapper(*args, **kwargs):
 
 class Table(TableBase):
 
+    @memory_ex_handler()
     def __init__(
         self, pm: PM, name: str, config: t.Optional[dict] = None
     ) -> None:
         self.pm = pm
         self.name = name
+        self.database = 'memory'
         self.set_config(config)
         self.key_attrs = get_key_attrs(self.config)
         self.items = self.config.get('items', {})

tests/data/google_mocked_creds.json

@@ -0,0 +1,9 @@
+{
+    "account": "",
+    "client_id": "foobar.apps.googleusercontent.com",
+    "client_secret": "foobar",
+    "refresh_token": "foobar",
+    "type": "authorized_user",
+    "universe_domain": "googleapis.com"
+}
+  

tests/test_firestore.py

@@ -1,11 +1,21 @@
 import os
 
 from mockfirestore import MockFirestore  # type: ignore
+import pytest
 from tests import common as cmn
 
+from abnosql import exceptions as ex
+from abnosql import table
+
+ROOT_DIR = os.path.dirname(os.path.dirname(__file__))
+DATA_DIR = os.path.join(ROOT_DIR, 'tests', 'data')
+
 
 def config(table='hash_range', extra={}):
     os.environ.update({
+        'GOOGLE_APPLICATION_CREDENTIALS': os.path.join(
+            DATA_DIR, 'google_mocked_creds.json'
+        ),
         'ABNOSQL_DB': 'firestore',
         'GOOGLE_CLOUD_PROJECT': 'foo',  # this is to just stop warnings on CLI
         'ABNOSQL_FIRESTORE_DATABASE': 'bar'
@@ -19,6 +29,20 @@ def config(table='hash_range', extra={}):
     return config
 
 
+def test_exceptions():
+    _config = config()
+    os.environ['GOOGLE_APPLICATION_CREDENTIALS'] = 'foobar'
+    with pytest.raises(ex.ConfigException) as e:
+        table('foobar', _config)
+    assert 'invalid config' in str(e.value)
+    assert e.value.to_problem() == {
+        'title': 'invalid config',
+        'detail': 'File foobar was not found.',
+        'status': 500,
+        'type': None
+    }
+
+
 def test_get_item():
     cmn.test_get_item(config('hash_range'), 'hash_range')
     cmn.test_get_item(config('hash_only'), 'hash_only')