def test_remove_checkpoint_group_members(pg_conn, ckp_a1, ckp_a2):
create_checkpoint_group(pg_conn, 'group-1')
add_checkpoint_group_member(pg_conn, 'group-1', ckp_a1)
add_checkpoint_group_member(pg_conn, 'group-1', ckp_a2)
remove_checkpoint_group_members(pg_conn, 'group-1', (ckp_a1, ))
rs = retrieve_checkpoint_group_members(pg_conn, 'group-1')
assert rs == [{'checkpoint_id': ckp_a2, 'checkpoint_name': 'ckp_a2'}]
def test_retrieve_checkpoint_groups_by_checkpoint_id(pg_conn, ckp_a1):
create_checkpoint_group(pg_conn, 'group-1')
create_checkpoint_group(pg_conn, 'group-2')
add_checkpoint_group_member(pg_conn, 'group-1', ckp_a1)
add_checkpoint_group_member(pg_conn, 'group-2', ckp_a1)
rs = retrieve_checkpoint_groups_by_checkpoint_id(pg_conn, ckp_a1)
assert rs == [{'group_name': 'group-1'}, {'group_name': 'group-2'}]
def test_remove_checkpoint_from_checkpoint_groups(pg_conn):
created_group_name = dql_binds.create_checkpoint_group(
pg_conn, group_name='group-1')
created_checkpoint_id = dql_binds.create_checkpoint(pg_conn, 'Entrance A')
dql_binds.add_checkpoint_group_member(pg_conn, created_group_name,
created_checkpoint_id)
result = dql_binds.remove_checkpoint_from_checkpoint_groups(
pg_conn, created_checkpoint_id, (created_group_name, ))
assert result is None
def test_retrieve_checkpoint_groups_by_checkpoint_id(pg_conn):
created_group_name = dql_binds.create_checkpoint_group(
pg_conn, group_name='group-1')
created_checkpoint_id = dql_binds.create_checkpoint(pg_conn, 'Entrance A')
dql_binds.add_checkpoint_group_member(pg_conn, created_group_name,
created_checkpoint_id)
records = dql_binds.retrieve_checkpoint_groups_by_checkpoint_id(
pg_conn, created_checkpoint_id)
assert len(records) == 1
assert records[0]['group_name'] == 'group-1'
def test_remove_checkpoint_from_checkpoint_groups(pg_conn, ckp_a1, ckp_a2):
create_checkpoint_group(pg_conn, 'group-1')
create_checkpoint_group(pg_conn, 'group-2')
create_checkpoint_group(pg_conn, 'group-3')
add_checkpoint_group_member(pg_conn, 'group-1', ckp_a1)
add_checkpoint_group_member(pg_conn, 'group-2', ckp_a1)
add_checkpoint_group_member(pg_conn, 'group-3', ckp_a1)
add_checkpoint_group_member(pg_conn, 'group-1', ckp_a2)
remove_checkpoint_from_checkpoint_groups(pg_conn, ckp_a1,
('group-1', 'group-3'))
rs = retrieve_checkpoint_groups_by_checkpoint_id(pg_conn, ckp_a1)
assert rs == [{'group_name': 'group-2'}]
rs = retrieve_checkpoint_groups_by_checkpoint_id(pg_conn, ckp_a2)
assert rs == [{'group_name': 'group-1'}]
def ckp_b1(pg_conn, ckpgrp_b):
ckp = create_checkpoint(pg_conn, 'ckp_b1')
add_checkpoint_group_member(pg_conn, 'ckpgrp_b', ckp)
return ckp
def test_add_checkpoint_group_member(pg_conn, ckp_a1):
create_checkpoint_group(pg_conn, 'group-1')
add_checkpoint_group_member(pg_conn, 'group-1', ckp_a1)
