Merge remote-tracking branch 'remotes/cohuck-gitlab/tags/s390x-20210305' into staging
[qemu.git] / tests / test-write-threshold.c
1 /*
2 * Test block device write threshold
3 *
4 * This work is licensed under the terms of the GNU LGPL, version 2 or later.
5 * See the COPYING.LIB file in the top-level directory.
6 *
7 */
8
9 #include "qemu/osdep.h"
10 #include "qapi/error.h"
11 #include "block/block_int.h"
12 #include "block/write-threshold.h"
13
14
15 static void test_threshold_not_set_on_init(void)
16 {
17 uint64_t res;
18 BlockDriverState bs;
19 memset(&bs, 0, sizeof(bs));
20
21 g_assert(!bdrv_write_threshold_is_set(&bs));
22
23 res = bdrv_write_threshold_get(&bs);
24 g_assert_cmpint(res, ==, 0);
25 }
26
27 static void test_threshold_set_get(void)
28 {
29 uint64_t threshold = 4 * 1024 * 1024;
30 uint64_t res;
31 BlockDriverState bs;
32 memset(&bs, 0, sizeof(bs));
33
34 bdrv_write_threshold_set(&bs, threshold);
35
36 g_assert(bdrv_write_threshold_is_set(&bs));
37
38 res = bdrv_write_threshold_get(&bs);
39 g_assert_cmpint(res, ==, threshold);
40 }
41
42 static void test_threshold_multi_set_get(void)
43 {
44 uint64_t threshold1 = 4 * 1024 * 1024;
45 uint64_t threshold2 = 15 * 1024 * 1024;
46 uint64_t res;
47 BlockDriverState bs;
48 memset(&bs, 0, sizeof(bs));
49
50 bdrv_write_threshold_set(&bs, threshold1);
51 bdrv_write_threshold_set(&bs, threshold2);
52 res = bdrv_write_threshold_get(&bs);
53 g_assert_cmpint(res, ==, threshold2);
54 }
55
56 static void test_threshold_not_trigger(void)
57 {
58 uint64_t amount = 0;
59 uint64_t threshold = 4 * 1024 * 1024;
60 BlockDriverState bs;
61 BdrvTrackedRequest req;
62
63 memset(&bs, 0, sizeof(bs));
64 memset(&req, 0, sizeof(req));
65 req.offset = 1024;
66 req.bytes = 1024;
67
68 bdrv_check_request(req.offset, req.bytes, &error_abort);
69
70 bdrv_write_threshold_set(&bs, threshold);
71 amount = bdrv_write_threshold_exceeded(&bs, &req);
72 g_assert_cmpuint(amount, ==, 0);
73 }
74
75
76 static void test_threshold_trigger(void)
77 {
78 uint64_t amount = 0;
79 uint64_t threshold = 4 * 1024 * 1024;
80 BlockDriverState bs;
81 BdrvTrackedRequest req;
82
83 memset(&bs, 0, sizeof(bs));
84 memset(&req, 0, sizeof(req));
85 req.offset = (4 * 1024 * 1024) - 1024;
86 req.bytes = 2 * 1024;
87
88 bdrv_check_request(req.offset, req.bytes, &error_abort);
89
90 bdrv_write_threshold_set(&bs, threshold);
91 amount = bdrv_write_threshold_exceeded(&bs, &req);
92 g_assert_cmpuint(amount, >=, 1024);
93 }
94
95 typedef struct TestStruct {
96 const char *name;
97 void (*func)(void);
98 } TestStruct;
99
100
101 int main(int argc, char **argv)
102 {
103 size_t i;
104 TestStruct tests[] = {
105 { "/write-threshold/not-set-on-init",
106 test_threshold_not_set_on_init },
107 { "/write-threshold/set-get",
108 test_threshold_set_get },
109 { "/write-threshold/multi-set-get",
110 test_threshold_multi_set_get },
111 { "/write-threshold/not-trigger",
112 test_threshold_not_trigger },
113 { "/write-threshold/trigger",
114 test_threshold_trigger },
115 { NULL, NULL }
116 };
117
118 g_test_init(&argc, &argv, NULL);
119 for (i = 0; tests[i].name != NULL; i++) {
120 g_test_add_func(tests[i].name, tests[i].func);
121 }
122 return g_test_run();
123 }