Improve code coverage for indexdigest/cli/script.py

macbre · Jun 11, 2021 · 21e46e3 · 21e46e3
1 parent 15ad544
commit 21e46e3
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 1 deletion.
diff --git a/indexdigest/cli/script.py b/indexdigest/cli/script.py
@@ -198,7 +198,7 @@ def filter_reports_by_table(reports, tables=None, skip_tables=None):
     return reports
 
 
-def main():
+def main():  # pragma: no cover
     """ Main entry point for CLI"""
     logger = logging.getLogger(__name__)
 

diff --git a/indexdigest/test/cli/__init__.py b/indexdigest/test/cli/__init__.py
diff --git a/indexdigest/test/cli/test_script.py b/indexdigest/test/cli/test_script.py
@@ -0,0 +1,8 @@
+from _pytest.monkeypatch import MonkeyPatch
+from indexdigest import VERSION
+from indexdigest.cli.script import get_version
+
+
+def test_get_version(monkeypatch: MonkeyPatch):
+    monkeypatch.setenv('COMMIT_SHA', '1234567890abc')
+    assert get_version() == f'{VERSION} (git 1234567)'