def test_fetch_by_refresh_token_no_data(self): collection_mock = Mock(spec=["find_one"]) collection_mock.find_one.return_value = None store = AccessTokenStore(collection=collection_mock) with self.assertRaises(AccessTokenNotFound): store.fetch_by_refresh_token(refresh_token="abcd")
def test_fetch_by_refresh_token_no_data(self): collection_mock = Mock(spec=["find_one"]) collection_mock.find_one.return_value = None store = AccessTokenStore(collection=collection_mock) with self.assertRaises(AccessTokenNotFound): store.fetch_by_refresh_token(refresh_token="abcd")
def test_fetch_by_refresh_token(self): refresh_token = "abcd" self.access_token_data["refresh_token"] = refresh_token collection_mock = Mock(spec=["find_one"]) collection_mock.find_one.return_value = self.access_token_data store = AccessTokenStore(collection=collection_mock) token = store.fetch_by_refresh_token(refresh_token=refresh_token) collection_mock.find_one.assert_called_with({"refresh_token": refresh_token}) self.assertTrue(isinstance(token, AccessToken)) self.assertDictEqual(token.__dict__, self.access_token_data)
def test_fetch_by_refresh_token(self): refresh_token = "abcd" self.access_token_data["refresh_token"] = refresh_token collection_mock = Mock(spec=["find_one"]) collection_mock.find_one.return_value = self.access_token_data store = AccessTokenStore(collection=collection_mock) token = store.fetch_by_refresh_token(refresh_token=refresh_token) collection_mock.find_one.assert_called_with( {"refresh_token": refresh_token}) self.assertTrue(isinstance(token, AccessToken)) self.assertDictEqual(token.__dict__, self.access_token_data)