blob: 97ca12f710870fc92838fbddc963d9b9de62a283 [file] [log] [blame]
Francesco Romanie2462112015-01-12 14:11:13 +01001/*
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
Peter Maydell681c28a2016-02-08 18:08:51 +00009#include "qemu/osdep.h"
Francesco Romanie2462112015-01-12 14:11:13 +010010#include "block/block_int.h"
11#include "block/write-threshold.h"
12
13
14static void test_threshold_not_set_on_init(void)
15{
16 uint64_t res;
17 BlockDriverState bs;
18 memset(&bs, 0, sizeof(bs));
19
20 g_assert(!bdrv_write_threshold_is_set(&bs));
21
22 res = bdrv_write_threshold_get(&bs);
23 g_assert_cmpint(res, ==, 0);
24}
25
26static void test_threshold_set_get(void)
27{
28 uint64_t threshold = 4 * 1024 * 1024;
29 uint64_t res;
30 BlockDriverState bs;
31 memset(&bs, 0, sizeof(bs));
32
33 bdrv_write_threshold_set(&bs, threshold);
34
35 g_assert(bdrv_write_threshold_is_set(&bs));
36
37 res = bdrv_write_threshold_get(&bs);
38 g_assert_cmpint(res, ==, threshold);
39}
40
41static void test_threshold_multi_set_get(void)
42{
43 uint64_t threshold1 = 4 * 1024 * 1024;
44 uint64_t threshold2 = 15 * 1024 * 1024;
45 uint64_t res;
46 BlockDriverState bs;
47 memset(&bs, 0, sizeof(bs));
48
49 bdrv_write_threshold_set(&bs, threshold1);
50 bdrv_write_threshold_set(&bs, threshold2);
51 res = bdrv_write_threshold_get(&bs);
52 g_assert_cmpint(res, ==, threshold2);
53}
54
55static void test_threshold_not_trigger(void)
56{
57 uint64_t amount = 0;
58 uint64_t threshold = 4 * 1024 * 1024;
59 BlockDriverState bs;
60 BdrvTrackedRequest req;
61
62 memset(&bs, 0, sizeof(bs));
63 memset(&req, 0, sizeof(req));
64 req.offset = 1024;
65 req.bytes = 1024;
66
67 bdrv_write_threshold_set(&bs, threshold);
68 amount = bdrv_write_threshold_exceeded(&bs, &req);
69 g_assert_cmpuint(amount, ==, 0);
70}
71
72
73static void test_threshold_trigger(void)
74{
75 uint64_t amount = 0;
76 uint64_t threshold = 4 * 1024 * 1024;
77 BlockDriverState bs;
78 BdrvTrackedRequest req;
79
80 memset(&bs, 0, sizeof(bs));
81 memset(&req, 0, sizeof(req));
82 req.offset = (4 * 1024 * 1024) - 1024;
83 req.bytes = 2 * 1024;
84
85 bdrv_write_threshold_set(&bs, threshold);
86 amount = bdrv_write_threshold_exceeded(&bs, &req);
87 g_assert_cmpuint(amount, >=, 1024);
88}
89
90typedef struct TestStruct {
91 const char *name;
92 void (*func)(void);
93} TestStruct;
94
95
96int main(int argc, char **argv)
97{
98 size_t i;
99 TestStruct tests[] = {
100 { "/write-threshold/not-set-on-init",
101 test_threshold_not_set_on_init },
102 { "/write-threshold/set-get",
103 test_threshold_set_get },
104 { "/write-threshold/multi-set-get",
105 test_threshold_multi_set_get },
106 { "/write-threshold/not-trigger",
107 test_threshold_not_trigger },
108 { "/write-threshold/trigger",
109 test_threshold_trigger },
110 { NULL, NULL }
111 };
112
113 g_test_init(&argc, &argv, NULL);
114 for (i = 0; tests[i].name != NULL; i++) {
115 g_test_add_func(tests[i].name, tests[i].func);
116 }
117 return g_test_run();
118}