diff mbox

[v1,3/8] throttle: Add throttle group infrastructure tests

Message ID 1412688279-8312-4-git-send-email-benoit.canet@nodalink.com
State New
Headers show

Commit Message

Benoît Canet Oct. 7, 2014, 1:24 p.m. UTC
Signed-off-by: Benoit Canet <benoit.canet@nodalink.com>
---
 tests/test-throttle.c | 51 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 51 insertions(+)

Comments

Fam Zheng Oct. 8, 2014, 6:20 a.m. UTC | #1
On Tue, 10/07 15:24, Benoît Canet wrote:
> Signed-off-by: Benoit Canet <benoit.canet@nodalink.com>
> ---
>  tests/test-throttle.c | 51 +++++++++++++++++++++++++++++++++++++++++++++++++++
>  1 file changed, 51 insertions(+)
> 
> diff --git a/tests/test-throttle.c b/tests/test-throttle.c
> index 3e52df3..ecb5504 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,55 @@ static void test_accounting(void)
>                                  (64.0 / 13)));
>  }
>  
> +static void test_groups(void)
> +{
> +    bool removed;
> +
> +    ThrottleState *ts_foo, *ts_bar, *tmp;
> +
> +    ts_bar = throttle_group_incref("bar");
> +    throttle_group_set_token(ts_bar, (BlockDriverState *) 0x5, false);

Why do you have the magic numbers cast to pointers instead of allocated objects
with bdrv_new?

Fam

> +    ts_foo = throttle_group_incref("foo");
> +
> +    tmp = throttle_group_incref("foo");
> +    throttle_group_set_token(tmp, (BlockDriverState *) 0x7, 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((int64_t) throttle_group_token(ts_bar, false) == 0x5);
> +    g_assert((int64_t) throttle_group_token(ts_foo, true) == 0x7);
> +
> +    removed = throttle_group_unref(ts_foo);
> +    g_assert(removed);
> +    removed = throttle_group_unref(ts_bar);
> +    g_assert(removed);
> +
> +    g_assert((int64_t) throttle_group_token(ts_foo, true) == 0x7);
> +
> +    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((int64_t) throttle_group_token(ts_bar, false) == 0x5);
> +
> +    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 +583,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.1
>
diff mbox

Patch

diff --git a/tests/test-throttle.c b/tests/test-throttle.c
index 3e52df3..ecb5504 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,55 @@  static void test_accounting(void)
                                 (64.0 / 13)));
 }
 
+static void test_groups(void)
+{
+    bool removed;
+
+    ThrottleState *ts_foo, *ts_bar, *tmp;
+
+    ts_bar = throttle_group_incref("bar");
+    throttle_group_set_token(ts_bar, (BlockDriverState *) 0x5, false);
+    ts_foo = throttle_group_incref("foo");
+
+    tmp = throttle_group_incref("foo");
+    throttle_group_set_token(tmp, (BlockDriverState *) 0x7, 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((int64_t) throttle_group_token(ts_bar, false) == 0x5);
+    g_assert((int64_t) throttle_group_token(ts_foo, true) == 0x7);
+
+    removed = throttle_group_unref(ts_foo);
+    g_assert(removed);
+    removed = throttle_group_unref(ts_bar);
+    g_assert(removed);
+
+    g_assert((int64_t) throttle_group_token(ts_foo, true) == 0x7);
+
+    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((int64_t) throttle_group_token(ts_bar, false) == 0x5);
+
+    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 +583,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();
 }