Ejemplo n.º 1
0
def test_make_dsn_from_env():
    env = {
        'PGHOST': 'localhost',
        'PGPORT': 5432,
        'PGUSER': '******',
        'PGPASSWORD': '******',
        'PGDATABASE': 'mydb'
    }

    conn_string = make_dsn_from_env(env)
    expected = 'host=localhost port=5432 user=foo password=secret dbname=mydb'
    assert conn_string == expected
Ejemplo n.º 2
0
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
Ejemplo n.º 3
0
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