fix: Add missing toolkit methods to mock_toolkit_fix

- Added get_fundamentals_openai
- Added get_finnhub_company_insider_sentiment
- Added get_finnhub_company_insider_transactions
- Improved docstring for patch_toolkit_in_test

🤖 Generated with Claude Code

Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
佐藤優一 2025-08-11 13:04:54 +09:00
parent af8863a024
commit 69501af220
1 changed files with 14 additions and 3 deletions

View File

@ -28,6 +28,9 @@ def create_mock_toolkit_with_tools():
"get_simfin_income_stmt",
"get_simfin_balance_sheet",
"get_finnhub_basic_financials",
"get_fundamentals_openai",
"get_finnhub_company_insider_sentiment",
"get_finnhub_company_insider_transactions",
]
# Create mock for each method with proper __name__ attribute
@ -57,8 +60,16 @@ def create_mock_toolkit_with_tools():
return toolkit
def patch_toolkit_in_test(mock_toolkit):
"""Configure the mock_toolkit patch to return a properly mocked instance."""
def patch_toolkit_in_test(mock_toolkit_class):
"""Configure the mock_toolkit patch to return a properly mocked instance.
Args:
mock_toolkit_class: The patched Toolkit class
Returns:
The mock toolkit instance that will be used
"""
mock_instance = create_mock_toolkit_with_tools()
mock_toolkit.return_value = mock_instance
# Ensure that Toolkit() constructor returns our mock instance
mock_toolkit_class.return_value = mock_instance
return mock_instance