psycopg/psycopg2
1
1
Fork 0
mirror of https://github.com/psycopg/psycopg2.git synced 2024-11-22 17:06:33 +03:00
Code Issues Packages Projects Releases Wiki Activity

Added function to drop a test table in the doctest.

Browse Source
This commit is contained in:
Daniele Varrazzo 2010-02-14 11:57:20 +00:00
parent a54336f7b4
commit f134bceb00
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
doc/conf.py
View File

@ -241,13 +241,17 @@ def test_connect():
conn = test_connect()
cur = conn.cursor()
def create_test_table():
def drop_test_table(name):
cur.execute("SAVEPOINT drop_test_table;")
try:
cur.execute("CREATE TABLE test (id SERIAL PRIMARY KEY, num INT, data TEXT)")
cur.execute("DROP TABLE %s;" % name)
except:
conn.rollback()
cur.execute("DELETE FROM test")
cur.execute("SELECT setval('test_id_seq', 1, False)")
cur.execute("ROLLBACK TO SAVEPOINT drop_test_table;")
conn.commit()
def create_test_table():
drop_test_table('test')
cur.execute("CREATE TABLE test (id SERIAL PRIMARY KEY, num INT, data TEXT)")
conn.commit()
"""