aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorAng Li <angli@google.com>2023-01-13 15:20:37 -0800
committerGitHub <noreply@github.com>2023-01-13 15:20:37 -0800
commit1c44fd97c968c5d1eba4ee6a1ef988b814a150d9 (patch)
treed36b051eb65c90d38e605a1d929d25ae9441eadf
parentf7f323e319d8ef66eb7ddf8ee7f5cb17d395b481 (diff)
downloadmobly-1c44fd97c968c5d1eba4ee6a1ef988b814a150d9.tar.gz
Ensure the termination of the `teardown_class` record. (#864)
-rw-r--r--mobly/base_test.py1
-rwxr-xr-xtests/mobly/base_test_test.py1
2 files changed, 2 insertions, 0 deletions
diff --git a/mobly/base_test.py b/mobly/base_test.py
index 547591d..e7da22c 100644
--- a/mobly/base_test.py
+++ b/mobly/base_test.py
@@ -471,6 +471,7 @@ class BaseTestClass:
records.TestSummaryEntryType.RECORD)
else:
if expects.recorder.has_error:
+ record.test_error()
record.update_record()
self.results.add_class_error(record)
self.summary_writer.dump(record.to_dict(),
diff --git a/tests/mobly/base_test_test.py b/tests/mobly/base_test_test.py
index d7e5ab5..be67437 100755
--- a/tests/mobly/base_test_test.py
+++ b/tests/mobly/base_test_test.py
@@ -1728,6 +1728,7 @@ class BaseTestTest(unittest.TestCase):
self.assertEqual(actual_record.test_name, 'teardown_class')
self.assertEqual(actual_record.details, MSG_EXPECTED_EXCEPTION)
self.assertEqual(actual_record.extras, MOCK_EXTRA)
+ self.assertIsNotNone(actual_record.end_time)
def test_expect_in_setup_test(self):
must_call = mock.Mock()