filter DNS answer to only include A records

bentsku · bentsku · commit f35b5a43c0d6 · 2024-12-20T16:39:43.000+01:00
diff --git a/localstack-core/localstack/aws/connect.py b/localstack-core/localstack/aws/connect.py
@@ -660,7 +660,11 @@ def resolve_dns_from_upstream(hostname: str) -> str:
     if len(response.answer) == 0:
         raise ValueError(f"No DNS response found for hostname '{hostname}'")
 
-    ip_addresses = list(response.answer[0].items.keys())
+    ip_addresses = []
+    for answer in response.answer:
+        if answer.match(dns.rdataclass.IN, dns.rdatatype.A, dns.rdatatype.TYPE0):
+            ip_addresses.extend(answer.items.keys())
+
     return choice(ip_addresses).address
 
 
