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 }