Run dockerhub and ecr together

Signed-off-by: Divya Madala <divyaasm@amazon.com>

src/validation_workflow/validation_args.py

@@ -14,7 +14,6 @@
 
 class ValidationArgs:
     SUPPORTED_PLATFORMS = ["linux", "windows"]
-    DOCKER_SOURCE = ["dockerhub", "ecr"]
 
     def __init__(self) -> None:
         parser = argparse.ArgumentParser(
@@ -74,11 +73,11 @@ def __init__(self) -> None:
         )
         parser.add_argument(
             "--docker-source",
-            type=str,
+            nargs='+',
             required=False,
-            choices=self.DOCKER_SOURCE,
-            help="(optional) Where to pull the docker image from, either DockerHub or ECR\n",
-            default="dockerhub",
+            help="(optional) Where to pull the docker image from, either DockerHub or ECR or Both \n",
+            choices=["dockerhub", "ecr"],
+            default=["dockerhub"],
         )
         parser.add_argument(
             "-v",

src/validation_workflow/validation_test_runner.py

@@ -29,4 +29,13 @@ class ValidationTestRunner:
 
     @classmethod
     def dispatch(cls, args: ValidationArgs, dist: str, work_dir: TemporaryDirectory) -> Validation:
-        return cls.RUNNERS[dist](args, work_dir)
+        if dist == "docker":
+            results = []
+            docker_source = args.docker_source
+            for source in docker_source:
+                new_args = args
+                new_args.docker_source = source
+                results.append(cls.RUNNERS[dist](new_args, work_dir))
+            return results
+        else:
+            return cls.RUNNERS[dist](args, work_dir)