feat: Adding an specific requisite check for pulling the TG image

Author: rboni-dk

dk-installer.py

@@ -986,9 +986,17 @@ def get_minikube_parser():
 )
 REQ_DOCKER_DAEMON = Requirement(
     "DOCKER_ENGINE",
-    ("docker", "info"),
+    ("docker", "system", "events", "--since=0m", "--until=0m"),
     ("The Docker engine is not running.", "Start the Docker engine and try again."),
 )
+REQ_TESTGEN_IMAGE = Requirement(
+    "TESTGEN_IMAGE",
+    ("docker", "manifest", "inspect", "{image}"),
+    (
+        "The Docker engine could not access the TestGen's image.",
+        "Make sure your networking policy allows Docker to pull the {image} image.",
+    ),
+)
 
 
 class AnalyticsMultiStepAction(MultiStepAction):
@@ -2046,7 +2054,7 @@ class TestgenInstallAction(TestgenActionMixin, AnalyticsMultiStepAction):
     intro_text = ["This process may take 5~10 minutes depending on your system resources and network speed."]
 
     args_cmd = "install"
-    requirements = [REQ_DOCKER, REQ_DOCKER_DAEMON]
+    requirements = [REQ_DOCKER, REQ_DOCKER_DAEMON, REQ_TESTGEN_IMAGE]
 
     def __init__(self):
         super().__init__()
@@ -2115,6 +2123,7 @@ def requirements(self):
         return [
             REQ_DOCKER,
             REQ_DOCKER_DAEMON,
+            REQ_TESTGEN_IMAGE,
             Requirement(
                 "TG_COMPOSE_FILE",
                 (

tests/test_tg_upgrade.py

@@ -60,7 +60,7 @@ def get_compose_content(*extra_vars):
 
 @pytest.mark.integration
 def test_tg_upgrade_compose_missing(tg_upgrade_action, args_mock, start_cmd_mock, console_msg_mock):
-    start_cmd_mock.__exit__.side_effect = [None, None, CommandFailed]
+    start_cmd_mock.__exit__.side_effect = [None, None, None, CommandFailed]
 
     with pytest.raises(AbortAction, match=""):
         tg_upgrade_action._check_requirements(args_mock)