def test_credentials_signed_session(self, mock_requests): creds = mock.create_autospec(InteractiveCredentials) creds._parse_token = lambda: AADMixin._parse_token(creds) creds.id = 'client_id' creds.token_uri = "token_uri" creds.resource = "resource" creds.token = {'expires_at':'1', 'expires_in':'2', 'refresh_token':"test"} AADMixin.signed_session(creds) mock_requests.OAuth2Session.assert_called_with( 'client_id', token=creds.token, auto_refresh_url='token_uri', auto_refresh_kwargs={'client_id':'client_id', 'resource':'resource'}, token_updater=creds._default_token_cache)
def test_credentials_signed_session(self, mock_requests): creds = mock.create_autospec(InteractiveCredentials) creds._parse_token = lambda: AADMixin._parse_token(creds) creds.id = 'client_id' creds.token_uri = "token_uri" creds.resource = "resource" creds.token = {'expires_at':'1', 'expires_in':'2', 'refresh_token':"test"} AADMixin.signed_session(creds) mock_requests.OAuth2Session.assert_called_with( 'client_id', token=creds.token, auto_refresh_url='token_uri', auto_refresh_kwargs={'client_id':'client_id', 'resource':'resource'}, token_updater=creds._default_token_cache)
def test_credentials_get_stored_auth(self, mock_keyring): mix = AADMixin() mix.cred_store = "store_name" mix.store_key = "client_id" mix.signed_session = mock.Mock() mock_keyring.get_password.return_value = None with self.assertRaises(ValueError): mix._retrieve_stored_token() mock_keyring.get_password.assert_called_with( "store_name", "client_id") mock_keyring.get_password.return_value = str( {'token_type':'1', 'access_token':'2'}) mix._retrieve_stored_token() mock_keyring.get_password.assert_called_with("store_name", "client_id")
def test_credentials_get_stored_auth(self, mock_keyring): mix = AADMixin() mix.cred_store = "store_name" mix.store_key = "client_id" mix.signed_session = mock.Mock() mock_keyring.get_password.return_value = None with self.assertRaises(ValueError): mix._retrieve_stored_token() mock_keyring.get_password.assert_called_with( "store_name", "client_id") mock_keyring.get_password.return_value = str( {'token_type':'1', 'access_token':'2'}) mix._retrieve_stored_token() mock_keyring.get_password.assert_called_with("store_name", "client_id")