Chapter 10 - Recipe 'Determining test coverage'

Python3 changes

Author: shalabhaggarwal

Diff for: test_app.py

@@ -4,6 +4,15 @@
 from unittest import mock
 import tempfile
 import geoip2.records
+import coverage
+
+cov = coverage.coverage(
+    omit = [
+        '/Users/shalabh.aggarwal/workspace/cookbook10/lib/python3.6/site-packages/*',
+        'app_tests.py'
+    ]
+)
+cov.start()
 
 
 class CatalogTestCase(unittest.TestCase):
@@ -147,4 +156,11 @@ def test_search_product(self):
 
 
 if __name__ == '__main__':
-    unittest.main()
+    try:
+        unittest.main()
+    finally:
+        cov.stop()
+        cov.save()
+        cov.report()
+        cov.html_report(directory = 'coverage')
+        cov.erase()