tasks/cephfs: write_n_mb optionally background

Signed-off-by: John Spray <john.spray@redhat.com>

diff --git a/tasks/cephfs/mount.py b/tasks/cephfs/mount.py
index 832d0002e54017637177cc5480cb76830681ca9a..f714238fffb93fbc4aed6eeb85abf2920a3b3784 100644 (file)
--- a/tasks/cephfs/mount.py
+++ b/tasks/cephfs/mount.py
@@ -281,17 +281,17 @@ class CephFSMount(object):
         self.background_procs.append(rproc)
         return rproc
 
-    def write_n_mb(self, filename, n_mb, seek=0):
+    def write_n_mb(self, filename, n_mb, seek=0, wait=True):
         """
         Write the requested number of megabytes to a file
         """
         assert(self.is_mounted())
 
-        self.run_shell(["dd", "if=/dev/urandom", "of={0}".format(filename),
-                        "bs=1M",
-                        "count={0}".format(n_mb),
-                        "seek={0}".format(seek)
-        ])
+        return self.run_shell(["dd", "if=/dev/urandom", "of={0}".format(filename),
+                               "bs=1M",
+                               "count={0}".format(n_mb),
+                               "seek={0}".format(seek)
+                               ], wait=wait)
 
     def write_test_pattern(self, filename, size):
         log.info("Writing {0} bytes to {1}".format(size, filename))