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 }
124