def trans(request, dsn): connection = psycopg2.connect(dsn) transaction = Transaction(connection, Node) node_ids.clear() transaction.install() transaction.commit() def fin(): transaction.uninstall() transaction.commit() request.addfinalizer(fin) return transaction
def test_install(mock1, mock2): transaction = Transaction(Mock(), Mock()) transaction.install() mock1.assert_called_with(transaction.cursor) mock2.assert_called_with(transaction.cursor)