示例#1
0
    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)
示例#2
0
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)
示例#3
0
    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
示例#4
0
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')
示例#5
0
def test_read_api_key_opt_none():
    with fake_creds():
        purge_api_key()

        assert read_api_key_opt() == (None, 'read')
示例#6
0
def test_creds_exist_false():
    with fake_creds():
        purge_api_key()

        assert creds_exist() == False
示例#7
0
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')
示例#8
0
def test_read_api_key_opt_none():
    with fake_creds('.jovian', 'credentials.json'):
        purge_api_key()

        assert read_api_key_opt() == (None, 'read')
示例#9
0
def test_creds_exist_false():
    with fake_creds('.jovian', 'credentials.json'):
        purge_api_key()

        assert creds_exist() == False