libqtest: Rename qmp_assert_error_class() to qmp_expect_error_and_unref()
[qemu.git] / tests / qtest / drive_del-test.c
1 /*
2 * blockdev.c test cases
3 *
4 * Copyright (C) 2013-2014 Red Hat Inc.
5 *
6 * Authors:
7 * Stefan Hajnoczi <stefanha@redhat.com>
8 *
9 * This work is licensed under the terms of the GNU LGPL, version 2.1 or later.
10 * See the COPYING.LIB file in the top-level directory.
11 */
12
13 #include "qemu/osdep.h"
14 #include "libqos/libqtest.h"
15 #include "libqos/virtio.h"
16 #include "qapi/qmp/qdict.h"
17
18 /* TODO actually test the results and get rid of this */
19 #define qmp_discard_response(q, ...) qobject_unref(qtest_qmp(q, __VA_ARGS__))
20
21 static void drive_add(QTestState *qts)
22 {
23 char *resp = qtest_hmp(qts, "drive_add 0 if=none,id=drive0");
24
25 g_assert_cmpstr(resp, ==, "OK\r\n");
26 g_free(resp);
27 }
28
29 static void drive_del(QTestState *qts)
30 {
31 char *resp = qtest_hmp(qts, "drive_del drive0");
32
33 g_assert_cmpstr(resp, ==, "");
34 g_free(resp);
35 }
36
37 static void device_del(QTestState *qts)
38 {
39 QDict *response;
40
41 /* Complication: ignore DEVICE_DELETED event */
42 qmp_discard_response(qts, "{'execute': 'device_del',"
43 " 'arguments': { 'id': 'dev0' } }");
44 response = qtest_qmp_receive(qts);
45 g_assert(response);
46 g_assert(qdict_haskey(response, "return"));
47 qobject_unref(response);
48 }
49
50 static void test_drive_without_dev(void)
51 {
52 QTestState *qts;
53
54 /* Start with an empty drive */
55 qts = qtest_init("-drive if=none,id=drive0");
56
57 /* Delete the drive */
58 drive_del(qts);
59
60 /* Ensure re-adding the drive works - there should be no duplicate ID error
61 * because the old drive must be gone.
62 */
63 drive_add(qts);
64
65 qtest_quit(qts);
66 }
67
68 /*
69 * qvirtio_get_dev_type:
70 * Returns: the preferred virtio bus/device type for the current architecture.
71 * TODO: delete this
72 */
73 static const char *qvirtio_get_dev_type(void)
74 {
75 const char *arch = qtest_get_arch();
76
77 if (g_str_equal(arch, "arm") || g_str_equal(arch, "aarch64")) {
78 return "device"; /* for virtio-mmio */
79 } else if (g_str_equal(arch, "s390x")) {
80 return "ccw";
81 } else {
82 return "pci";
83 }
84 }
85
86 static void test_after_failed_device_add(void)
87 {
88 char driver[32];
89 QDict *response;
90 QTestState *qts;
91
92 snprintf(driver, sizeof(driver), "virtio-blk-%s",
93 qvirtio_get_dev_type());
94
95 qts = qtest_init("-drive if=none,id=drive0");
96
97 /* Make device_add fail. If this leaks the virtio-blk device then a
98 * reference to drive0 will also be held (via qdev properties).
99 */
100 response = qtest_qmp(qts, "{'execute': 'device_add',"
101 " 'arguments': {"
102 " 'driver': %s,"
103 " 'drive': 'drive0'"
104 "}}", driver);
105 g_assert(response);
106 qmp_expect_error_and_unref(response, "GenericError");
107
108 /* Delete the drive */
109 drive_del(qts);
110
111 /* Try to re-add the drive. This fails with duplicate IDs if a leaked
112 * virtio-blk device exists that holds a reference to the old drive0.
113 */
114 drive_add(qts);
115
116 qtest_quit(qts);
117 }
118
119 static void test_drive_del_device_del(void)
120 {
121 QTestState *qts;
122
123 /* Start with a drive used by a device that unplugs instantaneously */
124 qts = qtest_initf("-drive if=none,id=drive0,file=null-co://,"
125 "file.read-zeroes=on,format=raw"
126 " -device virtio-scsi-%s"
127 " -device scsi-hd,drive=drive0,id=dev0",
128 qvirtio_get_dev_type());
129
130 /*
131 * Delete the drive, and then the device
132 * Doing it in this order takes notoriously tricky special paths
133 */
134 drive_del(qts);
135 device_del(qts);
136
137 qtest_quit(qts);
138 }
139
140 int main(int argc, char **argv)
141 {
142 g_test_init(&argc, &argv, NULL);
143
144 qtest_add_func("/drive_del/without-dev", test_drive_without_dev);
145
146 if (qvirtio_get_dev_type() != NULL) {
147 qtest_add_func("/drive_del/after_failed_device_add",
148 test_after_failed_device_add);
149 qtest_add_func("/blockdev/drive_del_device_del",
150 test_drive_del_device_del);
151 }
152
153 return g_test_run();
154 }