From 661a1a98da17b1d6c2c419a1c356da954d5592b6 Mon Sep 17 00:00:00 2001 From: Maximilian Fajnberg Date: Mon, 15 Nov 2021 15:48:31 +0100 Subject: [PATCH] full coverage --- tests/test_scrape.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/tests/test_scrape.py b/tests/test_scrape.py index 1873008..8f74b21 100644 --- a/tests/test_scrape.py +++ b/tests/test_scrape.py @@ -135,3 +135,22 @@ class ScrapeTestCase(IsolatedAsyncioTestCase): ]) mock_trs_from_page.assert_has_calls([call(mock_soup), call(mock_soup)]) mock_extract_row_data.assert_called_once_with(*mock_trs) + + @patch.object(scrape, 'get_data_from_category') + @patch.object(scrape, 'ClientSession') + async def test_get_all_data(self, mock_session_cls, mock_get_data_from_category): + mock_session = MagicMock() + mock_session_cls.return_value.__aenter__.return_value = mock_session + mock_result = ['foo'] + expected_output = len(scrape.CATEGORIES) * mock_result + mock_get_data_from_category.return_value = mock_result + output = await scrape.get_all_data(sequential=True) + self.assertListEqual(expected_output, output) + mock_get_data_from_category.assert_has_calls([ + call(category, mock_session) for category in scrape.CATEGORIES + ]) + output = await scrape.get_all_data(sequential=False) + self.assertListEqual(expected_output, output) + mock_get_data_from_category.assert_has_calls([ + call(category, mock_session) for category in scrape.CATEGORIES + ])