s/postgresql.driver.notifyman/postgresql.notifyman/g

James William Pye · James William Pye · commit 82c793aa05d0 · 2010-02-28T00:25:43.000-07:00
diff --git a/postgresql/test/test_driver.py b/postgresql/test/test_driver.py
@@ -1527,7 +1527,7 @@ def add(x):
 
 	# XXX: relocate some of these tests into their own module
 	def testNotificationManager(self):
-		from ..driver.notifyman import NotificationManager as NM
+		from ..notifyman import NotificationManager as NM
 		# signals each other
 		db = self.db
 		alt = self.db.clone()
@@ -1555,7 +1555,7 @@ def testNotificationManager(self):
 						self.fail("unknown connection received notify..")
 
 	def testNotificationManagerTimeout(self):
-		from ..driver.notifyman import NotificationManager as NM
+		from ..notifyman import NotificationManager as NM
 		nm = NM(self.db, timeout = 0.1)
 		self.db.listen('foo')
 		count = 0
@@ -1578,7 +1578,7 @@ def testNotificationManagerZeroTimeout(self):
 		# Zero-timeout means raise StopIteration when
 		# there are no notifications to emit.
 		# It checks the wire, but does *not* wait for data.
-		from ..driver.notifyman import NotificationManager as NM
+		from ..notifyman import NotificationManager as NM
 		nm = NM(self.db, timeout = 0)
 		self.db.listen('foo')
 		self.failUnlessEqual(list(nm), [])
@@ -1613,7 +1613,7 @@ def testNotificationManagerZeroTimeout(self):
 		# Zero-timeout means raise StopIteration when
 		# there are no notifications to emit.
 		# It checks the wire, but does *not* wait for data.
-		from ..driver.notifyman import NotificationManager as NM
+		from ..notifyman import NotificationManager as NM
 		self.db.listen('foo')
 		self.failUnlessEqual(list(self.db.wait(0)), [])
 		self.db.notify('foo')
