def test_get_handler_ids_for_case_post_save(self):
reminder1 = self._create_reminder(self.domain_1, self.case_type_1)
reminder2 = self._create_reminder(self.domain_1, self.case_type_2)
reminder3 = self._create_reminder(self.domain_2, self.case_type_1)
self.addCleanup(reminder1.delete)
self.addCleanup(reminder2.delete)
self.addCleanup(reminder3.delete)
self.assertEqual(
CaseReminderHandler.get_handler_ids_for_case_post_save(
self.domain_1, self.case_type_1), [reminder1._id])
self.assertEqual(
CaseReminderHandler.get_handler_ids_for_case_post_save(
self.domain_1, self.case_type_2), [reminder2._id])
self.assertEqual(
CaseReminderHandler.get_handler_ids_for_case_post_save(
self.domain_2, self.case_type_1), [reminder3._id])
# Test cache clear
reminder4 = self._create_reminder(self.domain_1, self.case_type_1)
self.addCleanup(reminder4.delete)
result = CaseReminderHandler.get_handler_ids_for_case_post_save(
self.domain_1, self.case_type_1)
self.assertEqual(len(result), 2)
self.assertEqual(set(result), set([reminder1._id, reminder4._id]))
