Пример #1
0
    def test_to_record(self):
        """Test that an entity can be converted into a record."""
        entity = TaskMapper.to_entity(self.url_task)
        record = TaskMapper.to_record(entity)

        for c in record.__table__.columns:
            self.assertEquals(getattr(record, c.name),
                              getattr(self.url_task, c.name))
Пример #2
0
    def test_write_record_new(self):
        """Test that we can create a new record."""
        entity = TaskEntity.get_mock_object()

        TaskRepository.write_record(TaskMapper.to_record(entity))

        task = TaskRepository.fetch_task_by_id(entity.id)
        self.assertEquals(task.id, entity.id)
Пример #3
0
 def test_to_entity(self):
     """Test that a record can be converted into an entity."""
     self.assertEquals(
         TaskMapper.to_entity(self.url_task).to_native(), {
             'id': self.url_task.id,
             'type': self.url_task.type,
             'config': self.url_task.config,
         })
Пример #4
0
    def fetch_task_by_id(cls, id):
        """Fetch TaskEntity by id.

        :param str id:
        :returns TaskEntity:
        """
        return TaskMapper.to_entity(
            TaskRepository.fetch_task_by_id(id),
        )
Пример #5
0
    def write_task(cls, task):
        """Writes TaskEntity to the repository.

        :param TaskEntity task:
        """
        task.validate()

        TaskRepository.write_record(
            TaskMapper.to_record(task),
        )
Пример #6
0
    def test_write_record_new(self):
        """Test that we can create a new record."""
        entity = TaskEntity.get_mock_object()

        TaskRepository.write_record(
            TaskMapper.to_record(entity)
        )

        task = TaskRepository.fetch_task_by_id(entity.id)
        self.assertEquals(task.id, entity.id)