made getGetInterface test work with ipv6 localhost

irmen · Oct 21, 2023 · 8db91b6 · 8db91b6
1 parent bc9cc9a
commit 8db91b6
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/tests/test_socketutil.py b/tests/test_socketutil.py
@@ -56,10 +56,15 @@ def testGetIP6(self):
 
     def testGetInterface(self):
         addr = socketutil.get_interface("localhost")
-        assert addr.version == 4
-        assert str(addr).startswith("127.")
-        assert str(addr.ip).startswith("127.0")
-        assert str(addr.network).startswith("127.0")
+        if addr.version == 6:
+            assert str(addr).startswith("::1")
+            assert str(addr.ip) == "::1"
+            assert str(addr.network).startswith("::1")
+        else:
+            assert addr.version == 4
+            assert str(addr).startswith("127.")
+            assert str(addr.ip).startswith("127.0")
+            assert str(addr.network).startswith("127.0")
         if has_ipv6:
             addr = socketutil.get_interface("::1")
             assert addr.version == 6