diff --git a/tests/test_extras_dictcursor.py b/tests/test_extras_dictcursor.py index dd746379..34e571d3 100755 --- a/tests/test_extras_dictcursor.py +++ b/tests/test_extras_dictcursor.py @@ -35,6 +35,16 @@ class ExtrasDictCursorTests(unittest.TestCase): def tearDown(self): self.conn.close() + def testDictConnCursorArgs(self): + self.conn.close() + self.conn = psycopg2.connect(dsn, connection_factory=psycopg2.extras.DictConnection) + cur = self.conn.cursor() + self.assert_(isinstance(cur, psycopg2.extras.DictCursor)) + self.assertEqual(cur.name, None) + # overridable + cur = self.conn.cursor('foo', cursor_factory=psycopg2.extras.NamedTupleCursor) + self.assertEqual(cur.name, 'foo') + self.assert_(isinstance(cur, psycopg2.extras.NamedTupleCursor)) def testDictCursorWithPlainCursorFetchOne(self): self._testWithPlainCursor(lambda curs: curs.fetchone()) @@ -219,6 +229,12 @@ class NamedTupleCursorTest(unittest.TestCase): if self.conn is not None: self.conn.close() + @skip_if_no_namedtuple + def test_cursor_args(self): + cur = self.conn.cursor('foo', cursor_factory=psycopg2.extras.DictCursor) + self.assertEqual(cur.name, 'foo') + self.assert_(isinstance(cur, psycopg2.extras.DictCursor)) + @skip_if_no_namedtuple def test_fetchone(self): curs = self.conn.cursor()