ProjectTaskTransformerTest

package com.talosdigital.timetracking.endpoint.transformer;

import static org.junit.Assert.assertEquals;
import static org.mockito.Matchers.any;
import static org.mockito.Mockito.when;

import java.util.Random;

import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.runners.MockitoJUnitRunner;

import com.talosdigital.timetracking.endpoint.dto.ProjectTaskDto;
import com.talosdigital.timetracking.persistence.dao.ProjectDao;
import com.talosdigital.timetracking.persistence.entity.Project;
import com.talosdigital.timetracking.persistence.entity.ProjectTask;

@RunWith(MockitoJUnitRunner.class)
public class ProjectTaskTransformerTest implements TransformerTest {

    private static final Random RANDOM_GENERATOR = new Random();

    @InjectMocks
    private ProjectTaskTransformer projectTastTransformer;

    @Mock
    private ProjectDao projectDao;

    private ProjectTask projectTask;
    private ProjectTaskDto projectTaskDto;

    @Before
    public void setUp() {
        Project project = new Project();
        project.setId(RANDOM_GENERATOR.nextInt());
        project.setAbbreviation("PJT");

        projectTask = new ProjectTask();
        projectTask.setId(RANDOM_GENERATOR.nextInt());
        projectTask.setName("Project task name");
        projectTask.setEnabled(true);
        projectTask.setProject(project);

        projectTaskDto = new ProjectTaskDto();
        projectTaskDto.setId(RANDOM_GENERATOR.nextInt());
        projectTaskDto.setName("Project task dto name");
        projectTaskDto.setEnabled(true);
        projectTaskDto.setProjectId(project.getId());

        when(projectDao.get(any(int.class))).thenReturn(project);
    }

    @Test
    @Override
    public void toDto() {
        ProjectTaskDto projectTaskDto = projectTastTransformer.toDto(projectTask);
        assertEquals(projectTask.getId(), projectTaskDto.getId());
        assertEquals(generateProjectTaskDtoName(projectTask), projectTaskDto.getName());
        assertEquals(projectTask.isEnabled(), projectTaskDto.isEnabled());
        assertEquals(projectTask.getProject().getId(), projectTaskDto.getProjectId());
    }

    @Test
    @Override
    public void toEntity() {
        ProjectTask projectTask = projectTastTransformer.toEntity(projectTaskDto);
        assertEquals(projectTaskDto.getId(), projectTask.getId());
        assertEquals(projectTaskDto.getName(), generateProjectTaskDtoName(projectTask));
        assertEquals(projectTaskDto.isEnabled(), projectTask.isEnabled());
        assertEquals(projectTaskDto.getProjectId(), projectTask.getProject().getId());
    }

    private static String generateProjectTaskDtoName(ProjectTask projectTask) {
        return projectTask.getProject().getAbbreviation() + "-" + projectTask.getName();
    }
}