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

function drafts

Browse Source
This commit is contained in:
Maximilian Fajnberg
2021-11-26 23:37:46 +01:00
parent 44dee0b762
commit 25be101cf9
2 changed files with 18 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/test_functions.py
View File

@ -38,8 +38,6 @@ class FunctionsTestCase(IsolatedAsyncioTestCase):
test_html = '<b>foo</b>'
mock_session_cls.return_value = mock_session_obj = self.get_mock_session(test_html)
expected_output = BeautifulSoup(test_html, 'html.parser')
output = await functions.soup_from_url('baz')
self.assertEqual(expected_output, output)
output = await functions.soup_from_url('baz', mock_session_obj)
self.assertEqual(expected_output, output)
@ -59,10 +57,11 @@ class FunctionsTestCase(IsolatedAsyncioTestCase):
@patch.object(functions, 'is_relevant_table_row')
def test_find_relevant_table_rows(self, mock_is_relevant_table_row):
mock_is_relevant_table_row.return_value = True
expected_output = self.test_soup.find('div', attrs={'class': 'financials'}).div.div.table.tbody.find_all('tr')
test_table = self.test_soup.find('div', attrs={'class': 'financials'})
expected_output = test_table.find_all('tr')[1:-1]
output = functions.find_relevant_table_rows(self.test_soup)
self.assertListEqual(expected_output, output)
mock_is_relevant_table_row.assert_has_calls([call(expected_output[0]), call(expected_output[1])])
mock_is_relevant_table_row.assert_called_once_with(expected_output[0])
def test_extract_row_data(self):
test_table = self.test_soup.find('div', attrs={'class': 'financials'}).div.div.table