From 74eee3496c5c8e13582e2ce2f0794157a3dcc18b Mon Sep 17 00:00:00 2001
From: Andrey Smirnov <Smirnov.Andrey@gmail.com>
Date: Tue, 26 Aug 2014 23:58:25 +0400
Subject: [PATCH] Capture test results in prepared format. #81

---
 system/lib.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/system/lib.py b/system/lib.py
index a68d4cc2..ad787d9e 100644
--- a/system/lib.py
+++ b/system/lib.py
@@ -195,6 +195,8 @@ class BaseTest(object):
             self.verify_match(self.get_gold(), self.output, match_prepare=self.outputMatchPrepare)
         except:
             if self.captureResults:
+                if self.outputMatchPrepare is not None:
+                    self.output = self.outputMatchPrepare(self.output)
                 with open(self.get_gold_filename(), "w") as f:
                     f.write(self.output)
             else:
@@ -206,6 +208,8 @@ class BaseTest(object):
             self.verify_match(self.get_gold(gold_name), output, match_prepare)
         except:
             if self.captureResults:
+                if match_prepare is not None:
+                    output = match_prepare(output)
                 with open(self.get_gold_filename(gold_name), "w") as f:
                     f.write(output)
             else: