test-announce-self: convert to qgraph

[mirror_qemu.git] / tests / virtio-rng-test.c

diff --git a/tests/virtio-rng-test.c b/tests/virtio-rng-test.c
index 402c2060da570de8fb2294499dae36f0eb03a50e..5309c7c8ab6fae332bf5b21869fe12a538d6ee44 100644 (file)
--- a/tests/virtio-rng-test.c
+++ b/tests/virtio-rng-test.c
@@ -7,27 +7,28 @@
  * See the COPYING file in the top-level directory.
  */
 
-#include <glib.h>
-#include <string.h>
-#include "libqtest.h"
 #include "qemu/osdep.h"
+#include "libqtest.h"
+#include "libqos/qgraph.h"
+#include "libqos/virtio-rng.h"
 
-/* Tests only initialization so far. TODO: Replace with functional tests */
-static void pci_nop(void)
-{
-}
+#define PCI_SLOT_HP             0x06
 
-int main(int argc, char **argv)
+static void rng_hotplug(void *obj, void *data, QGuestAllocator *alloc)
 {
-    int ret;
+    const char *arch = qtest_get_arch();
 
-    g_test_init(&argc, &argv, NULL);
-    qtest_add_func("/virtio/rng/pci/nop", pci_nop);
+    qtest_qmp_device_add("virtio-rng-pci", "rng1",
+                         "{'addr': %s}", stringify(PCI_SLOT_HP));
 
-    qtest_start("-device virtio-rng-pci");
-    ret = g_test_run();
-
-    qtest_end();
+    if (strcmp(arch, "i386") == 0 || strcmp(arch, "x86_64") == 0) {
+        qpci_unplug_acpi_device_test("rng1", PCI_SLOT_HP);
+    }
+}
 
-    return ret;
+static void register_virtio_rng_test(void)
+{
+    qos_add_test("hotplug", "virtio-rng-pci", rng_hotplug, NULL);
 }
+
+libqos_init(register_virtio_rng_test);