Add tests for device manager use in CLI

Author: tpoliaw

tests/test_cli.py

@@ -1,5 +1,5 @@
 import os
-from unittest.mock import patch
+from unittest.mock import Mock, patch
 
 import pytest
 from bluesky import RunEngine
@@ -13,6 +13,7 @@
 
 from dodal import __version__
 from dodal.cli import main
+from dodal.device_manager import DeviceManager
 from dodal.utils import AnyDevice, OphydV1Device, OphydV2Device
 
 # Test with an example beamline, device instantiation is already tested
@@ -297,6 +298,34 @@ def test_cli_connect_when_devices_error(
         )
 
 
+@patch.dict(os.environ, clear=True)
+def test_missing_device_manager(runner: CliRunner):
+    with patch("dodal.cli.importlib"):
+        with pytest.raises(
+            ValueError,
+            match="Name 'devices' could not be found or is not a DeviceManager",
+        ):
+            print("Invoking connect")
+            runner.invoke(
+                main, ["connect", "-n", "devices", "i22"], catch_exceptions=False
+            )
+            print("Ran connect")
+
+
+@patch.dict(os.environ, clear=True)
+@pytest.mark.parametrize("mock", [True, False], ids=["live", "sim"].__getitem__)
+def test_device_manager_init(runner: CliRunner, mock: bool):
+    with patch("dodal.cli.importlib") as mock_import:
+        dm = mock_import.import_module("dodal.beamlines.i22")
+        dm.devices = Mock(spec=DeviceManager)
+        mock_import.reset_mock()
+        runner.invoke(
+            main, ["connect", "-n", "devices", "i22"] + (["-s"] if mock else [])
+        )
+        mock_import.import_module.assert_called_once_with("dodal.beamlines.i22")
+        dm.devices.build_and_connect.assert_called_once_with(mock=mock)
+
+
 def _mock_connect(
     *args,
     runner: CliRunner,