Python retrieve_processes_by_ids Examples

Example #1

File: test_crud_process.py Project: ccortezia/unigator

def test_create_process_should_use_auth_flow_setting(pg_conn):
    pk = create_process(pg_conn)
    rs = retrieve_processes_by_ids(pg_conn, (pk,))
    active_flow = retrieve_current_active_auth_flow(pg_conn)
    assert rs[0]['auth_flow'] == active_flow['auth_flow_id']

    auth_flow_1 = create_auth_flow(pg_conn, 'Flow 1')
    update_current_active_auth_flow(pg_conn, auth_flow_1)
    pk = create_process(pg_conn)
    rs = retrieve_processes_by_ids(pg_conn, (pk,))
    assert rs[0]['auth_flow'] == auth_flow_1

    auth_flow_2 = create_auth_flow(pg_conn, 'Flow 2')
    update_current_active_auth_flow(pg_conn, auth_flow_2)
    pk = create_process(pg_conn)
    rs = retrieve_processes_by_ids(pg_conn, (pk,))
    assert rs[0]['auth_flow'] == auth_flow_2

Example #2

File: test_crud_process.py Project: ccortezia/unigator

def test_terminate_process_no_vhc(pg_conn):
    orig = create_location(pg_conn, 'Origin')
    dest = create_location(pg_conn, 'Dest')
    ctc = create_contact(pg_conn, 'Subject')
    ckp = create_checkpoint(pg_conn, 'Gate 1')
    pk = create_process(pg_conn)
    time.sleep(0.01)
    update_process(pg_conn, pk, ctc, orig, dest, ckp, None)
    terminate_process(pg_conn, pk)
    rs = retrieve_processes_by_ids(pg_conn, (pk,))
    assert rs[0]['finished_at']

Example #3

File: test_crud_process.py Project: ccortezia/unigator

def test_update_process(pg_conn):
    orig = create_location(pg_conn, 'Origin')
    dest = create_location(pg_conn, 'Dest')
    ctc = create_contact(pg_conn, 'Subject')
    ckp = create_checkpoint(pg_conn, 'Gate 1')
    vhc = create_vehicle(pg_conn, 'UUU0000')
    pk = create_process(pg_conn)
    update_process(pg_conn, pk, ctc, orig, dest, ckp, vhc)
    rs = retrieve_processes_by_ids(pg_conn, (pk,))
    assert len(rs) == 1
    assert rs[0]['proc_id'] == pk
    assert rs[0]['ctc'] == ctc
    assert rs[0]['vhc'] == vhc
    assert rs[0]['ckp'] == ckp
    assert rs[0]['loc_orig'] == orig
    assert rs[0]['loc_dest'] == dest

Example #4

File: test_crud_process.py Project: ccortezia/unigator

def test_retrieve_processes_by_ids(pg_conn):
    pks = [create_process(pg_conn), create_process(pg_conn), create_process(pg_conn), create_process(pg_conn)]
    rs = retrieve_processes_by_ids(pg_conn, tuple(pks))
    assert len(rs) == 4
    assert [r['proc_id'] for r in rs] == pks

Example #5

File: test_crud_process.py Project: ccortezia/unigator

def test_create_process(pg_conn):
    pk = create_process(pg_conn)
    rs = retrieve_processes_by_ids(pg_conn, (pk,))
    assert len(rs) == 1
    assert rs[0]['proc_id'] == pk

Example #6

File: test_crud_process.py Project: ccortezia/unigator

def test_delete_processes_by_id(pg_conn):
    pks = [create_process(pg_conn), create_process(pg_conn), create_process(pg_conn), create_process(pg_conn)]
    delete_processes_by_ids(pg_conn, tuple(pks))
    rs = retrieve_processes_by_ids(pg_conn, tuple(pks))
    assert len(rs) == 0

Example #7

File: test_view_process.py Project: ccortezia/unigator

def test_retrieve_all_running_process_snapshots_blank_procs(pg_conn):
    pks = [create_process(pg_conn), create_process(pg_conn), create_process(pg_conn), create_process(pg_conn)]
    retrieve_all_running_process_snapshots(pg_conn)
    rs = retrieve_processes_by_ids(pg_conn, tuple(pks))
    assert len(rs) == 4
    assert [r['proc_id'] for r in rs] == pks

Example #8

File: test_all.py Project: ccortezia/unigator

def test_retrieve_processes_by_ids(pg_conn):
    created_proc_id = dql_binds.create_process(pg_conn)
    records = dql_binds.retrieve_processes_by_ids(pg_conn, (created_proc_id, ))
    assert len(records) == 1