Briq
/
jaydebeapi
6
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add integration tests for with statement

master
Marcin Szczepanski 2020-05-20 15:32:33 +02:00
parent e6a25206ee
commit 34be9556a9
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
test/test_integration.py
View File

@ -207,6 +207,17 @@ class IntegrationTestBase(object):
cursor.execute("select * from ACCOUNT") cursor.execute("select * from ACCOUNT")
self.assertEqual(cursor.rowcount, -1) self.assertEqual(cursor.rowcount, -1)
def test_connection_with_statement(self):
with self.connect() as conn:
self.assertEqual(conn._closed, False)
self.assertEqual(conn._closed, True)
def test_cursor_with_statement(self):
with self.conn.cursor() as cursor:
cursor.execute("select 1 from ACCOUNT")
self.assertIsNotNone(cursor._connection)
self.assertIsNone(cursor._connection)
class SqliteTestBase(IntegrationTestBase): class SqliteTestBase(IntegrationTestBase):
def setUpSql(self): def setUpSql(self):