Ejemplos de Transaction en Python

Ejemplo n.º 1

Archivo: conftest.py Proyecto: pombredanne/libtree

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

Ejemplo n.º 2

Archivo: conftest.py Proyecto: pombredanne/libtree

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

Ejemplo n.º 3

def test_get_node(mock):
    transaction = Transaction(Mock(), Mock())
    transaction.get_node(1337)
    mock.assert_called_with(transaction.cursor, 1337)

Ejemplo n.º 4

Archivo: test_transaction.py Proyecto: pombredanne/libtree

def test_get_nodes_by_property_dict(mock):
    mock.return_value = [Mock(id=1), Mock(id=2)]
    transaction = Transaction(Mock(), Mock())
    query = {'key': 'value'}
    transaction.get_nodes_by_property_dict(query)
    mock.assert_called_with(transaction.cursor, query)

Ejemplo n.º 5

Archivo: test_transaction.py Proyecto: pombredanne/libtree

def test_get_node(mock):
    transaction = Transaction(Mock(), Mock())
    transaction.get_node(1337)
    mock.assert_called_with(transaction.cursor, 1337)

Ejemplo n.º 6

def test_get_nodes_by_property_value(mock):
    mock.return_value = [Mock(id=1), Mock(id=2)]
    transaction = Transaction(Mock(), Mock())
    transaction.get_nodes_by_property_value('key', 'value')
    mock.assert_called_with(transaction.cursor, 'key', 'value')

Ejemplo n.º 7

Archivo: test_transaction.py Proyecto: pombredanne/libtree

def test_clear(mock):
    transaction = Transaction(Mock(), Mock())
    transaction.clear()
    mock.assert_called_with(transaction.cursor)

Ejemplo n.º 8

Archivo: test_transaction.py Proyecto: pombredanne/libtree

def test_get_tree_size(mock):
    transaction = Transaction(Mock(), Mock())
    transaction.get_tree_size()
    mock.assert_called_with(transaction.cursor)

Ejemplo n.º 9

def test_uninstall(mock):
    transaction = Transaction(Mock(), Mock())
    transaction.uninstall()
    mock.assert_called_with(transaction.cursor)

Ejemplo n.º 10

Archivo: test_transaction.py Proyecto: pombredanne/libtree

def test_install(mock1, mock2):
    transaction = Transaction(Mock(), Mock())
    transaction.install()
    mock1.assert_called_with(transaction.cursor)
    mock2.assert_called_with(transaction.cursor)

Ejemplo n.º 11

def test_is_compatible_postgres_version(mock):
    transaction = Transaction(Mock(), Mock())
    transaction.is_compatible_postgres_version()
    mock.assert_called_with(transaction.cursor)

Ejemplo n.º 12

def test_install(mock1, mock2):
    transaction = Transaction(Mock(), Mock())
    transaction.install()
    mock1.assert_called_with(transaction.cursor)
    mock2.assert_called_with(transaction.cursor)

Ejemplo n.º 13

def test_rollback():
    conn = Mock()
    Transaction(conn, Mock()).rollback()
    assert conn.rollback.called

Ejemplo n.º 14

def test_commit():
    conn = Mock()
    Transaction(conn, Mock()).commit()
    assert conn.commit.called

Ejemplo n.º 15

def test_it_creates_a_cursor():
    conn = Mock()
    Transaction(conn, Mock())
    assert conn.cursor.called

Ejemplo n.º 16

Archivo: test_transaction.py Proyecto: pombredanne/libtree

def test_get_nodes_by_property_value(mock):
    mock.return_value = [Mock(id=1), Mock(id=2)]
    transaction = Transaction(Mock(), Mock())
    transaction.get_nodes_by_property_value('key', 'value')
    mock.assert_called_with(transaction.cursor, 'key', 'value')

Ejemplo n.º 17

def test_clear(mock):
    transaction = Transaction(Mock(), Mock())
    transaction.clear()
    mock.assert_called_with(transaction.cursor)

Ejemplo n.º 18

Archivo: test_transaction.py Proyecto: pombredanne/libtree

def test_is_compatible_postgres_version(mock):
    transaction = Transaction(Mock(), Mock())
    transaction.is_compatible_postgres_version()
    mock.assert_called_with(transaction.cursor)

Ejemplo n.º 19

def test_it_takes_a_connection():
    conn = Mock()
    assert Transaction(conn, Mock()).connection is conn

Ejemplo n.º 20

Archivo: test_transaction.py Proyecto: pombredanne/libtree

def test_uninstall(mock):
    transaction = Transaction(Mock(), Mock())
    transaction.uninstall()
    mock.assert_called_with(transaction.cursor)

Ejemplo n.º 21

def test_print_tree(mock):
    transaction = Transaction(Mock(), Mock())
    transaction.print_tree()
    mock.assert_called_with(transaction.cursor)

Ejemplo n.º 22

Archivo: test_transaction.py Proyecto: pombredanne/libtree

def test_print_tree(mock):
    transaction = Transaction(Mock(), Mock())
    transaction.print_tree()
    mock.assert_called_with(transaction.cursor)

Ejemplo n.º 23

def test_get_nodes_by_property_dict(mock):
    mock.return_value = [Mock(id=1), Mock(id=2)]
    transaction = Transaction(Mock(), Mock())
    query = {'key': 'value'}
    transaction.get_nodes_by_property_dict(query)
    mock.assert_called_with(transaction.cursor, query)

Ejemplo n.º 24

Archivo: test_transaction.py Proyecto: pombredanne/libtree

def test_insert_root_node(mock):
    transaction = Transaction(Mock(), Mock())
    transaction.insert_root_node()
    mock.assert_called_with(transaction.cursor, None)

Ejemplo n.º 25

def test_insert_root_node(mock):
    transaction = Transaction(Mock(), Mock())
    transaction.insert_root_node()
    mock.assert_called_with(transaction.cursor, None)

Ejemplo n.º 26

def test_get_tree_size(mock):
    transaction = Transaction(Mock(), Mock())
    transaction.get_tree_size()
    mock.assert_called_with(transaction.cursor)

Ejemplo n.º 27

def test_it_disables_autocommit():
    conn = Mock()
    transaction = Transaction(conn, Mock())
    assert transaction.connection.autocommit is False