diff --git a/trove/tests/unittests/trove_testtools.py b/trove/tests/unittests/trove_testtools.py index 97d7899..9447537 100644 --- a/trove/tests/unittests/trove_testtools.py +++ b/trove/tests/unittests/trove_testtools.py @@ -19,11 +19,18 @@ import mock import os import sys import testtools +import time from trove.common.context import TroveContext from trove.common.notification import DBaaSAPINotification from trove.tests import root_logger +import atexit +TIME_LOST_IN_DETECTION = 0.0 +def goodbye(): + print("Time lost in detection: %.1f sec" % TIME_LOST_IN_DETECTION) +atexit.register(goodbye) + def patch_notifier(test_case): notification_notify = mock.patch.object( @@ -117,10 +124,14 @@ class TestCase(testtools.TestCase): self.fail(self._NEWLINE + self._NEWLINE.join(messages)) def _find_mock_refs(self): + global TIME_LOST_IN_DETECTION + start = time.time() discovered_mocks = set() for module_name, module in self._get_loaded_modules().items(): self._find_mocks(module_name, module, discovered_mocks, 1) + dt = time.time() - start + TIME_LOST_IN_DETECTION += dt return discovered_mocks def _find_mocks(self, parent_name, parent, container, depth):