Index: client/tests/kvm/tests/file_transfer.py |
diff --git a/client/tests/kvm/tests/file_transfer.py b/client/tests/kvm/tests/file_transfer.py |
index f02078e854547d50e6dd7976aaa3ee0e6cb6b973..e872beda17569d8644b521085b89445bda73584f 100644 |
--- a/client/tests/kvm/tests/file_transfer.py |
+++ b/client/tests/kvm/tests/file_transfer.py |
@@ -24,17 +24,13 @@ def run_file_transfer(test, params, env): |
if not session: |
raise error.TestFail("Could not log into guest '%s'" % vm.name) |
- dir_name = test.tmpdir |
+ dir = test.tmpdir |
transfer_timeout = int(params.get("transfer_timeout")) |
transfer_type = params.get("transfer_type") |
tmp_dir = params.get("tmp_dir", "/tmp/") |
clean_cmd = params.get("clean_cmd", "rm -f") |
filesize = int(params.get("filesize", 4000)) |
- count = int(filesize / 10) |
- if count == 0: |
- count = 1 |
- cmd = "dd if=/dev/zero of=%s/a.out bs=10M count=%d" % (dir_name, |
- count) |
+ cmd = "dd if=/dev/urandom of=%s/a.out bs=1M count=%d" % (dir, filesize) |
guest_path = tmp_dir + "b.out" |
try: |
@@ -45,7 +41,7 @@ def run_file_transfer(test, params, env): |
logging.info("Transfering file host -> guest, timeout: %ss", |
transfer_timeout) |
t_begin = time.time() |
- success = vm.copy_files_to("%s/a.out" % dir_name, guest_path, |
+ success = vm.copy_files_to("%s/a.out" % dir, guest_path, |
timeout=transfer_timeout) |
t_end = time.time() |
throughput = filesize / (t_end - t_begin) |
@@ -57,7 +53,7 @@ def run_file_transfer(test, params, env): |
logging.info("Transfering file guest -> host, timeout: %ss", |
transfer_timeout) |
t_begin = time.time() |
- success = vm.copy_files_from(guest_path, "%s/c.out" % dir_name, |
+ success = vm.copy_files_from(guest_path, "%s/c.out" % dir, |
timeout=transfer_timeout) |
t_end = time.time() |
throughput = filesize / (t_end - t_begin) |
@@ -70,12 +66,12 @@ def run_file_transfer(test, params, env): |
transfer_type) |
for f in ['a.out', 'c.out']: |
- p = os.path.join(dir_name, f) |
+ p = os.path.join(dir, f) |
size = os.path.getsize(p) |
logging.debug('Size of %s: %sB', f, size) |
- md5_orig = utils.hash_file("%s/a.out" % dir_name, method="md5") |
- md5_new = utils.hash_file("%s/c.out" % dir_name, method="md5") |
+ md5_orig = utils.hash_file("%s/a.out" % dir, method="md5") |
+ md5_new = utils.hash_file("%s/c.out" % dir, method="md5") |
if md5_orig != md5_new: |
raise error.TestFail("File changed after transfer host -> guest " |
@@ -84,11 +80,11 @@ def run_file_transfer(test, params, env): |
finally: |
logging.info('Cleaning temp file on guest') |
clean_cmd += " %s" % guest_path |
- session.cmd(clean_cmd) |
+ s, o = session.get_command_status_output(clean_cmd) |
+ if s: |
+ logging.warning("Failed to clean remote file %s, output:%s", |
+ guest_path, o) |
logging.info('Cleaning temp files on host') |
- try: |
- os.remove('%s/a.out' % dir_name) |
- os.remove('%s/c.out' % dir_name) |
- except OSError: |
- pass |
+ os.remove('%s/a.out' % dir) |
+ os.remove('%s/c.out' % dir) |
session.close() |