Tests: wait for limit to be set in set_disk_free_limit_command_test

Author: dcorbacho

deps/rabbitmq_cli/test/ctl/set_disk_free_limit_command_test.exs

@@ -104,8 +104,12 @@ defmodule SetDiskFreeLimitCommandTest do
   test "run: a valid integer input returns an ok and sets the disk free limit", context do
     set_disk_free_limit(@default_limit)
     assert @command.run([context[:limit]], context[:opts]) == :ok
-    Process.sleep(500)
-    assert status()[:disk_free_limit] === context[:limit]
+    await_condition(
+        fn ->
+           status()[:disk_free_limit] === context[:limit]
+        end,
+        60000
+    )
 
     set_disk_free_limit(@default_limit)
   end
@@ -115,8 +119,12 @@ defmodule SetDiskFreeLimitCommandTest do
        context do
     set_disk_free_limit(@default_limit)
     assert @command.run([context[:limit]], context[:opts]) == :ok
-    Process.sleep(500)
-    assert status()[:disk_free_limit] === round(context[:limit])
+    await_condition(
+        fn ->
+           status()[:disk_free_limit] === round(context[:limit])
+        end,
+        60000
+    )
 
     set_disk_free_limit(@default_limit)
   end
@@ -126,8 +134,12 @@ defmodule SetDiskFreeLimitCommandTest do
        context do
     set_disk_free_limit(@default_limit)
     assert @command.run([context[:limit]], context[:opts]) == :ok
-    Process.sleep(500)
-    assert status()[:disk_free_limit] === context[:limit] |> Float.floor() |> round
+    await_condition(
+        fn ->
+           status()[:disk_free_limit] === context[:limit] |> Float.floor() |> round
+        end,
+        60000
+    )
 
     set_disk_free_limit(@default_limit)
   end
@@ -136,17 +148,25 @@ defmodule SetDiskFreeLimitCommandTest do
   test "run: an integer string input returns an ok and sets the disk free limit", context do
     set_disk_free_limit(@default_limit)
     assert @command.run([context[:limit]], context[:opts]) == :ok
-    Process.sleep(500)
-    assert status()[:disk_free_limit] === String.to_integer(context[:limit])
+    await_condition(
+        fn ->
+           status()[:disk_free_limit] === String.to_integer(context[:limit])
+        end,
+        60000
+    )
 
     set_disk_free_limit(@default_limit)
   end
 
   @tag limit: "2MB"
   test "run: an valid unit string input returns an ok and changes the limit", context do
     assert @command.run([context[:limit]], context[:opts]) == :ok
-    Process.sleep(500)
-    assert status()[:disk_free_limit] === 2_000_000
+    await_condition(
+        fn ->
+           status()[:disk_free_limit] === 2_000_000
+        end,
+        60000
+    )
 
     set_disk_free_limit(@default_limit)
   end