Close all sessions to prevent tests hanging. Cleanup logging.

common/test/__init__.py

@@ -1,13 +1,13 @@
 import unittest
 import json
 import yaml
+import logging
 import importlib
 import http.client
 
 from os import path
 from operator import itemgetter
 from flask.json import dumps
-from flask.logging import create_logger
 from flask.testing import FlaskClient
 from flask.wrappers import Response
 from contextlib import wraps
@@ -87,7 +87,7 @@ def setUp(self):
         assert hasattr(self, 'app'), 'Please provide a reference to the Flask app'
         assert hasattr(self, 'db'), 'Please provide a reference to the Flask SQLAlchemy object'
 
-        self.logger = create_logger(self.app)
+        self.logger = logging.getLogger(self.__class__.__name__)
 
         # Configure app for testing
         self.app.testing = True
@@ -160,10 +160,9 @@ def tearDown(self):
         up the SQLAlchemy session.
         """
 
-        # Need to manually rollback the transaction because, of course, Flask does not do this
-        # automatically (on error or any other reason). Flask manages the session but has no opinion on
-        # transaction handling.
-        self.db.session.rollback()
+        # Tearing down all sessions seems drastic, but simply rolling back the current transaction does not
+        # terminate all database connections, which can cause the tests to hang.
+        self.db.session.close_all()
         self.context.pop()
 
     def assertEntitiesContain(self, actual_entities, expected_entities):

setup.py

@@ -1,7 +1,7 @@
 from setuptools import setup
 
 setup(name='common',
-      version='0.0.9',
+      version='0.1.1',
       author='Unascribed',
       author_email='[email protected]',
       description='Code intended to be used across Polymath Ventures repositories.',