Use createami_custom_node_package in runtime_bake tests

Signed-off-by: Rex <[email protected]>

Author: Rex

tests/integration-tests/tests/runtime_bake/test_runtime_bake.py

@@ -49,8 +49,9 @@ def test_runtime_bake(
     else:
         custom_ami = retrieve_latest_ami(region, os, architecture=architecture)
 
-    # Custom Cookbook
+    # Custom Cookbook and Node
     custom_cookbook = request.config.getoption("createami_custom_chef_cookbook")
+    custom_node = request.config.getoption("createami_custom_node_package")
 
     # Create S3 bucket for pre install scripts, to remove epel package if it is installed
     bucket_name = s3_bucket_factory()
@@ -61,6 +62,7 @@ def test_runtime_bake(
         bucket_name=bucket_name,
         custom_ami=custom_ami,
         custom_cookbook=custom_cookbook if custom_cookbook else "",
+        custom_node=custom_node if custom_node else "",
     )
     cluster = clusters_factory(cluster_config)
     remote_command_executor = RemoteCommandExecutor(cluster)

tests/integration-tests/tests/runtime_bake/test_runtime_bake/test_runtime_bake/pcluster.config.ini

@@ -24,7 +24,11 @@ s3_read_resource = arn:aws-us-gov:s3:::{{ bucket_name }}/scripts/*
 s3_read_resource = arn:aws:s3:::{{ bucket_name }}/scripts/*
 {% endif %}
 pre_install = s3://{{ bucket_name }}/scripts/pre-install.sh
+{% if custom_node %}
+extra_json = { "cluster" : { "skip_install_recipes" : "no", "custom_node_package" : "{{ custom_node }}" } }
+{% else %}
 extra_json = { "cluster" : { "skip_install_recipes" : "no" } }
+{% endif %}
 {% if custom_cookbook %}
 custom_chef_cookbook = {{ custom_cookbook }}
 {% endif %}