From 0033855c434034dd2f3c99dfad10b6a41c55f1e4 Mon Sep 17 00:00:00 2001
From: Peter Zhu <zhujiaxi@amazon.com>
Date: Wed, 28 Aug 2024 18:19:47 -0400
Subject: [PATCH] Add more tests

Signed-off-by: Peter Zhu <zhujiaxi@amazon.com>
---
 .../test_input_manifests.py                   | 58 +++++++++++++++++++
 1 file changed, 58 insertions(+)

diff --git a/tests/tests_manifests_workflow/test_input_manifests.py b/tests/tests_manifests_workflow/test_input_manifests.py
index 445b64f741..2f5afbb06d 100644
--- a/tests/tests_manifests_workflow/test_input_manifests.py
+++ b/tests/tests_manifests_workflow/test_input_manifests.py
@@ -58,6 +58,20 @@ def test_create_manifest_opensearch_template(self) -> None:
                                 "checks": ["gradle:publish", "gradle:properties:version"]}]
             }
         )
+        self.assertEqual(
+            input_manifest[1].to_dict(),
+            {
+                'schema-version': '1.0',
+                'name': 'OpenSearch',
+                'ci': {'image': {'name': 'opensearchstaging/ci-runner:ci-runner-centos7-opensearch-build-v3',
+                                 'args': '-e JAVA_HOME=/opt/java/openjdk-11'}},
+                'components': [{'name': 'index-management',
+                                'integ-test': {'build-dependencies': ['job-scheduler'],
+                                               'test-configs': ['with-security', 'without-security'],
+                                               'additional-cluster-configs': {'path.repo': ['/tmp']}},
+                                                                              'bwc-test': {'test-configs': ['with-security']}}]
+            }
+        )
 
     def test_create_manifest_opensearch_default_template(self) -> None:
         input_manifests = InputManifests("OpenSearch")
@@ -75,6 +89,20 @@ def test_create_manifest_opensearch_default_template(self) -> None:
                                 "checks": ["gradle:publish", "gradle:properties:version"]}]
             }
         )
+        self.assertEqual(
+            input_manifest[1].to_dict(),
+            {
+                'schema-version': '1.0',
+                'name': 'OpenSearch',
+                'ci': {'image': {'name': 'opensearchstaging/ci-runner:ci-runner-al2-opensearch-build-v1',
+                                 'args': '-e JAVA_HOME=/opt/java/openjdk-21'}},
+                'components': [{'name': 'index-management',
+                                'integ-test': {'build-dependencies': ['job-scheduler'],
+                                               'test-configs': ['with-security', 'without-security'],
+                                               'additional-cluster-configs': {'path.repo': ['/tmp']}},
+                                                                              'bwc-test': {'test-configs': ['with-security']}}]
+            }
+        )
 
     def test_create_manifest_opensearch_previous_base_version(self) -> None:
         input_manifests = InputManifests("OpenSearch")
@@ -97,6 +125,16 @@ def test_create_manifest_opensearch_dashboards_template(self) -> None:
                                 "ref": "1.x"}]
             }
         )
+        self.assertEqual(
+            input_manifest[1].to_dict(),
+            {
+                'schema-version': '1.0',
+                'name': 'OpenSearch Dashboards',
+                'ci': {'image': {'name': 'opensearchstaging/ci-runner:ci-runner-rockylinux8-opensearch-dashboards-integtest-v4'}},
+                'components': [{'name': 'indexManagementDashboards',
+                                'integ-test': {'test-configs': ['with-security', 'without-security']}}]
+            }
+        )
 
     def test_create_manifest_opensearch_dashboards_default_template(self) -> None:
         input_manifests = InputManifests("OpenSearch Dashboards")
@@ -113,6 +151,26 @@ def test_create_manifest_opensearch_dashboards_default_template(self) -> None:
                                 "checks": ["npm:package:version"]}]
             }
         )
+        self.assertEqual(
+            input_manifest[1].to_dict(),
+            {
+                'schema-version': '1.0',
+                'name': 'OpenSearch Dashboards',
+                'ci': {'image': {'name': 'opensearchstaging/ci-runner:ci-runner-almalinux8-opensearch-dashboards-integtest-v1'}},
+                'components': [{'name': 'OpenSearch-Dashboards',
+                                'integ-test': {'test-configs': ['with-security', 'without-security'],
+                                                'additional-cluster-configs': {'vis_builder.enabled': True,
+                                                                               'data_source.enabled': True,
+                                                                               'savedObjects.maxImportPayloadBytes': 10485760,
+                                                                               'server.maxPayloadBytes': 1759977,
+                                                                               'logging.json': False,
+                                                                               'data.search.aggs.shardDelay.enabled': True,
+                                                                               'csp.warnLegacyBrowsers': False},
+                                                'ci-groups': 9}},
+                                {'name': 'indexManagementDashboards',
+                                 'integ-test': {'test-configs': ['with-security', 'without-security']}}]
+            }
+        )
 
     def test_create_manifest_opensearch_dashboards_previous_base_version(self) -> None:
         input_manifests = InputManifests("OpenSearch-Dashboards")