monkeypatch and test the warning pop

Author: unkcpz

aiidalab_launch/__main__.py

@@ -21,13 +21,7 @@
 from .core import LOGGER
 from .instance import AiidaLabInstance
 from .profile import DEFAULT_IMAGE, DEFAULT_PORT, Profile
-from .util import (
-    confirm_with_value,
-    get_latest_version,
-    image_is_latest,
-    spinner,
-    webbrowser_available,
-)
+from .util import confirm_with_value, get_latest_version, spinner, webbrowser_available
 from .version import __version__
 
 MSG_MOUNT_POINT_CONFLICT = """Warning: There is at least one other running
@@ -353,11 +347,13 @@ async def _async_start(
         with spinner(msg):
             instance.pull()
     else:
+        from aiidalab_launch import util
+
         # use local image
         msg = f"Using local image '{profile.image}'."
 
         # check if local image is outdated and pull latest version if so
-        if not image_is_latest(instance.client, profile.image):
+        if not util.image_is_latest(instance.client, profile.image):
             click.secho(
                 "Warning! Local image is outdated, please run with --pull to update.",
                 fg="yellow",

tests/test_cli.py

@@ -208,7 +208,7 @@ def test_remove_running_profile(self):
 @pytest.mark.slow
 @pytest.mark.trylast
 class TestInstanceLifecycle:
-    def test_start_stop_reset(self, instance, docker_client, caplog):
+    def test_start_stop_reset(self, instance, docker_client, caplog, monkeypatch):
         caplog.set_level(logging.DEBUG)
 
         def get_volume(volume_name):
@@ -260,6 +260,20 @@ def assert_status_down():
         assert result.exit_code == 0
         assert_status_up()
 
+        # test the warning message of image not the latest is not raised
+        assert "Warning!" not in result.output.strip()
+
+        # Then by monkeypatching the image_is_latest function, we can test that
+        # the warning message is raised
+        def image_is_latest(docker_client, image_name):
+            return False
+
+        monkeypatch.setattr("aiidalab_launch.util.image_is_latest", image_is_latest)
+        result: Result = runner.invoke(
+            cli.cli, ["start", "--no-browser", "--no-pull", "--wait=300"]
+        )
+        assert "Warning!" in result.output.strip()
+
         # Restart instance.
         # TODO: This test is currently disabled, because it is too flaky.  For
         # a currently unknown reason, the docker client will not be able to