From: Karl O. Pinc <kop@meme.com>
Date: Fri, 2 Nov 2018 00:51:37 +0000 (-0500)
Subject: Use context manager
X-Git-Url: https://papio.biology.duke.edu/gitweb/?a=commitdiff_plain;h=bf84a48501242800ba975240fd8a4327bc51b6ef;p=pgwui_server

Use context manager
---

diff --git a/src/pgwui_server/__init__.py b/src/pgwui_server/__init__.py
index eaab761..dde886e 100644
--- a/src/pgwui_server/__init__.py
+++ b/src/pgwui_server/__init__.py
@@ -27,9 +27,8 @@ from pyramid.config import Configurator
 
 def pgwui_server_config(settings):
     rp = settings.get('pgwui.route_prefix')
-    config = Configurator(settings=settings,
-                          route_prefix=rp)
-    config.include('pgwui_common')
+    with Configurator(settings=settings, route_prefix=rp) as config:
+        config.include('pgwui_common')
     return config
 
 
diff --git a/tests/test___init__.py b/tests/test___init__.py
index 8a54296..3c0e666 100644
--- a/tests/test___init__.py
+++ b/tests/test___init__.py
@@ -22,10 +22,17 @@
 import pgwui_server.__init__ as pgwui_server_init
 
 
+# Use contextlib.AbstractContextManager for Python >= 3.6
 class MockConfigurator():
     def __init__(self, **kwargs):
         pass
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, *args):
+        pass
+
     def make_wsgi_app(self):
         return 'wsgi_app'