diff --git a/corporate/tests/test_stripe.py b/corporate/tests/test_stripe.py index 52a115aeaf8d0..cefa3eac7e6d1 100644 --- a/corporate/tests/test_stripe.py +++ b/corporate/tests/test_stripe.py @@ -5,7 +5,7 @@ import os import re import sys -from typing import Any, Callable, Dict, List, Optional, TypeVar, Tuple +from typing import Any, Callable, Dict, List, Optional, TypeVar, Tuple, cast import ujson import json @@ -194,8 +194,8 @@ def normalize_fixtures(decorated_function: CallableT) -> None: # nocoverage f.write(file_content) def mock_stripe(*mocked_function_names: str, - generate: Optional[bool]=None) -> Callable[[CallableT], Callable[..., Any]]: - def _mock_stripe(decorated_function: CallableT) -> Callable[..., Any]: + generate: Optional[bool]=None) -> Callable[[CallableT], CallableT]: + def _mock_stripe(decorated_function: CallableT) -> CallableT: generate_fixture = generate if generate_fixture is None: generate_fixture = GENERATE_STRIPE_FIXTURES @@ -215,7 +215,7 @@ def wrapped(*args: Any, **kwargs: Any) -> Any: if generate_fixture: # nocoverage normalize_fixtures(decorated_function) return val - return wrapped + return cast(CallableT, wrapped) return _mock_stripe # A Kandra is a fictional character that can become anything. Used as a