def test_actions_are_claimed(purge_accounts_and_actions, patched_worker): with session_scope_by_shard_id(0) as db_session: account = make_imap_account(db_session, '{}@test.com'.format(0)) schedule_test_action(db_session, account) with session_scope_by_shard_id(1) as db_session: account = make_imap_account(db_session, '{}@test.com'.format(1)) schedule_test_action(db_session, account) service = SyncbackService(cpu_id=1, total_cpus=2) service.workers = set() service._process_log() gevent.joinall(list(service.workers)) with session_scope_by_shard_id(0) as db_session: q = db_session.query(ActionLog) assert q.count() == 1 assert all(a.status == 'pending' for a in q) with session_scope_by_shard_id(1) as db_session: q = db_session.query(ActionLog) assert q.count() == 1 assert all(a.status != 'pending' for a in q)
def test_actions_claimed_by_a_single_service(purge_accounts_and_actions, patched_worker): actionlogs = [] for key in (0, 1): with session_scope_by_shard_id(key) as db_session: account = make_imap_account(db_session, '{}@test.com'.format(key)) schedule_test_action(db_session, account) actionlogs += [db_session.query(ActionLog).one().id] services = [] for cpu_id in (0, 1): service = SyncbackService(cpu_id=cpu_id, total_cpus=2) service.workers = set() service._process_log() services.append(service) for i, service in enumerate(services): assert len(service.workers) == 1 assert list(service.workers)[0].action_log_id == actionlogs[i] gevent.joinall(list(service.workers))