Filter regions based on account availabilities in get_regions function

test/integration/conftest.py

@@ -1,4 +1,5 @@
 import ipaddress
+import logging
 import os
 import random
 import time
@@ -26,6 +27,7 @@
     PlacementGroupType,
     PostgreSQLDatabase,
 )
+from linode_api4.errors import ApiError
 from linode_api4.linode_client import LinodeClient, MonitorClient
 from linode_api4.objects import Region
 
@@ -36,6 +38,15 @@
 RUN_LONG_TESTS = "RUN_LONG_TESTS"
 SKIP_E2E_FIREWALL = "SKIP_E2E_FIREWALL"
 
+ALL_ACCOUNT_AVAILABILITIES = {
+    "Linodes",
+    "NodeBalancers",
+    "Block Storage",
+    "Kubernetes",
+}
+
+logger = logging.getLogger(__name__)
+
 
 def get_token():
     return os.environ.get(ENV_TOKEN_NAME, None)
@@ -58,9 +69,40 @@ def get_regions(
 
     regions = client.regions()
 
+    account_regional_availabilities = {}
+    try:
+        account_availabilities = client.account.availabilities()
+        for availability in account_availabilities:
+            account_regional_availabilities[availability.region] = (
+                availability.available
+            )
+    except ApiError:
+        logger.warning(
+            "Failed to retrieve account availabilities for regions. "
+            "Assuming required capabilities are available in all regions for this account. "
+            "Tests may fail if the account lacks access to necessary capabilities in the selected region."
+        )
+
     if capabilities is not None:
+        required_capabilities = set(capabilities)
+        required_account_capabilities = required_capabilities.intersection(
+            ALL_ACCOUNT_AVAILABILITIES
+        )
+
         regions = [
-            v for v in regions if set(capabilities).issubset(v.capabilities)
+            v
+            for v in regions
+            if required_capabilities.issubset(v.capabilities)
+            and required_account_capabilities.issubset(
+                account_regional_availabilities.get(
+                    v.id,
+                    (
+                        []
+                        if account_regional_availabilities
+                        else ALL_ACCOUNT_AVAILABILITIES
+                    ),
+                )
+            )
         ]
 
     if site_type is not None: