def test_put_for_task_entries_which_do_not_exist_creates_new_models(self):
open_task = self._new_open_task(state_name='Start')
obsolete_task = self._new_obsolete_task(state_name='Middle')
resolved_task = self._new_resolved_task(state_name='End')
improvements_services.put_tasks(
[open_task, obsolete_task, resolved_task])
open_task_model = (improvements_models.TaskEntryModel.get_by_id(
open_task.task_id))
obsolete_task_model = (improvements_models.TaskEntryModel.get_by_id(
obsolete_task.task_id))
resolved_task_model = (improvements_models.TaskEntryModel.get_by_id(
resolved_task.task_id))
self.assertEqual(
open_task.to_dict(),
improvements_services.get_task_entry_from_model(
open_task_model).to_dict())
self.assertEqual(
obsolete_task.to_dict(),
improvements_services.get_task_entry_from_model(
obsolete_task_model).to_dict())
self.assertEqual(
resolved_task.to_dict(),
improvements_services.get_task_entry_from_model(
resolved_task_model).to_dict())
def test_returns_same_fields_as_model(self):
task_id = improvements_models.TaskEntryModel.create(
improvements_models.TASK_ENTITY_TYPE_EXPLORATION, self.EXP_ID, 1,
improvements_models.TASK_TYPE_HIGH_BOUNCE_RATE,
improvements_models.TASK_TARGET_TYPE_STATE,
feconf.DEFAULT_INIT_STATE_NAME, 'issue description',
improvements_models.TASK_STATUS_RESOLVED, self.owner_id,
self.MOCK_DATE)
task_entry_model = improvements_models.TaskEntryModel.get_by_id(
task_id)
task_entry = (
improvements_services.get_task_entry_from_model(task_entry_model))
self.assertEqual(task_entry.task_id, task_entry_model.id)
self.assertEqual(task_entry.composite_entity_id,
task_entry_model.composite_entity_id)
self.assertEqual(task_entry.entity_type, task_entry_model.entity_type)
self.assertEqual(task_entry.entity_version,
task_entry_model.entity_version)
self.assertEqual(task_entry.task_type, task_entry_model.task_type)
self.assertEqual(task_entry.target_type, task_entry_model.target_type)
self.assertEqual(task_entry.target_id, task_entry_model.target_id)
self.assertEqual(task_entry.issue_description,
task_entry_model.issue_description)
self.assertEqual(task_entry.status, task_entry_model.status)
self.assertEqual(task_entry.resolver_id, task_entry_model.resolver_id)
self.assertEqual(task_entry.resolved_on, task_entry_model.resolved_on)