def test_has_reference_to_user_id(self) -> None:
feedback_models.UnsentFeedbackEmailModel(id='user_id').put()
self.assertTrue(
feedback_models.UnsentFeedbackEmailModel.has_reference_to_user_id(
'user_id'))
self.assertFalse(
feedback_models.UnsentFeedbackEmailModel.has_reference_to_user_id(
'id_x'))
def test_new_instances_stores_correct_data(self) -> None:
user_id = 'A'
message_reference_dict = {
'exploration_id': 'ABC123',
'thread_id': 'thread_id1',
'message_id': 'message_id1'
}
email_instance = feedback_models.UnsentFeedbackEmailModel(
id=user_id, feedback_message_references=[message_reference_dict])
email_instance.update_timestamps()
email_instance.put()
retrieved_instance = (
feedback_models.UnsentFeedbackEmailModel.get_by_id(id=user_id))
self.assertEqual(retrieved_instance.feedback_message_references,
[message_reference_dict])
self.assertEqual(retrieved_instance.retries, 0)
def setUp(self) -> None:
super().setUp()
feedback_models.UnsentFeedbackEmailModel(id='user_id').put()
