From 51c7005dc5cc04a9ed24fbfa290e50df6fb70369 Mon Sep 17 00:00:00 2001 From: Mike Bayer Date: Sat, 30 Aug 2014 00:11:32 -0400 Subject: - clean up zoomark a little and try to get new profiles written --- lib/sqlalchemy/testing/replay_fixture.py | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) (limited to 'lib/sqlalchemy/testing') diff --git a/lib/sqlalchemy/testing/replay_fixture.py b/lib/sqlalchemy/testing/replay_fixture.py index b8a0f6df1..b50f52e3d 100644 --- a/lib/sqlalchemy/testing/replay_fixture.py +++ b/lib/sqlalchemy/testing/replay_fixture.py @@ -29,9 +29,11 @@ class ReplayFixtureTest(fixtures.TestBase): self.session = Session(engine) self.setup_engine() - self._run_steps(ctx=self._dummy_ctx) - self.teardown_engine() - engine.dispose() + try: + self._run_steps(ctx=self._dummy_ctx) + finally: + self.teardown_engine() + engine.dispose() player = lambda: dbapi_session.player() engine = create_engine( @@ -43,8 +45,11 @@ class ReplayFixtureTest(fixtures.TestBase): self.session = Session(engine) self.setup_engine() - self._run_steps(ctx=profiling.count_functions) - self.teardown_engine() + try: + self._run_steps(ctx=profiling.count_functions) + finally: + self.session.close() + engine.dispose() def setup_engine(self): pass -- cgit v1.2.1