aboutsummaryrefslogtreecommitdiff
path: root/test/py/test_setup_region.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/py/test_setup_region.py')
-rw-r--r--test/py/test_setup_region.py36
1 files changed, 18 insertions, 18 deletions
diff --git a/test/py/test_setup_region.py b/test/py/test_setup_region.py
index d00de68..05e6457 100644
--- a/test/py/test_setup_region.py
+++ b/test/py/test_setup_region.py
@@ -165,13 +165,13 @@ def test_setup_region_cfg_always_cb():
ret = vfu_realize_ctx(ctx)
assert ret == 0
- sock = connect_client(ctx)
+ client = connect_client(ctx)
- payload = read_region(ctx, sock, VFU_PCI_DEV_CFG_REGION_IDX,
+ payload = read_region(ctx, client.sock, VFU_PCI_DEV_CFG_REGION_IDX,
offset=0, count=2)
assert payload == b'\xcc\xcc'
- disconnect_client(ctx, sock)
+ client.disconnect(ctx)
def test_region_offset_overflow():
@@ -185,12 +185,12 @@ def test_region_offset_overflow():
ret = vfu_realize_ctx(ctx)
assert ret == 0
- sock = connect_client(ctx)
+ client = connect_client(ctx)
- read_region(ctx, sock, VFU_PCI_DEV_CFG_REGION_IDX,
+ read_region(ctx, client.sock, VFU_PCI_DEV_CFG_REGION_IDX,
offset=UINT64_MAX, count=256, expect=errno.EINVAL)
- disconnect_client(ctx, sock)
+ client.disconnect(ctx)
def test_access_region_zero_count():
@@ -203,40 +203,40 @@ def test_access_region_zero_count():
ret = vfu_realize_ctx(ctx)
assert ret == 0
- sock = connect_client(ctx)
+ client = connect_client(ctx)
- payload = read_region(ctx, sock, VFU_PCI_DEV_BAR0_REGION_IDX, offset=0,
- count=0)
+ payload = read_region(ctx, client.sock, VFU_PCI_DEV_BAR0_REGION_IDX,
+ offset=0, count=0)
assert payload == b''
- write_region(ctx, sock, VFU_PCI_DEV_BAR0_REGION_IDX, offset=0, count=0,
- data=payload)
+ write_region(ctx, client.sock, VFU_PCI_DEV_BAR0_REGION_IDX, offset=0,
+ count=0, data=payload)
- disconnect_client(ctx, sock)
+ client.disconnect(ctx)
def test_access_region_large_count():
global ctx
- sock = connect_client(ctx)
+ client = connect_client(ctx)
- read_region(ctx, sock, VFU_PCI_DEV_CFG_REGION_IDX, offset=0,
+ read_region(ctx, client.sock, VFU_PCI_DEV_CFG_REGION_IDX, offset=0,
count=SERVER_MAX_DATA_XFER_SIZE + 8, expect=errno.EINVAL)
- disconnect_client(ctx, sock)
+ client.disconnect(ctx)
def test_region_offset_too_short():
global ctx
- sock = connect_client(ctx)
+ client = connect_client(ctx)
payload = struct.pack("Q", 0)
- msg(ctx, sock, VFIO_USER_REGION_WRITE, payload,
+ msg(ctx, client.sock, VFIO_USER_REGION_WRITE, payload,
expect=errno.EINVAL)
- disconnect_client(ctx, sock)
+ client.disconnect(ctx)
def test_setup_region_cleanup():