def test_terminate_process_no_ckp(pg_conn):
orig = create_location(pg_conn, 'Origin')
dest = create_location(pg_conn, 'Dest')
ctc = create_contact(pg_conn, 'Subject')
vhc = create_vehicle(pg_conn, 'UUU0000')
pk = create_process(pg_conn)
time.sleep(0.01)
update_process(pg_conn, pk, ctc, orig, dest, None, vhc)
with pytest.raises(Exception):
terminate_process(pg_conn, pk)
def test_terminate_process_no_ctc(pg_conn):
orig = create_location(pg_conn, 'Origin')
dest = create_location(pg_conn, 'Dest')
ckp = create_checkpoint(pg_conn, 'Gate 1')
vhc = create_vehicle(pg_conn, 'UUU0000')
pk = create_process(pg_conn)
time.sleep(0.01)
update_process(pg_conn, pk, None, orig, dest, ckp, vhc)
with pytest.raises(Exception):
terminate_process(pg_conn, pk)
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']
def test_create_process_auth_events_on_finished_process_err(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')
proc_id = create_process(pg_conn)
time.sleep(0.1)
update_process(pg_conn, proc_id, ctc, orig, dest, ckp, vhc)
terminate_process(pg_conn, proc_id)
with pytest.raises(Exception):
create_process_auth_event(pg_conn, proc_id, 0, 'sys/none')
def test_update_process_finished_err(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)
time.sleep(0.01)
update_process(pg_conn, pk, ctc, orig, dest, ckp, vhc)
terminate_process(pg_conn, pk)
with pytest.raises(Exception):
update_process(pg_conn, pk, ctc, orig, dest, ckp, vhc)
def test_process_archive_with_unresolved_terminated_process_should_archive_none(
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)
time.sleep(0.1)
update_process(pg_conn, pk, ctc, orig, dest, ckp, vhc)
terminate_process(pg_conn, pk)
results = archive_finished_processes(pg_conn)
assert results == []
def test_process_archive_with_resolved_terminated_process_should_archive_one(
pg_conn, basic_flow):
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)
time.sleep(0.1)
update_process(pg_conn, pk, ctc, orig, dest, ckp, vhc)
create_process_auth_event(pg_conn, pk, ctc, 'sys/acc')
terminate_process(pg_conn, pk)
results = archive_finished_processes(pg_conn)
assert results[0]['archived_proc_id'] == pk
def test_terminate_process(pg_conn):
created_proc_id = dql_binds.create_process(pg_conn)
created_ctc_id = dql_binds.create_contact(pg_conn, 'Joe')
created_orig_id = dql_binds.create_location(pg_conn, 'Addr 1')
created_dest_id = dql_binds.create_location(pg_conn, 'Addr 2')
created_ckp_id = dql_binds.create_checkpoint(pg_conn, 'Gate 1')
result = dql_binds.update_process(pg_conn,
created_proc_id,
ctc=created_ctc_id,
loc_orig=created_orig_id,
loc_dest=created_dest_id,
ckp=created_ckp_id)
result = dql_binds.terminate_process(pg_conn, created_proc_id)
assert isinstance(result, datetime.datetime)
def test_add_location_group_member(pg_conn):
created_group_name = dql_binds.create_location_group(pg_conn,
group_name='group-1')
created_location_id = dql_binds.create_location(pg_conn, 'Address 1')
result = dql_binds.add_location_group_member(pg_conn, created_group_name,
created_location_id)
assert result is None
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
def test_remove_location_from_location_groups(pg_conn):
created_group_name = dql_binds.create_location_group(pg_conn,
group_name='group-1')
created_location_id = dql_binds.create_location(pg_conn, 'Address 1')
dql_binds.add_location_group_member(pg_conn, created_group_name,
created_location_id)
result = dql_binds.remove_location_from_location_groups(
pg_conn, created_location_id, (created_group_name, ))
assert result is None
def test_create_location(pg_conn):
created_id = dql_binds.create_location(pg_conn,
addr_1='Address 1',
addr_2='Address 2',
addr_3='Address 3',
addr_4='Address 4',
addr_5='Address 5',
addr_6='Address 6')
assert (isinstance(created_id, int))
def test_terminate_process_no_orig(pg_conn):
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)
time.sleep(0.01)
update_process(pg_conn, pk, ctc, None, dest, ckp, vhc)
with pytest.raises(Exception):
terminate_process(pg_conn, pk)
def test_assign_contact_to_location(pg_conn):
created_location_id = dql_binds.create_location(pg_conn, 'Address 1')
created_contact_id = dql_binds.create_contact(pg_conn,
contact_name='Joe Manzana')
created_assign_type = dql_binds.create_location_assign_type(
pg_conn, 'Owner')
result = dql_binds.assign_contact_to_location(pg_conn, created_location_id,
created_contact_id,
created_assign_type)
assert result is None
def test_retrieve_location_groups_by_location_id(pg_conn):
created_group_name = dql_binds.create_location_group(pg_conn,
group_name='group-1')
created_location_id = dql_binds.create_location(pg_conn, 'Address 1')
dql_binds.add_location_group_member(pg_conn, created_group_name,
created_location_id)
records = dql_binds.retrieve_location_groups_by_location_id(
pg_conn, created_location_id)
assert len(records) == 1
assert records[0]['group_name'] == 'group-1'
def test_update_location(pg_conn):
created_id = dql_binds.create_location(pg_conn, 'Address 1')
result = dql_binds.update_location(pg_conn,
addr_1=created_id,
addr_2=SQLDefault(),
addr_3=SQLDefault(),
addr_4=SQLDefault(),
addr_5=SQLDefault(),
addr_6=SQLDefault())
assert result is None
def test_retrieve_locations_by_ids(pg_conn):
created_id = dql_binds.create_location(pg_conn, 'Address 1')
records = dql_binds.retrieve_locations_by_ids(pg_conn, (created_id, ))
assert (len(records) == 1)
assert (records[0]['location_id'] == created_id)
assert (records[0]['addr_1'] == 'Address 1')
assert (records[0]['addr_2'] is None)
assert (records[0]['addr_3'] is None)
assert (records[0]['addr_4'] is None)
assert (records[0]['addr_5'] is None)
assert (records[0]['addr_6'] is None)
def test_retrieve_process_histories_by_ids(pg_conn):
dql_binds.create_auth_level(pg_conn, 'sys/acc', auth_grant=True)
dql_binds.create_auth_flow_st(pg_conn, 0, 'sys/acc', st_term=True)
dql_binds.create_auth_flow_tr(pg_conn, 0, 'sys/acc', 'sys/none', 'sys/acc')
created_proc_id = dql_binds.create_process(pg_conn)
ctc = dql_binds.create_contact(pg_conn, 'Joe')
loc_orig = dql_binds.create_location(pg_conn, 'Addr 1')
loc_dest = dql_binds.create_location(pg_conn, 'Addr 2')
ckp = dql_binds.create_checkpoint(pg_conn, 'Gate 1')
dql_binds.update_process(pg_conn,
created_proc_id,
ctc=ctc,
loc_orig=loc_orig,
loc_dest=loc_dest,
ckp=ckp)
dql_binds.create_process_auth_event(pg_conn, created_proc_id, ctc,
'sys/acc')
result = dql_binds.retrieve_process_histories_by_ids(
pg_conn, (created_proc_id, ))
assert result
def test_archive_finished_processes(pg_conn):
dql_binds.create_auth_level(pg_conn, 'sys/acc', auth_grant=True)
dql_binds.create_auth_flow_st(pg_conn, 0, 'sys/acc', st_term=True)
dql_binds.create_auth_flow_tr(pg_conn, 0, 'sys/acc', 'sys/none', 'sys/acc')
created_proc_id = dql_binds.create_process(pg_conn)
ctc = dql_binds.create_contact(pg_conn, 'Joe')
loc_orig = dql_binds.create_location(pg_conn, 'Addr 1')
loc_dest = dql_binds.create_location(pg_conn, 'Addr 2')
ckp = dql_binds.create_checkpoint(pg_conn, 'Gate 1')
dql_binds.update_process(pg_conn,
created_proc_id,
ctc=ctc,
loc_orig=loc_orig,
loc_dest=loc_dest,
ckp=ckp)
dql_binds.create_process_auth_event(pg_conn, created_proc_id, ctc,
'sys/acc')
time.sleep(0.1)
dql_binds.terminate_process(pg_conn, created_proc_id)
result = dql_binds.archive_finished_processes(pg_conn)
assert result
def test_retrieve_location_group_members(pg_conn):
created_group_name = dql_binds.create_location_group(pg_conn,
group_name='group-1')
created_location_id = dql_binds.create_location(pg_conn, 'Address 1')
dql_binds.add_location_group_member(pg_conn, created_group_name,
created_location_id)
records = dql_binds.retrieve_location_group_members(
pg_conn, created_group_name)
assert len(records) == 1
assert records[0]['addr_1'] == 'Address 1'
assert records[0]['addr_2'] is None
assert records[0]['addr_3'] is None
assert records[0]['addr_4'] is None
assert records[0]['addr_5'] is None
assert records[0]['addr_6'] is None
def test_retrieve_location_contact_assignments_by_contact_id(pg_conn):
created_location_id = dql_binds.create_location(pg_conn, 'Address 1')
created_contact_id = dql_binds.create_contact(pg_conn,
contact_name='Joe Manzana')
created_assign_type = dql_binds.create_location_assign_type(
pg_conn, 'Owner')
dql_binds.assign_contact_to_location(pg_conn, created_location_id,
created_contact_id,
created_assign_type)
records = dql_binds.retrieve_location_contact_assignments_by_contact_id(
pg_conn, created_contact_id)
assert len(records) == 1
assert records[0]['location_id'] == created_location_id
assert records[0]['addr_1'] == 'Address 1'
assert records[0]['assign_type'] == 'Owner'
def loc_nogrp(pg_conn, locgrp_b):
return create_location(pg_conn, 'loc_nogrp')
def test_retrieve_locations_by_ids(pg_conn):
addrs = ['Place 1', 'Place 2', 'Place 3', 'Place 4', 'Place 5']
pks = [create_location(pg_conn, addr) for addr in addrs]
rs = retrieve_locations_by_ids(pg_conn, tuple(pks))
assert [r['location_id'] for r in rs] == pks
assert [r['addr_1'] for r in rs] == addrs
def loc_a2(pg_conn):
return create_location(pg_conn, 'loc_a2')
def test_delete_locations_by_ids(pg_conn):
created_id = dql_binds.create_location(pg_conn, 'Address 1')
result = dql_binds.delete_locations_by_ids(pg_conn, (created_id, ))
assert result is None
def loc_b1(pg_conn, locgrp_b):
loc = create_location(pg_conn, 'loc_b1')
add_location_group_member(pg_conn, 'locgrp_b', loc)
return loc
def test_update_location(pg_conn):
pk = create_location(pg_conn, 'Place 1')
update_location(pg_conn, pk, addr_1='Nowhere')
rs = retrieve_locations_by_ids(pg_conn, (pk, ))
assert rs[0]['addr_1'] == 'Nowhere'
def test_update_location_pk_should_err(pg_conn):
pk = create_location(pg_conn, 'Place 1')
with pytest.raises(Exception):
pg_conn.execute(
'update ac_locations set location_id=1088 where location_id = %(pk)s',
{'pk': pk})
def test_delete_locations_by_ids(pg_conn):
pk = create_location(pg_conn, 'Place 1')
delete_locations_by_ids(pg_conn, (pk, ))
rs = retrieve_locations_by_ids(pg_conn, (pk, ))
assert len(rs) == 0
def test_create_location(pg_conn):
pk = create_location(pg_conn, 'Place 1')
rs = retrieve_locations_by_ids(pg_conn, (pk, ))
assert len(rs) == 1
assert rs[0]['location_id'] == pk
assert rs[0]['addr_1'] == 'Place 1'
