daniil-berg/mwfin
2
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

adjusted tests to call the new function

Browse Source
This commit is contained in:
Maximilian Fajnberg
2021-12-03 14:32:52 +01:00
parent ca360413c3
commit a7dbdcf917
1 changed files with 14 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
tests/test_functions.py
View File

@@ -141,9 +141,9 @@ class FunctionsTestCase(IsolatedAsyncioTestCase):
call(statement, symbol2, quarterly, mock_session) call(statement, symbol2, quarterly, mock_session)
]) ])
async def _helper_test_get_any_statement(self, statement: str, mock__get_single_company_fin_stmt): async def _helper_test_get_any_statement(self, statement: str, mock__get_multi_companies_fin_stmt):
symbol, quarterly, mock_session = 'foo', False, MagicMock() symbol1, symbol2, quarterly, mock_session = 'foo', 'bar', False, MagicMock()
mock__get_single_company_fin_stmt.return_value = expected_output = 'bar' mock__get_multi_companies_fin_stmt.return_value = expected_output = 'baz'
if statement == BS: if statement == BS:
function = functions.get_balance_sheet function = functions.get_balance_sheet
elif statement == IS: elif statement == IS:
@@ -152,31 +152,21 @@ class FunctionsTestCase(IsolatedAsyncioTestCase):
function = functions.get_cash_flow_statement function = functions.get_cash_flow_statement
else: else:
raise ValueError raise ValueError
output = await function(symbol, quarterly=quarterly, session=mock_session)
self.assertEqual(expected_output, output)
mock__get_single_company_fin_stmt.assert_called_once_with(statement, symbol, quarterly, mock_session)
mock__get_single_company_fin_stmt.reset_mock()
symbol1, symbol2 = 'x', 'y'
expected_output = {symbol1: expected_output, symbol2: expected_output}
output = await function(symbol1, symbol2, quarterly=quarterly, session=mock_session) output = await function(symbol1, symbol2, quarterly=quarterly, session=mock_session)
self.assertDictEqual(expected_output, output) self.assertEqual(expected_output, output)
mock__get_single_company_fin_stmt.assert_has_calls([ mock__get_multi_companies_fin_stmt.assert_called_once_with(statement, symbol1, symbol2, quarterly, mock_session)
call(statement, symbol1, quarterly, mock_session),
call(statement, symbol2, quarterly, mock_session),
])
@patch.object(functions, '_get_single_company_fin_stmt') @patch.object(functions, '_get_multi_companies_fin_stmt')
async def test_get_balance_sheet(self, mock__get_single_company_fin_stmt): async def test_get_balance_sheet(self, mock__get_multi_companies_fin_stmt):
await self._helper_test_get_any_statement(BS, mock__get_single_company_fin_stmt) await self._helper_test_get_any_statement(BS, mock__get_multi_companies_fin_stmt)
@patch.object(functions, '_get_single_company_fin_stmt') @patch.object(functions, '_get_multi_companies_fin_stmt')
async def test_get_income_statement(self, mock__get_single_company_fin_stmt): async def test_get_income_statement(self, mock__get_multi_companies_fin_stmt):
await self._helper_test_get_any_statement(IS, mock__get_single_company_fin_stmt) await self._helper_test_get_any_statement(IS, mock__get_multi_companies_fin_stmt)
@patch.object(functions, '_get_single_company_fin_stmt') @patch.object(functions, '_get_multi_companies_fin_stmt')
async def test_get_cash_flow_statement(self, mock__get_single_company_fin_stmt): async def test_get_cash_flow_statement(self, mock__get_multi_companies_fin_stmt):
await self._helper_test_get_any_statement(CF, mock__get_single_company_fin_stmt) await self._helper_test_get_any_statement(CF, mock__get_multi_companies_fin_stmt)
@patch.object(functions, '_get_single_company_fin_stmt') @patch.object(functions, '_get_single_company_fin_stmt')
async def test__get_single_company_financials(self, mock__get_single_company_fin_stmt): async def test__get_single_company_financials(self, mock__get_single_company_fin_stmt):