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())