diff --git a/HtmlTestRunner/result.py b/HtmlTestRunner/result.py index 96fb431..9ce03cc 100644 --- a/HtmlTestRunner/result.py +++ b/HtmlTestRunner/result.py @@ -144,6 +144,7 @@ def __init__(self, stream, descriptions, verbosity): self.callback = None self.infoclass = _TestInfo self.report_files = [] + self.images = {} def _prepare_callback(self, test_info, target_list, verbose_str, short_str): @@ -192,6 +193,9 @@ def stopTest(self, test): if self.callback and callable(self.callback): self.callback() self.callback = None + + if hasattr(test, 'screenshot'): + self.images[test._testMethodName] = test.screenshot def addSuccess(self, test): """ Called when a test executes successfully. """ diff --git a/HtmlTestRunner/template/report_template.html b/HtmlTestRunner/template/report_template.html index 4c2256b..5d5a8ee 100644 --- a/HtmlTestRunner/template/report_template.html +++ b/HtmlTestRunner/template/report_template.html @@ -55,9 +55,11 @@