def __add_entry_to_case_wrapper(sender, data, event_type, entry_message, message_name): if isinstance(sender, dict): originator = sender['uid'] else: originator = sender.uid cases_to_add = case_subscription.get_cases_subscribed( originator, message_name) if cases_to_add: if not isinstance(data, string_types): try: data = json.dumps(data) except TypeError: data = str(data) event = Event(type=event_type, timestamp=datetime.datetime.utcnow(), originator=originator, message=entry_message, data=data) database.case_db.add_event(event, cases_to_add)
def test_get_cases_subscribed_multiple_cases_one_event_found(self): subs.set_subscriptions(self.cases1) self.assertSetEqual(set(subs.get_cases_subscribed('uid1', 'e2')), {'case1', 'case2'})
def test_get_cases_subscribed_no_event_found(self): subs.set_subscriptions(self.cases1) self.assertSetEqual(set(subs.get_cases_subscribed('uid1', 'invalid')), set())
def test_get_cases_subscribed_one_case_one_case_found(self): subs.set_subscriptions(self.cases2) self.assertSetEqual(set(subs.get_cases_subscribed('uid4', 'd')), {'case3'})
def test_get_cases_subscribed_no_cases(self): self.assertSetEqual(set(subs.get_cases_subscribed('uid1', 'e')), set())