def test_returned_transaction_uses_connection_from_pool():
pool, conn = Mock(), Mock()
pool.getconn.return_value = conn
tree = Tree(pool=pool)
transaction = tree.make_transaction()
assert transaction.connection is conn
def test_make_transaction_returns_a_transaction_object():
conn = Mock()
tree = Tree(connection=conn)
transaction = tree.make_transaction()
assert transaction.__class__ == Transaction
def test_returned_transaction_uses_assigned_transaction_object():
conn = Mock()
tree = Tree(connection=conn)
transaction = tree.make_transaction()
assert transaction.connection is conn
def test_make_transaction_returns_a_transaction_object():
conn = Mock()
tree = Tree(connection=conn)
transaction = tree.make_transaction()
assert transaction.__class__ == Transaction
def test_returned_transaction_uses_assigned_transaction_object():
conn = Mock()
tree = Tree(connection=conn)
transaction = tree.make_transaction()
assert transaction.connection is conn
def test_it_requires_either_connection_or_pool():
with pytest.raises(TypeError):
Tree()
def test_it_cant_deal_with_both_connection_and_pool():
with pytest.raises(TypeError):
Tree(connection=Mock(), pool=Mock())
def test_close_connection():
conn = Mock()
Tree(connection=conn).close()
assert conn.close.called
def test_it_takes_a_connection_pool(dsn):
pool = Mock()
assert Tree(pool=pool).pool is pool
def test_it_takes_a_connection():
conn = Mock()
assert Tree(connection=conn).connection is conn
def test_cm_commits_transaction():
conn = Mock()
tree = Tree(connection=conn)
with tree() as transaction: # noqa
pass
assert conn.commit.called
def test_returned_transaction_uses_connection_from_pool():
pool, conn = Mock(), Mock()
pool.getconn.return_value = conn
tree = Tree(pool=pool)
transaction = tree.make_transaction()
assert transaction.connection is conn
def test_make_transaction_returns_a_read_write_transaction_object():
conn = Mock()
tree = Tree(connection=conn)
transaction = tree.make_transaction(write=True)
assert transaction.__class__ == ReadWriteTransaction
def test_get_connection_returns_connection_from_pool():
pool, conn = Mock(), Mock()
pool.getconn.return_value = conn
tree = Tree(pool=pool)
assert tree.get_connection() is conn
assert pool.getconn.called
def test_get_connection_returns_the_assigned_one():
conn = Mock()
tree = Tree(connection=conn)
assert tree.get_connection() is conn
def test_close_pool():
pool = Mock()
Tree(pool=pool).close()
assert pool.closeall.called