From: BenoƮt Canet <benoit.ca...@nodalink.com> Signed-off-by: Benoit Canet <benoit.ca...@nodalink.com> Signed-off-by: Alberto Garcia <be...@igalia.com> --- tests/test-throttle.c | 56 +++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 56 insertions(+)
diff --git a/tests/test-throttle.c b/tests/test-throttle.c index 458f577..0b649b1 100644 --- a/tests/test-throttle.c +++ b/tests/test-throttle.c @@ -15,6 +15,7 @@ #include "block/aio.h" #include "qemu/throttle.h" #include "qemu/error-report.h" +#include "block/throttle-groups.h" static AioContext *ctx; static LeakyBucket bkt; @@ -500,6 +501,60 @@ static void test_accounting(void) (64.0 / 13))); } +static void test_groups(void) +{ + bool removed; + + ThrottleState *ts_foo, *ts_bar, *tmp; + BlockDriverState *bdrv1, *bdrv2; + + bdrv1 = bdrv_new(); + bdrv2 = bdrv_new(); + + ts_bar = throttle_group_incref("bar"); + ts_foo = throttle_group_incref("foo"); + tmp = throttle_group_incref("foo"); + + throttle_group_set_token(ts_bar, bdrv1, false); + throttle_group_set_token(tmp, bdrv2, true); + + g_assert(tmp == ts_foo); + + tmp = throttle_group_incref("bar"); + g_assert(tmp == ts_bar); + + tmp = throttle_group_incref("bar"); + g_assert(tmp == ts_bar); + + g_assert(throttle_group_token(ts_bar, false) == bdrv1); + g_assert(throttle_group_token(ts_foo, true) == bdrv2); + + removed = throttle_group_unref(ts_foo); + g_assert(removed); + removed = throttle_group_unref(ts_bar); + g_assert(removed); + + g_assert(throttle_group_token(ts_foo, true) == bdrv2); + + removed = throttle_group_unref(ts_foo); + g_assert(removed); + removed = throttle_group_unref(ts_bar); + g_assert(removed); + + /* "foo" group should be destroyed when reaching this */ + removed = throttle_group_unref(ts_foo); + g_assert(!removed); + + g_assert(throttle_group_token(ts_bar, false) == bdrv1); + + removed = throttle_group_unref(ts_bar); + g_assert(removed); + + /* "bar" group should be destroyed when reaching this */ + removed = throttle_group_unref(ts_bar); + g_assert(!removed); +} + int main(int argc, char **argv) { GSource *src; @@ -533,6 +588,7 @@ int main(int argc, char **argv) g_test_add_func("/throttle/config/is_valid", test_is_valid); g_test_add_func("/throttle/config_functions", test_config_functions); g_test_add_func("/throttle/accounting", test_accounting); + g_test_add_func("/throttle/groups", test_groups); return g_test_run(); } -- 2.1.4