Introduce client object in python tests (#772)

Thus far, the client end of the socket is the only piece of client state
tracked in tests, for which a global `socket` variable has been used. In
preparation to add more state, replace the `socket` global with a
`client` global object that groups all client state.

Signed-off-by: Mattias Nissler <mnissler@rivosinc.com>
Reviewed-by: John Levon <john.levon@nutanix.com>
Reviewed-by: Thanos Makatos <thanos.makatos@nutanix.com>

1 files changed, 6 insertions, 6 deletions

diff --git a/test/py/test_device_get_region_info_zero_size.py b/test/py/test_device_get_region_info_zero_size.py
index 42a6ae0..146e812 100644
--- a/test/py/test_device_get_region_info_zero_size.py
+++ b/test/py/test_device_get_region_info_zero_size.py
@@ -30,13 +30,13 @@
 from libvfio_user import *
 
 ctx = None
-sock = None
+client = None
 
 argsz = len(vfio_region_info())
 
 
 def test_device_get_region_info_setup():
-    global ctx, sock
+    global ctx, client
 
     ctx = vfu_create_ctx(flags=LIBVFIO_USER_FLAG_ATTACH_NB)
     assert ctx is not None
@@ -44,13 +44,13 @@ def test_device_get_region_info_setup():
     ret = vfu_realize_ctx(ctx)
     assert ret == 0
 
-    sock = connect_client(ctx)
+    client = connect_client(ctx)
 
 
 def test_device_get_region_info_zero_sized_region():
     """Tests that a zero-sized region has no caps."""
 
-    global sock
+    global client
 
     for index in [VFU_PCI_DEV_BAR1_REGION_IDX, VFU_PCI_DEV_MIGR_REGION_IDX]:
         payload = vfio_region_info(argsz=argsz, flags=0,
@@ -59,9 +59,9 @@ def test_device_get_region_info_zero_sized_region():
 
         hdr = vfio_user_header(VFIO_USER_DEVICE_GET_REGION_INFO,
                                size=len(payload))
-        sock.send(hdr + payload)
+        client.sock.send(hdr + payload)
         vfu_run_ctx(ctx)
-        result = get_reply(sock)
+        result = get_reply(client.sock)
 
         assert len(result) == argsz