@Before protected void setUp() throws Exception { Mockery contractServiceMockery = new Mockery(); final IContractService contractService = contractServiceMockery.mock(IContractService.class); contractServiceMockery.checking( new Expectations() { { allowing(contractService).searchOperaLogByWorkflowId("1"); List<ContractOperatorLog> contractLogList = new ArrayList<ContractOperatorLog>(); ContractOperatorLog c = new ContractOperatorLog(); c.setCreateUser("1"); contractLogList.add(c); will(returnValue(contractLogList)); } }); contractWorkflowManager.setContractService(contractService); Mockery contractWorkflowServiceMockery = new Mockery(); final IContractWorkflowService contractWorkflowService = contractWorkflowServiceMockery.mock(IContractWorkflowService.class); contractWorkflowServiceMockery.checking( new Expectations() { { allowing(contractWorkflowService) .queryContractWorkflowInfoByBusino(with(any(String.class))); will(returnValue(new ContractWorkflowInfo())); } }); contractWorkflowManager.setContractWorkflowService(contractWorkflowService); Mockery employeeServiceMockery = new Mockery(); final IEmployeeService employeeService = employeeServiceMockery.mock(IEmployeeService.class); employeeServiceMockery.checking( new Expectations() { { allowing(employeeService).getEmpById("1"); Employee employee = new Employee(); employee.setEmpCode("1"); will(returnValue(employee)); } }); contractWorkflowManager.setEmployeeService(employeeService); Mockery userServiceMockery = new Mockery(); final IUserService userService = userServiceMockery.mock(IUserService.class); userServiceMockery.checking( new Expectations() { { allowing(userService).findByLoginName("1"); User user = new User(); user.setId("1"); will(returnValue(user)); allowing(userService).getUserRoleFunDeptById("1"); will(returnValue(user)); } }); contractWorkflowManager.setUserService(userService); }
@Test public void testFindContractWorkflowInfoByWorkNo() { contractWorkflowManager.findContractWorkflowInfoByWorkNo(null, "1"); contractWorkflowManager.findContractWorkflowInfoByWorkNo("1", "1"); contractWorkflowManager.getContractService(); contractWorkflowManager.getContractWorkflowService(); contractWorkflowManager.getUserService(); contractWorkflowManager.getEmployeeService(); }
public void testSearchOperaDeptByWorkflowId() { contractWorkflowManager.searchOperaDeptByWorkflowId("1"); }