def trans(request):
dsn = make_dsn_from_env(os.environ)
connection = psycopg2.connect(dsn)
transaction = ReadWriteTransaction(connection, Node)
if transaction.is_installed():
transaction.uninstall()
node_ids.clear()
transaction.install()
transaction.commit()
def fin():
transaction.uninstall()
transaction.commit()
request.addfinalizer(fin)
return transaction
def trans(request):
dsn = make_dsn_from_env(os.environ)
connection = psycopg2.connect(dsn)
transaction = ReadWriteTransaction(connection, Node)
if transaction.is_installed():
transaction.uninstall()
node_ids.clear()
transaction.install()
transaction.commit()
def fin():
transaction.uninstall()
transaction.commit()
request.addfinalizer(fin)
return transaction
def test_uninstall(mock):
transaction = ReadWriteTransaction(Mock(), Mock())
transaction.uninstall()
mock.assert_called_with(transaction.cursor)
def test_uninstall(mock):
transaction = ReadWriteTransaction(Mock(), Mock())
transaction.uninstall()
mock.assert_called_with(transaction.cursor)
