示例#1
0
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
示例#2
0
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
示例#3
0
def test_install(mock1, mock2):
    transaction = Transaction(Mock(), Mock())
    transaction.install()
    mock1.assert_called_with(transaction.cursor)
    mock2.assert_called_with(transaction.cursor)
示例#4
0
def test_install(mock1, mock2):
    transaction = Transaction(Mock(), Mock())
    transaction.install()
    mock1.assert_called_with(transaction.cursor)
    mock2.assert_called_with(transaction.cursor)