Fix CI on 3.8+

oremanj · oremanj · commit c7fd3e5485b0 · 2022-01-13T17:25:29.000-07:00
diff --git a/tests/test_basic.py b/tests/test_basic.py
@@ -138,9 +138,9 @@ async def listen_for_packets():
                 await harness.send(2, b"one", b"two")
                 await harness.expect(2, b"one", b"two")
             async with harness.enqueue_packets_to(2, queue_num, forwarded=False):
-                sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
-                for payload in (b"three", b"four"):
-                    sock.sendto(payload, harness.dest_addr[2])
+                with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as sock:
+                    for payload in (b"three", b"four"):
+                        sock.sendto(payload, harness.dest_addr[2])
                 with trio.fail_after(1):
                     while len(hwaddrs) < 4:
                         await trio.sleep(0.1)
