[Test] In test_update_slurm, use flexible instance types for spot ins…

…tances to reduce the risk of ICEs. Signed-off-by: Giacomo Marciani <[email protected]>
aws · Nov 28, 2024 · e8563ce · e8563ce
1 parent 3119557
commit e8563ce
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 0 deletions.
diff --git a/tests/integration-tests/tests/update/test_update.py b/tests/integration-tests/tests/update/test_update.py
@@ -133,6 +133,15 @@ def test_update_slurm(region, pcluster_config_reader, s3_bucket_factory, cluster
                     "instances": [
                         {
                             "instance_type": "t3.small",
+                        },
+                        {
+                            "instance_type": "t3a.small",
+                        },
+                        {
+                            "instance_type": "t3.medium",
+                        },
+                        {
+                            "instance_type": "t3a.medium",
                         }
                     ],
                     "expected_running_instances": 1,
@@ -240,6 +249,15 @@ def test_update_slurm(region, pcluster_config_reader, s3_bucket_factory, cluster
                     "instances": [
                         {
                             "instance_type": "t3.small",
+                        },
+                        {
+                            "instance_type": "t3a.small",
+                        },
+                        {
+                            "instance_type": "t3.medium",
+                        },
+                        {
+                            "instance_type": "t3a.medium",
                         }
                     ],
                     "expected_running_instances": 0,

diff --git a/.../integration-tests/tests/update/test_update/test_update_slurm/pcluster.config.update.yaml b/.../integration-tests/tests/update/test_update/test_update_slurm/pcluster.config.update.yaml
@@ -59,6 +59,9 @@ Scheduling:
         - Name: queue1-i3 # New compute resource
           Instances:
             - InstanceType: t3.small
+            - InstanceType: t3a.small
+            - InstanceType: t3.medium
+            - InstanceType: t3a.medium
               # Removed MinCount
       Networking:
         SubnetIds:

diff --git a/tests/integration-tests/tests/update/test_update/test_update_slurm/pcluster.config.yaml b/tests/integration-tests/tests/update/test_update/test_update_slurm/pcluster.config.yaml
@@ -51,6 +51,9 @@ Scheduling:
         - Name: queue1-i2
           Instances:
             - InstanceType: t3.small
+            - InstanceType: t3a.small
+            - InstanceType: t3.medium
+            - InstanceType: t3a.medium
           MinCount: 1
       Networking:
         SubnetIds: