Fix imports in testing.py

Shoobx · May 6, 2021 · 5264a5e · 5264a5e
1 parent 15fc99a
commit 5264a5e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -6,7 +6,7 @@ CHANGES
 3.0.1 (unreleased)
 ------------------
 
-- Nothing changed yet.
+- Fix imports in `testing.py`
 
 
 3.0.0 (2021-02-22)

diff --git a/src/pjpersist/testing.py b/src/pjpersist/testing.py
@@ -18,6 +18,7 @@
 import os
 import psycopg2
 import psycopg2.extras
+import psycopg2.pool
 import re
 import sys
 import threading
@@ -28,7 +29,7 @@
 import zope.component
 from zope.testing import module, renormalizing
 
-from pjpersist import datamanager, serialize, serializers, interfaces
+from pjpersist import datamanager, serialize, interfaces
 
 py3checkers = [
     # Mangle unicode strings
@@ -67,7 +68,7 @@ def __init__(self, conn):
 
     def getconn(self):
         if self._available is None:
-            raise PoolError("Connection is already taken")
+            raise psycopg2.pool.PoolError("Connection is already taken")
         self._available.reset()
         self._available.set_isolation_level(
             psycopg2.extensions.ISOLATION_LEVEL_SERIALIZABLE)