aboutsummaryrefslogtreecommitdiff
path: root/test/unit-tests.c
diff options
context:
space:
mode:
Diffstat (limited to 'test/unit-tests.c')
-rw-r--r--test/unit-tests.c14
1 files changed, 14 insertions, 0 deletions
diff --git a/test/unit-tests.c b/test/unit-tests.c
index c10ce5a..d4ac1ef 100644
--- a/test/unit-tests.c
+++ b/test/unit-tests.c
@@ -483,6 +483,19 @@ test_pci_caps(void **state __attribute__((unused)))
assert_memory_equal(&config_space.raw[off], vsc[0]->vsc.data, 5);
}
+static void
+test_device_get_info(void **state __attribute__((unused)))
+{
+ vfu_ctx_t vfu_ctx = { .nr_regions = 0xdeadbeef};
+ struct vfio_device_info d = { 0 };
+
+ assert_int_equal(0, handle_device_get_info(&vfu_ctx, sizeof d, &d));
+ assert_int_equal(sizeof d, d.argsz);
+ assert_int_equal(VFIO_DEVICE_FLAGS_PCI | VFIO_DEVICE_FLAGS_RESET, d.flags);
+ assert_int_equal(vfu_ctx.nr_regions, d.num_regions);
+ assert_int_equal(VFU_DEV_NUM_IRQS, d.num_irqs);
+}
+
/*
* FIXME we shouldn't have to specify a setup function explicitly for each unit
* test, cmocka should provide that. E.g. cmocka_run_group_tests enables us to
@@ -511,6 +524,7 @@ int main(void)
cmocka_unit_test_setup(test_run_ctx, setup),
cmocka_unit_test_setup(test_vfu_ctx_create, setup),
cmocka_unit_test_setup(test_pci_caps, setup),
+ cmocka_unit_test_setup(test_device_get_info, setup),
};
return cmocka_run_group_tests(tests, NULL, NULL);