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 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_get_node(mock):
transaction = Transaction(Mock(), Mock())
transaction.get_node(1337)
mock.assert_called_with(transaction.cursor, 1337)
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)
def test_get_node(mock):
transaction = Transaction(Mock(), Mock())
transaction.get_node(1337)
mock.assert_called_with(transaction.cursor, 1337)
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')
def test_clear(mock):
transaction = Transaction(Mock(), Mock())
transaction.clear()
mock.assert_called_with(transaction.cursor)
def test_get_tree_size(mock):
transaction = Transaction(Mock(), Mock())
transaction.get_tree_size()
mock.assert_called_with(transaction.cursor)
def test_uninstall(mock):
transaction = Transaction(Mock(), Mock())
transaction.uninstall()
mock.assert_called_with(transaction.cursor)
def test_install(mock1, mock2):
transaction = Transaction(Mock(), Mock())
transaction.install()
mock1.assert_called_with(transaction.cursor)
mock2.assert_called_with(transaction.cursor)
def test_is_compatible_postgres_version(mock):
transaction = Transaction(Mock(), Mock())
transaction.is_compatible_postgres_version()
mock.assert_called_with(transaction.cursor)
def test_install(mock1, mock2):
transaction = Transaction(Mock(), Mock())
transaction.install()
mock1.assert_called_with(transaction.cursor)
mock2.assert_called_with(transaction.cursor)
def test_rollback():
conn = Mock()
Transaction(conn, Mock()).rollback()
assert conn.rollback.called
def test_commit():
conn = Mock()
Transaction(conn, Mock()).commit()
assert conn.commit.called
def test_it_creates_a_cursor():
conn = Mock()
Transaction(conn, Mock())
assert conn.cursor.called
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')
def test_clear(mock):
transaction = Transaction(Mock(), Mock())
transaction.clear()
mock.assert_called_with(transaction.cursor)
def test_is_compatible_postgres_version(mock):
transaction = Transaction(Mock(), Mock())
transaction.is_compatible_postgres_version()
mock.assert_called_with(transaction.cursor)
def test_it_takes_a_connection():
conn = Mock()
assert Transaction(conn, Mock()).connection is conn
def test_uninstall(mock):
transaction = Transaction(Mock(), Mock())
transaction.uninstall()
mock.assert_called_with(transaction.cursor)
def test_print_tree(mock):
transaction = Transaction(Mock(), Mock())
transaction.print_tree()
mock.assert_called_with(transaction.cursor)
def test_print_tree(mock):
transaction = Transaction(Mock(), Mock())
transaction.print_tree()
mock.assert_called_with(transaction.cursor)
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)
def test_insert_root_node(mock):
transaction = Transaction(Mock(), Mock())
transaction.insert_root_node()
mock.assert_called_with(transaction.cursor, None)
def test_insert_root_node(mock):
transaction = Transaction(Mock(), Mock())
transaction.insert_root_node()
mock.assert_called_with(transaction.cursor, None)
def test_get_tree_size(mock):
transaction = Transaction(Mock(), Mock())
transaction.get_tree_size()
mock.assert_called_with(transaction.cursor)
def test_it_disables_autocommit():
conn = Mock()
transaction = Transaction(conn, Mock())
assert transaction.connection.autocommit is False
