From edbb32649960db29761e765fc74c025e9999c7dd Mon Sep 17 00:00:00 2001 From: Mike Bayer Date: Tue, 15 Mar 2016 18:01:23 -0400 Subject: - ensure engine is restored even if we had a failure in cleanup for stray connection --- lib/sqlalchemy/testing/plugin/plugin_base.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) (limited to 'lib/sqlalchemy/testing/plugin') diff --git a/lib/sqlalchemy/testing/plugin/plugin_base.py b/lib/sqlalchemy/testing/plugin/plugin_base.py index 22b79d1b3..fd33c56a0 100644 --- a/lib/sqlalchemy/testing/plugin/plugin_base.py +++ b/lib/sqlalchemy/testing/plugin/plugin_base.py @@ -428,9 +428,11 @@ def stop_test_class(cls): #from sqlalchemy import inspect #assert not inspect(testing.db).get_table_names() engines.testing_reaper._stop_test_ctx() - if not options.low_connections: - assertions.global_cleanup_assertions() - _restore_engine() + try: + if not options.low_connections: + assertions.global_cleanup_assertions() + finally: + _restore_engine() def _restore_engine(): -- cgit v1.2.1