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_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_print_tree(mock): transaction = Transaction(Mock(), Mock()) transaction.print_tree() mock.assert_called_with(transaction.cursor)
def test_it_takes_a_connection(): conn = Mock() assert Transaction(conn, Mock()).connection is conn
def test_clear(mock): transaction = Transaction(Mock(), Mock()) transaction.clear() 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_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_it_disables_autocommit(): conn = Mock() transaction = Transaction(conn, Mock()) assert transaction.connection.autocommit is False
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)