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)