Improve codecov

Author: olliesilvester

src/daq_config_server/__main__.py

@@ -2,29 +2,33 @@
 
 from . import __version__
 
-try:
-    import uvicorn  # noqa
-    from fastapi import FastAPI  # noqa
+__all__ = ["main"]
 
-    server_dependencies_exist = True
-except ImportError:
-    server_dependencies_exist = False
+INSUFFICIENT_DEPENDENCIES_MESSAGE = "To do anything other than print the version and be\
+    available for importing the client, you must install this package with [server]\
+    optional dependencies"
 
 
-__all__ = ["main"]
+def check_server_dependencies():
+    try:
+        import uvicorn  # noqa
+        from fastapi import FastAPI  # noqa
+
+        server_dependencies_exist = True
+
+    except ImportError:
+        server_dependencies_exist = False
+
+    return server_dependencies_exist
 
 
 def main():
     parser = ArgumentParser()
     parser.add_argument("-v", "--version", action="version", version=__version__)
     parser.parse_args()
 
-    if not server_dependencies_exist:
-        print(
-            "To do anything other than print the version and be available for "
-            "importing the client, you must install this package with [server] "
-            "optional dependencies"
-        )
+    if not check_server_dependencies():
+        print(INSUFFICIENT_DEPENDENCIES_MESSAGE)
     else:
         from .app import main
 

tests/unit_tests/test_cli.py

@@ -1,9 +1,33 @@
 import subprocess
 import sys
+from unittest.mock import MagicMock, patch
 
-from daq_config_server import __version__
+from daq_config_server.__main__ import (
+    INSUFFICIENT_DEPENDENCIES_MESSAGE,
+    __version__,
+    main,
+)
 
 
 def test_cli_version():
     cmd = [sys.executable, "-m", "daq_config_server", "--version"]
     assert subprocess.check_output(cmd).decode().strip() == __version__
+
+
+@patch("daq_config_server.app.main")
+@patch("daq_config_server.__main__.print")
+@patch("daq_config_server.__main__.ArgumentParser.parse_args")
+@patch.dict("sys.modules", {"uvicorn": None, "fastapi": None})
+def test_print_and_exit_if_incorrect_dependencies(
+    mock_parse_args, mock_print: MagicMock, mock_main: MagicMock
+):
+    main()
+    mock_print.assert_called_once_with(INSUFFICIENT_DEPENDENCIES_MESSAGE)
+    mock_main.assert_not_called()
+
+
+@patch("daq_config_server.app.main")
+@patch("daq_config_server.__main__.ArgumentParser.parse_args")
+def test_main_runs_with_correct_dependencies(mock_parse_args, mock_main: MagicMock):
+    main()
+    mock_main.assert_called_once()