From 0fd5be444372d3de64f1ebd2e88c57dbcb3d6b12 Mon Sep 17 00:00:00 2001
From: Penn Bauman <me@pennbauman.com>
Date: Fri, 29 Nov 2024 01:29:21 -0800
Subject: [PATCH] test/system: Fix issue with python resolver IP type

The existing code would never properly select the IPv4 address
instead a fallback value was selected when the IPv4 option was
used, the script is updated to that both IPv4 and IPv6 will be
properly selected.

Signed-off-by: Penn Bauman <me@pennbauman.com>
---
 test/system/203-network.bats | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/test/system/203-network.bats b/test/system/203-network.bats
index e29cdeab4..ae2a236c8 100644
--- a/test/system/203-network.bats
+++ b/test/system/203-network.bats
@@ -24,9 +24,8 @@ load 'libs/helpers'
 readonly RESOLVER_PYTHON3='\
 import socket; \
 import sys; \
-family = socket.AddressFamily.AF_INET if sys.argv[1] == "A" else 0; \
-family = socket.AddressFamily.AF_INET6 if sys.argv[1] == "AAAA" else 0; \
-addr = socket.getaddrinfo(sys.argv[2], None, family, socket.SocketKind.SOCK_RAW)[0][4][0]; \
+family = {"A": socket.AddressFamily.AF_INET, "AAAA": socket.AddressFamily.AF_INET6}
+addr = socket.getaddrinfo(sys.argv[2], None, family[sys.argv[1]], socket.SocketKind.SOCK_RAW)[0][4][0]; \
 print(addr)'
 
 # shellcheck disable=SC2016