48 lines
1.7 KiB
Python
48 lines
1.7 KiB
Python
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
|