diff options
Diffstat (limited to 'test/py/test_setup_region.py')
-rw-r--r-- | test/py/test_setup_region.py | 36 |
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(): |