private void registerServices() throws FalconException { mockTimeService = Mockito.mock(AlarmService.class); Mockito.when(mockTimeService.getName()).thenReturn("AlarmService"); Mockito.when( mockTimeService.createRequestBuilder( Mockito.any(NotificationHandler.class), Mockito.any(ID.class))) .thenCallRealMethod(); mockDataService = Mockito.mock(DataAvailabilityService.class); Mockito.when(mockDataService.getName()).thenReturn("DataAvailabilityService"); Mockito.when( mockDataService.createRequestBuilder( Mockito.any(NotificationHandler.class), Mockito.any(ID.class))) .thenCallRealMethod(); dagEngine = Mockito.mock(OozieDAGEngine.class); Mockito.doNothing().when(dagEngine).resume(Mockito.any(ExecutionInstance.class)); mockSchedulerService = Mockito.mock(SchedulerService.class); Mockito.when(mockSchedulerService.getName()).thenReturn("JobSchedulerService"); StartupProperties.get().setProperty("dag.engine.impl", MockDAGEngine.class.getName()); StartupProperties.get() .setProperty("execution.service.impl", FalconExecutionService.class.getName()); dagEngine = Mockito.spy(DAGEngineFactory.getDAGEngine("testCluster")); Mockito.when( mockSchedulerService.createRequestBuilder( Mockito.any(NotificationHandler.class), Mockito.any(ID.class))) .thenCallRealMethod(); mockCompletionService = Mockito.mock(JobCompletionService.class); Mockito.when(mockCompletionService.getName()).thenReturn("JobCompletionService"); Mockito.when( mockCompletionService.createRequestBuilder( Mockito.any(NotificationHandler.class), Mockito.any(ID.class))) .thenCallRealMethod(); Services.get().register(mockTimeService); Services.get().register(mockDataService); Services.get().register(mockSchedulerService); Services.get().register(mockCompletionService); }