def test_get_all_case_owner_ids(self): self.assertEqual( get_all_case_owner_ids(self.domain), set(case.user_id for case in self.cases if case.domain == self.domain) ) # sanity check! self.assertEqual( get_all_case_owner_ids(self.domain), {'XXX', 'ZZZ'}, )
def get_all_case_ids_for_domain(domain, start, end): case_ids = [] for owner_id in with_progress_bar(get_all_case_owner_ids(domain)): case_ids.extend( get_case_ids_modified_with_owner_since(domain, owner_id, start, end)) return case_ids
def set_cleanliness_flags_for_domain(domain, force_full=False): """ Sets all cleanliness flags for an entire domain. """ for owner_id in get_all_case_owner_ids(domain): if owner_id and owner_id not in WEIRD_USER_IDS: set_cleanliness_flags(domain, owner_id, force_full=force_full)
def set_cleanliness_flags_for_domain(domain, force_full=False): """ Sets all cleanliness flags for an entire domain. """ for owner_id in get_all_case_owner_ids(domain): if owner_id and owner_id not in WEIRD_USER_IDS: try: set_cleanliness_flags(domain, owner_id, force_full=force_full) except InvalidOwnerIdError as e: notify_exception(None, unicode(e))
def set_cleanliness_flags_for_domain(domain, force_full=False): """ Sets all cleanliness flags for an entire domain. """ for owner_id in get_all_case_owner_ids(domain): if owner_id and owner_id not in WEIRD_USER_IDS: try: set_cleanliness_flags(domain, owner_id, force_full=force_full) except InvalidOwnerIdError as e: notify_exception(None, unicode(e))
def get_case_owner_ids(domain): return get_all_case_owner_ids(domain)
def get_case_owner_ids(domain): return get_all_case_owner_ids(domain)
def test_get_all_case_owner_ids(self): self.assertEqual( get_all_case_owner_ids(self.domain), set(case.user_id for case in self.cases if case.domain == self.domain))
owner_id='XXX', closed=True)), { case.get_id for case in self.cases if case.domain == self.domain and case.user_id == 'XXX' and case.closed is True }) def test_get_all_case_owner_ids(self): self.assertEqual( get_all_case_owner_ids(self.domain), set(case.user_id for case in self.cases if case.domain == self.domain)) # sanity check! self.assertEqual( get_all_case_owner_ids(self.domain), {'XXX', 'ZZZ'}, ) TEST_ES_META = {CASE_INDEX_INFO.index: CASE_INDEX_INFO} class ESAccessorsTest(TestCase): domain = 'hqadmin-es-accessor' def setUp(self): super(ESAccessorsTest, self).setUp() with trap_extra_setup(ConnectionError): self.elasticsearch = get_es_new() initialize_index_and_mapping(self.elasticsearch, CASE_INDEX_INFO)
def get_all_case_ids_for_domain(domain, start, end): case_ids = [] for owner_id in with_progress_bar(get_all_case_owner_ids(domain)): case_ids.extend(get_case_ids_modified_with_owner_since(domain, owner_id, start, end)) return case_ids