LXR qemu/tests/unit/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
  15static 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
  27static 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
  42static 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
  56static 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
  76static 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
  95typedef struct TestStruct {
  96    const char *name;
  97    void (*func)(void);
  98} TestStruct;
  99
 100
 101int 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