From 9fb8a0ea4b62bae145fd3f8df04f4cc74eba29a8 Mon Sep 17 00:00:00 2001 From: Andrey Smirnov Date: Sat, 12 Jul 2014 21:53:47 +0400 Subject: [PATCH] Capturing results for other command output. --- system/lib.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/system/lib.py b/system/lib.py index 6a447a1b..56652338 100644 --- a/system/lib.py +++ b/system/lib.py @@ -200,7 +200,15 @@ class BaseTest(object): raise def check_cmd_output(self, command, gold_name, match_prepare=None, expected_code=0): - self.verify_match(self.get_gold(gold_name), self.run_cmd(command, expected_code=expected_code), match_prepare) + try: + output = self.run_cmd(command, expected_code=expected_code) + self.verify_match(self.get_gold(gold_name), output, match_prepare) + except: + if self.captureResults: + with open(self.get_gold_filename(gold_name), "w") as f: + f.write(output) + else: + raise def read_file(self, path): with open(os.path.join(os.environ["HOME"], ".aptly", path), "r") as f: