def run_tests(self, *args, **kwargs): register_context_manager(EnvContextManager) # Ensure our db setup/teardown manager is registered if not (self.distributed or self.multiprocess): register_context_manager(DatabaseContextManager) return MuleTestLoader.run_tests(self, *args, **kwargs)