def _save_token(token, db_session=None, farm=None):
logging.debug("Saving new token for farm: " + str(farm.id))
token_in = FarmTokenCreate(farm_id=farm.id, **token)
# Make sure we have a DB session and Farm object.
if db_session is not None and farm is not None:
# Update the farm token if it exists.
if farm.token is not None:
update_farm_token(db_session, farm.token, token_in)
else:
create_farm_token(db_session, token_in)
def _save_token(token, db=None, farm=None):
logging.debug("Saving new token for farm: " + str(farm.id))
token_in = FarmTokenCreate(farm_id=farm.id, **token)
# Make sure we have a DB session and Farm object.
if db is not None and farm is not None:
# Update the farm token if it exists.
if farm.token is not None:
update_farm_token(db, farm.token, token_in)
else:
create_farm_token(db, token_in)
# Update the Farm.scope attribute based on what the server returned.
if 'scope' in token:
crud.farm.update_scope(db, farm=farm, scope=token['scope'])
def test_create_farm_token(db: Session, test_farm):
token = FarmTokenCreate(farm_id=test_farm.id,
access_token=random_lower_string(),
expires_in=random_lower_string(),
refresh_token=random_lower_string(),
expires_at=1581363344.0651991)
# Check for existing token
old_token = farm_token.get_farm_token(db, test_farm.id)
if old_token is None:
farm_token.create_farm_token(db, token=token)
else:
farm_token.update_farm_token(db, token=old_token, token_in=token)
db_token = farm_token.get_farm_token(db, farm_id=test_farm.id)
assert db_token.farm_id == token.farm_id == test_farm.id
assert db_token.access_token == token.access_token
assert db_token.expires_in == token.expires_in
assert db_token.refresh_token == token.refresh_token
assert float(db_token.expires_at) == token.expires_at