commit b540954e20a82c69ac5704de78f1855e47c1247e
author Hoa Nguyen <hoanguyen@ucdavis.edu> Thu Sep 03 01:35:24 2020 -0700
committer Hoa Nguyen <hoanguyen@ucdavis.edu> Tue Sep 08 18:28:17 2020 +0000
tree 3fdd62d007389a3fe2dbcdce7f8779cd9bf1c995
parent c207628d1e45741f3874286b3177d5cc00204ead

ext: Add post_test_procedure to testlib runner

This procedure is trigger after a test has finished and before
the tearing down process kicks in.

Change-Id: I58ce10814fbc80d96f2f72565491b18de0ec290a
Signed-off-by: Hoa Nguyen <hoanguyen@ucdavis.edu>
Reviewed-on: https://gem5-review.googlesource.com/c/public/gem5/+/33997
Tested-by: kokoro <noreply+kokoro@google.com>
Reviewed-by: Jason Lowe-Power <power.jg@gmail.com>
Maintainer: Jason Lowe-Power <power.jg@gmail.com>

ext/testlib/fixture.py

diff --git a/ext/testlib/fixture.py b/ext/testlib/fixture.py
index 79b57c0..bcd22d9 100644
--- a/ext/testlib/fixture.py
+++ b/ext/testlib/fixture.py
@@ -70,6 +70,9 @@
     def setup(self, testitem):
         pass
 
+    def post_test_procedure(self, testitem):
+        pass
+
     def teardown(self, testitem):
         pass
 

ext/testlib/runner.py

diff --git a/ext/testlib/runner.py b/ext/testlib/runner.py
index a59aca3..ee658c9 100644
--- a/ext/testlib/runner.py
+++ b/ext/testlib/runner.py
@@ -128,6 +128,7 @@
             self.testable.status = Status.Running
             self.test()
         finally:
+            self.builder.post_test_procedure(self.testable)
             self.testable.status = Status.TearingDown
             self.builder.teardown(self.testable)
 
@@ -209,6 +210,10 @@
                 raise BrokenFixtureException(fixture, testitem,
                         traceback.format_exc())
 
+    def post_test_procedure(self, testitem):
+        for fixture in self.built_fixtures:
+            fixture.post_test_procedure(testitem)
+
     def teardown(self, testitem):
         for fixture in self.built_fixtures:
             try: