def test_install_already_installed(mock1, mock2, mock3):
transaction = ReadWriteTransaction(Mock(), Mock())
transaction.install()
assert mock1.not_called()
assert mock2.not_called()
mock3.assert_called()
def test_install_already_installed(mock1, mock2, mock3):
transaction = ReadWriteTransaction(Mock(), Mock())
transaction.install()
assert mock1.not_called()
assert mock2.not_called()
mock3.assert_called()
def trans(request):
dsn = make_dsn_from_env(os.environ)
connection = psycopg2.connect(dsn)
transaction = ReadWriteTransaction(connection, Node)
if transaction.is_installed():
transaction.uninstall()
node_ids.clear()
transaction.install()
transaction.commit()
def fin():
transaction.uninstall()
transaction.commit()
request.addfinalizer(fin)
return transaction
def trans(request):
dsn = make_dsn_from_env(os.environ)
connection = psycopg2.connect(dsn)
transaction = ReadWriteTransaction(connection, Node)
if transaction.is_installed():
transaction.uninstall()
node_ids.clear()
transaction.install()
transaction.commit()
def fin():
transaction.uninstall()
transaction.commit()
request.addfinalizer(fin)
return transaction
def test_commit():
conn = Mock()
ReadWriteTransaction(conn, Mock()).commit()
assert conn.commit.called
def test_rollback():
conn = Mock()
ReadWriteTransaction(conn, Mock()).rollback()
assert conn.rollback.called
def test_it_sets_read_write_mode():
conn = Mock()
ReadWriteTransaction(conn, Mock())
conn.set_session.assert_called_with(readonly=False)
def test_it_creates_a_cursor():
conn = Mock()
ReadWriteTransaction(conn, Mock())
assert conn.cursor.called
def test_install(mock1, mock2, mock3):
transaction = ReadWriteTransaction(Mock(), Mock())
transaction.install()
mock1.assert_called_with(transaction.cursor)
mock2.assert_called_with(transaction.cursor)
mock3.assert_called()
def test_get_nodes_by_property_value(mock):
mock.return_value = [Mock(id=1), Mock(id=2)]
transaction = ReadWriteTransaction(Mock(), Mock())
transaction.get_nodes_by_property_value('key', 'value')
mock.assert_called_with(transaction.cursor, 'key', 'value')
def test_uninstall(mock):
transaction = ReadWriteTransaction(Mock(), Mock())
transaction.uninstall()
mock.assert_called_with(transaction.cursor)
def test_it_takes_a_connection():
conn = Mock()
assert ReadWriteTransaction(conn, Mock()).connection is conn
def test_get_tree_size(mock):
transaction = ReadWriteTransaction(Mock(), Mock())
transaction.get_tree_size()
mock.assert_called_with(transaction.cursor)
def test_get_nodes_by_property_dict(mock):
mock.return_value = [Mock(id=1), Mock(id=2)]
transaction = ReadWriteTransaction(Mock(), Mock())
query = {'key': 'value'}
transaction.get_nodes_by_property_dict(query)
mock.assert_called_with(transaction.cursor, query)
def test_clear(mock):
transaction = ReadWriteTransaction(Mock(), Mock())
transaction.clear()
mock.assert_called_with(transaction.cursor)
def test_uninstall(mock):
transaction = ReadWriteTransaction(Mock(), Mock())
transaction.uninstall()
mock.assert_called_with(transaction.cursor)
def test_get_nodes_by_property_value(mock):
mock.return_value = [Mock(id=1), Mock(id=2)]
transaction = ReadWriteTransaction(Mock(), Mock())
transaction.get_nodes_by_property_value('key', 'value')
mock.assert_called_with(transaction.cursor, 'key', 'value')
def test_is_compatible_postgres_version(mock):
transaction = ReadWriteTransaction(Mock(), Mock())
transaction.is_compatible_postgres_version()
mock.assert_called_with(transaction.cursor)
def test_get_tree_size(mock):
transaction = ReadWriteTransaction(Mock(), Mock())
transaction.get_tree_size()
mock.assert_called_with(transaction.cursor)
def test_install(mock1, mock2, mock3):
transaction = ReadWriteTransaction(Mock(), Mock())
transaction.install()
mock1.assert_called_with(transaction.cursor)
mock2.assert_called_with(transaction.cursor)
mock3.assert_called()
def test_get_node(mock):
transaction = ReadWriteTransaction(Mock(), Mock())
transaction.get_node(1337)
mock.assert_called_with(transaction.cursor, 1337)
def test_insert_root_node(mock):
transaction = ReadWriteTransaction(Mock(), Mock())
transaction.insert_root_node()
mock.assert_called_with(transaction.cursor, None, None)
def test_is_compatible_postgres_version(mock):
transaction = ReadWriteTransaction(Mock(), Mock())
transaction.is_compatible_postgres_version()
mock.assert_called_with(transaction.cursor)
def test_clear(mock):
transaction = ReadWriteTransaction(Mock(), Mock())
transaction.clear()
mock.assert_called_with(transaction.cursor)
def test_get_node(mock):
transaction = ReadWriteTransaction(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 = ReadWriteTransaction(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 = ReadWriteTransaction(Mock(), Mock())
transaction.insert_root_node()
mock.assert_called_with(transaction.cursor, None, None)
