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)