Python Tree.get_connectionの例

コード例 #1

ファイル: test_tree.py プロジェクト: vortec/libtree

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

コード例 #2

ファイル: test_tree.py プロジェクト: vortec/libtree

def test_get_connection_returns_the_assigned_one():
    conn = Mock()
    tree = Tree(connection=conn)
    assert tree.get_connection() is conn