def test_save_token(self): collection_mock = mock.Mock(spec=["insert_one"]) store = MongodbAccessTokenStore(collection=collection_mock) store.save_token(self.access_token) collection_mock.insert_one.assert_called_with(self.access_token)
def test_fetch_by_token(self): collection_mock = mock.Mock(spec=["find_one"]) collection_mock.find_one.return_value = self.access_token store = MongodbAccessTokenStore(collection=collection_mock) access_token = store.fetch_by_token(token=self.access_token.token) collection_mock.find_one.assert_called_with({"token": self.access_token.token}) assert isinstance(access_token, AccessToken)