Merge pull request #30 from drmobile/develop

Develop

Author: calvin620707

flask_soocii_auth/__init__.py

@@ -112,6 +112,10 @@ def authorize_user():
             g.user = users.AnonymousUser()
             return
 
+        if g.access_token['soocii_id'] == 'soocii_guest':
+            g.user = users.SoociiGuest()
+            return
+
         if g.access_token['uid'] == '':
             g.user = users.BackstageUser()
             return

flask_soocii_auth/users.py

@@ -8,6 +8,10 @@ class AnonymousUser(User):
     pass
 
 
+class SoociiGuest(User):
+    pass
+
+
 class BackstageUser(User):
     @property
     def is_backstage(self):

setup.py

@@ -4,7 +4,7 @@
 
 setup(
     name='soocii-services-lib',
-    version='1.2.1',
+    version='1.2.2',
     packages=find_packages(exclude=['tests']),
     url='https://github.com/drmobile/services-lib',
     license='Apache Software License',

tests/test_flask_soocii_auth.py

@@ -84,6 +84,16 @@ def fake_func(_):
             assert resp.status_code == 200
             assert isinstance(g.user, users.AnonymousUser)
 
+    def test_user_soocii_guest(self):
+        with self.app.test_client() as c:
+            token = auth.generate_access_token(
+                '', uuid.uuid4().hex, 1, uuid.uuid4().hex, soocii_id="soocii_guest"
+            ).decode('utf-8')
+            SoociiAuthenticator(self.app)
+            resp = c.get('/', headers={'Authorization': 'Bearer {}'.format(token)})
+            assert resp.status_code == 200
+            assert isinstance(g.user, users.SoociiGuest)
+
     def test_env_vars_invalid(self):
         os.environ['ACCESS_TOKEN_SECRET'] = 'None'
         os.environ['REFRESH_TOKEN_SECRET'] = 'None'