@Test public void testGetEndWorkflowActions() { // given final List<WorkflowActionModel> actions = new ArrayList<WorkflowActionModel>(); actions.add(mock(WorkflowActionModel.class)); when(workflowActionDao.findEndWorkflowActions(workflow)).thenReturn(actions); // when final List<WorkflowActionModel> endWorkflowActions = workflowActionService.getEndWorkflowActions(workflow); // then assertThat(endWorkflowActions).isSameAs(actions); }