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