def test_create_from_invalid_jwt_token_no_token(session):
"""Assert User is not created from an empty token."""
token = b'invalidtoken'
with pytest.raises(AttributeError) as excinfo:
User.create_from_jwt_token(token)
assert excinfo.value.args[0] == "'bytes' object has no attribute 'get'"
def test_create_from_jwt_token(session):
"""Assert User is created from the JWT fields."""
token = {'username': '******',
'given_name': 'given_name',
'family_name': 'family_name',
'iss': 'iss',
'sub': 'sub'
}
u = User.create_from_jwt_token(token)
assert u.id is not None
def test_save_filing_with_colin_id(session):
"""Assert that saving a filing from the coops-updater-job user is set to paid and source is colin."""
from legal_api.models import Filing
# setup
filing = Filing()
filing.filing_json = ANNUAL_REPORT
user = User.create_from_jwt_token({'username': '******', 'iss': 'test', 'sub': 'test'})
filing.submitter_id = user.id
filing.save()
# test
assert filing.source == Filing.Source.COLIN.value
assert filing.status == Filing.Status.PAID.value
def test_create_from_jwt_token_no_token(session):
"""Assert User is not created from an empty token."""
token = None
u = User.create_from_jwt_token(token)
assert u is None
