From 24cdf520fd49e16e78f47935043c652ab7515f73 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=C3=89loi=20Rivard?= Date: Thu, 24 Oct 2024 09:53:09 +0200 Subject: [PATCH] =?UTF-8?q?tests:=C2=A0properly=20teardown=20connections?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/test_main.py | 39 +++++++++++++++++++++++++++------------ 1 file changed, 27 insertions(+), 12 deletions(-) diff --git a/tests/test_main.py b/tests/test_main.py index 20e99f3..5574681 100755 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -94,11 +94,14 @@ def _fill(self, sess): class QuerySelectFieldTest(TestBase): def setUp(self): - engine = create_engine("sqlite:///:memory:", echo=False) - self.Session = sessionmaker(bind=engine) + self.engine = create_engine("sqlite:///:memory:", echo=False) + self.Session = sessionmaker(bind=self.engine) from sqlalchemy.orm import mapper - self._do_tables(mapper, engine) + self._do_tables(mapper, self.engine) + + def tearDown(self): + self.engine.dispose() def test_without_factory(self): sess = self.Session() @@ -239,12 +242,16 @@ class QuerySelectMultipleFieldTest(TestBase): def setUp(self): from sqlalchemy.orm import mapper - engine = create_engine("sqlite:///:memory:", echo=False) - Session = sessionmaker(bind=engine) - self._do_tables(mapper, engine) + self.engine = create_engine("sqlite:///:memory:", echo=False) + Session = sessionmaker(bind=self.engine) + self._do_tables(mapper, self.engine) self.sess = Session() self._fill(self.sess) + def tearDown(self): + self.sess.close() + self.engine.dispose() + class F(Form): a = QuerySelectMultipleField(get_label="name", widget=LazySelect()) @@ -343,12 +350,16 @@ class Student(Model): self.Student = Student self.Course = Course - engine = create_engine("sqlite:///:memory:", echo=False) - Session = sessionmaker(bind=engine) + self.engine = create_engine("sqlite:///:memory:", echo=False) + Session = sessionmaker(bind=self.engine) self.metadata = Model.metadata - self.metadata.create_all(bind=engine) + self.metadata.create_all(bind=self.engine) self.sess = Session() + def tearDown(self): + self.sess.close() + self.engine.dispose() + def test_auto_validators(self): course_form = model_form(self.Course, self.sess)() student_form = model_form(self.Student, self.sess)() @@ -449,12 +460,16 @@ class StudentDefaultScoreScalar(Model): self.StudentDefaultScoreCallable = StudentDefaultScoreCallable self.StudentDefaultScoreScalar = StudentDefaultScoreScalar - engine = create_engine("sqlite:///:memory:", echo=False) - Session = sessionmaker(bind=engine) + self.engine = create_engine("sqlite:///:memory:", echo=False) + Session = sessionmaker(bind=self.engine) self.metadata = Model.metadata - self.metadata.create_all(bind=engine) + self.metadata.create_all(bind=self.engine) self.sess = Session() + def tearDown(self): + self.sess.close() + self.engine.dispose() + def test_column_default_callable(self): student_form = model_form(self.StudentDefaultScoreCallable, self.sess)() self.assertEqual(student_form._fields["score"].default, 5)