Fix tests

Author: r4victor

src/tests/_internal/core/backends/aws/test_configurator.py

@@ -17,7 +17,7 @@ def test_validate_config_valid(self):
         )
         with (
             patch("dstack._internal.core.backends.aws.auth.authenticate"),
-            patch("dstack._internal.core.backends.aws.compute.get_vpc_id_subnet_id_or_error"),
+            patch("dstack._internal.core.backends.aws.compute.get_vpc_id_subnets_ids_or_error"),
         ):
             AWSConfigurator().validate_config(config, default_creds_enabled=True)
 

src/tests/_internal/server/routers/test_backends.py

@@ -141,7 +141,7 @@ async def test_creates_aws_backend(self, test_db, session: AsyncSession, client:
         }
         with (
             patch("dstack._internal.core.backends.aws.auth.authenticate"),
-            patch("dstack._internal.core.backends.aws.compute.get_vpc_id_subnet_id_or_error"),
+            patch("dstack._internal.core.backends.aws.compute.get_vpc_id_subnets_ids_or_error"),
         ):
             response = await client.post(
                 f"/api/project/{project.name}/backends/create",
@@ -542,7 +542,7 @@ async def test_returns_400_if_backend_exists(
         }
         with (
             patch("dstack._internal.core.backends.aws.auth.authenticate"),
-            patch("dstack._internal.core.backends.aws.compute.get_vpc_id_subnet_id_or_error"),
+            patch("dstack._internal.core.backends.aws.compute.get_vpc_id_subnets_ids_or_error"),
         ):
             response = await client.post(
                 f"/api/project/{project.name}/backends/create",
@@ -605,7 +605,7 @@ async def test_updates_backend(self, test_db, session: AsyncSession, client: Asy
         }
         with (
             patch("dstack._internal.core.backends.aws.auth.authenticate"),
-            patch("dstack._internal.core.backends.aws.compute.get_vpc_id_subnet_id_or_error"),
+            patch("dstack._internal.core.backends.aws.compute.get_vpc_id_subnets_ids_or_error"),
         ):
             response = await client.post(
                 f"/api/project/{project.name}/backends/update",
@@ -857,7 +857,7 @@ async def test_creates_aws_backend(self, test_db, session: AsyncSession, client:
         body = {"config_yaml": yaml.dump(config_dict)}
         with (
             patch("dstack._internal.core.backends.aws.auth.authenticate"),
-            patch("dstack._internal.core.backends.aws.compute.get_vpc_id_subnet_id_or_error"),
+            patch("dstack._internal.core.backends.aws.compute.get_vpc_id_subnets_ids_or_error"),
         ):
             response = await client.post(
                 f"/api/project/{project.name}/backends/create_yaml",
@@ -945,7 +945,7 @@ async def test_updates_aws_backend(self, test_db, session: AsyncSession, client:
         body = {"config_yaml": yaml.dump(config_dict)}
         with (
             patch("dstack._internal.core.backends.aws.auth.authenticate"),
-            patch("dstack._internal.core.backends.aws.compute.get_vpc_id_subnet_id_or_error"),
+            patch("dstack._internal.core.backends.aws.compute.get_vpc_id_subnets_ids_or_error"),
         ):
             response = await client.post(
                 f"/api/project/{project.name}/backends/update_yaml",

src/tests/_internal/server/services/test_config.py

@@ -106,7 +106,9 @@ async def test_creates_backend(self, test_db, session: AsyncSession, tmp_path: P
             with (
                 patch("boto3.session.Session"),
                 patch.object(settings, "SERVER_CONFIG_FILE_PATH", config_filepath),
-                patch("dstack._internal.core.backends.aws.compute.get_vpc_id_subnet_id_or_error"),
+                patch(
+                    "dstack._internal.core.backends.aws.compute.get_vpc_id_subnets_ids_or_error"
+                ),
             ):
                 manager = ServerConfigManager()
                 manager.load_config()