added package for common API testing logic

shared/selene/api/testing/__init__.py

@@ -0,0 +1 @@
+from .authentication import get_account, validate_token_cookies

shared/selene/api/testing/authentication.py

@@ -0,0 +1,47 @@
+from hamcrest import assert_that, equal_to, has_item
+
+from selene.account import Account, AccountRepository
+from selene.util.db import get_db_connection
+
+ACCESS_TOKEN_COOKIE_KEY = 'seleneAccess'
+REFRESH_TOKEN_COOKIE_KEY = 'seleneRefresh'
+
+
+def validate_token_cookies(context, expired=False):
+    for cookie in context.response.headers.getlist('Set-Cookie'):
+        ingredients = _parse_cookie(cookie)
+        ingredient_names = list(ingredients.keys())
+        if ACCESS_TOKEN_COOKIE_KEY in ingredient_names:
+            context.access_token = ingredients[ACCESS_TOKEN_COOKIE_KEY]
+        elif REFRESH_TOKEN_COOKIE_KEY in ingredient_names:
+            context.refresh_token = ingredients[REFRESH_TOKEN_COOKIE_KEY]
+        for ingredient_name in ('Domain', 'Expires', 'Max-Age'):
+            assert_that(ingredient_names, has_item(ingredient_name))
+        if expired:
+            assert_that(ingredients['Max-Age'], equal_to('0'))
+
+    assert hasattr(context, 'access_token'), 'no access token in response'
+    assert hasattr(context, 'refresh_token'), 'no refresh token in response'
+    if expired:
+        assert_that(context.access_token, equal_to(''))
+        assert_that(context.refresh_token, equal_to(''))
+
+
+def _parse_cookie(cookie: str) -> dict:
+    ingredients = {}
+    for ingredient in cookie.split('; '):
+        if '=' in ingredient:
+            key, value = ingredient.split('=')
+            ingredients[key] = value
+        else:
+            ingredients[ingredient] = None
+
+    return ingredients
+
+
+def get_account(context) -> Account:
+    with get_db_connection(context.db_pool) as db:
+        acct_repository = AccountRepository(db)
+        account = acct_repository.get_account_by_id(context.account.id)
+
+    return account