adjust test

Author: sanderegg

packages/aws-library/tests/test_ec2_client.py

@@ -827,9 +827,12 @@ async def mock_run_instances(*args, **kwargs):
     assert call_count == 2
 
     # Verify the error contains the expected information
+    subnet_desc = await ec2_client.describe_subnets(SubnetIds=[aws_subnet_id])
     assert hasattr(exc_info.value, "instance_type")
     assert exc_info.value.instance_type == fake_ec2_instance_type.name  # type: ignore
-    assert exc_info.value.subnet_ids == [aws_subnet_id]  # type: ignore
+    assert exc_info.value.availability_zones == [  # type: ignore
+        subnet_desc["Subnets"][0]["AvailabilityZone"]  # type: ignore
+    ]
 
     # Verify still only 2 instances exist (no new ones were created)
     await _assert_instances_in_ec2(