[Test] In test_pcluster_configure, assert that head node and compute instance types are taken from the current free tier.

Author: gmarciani

tests/integration-tests/tests/configure/test_pcluster_configure.py

@@ -10,6 +10,7 @@
 # This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, express or implied.
 # See the License for the specific language governing permissions and limitations under the License.
 import logging
+import re
 from datetime import datetime
 from os import environ
 
@@ -21,14 +22,24 @@
 from cfn_stacks_factory import CfnVpcStack
 from conftest import inject_additional_config_settings
 from conftest_networking import CIDR_FOR_CUSTOM_SUBNETS
-from utils import get_instance_info
+from utils import get_instance_info, get_free_tier_instance_types
+
+# REGEX_FREE_TIER_INSTANCE_TYPES = '|'.join(map(re.escape, get_free_tier_instance_types()))
+REGEX_FREE_TIER_INSTANCE_TYPES = '|'.join(map(re.escape, [
+    "c7i-flex.large",
+    "t4g.small",
+    "t4g.micro",
+    "t3.micro",
+    "t3.small",
+    "m7i-flex.large"
+]))
 
 PROMPTS = {
     "region": lambda region: {"prompt": r"AWS Region ID \[.*\]: ", "response": region},
     "key_pair": lambda key_name: {"prompt": r"EC2 Key Pair Name \[.*\]: ", "response": key_name},
     "scheduler": lambda scheduler: {"prompt": r"Scheduler \[slurm\]: ", "response": scheduler},
     "os": lambda os: {"prompt": r"Operating System \[alinux2\]: ", "response": os, "skip_for_batch": True},
-    "head_instance_type": lambda instance: {"prompt": r"Head node instance type \[t.\.micro\]: ", "response": instance},
+    "head_instance_type": lambda instance: {"prompt": rf"Head node instance type \[({REGEX_FREE_TIER_INSTANCE_TYPES})\]: ", "response": instance},
     "no_of_queues": lambda n: {"prompt": rf"Number of queues \[{n}\]: ", "response": f"{n}", "skip_for_batch": True},
     "queue_name": lambda queue, name: {"prompt": rf"Name of queue {queue} \[queue{queue}\]: ", "response": name},
     "no_of_compute_resources": lambda queue_name, queue, n: {
@@ -37,7 +48,7 @@
         "skip_for_batch": True,
     },
     "compute_instance_type": lambda resource, queue_name, instance: {
-        "prompt": rf"Compute instance type for compute resource {resource} in {queue_name} \[t.\.micro\]: ",
+        "prompt": rf"Compute instance type for compute resource {resource} in {queue_name} \[({REGEX_FREE_TIER_INSTANCE_TYPES})\]: ",
         "response": instance,
         "skip_for_batch": True,
     },

tests/integration-tests/utils.py

@@ -908,3 +908,14 @@ def find_stack_by_tag(tag, region, stack_prefix):
             logging.info(f"Found stack: {name} (created on {creation_date})")
             return name
     return None
+
+
+def get_free_tier_instance_types(region: str = None):
+    ec2 = boto3.client('ec2', region_name=region)
+    response = ec2.describe_instance_types(
+        Filters=[
+            {'Name': 'free-tier-eligible', 'Values': ['true']},
+            {"Name": "current-generation", "Values": ["true"]},
+        ]
+    )
+    return [instance['InstanceType'] for instance in response['InstanceTypes']]