From 882442dfccac8bb1b599d4aee6939a32fdd1ebb0 Mon Sep 17 00:00:00 2001 From: Kamil Páral Date: Feb 24 2012 13:41:06 +0000 Subject: rats_sanity: fix crash --- diff --git a/tests/rats_sanity/rats_sanity.py b/tests/rats_sanity/rats_sanity.py index 6440da0..87d7078 100644 --- a/tests/rats_sanity/rats_sanity.py +++ b/tests/rats_sanity/rats_sanity.py @@ -19,7 +19,6 @@ import os import pickle -from autoqa import util from autotest_lib.client.bin import utils from autotest_lib.client.common_lib import error from autoqa.test import AutoQATest @@ -60,6 +59,9 @@ class rats_sanity(AutoQATest): self.detail.result = 'FAILED' # grab extra key = val data - keyval_file = open(os.path.join(self.resultsdir, 'keyval')) - keyvals = pickle.load(keyval_file) - self.write_test_keyval(keyvals) + keyval = os.path.join(self.resultsdir, 'keyval') + if os.path.exists(keyval): + keyval_file = open(keyval) + keyvals = pickle.load(keyval_file) + self.write_test_keyval(keyvals) + keyval_file.close() diff --git a/tests/rats_sanity/sanity.py b/tests/rats_sanity/sanity.py index a46cf5d..52ba6af 100755 --- a/tests/rats_sanity/sanity.py +++ b/tests/rats_sanity/sanity.py @@ -23,7 +23,6 @@ import os import yum -import sys import optparse import pickle from autoqa import util @@ -242,6 +241,6 @@ if __name__ == '__main__': print "\n== testing complete ==" # save keyvals - keyval_file = open(opt.keyval_file) + keyval_file = open(opt.keyval_file, 'w') pickle.dump(keyvals, keyval_file) keyval_file.close()