Allow user to provide custom-manifests

ods_ci/tasks/Resources/Files/dsc_template.yml

@@ -5,22 +5,31 @@ metadata:
 spec:
   components:
     codeflare:
+      devFlags: <codeflare_devflags>
       managementState: <codeflare_value>
     dashboard:
+      devFlags: <dashboard_devflags>
       managementState: <dashboard_value>
     datasciencepipelines:
+      devFlags: <datasciencepipelines_devflags>
       managementState: <datasciencepipelines_value>
     kserve:
+      devFlags: <kserve_devflags>
       defaultDeploymentMode: Serverless
       managementState: <kserve_value>
     modelmeshserving:
+      devFlags: <modelmeshserving_devflags>
       managementState: <modelmeshserving_value>
     ray:
+      devFlags: <ray_devflags>
       managementState: <ray_value>
     kueue:
+      devFlags: <kueue_devflags>
       managementState: <kueue_value>
     trustyai:
+      devFlags: <trustyai_devflags>
       managementState: <trustyai_value>
     workbenches:
+      devFlags: <workbenches_devflags>
       managementState: <workbenches_value>
 

ods_ci/tasks/Resources/RHODS_OLM/install/oc_install.robot

@@ -274,6 +274,7 @@ Apply DataScienceCluster CustomResource
         END
     END
     Create DataScienceCluster CustomResource Using Test Variables
+    Apply Custom Manifest in DataScienceCluster CustomResource Using Test Variables
     ${yml} =    Get File    ${file_path}dsc_apply.yml
     Log To Console    Applying DSC yaml
     Log To Console    ${yml}
@@ -307,6 +308,21 @@ Create DataScienceCluster CustomResource Using Test Variables
         END
     END
 
+Apply Custom Manifest in DataScienceCluster CustomResource Using Test Variables
+    [Documentation]    Apply custom manifests to a DSC file
+    Log To Console    Applying Custom Manifests
+
+    ${file_path} =    Set Variable    tasks/Resources/Files/
+    FOR    ${cmp}    IN    @{COMPONENT_LIST}
+        IF    $cmp in $CUSTOM_MANIFESTS
+            ${manifest_string}=    Convert To String    ${CUSTOM_MANIFESTS}[${cmp}]
+            # Use sed to replace the placeholder with the YAML string
+            Run    sed -i "s|<${cmp}_devflags>|${manifest_string}|g" ${file_path}dsc_apply.yml
+        ELSE
+            Run    sed -i "s|<${cmp}_devflags>||g" ${file_path}dsc_apply.yml
+        END
+    END
+
 Component Should Be Enabled
     [Arguments]    ${component}    ${dsc_name}=${DSC_NAME}
     ${status} =    Is Component Enabled    ${component}    ${dsc_name}

ods_ci/utils/scripts/testconfig/generateTestConfigFile.py

@@ -101,6 +101,14 @@ def parse_args():
         dest="components",
         help="Comma-separated list of components and their states (component1:state1,component2:state2,...)",
     )
+    parser.add_argument(
+        "--custom_manifests",
+        type=str,
+        dest="custom_manifests",
+        help="Comma-separated list of custom_manifests("
+             "component1:repo_org:repo_name:branch_name:context_dir:source_path,"
+             "component2:repo_org:repo_name:branch_name:context_dir:source_path,...)",
+    )
 
     return parser.parse_args()
 
@@ -119,6 +127,27 @@ def change_component_state(components):
     return component_states
 
 
+def initialize_custom_manifest(custom_manifests):
+    # Parse and convert the custom manifest argument into a dictionary
+    manifest_details = {}
+    custom_manifest_list = custom_manifests.split(",")
+
+    for custom_manifest in custom_manifest_list:
+        comp, repo_org, repo_name, branch_name, context_dir, source_path = custom_manifest.split(":")
+        comp_manifest = {
+            "uri": f"https://github.com/{repo_org}/{repo_name}/tarball/{branch_name}",
+            "contextDir": context_dir,
+            "sourcePath": source_path if len(source_path) > 0 else ""
+        }
+
+        if comp not in manifest_details:
+            manifest_details[comp] = {"manifests": []}
+
+        manifest_details[comp]["manifests"].append(comp_manifest)
+
+    return manifest_details
+
+
 def get_prometheus_token(project):
     """
     Get prometheus token for the cluster.
@@ -156,6 +185,7 @@ def generate_test_config_file(
     set_prometheus_config,
     set_dashboard_url,
     components=None,
+    custom_manifests=None,
 ):
     """
     Generates test config file dynamically by
@@ -246,6 +276,11 @@ def generate_test_config_file(
         print(components)
         data["COMPONENTS"] = change_component_state(components)
 
+    if custom_manifests:
+        print("Setting custom_manifest")
+        print(custom_manifests)
+        data["CUSTOM_MANIFESTS"] = initialize_custom_manifest(custom_manifests)
+
     # Login to test cluster using oc command
     oc_login(
         data["OCP_CONSOLE_URL"],
@@ -301,6 +336,7 @@ def main():
         args.set_prometheus_config,
         args.set_dashboard_url,
         components=args.components,
+        custom_manifests=args.custom_manifests,
     )
     print("Done generating config file")