more tests; coverage script and settings

.coveragerc

@@ -0,0 +1,10 @@
+[run]
+source = .
+branch = true
+omit =
+    .venv/*
+
+[report]
+fail_under = 100
+show_missing = True
+skip_covered = True

.gitignore

@@ -3,4 +3,8 @@
 # PyCharm:
 /.idea/
 # Distribution / packaging:
-*.egg-info/
+*.egg-info/
+# Python cache:
+__pycache__/
+# Tests:
+.coverage

coverage.sh

@@ -0,0 +1,3 @@
+#!/usr/bin/env sh
+
+coverage erase && coverage run -m unittest discover && coverage report

setup.cfg

@@ -19,7 +19,7 @@ keywords = webscraping, html, markup, dom, scraper, attributes, tags, stocks, fi
 package_dir =
     = src
 packages = find:
-python_requires = >=3.7
+python_requires = >=3.8
 install_requires =
     beautifulsoup4
     aiohttp

src/stocksymbolscraper/__main__.py

@@ -8,7 +8,7 @@ from pathlib import Path
 from . import get_all_data, log
 
 
-def main() -> None:
+async def main() -> None:
     parser = ArgumentParser(description="Scrape all stock symbols")
     parser.add_argument(
         '-v', '--verbose',
@@ -29,7 +29,7 @@ def main() -> None:
     if args.verbose:
         log.setLevel(logging.DEBUG)
 
-    data = asyncio.run(get_all_data(args.sequential))
+    data = await get_all_data(args.sequential)
 
     if args.to_file is None:
         csv.writer(sys.stdout).writerows(data)
@@ -39,4 +39,4 @@ def main() -> None:
 
 
 if __name__ == '__main__':
-    main()
+    asyncio.run(main())

tests/test_scrape.py

@@ -1,12 +1,13 @@
-from unittest import TestCase
-from unittest.mock import patch, MagicMock, call
+import logging
+from unittest import IsolatedAsyncioTestCase
+from unittest.mock import patch, MagicMock, AsyncMock, call
 
 from bs4 import BeautifulSoup
 
 from stocksymbolscraper import scrape
 
 
-class ScrapeTestCase(TestCase):
+class ScrapeTestCase(IsolatedAsyncioTestCase):
 
     @patch.object(scrape, 'get_single_tr_data')
     def test_extract_row_data(self, mock_get_single_tr_data: MagicMock):
@@ -53,8 +54,36 @@ class ScrapeTestCase(TestCase):
         output = scrape.get_str_from_td(test_td)
         self.assertEqual(expected_output, output)
 
+    @patch.object(scrape, 'ClientSession')
+    async def test_soup_from_url(self, mock_session_cls):
+        test_html = '<b>foo</b>'
+        mock_response = MagicMock()
+        mock_response.text = AsyncMock(return_value=test_html)
+        mock_get_return = MagicMock()
+        mock_get_return.__aenter__ = AsyncMock(return_value=mock_response)
+        mock_session_obj = MagicMock()
+        mock_session_obj.get = MagicMock(return_value=mock_get_return)
+        mock_session_cls.return_value = mock_session_obj
+        output = await scrape.soup_from_url('foo')
+        expected_output = BeautifulSoup(test_html, scrape.HTML_PARSER)
+        self.assertEqual(expected_output, output)
+
+        output = await scrape.soup_from_url('foo', mock_session_obj)
+        self.assertEqual(expected_output, output)
+
     def test_trs_from_page(self):
-        # Tested function takes URL as argument (GET request is issued)
-        # HTML to be parsed could be substituted
-        #
-        pass
+        tr1_text, tr2_text = '<tr>foo</tr>', '<tr>bar</tr>'
+        test_html = f'<div id="marketsindex"><table><tbody>{tr1_text}{tr2_text}</tbody></table></div>'
+        test_soup = BeautifulSoup(test_html, scrape.HTML_PARSER)
+        output = scrape.trs_from_page(test_soup)
+        expected_output = test_soup.find_all('tr')
+        self.assertSequenceEqual(expected_output, output)
+
+        logging.disable(logging.CRITICAL)
+        test_html = f'<div id="marketsindex"><table>garbage</table></div>'
+        test_soup = BeautifulSoup(test_html, scrape.HTML_PARSER)
+        with patch.object(scrape, 'open') as mock_open:
+            self.assertRaises(scrape.UnexpectedMarkupError, scrape.trs_from_page, test_soup)
+            mock_open.assert_called_once()
+            mock_open.return_value.__enter__.return_value.write.assert_called_once_with(test_soup.prettify())
+        logging.disable(logging.NOTSET)