def test_purge_api_key(self): with fake_creds('.jovian', 'credentials.json'): assert read_cred(API_TOKEN_KEY) == 'fake_api_key' purge_api_key() with self.assertRaises(KeyError): read_cred(API_TOKEN_KEY)
def test_purge_api_key(): with fake_creds(): assert read_cred(API_TOKEN_KEY) == 'fake_api_key' purge_api_key() with pytest.raises(KeyError): read_cred(API_TOKEN_KEY)
def _request_wrapper(*args, **kwargs): for i in range(2): res = request(*args, **kwargs) if res.status_code == 401: log('The current API key is invalid or expired.', error=True) purge_api_key() # This will ensure that fresh api token is requested if 'headers' in kwargs: kwargs['headers'][ 'Authorization'] = "Bearer " + get_api_key() else: return res return res
def test_read_or_request_api_key_none(mock_prompt): with fake_creds(): purge_api_key() assert read_or_request_api_key() == ('fake_api_key', 'request')
def test_read_api_key_opt_none(): with fake_creds(): purge_api_key() assert read_api_key_opt() == (None, 'read')
def test_creds_exist_false(): with fake_creds(): purge_api_key() assert creds_exist() == False
def test_read_or_request_api_key_none(mock_prompt): with fake_creds('.jovian', 'credentials.json'): purge_api_key() assert read_or_request_api_key() == ('fake_api_key', 'request')
def test_read_api_key_opt_none(): with fake_creds('.jovian', 'credentials.json'): purge_api_key() assert read_api_key_opt() == (None, 'read')
def test_creds_exist_false(): with fake_creds('.jovian', 'credentials.json'): purge_api_key() assert creds_exist() == False