add print statements

Signed-off-by: Divya Madala <[email protected]>
Divyaasm · Sep 20, 2023 · aedbf8c · aedbf8c
1 parent 63d207d
commit aedbf8c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/validation_workflow/docker/validation_docker.py b/src/validation_workflow/docker/validation_docker.py
@@ -41,6 +41,8 @@ def download_artifacts(self) -> bool:
                 self.args.version if not self.args.using_staging_artifact_only else ValidationArgs().stg_tag(product).replace(" ", ""))
             self.image_ids = {key: value for key, value in zip(product_names, list(map(get_image_id, product_names)))}
             print(self.image_ids)
+            self.image_ids = {key: value.strip() for key, value in self.image_ids.items()}
+            print(self.image_ids)
 
             return True
 

diff --git a/tests/tests_validation_workflow/test_validation_docker.py b/tests/tests_validation_workflow/test_validation_docker.py
@@ -60,7 +60,10 @@ def test_validation(self, mock_time_sleep: Mock, mock_digest: Mock, mock_contain
 
         # Create instance of ValidateDocker class
         validate_docker = ValidateDocker(mock_validation_args.return_value)
+
+        validate_docker.image_names_list['opensearchproject/opensearch']
         validate_docker.image_ids = {'opensearch': 'images_id_0'}
+        validate_docker.replacements = [('opensearchproject/opensearch:1', 'sha256:774c2aae595fe9d725c10ee8918fcea90c0fbcda3146eec0f2b1b5d340594649')]
 
         # Call validation method and assert the result
         result = validate_docker.validation()