@Override protected CollectorNodemanagerProtocol getNMCollectorService() { CollectorNodemanagerProtocol protocol = mock(CollectorNodemanagerProtocol.class); try { GetTimelineCollectorContextResponse response = GetTimelineCollectorContextResponse.newInstance( UserGroupInformation.getCurrentUser().getShortUserName(), "test_flow_name", "test_flow_version", 1L); when(protocol.getTimelineCollectorContext(any( GetTimelineCollectorContextRequest.class))).thenReturn(response); } catch (YarnException | IOException e) { fail(); } return protocol; } }
@Override protected CollectorNodemanagerProtocol getNMCollectorService() { CollectorNodemanagerProtocol protocol = mock(CollectorNodemanagerProtocol.class); try { GetTimelineCollectorContextResponse response = GetTimelineCollectorContextResponse.newInstance( UserGroupInformation.getCurrentUser().getUserName(), "test_flow_name", "test_flow_version", 1L); when(protocol.getTimelineCollectorContext(any( GetTimelineCollectorContextRequest.class))).thenReturn(response); } catch (YarnException | IOException e) { fail(); } return protocol; } }
private NodeTimelineCollectorManager createCollectorManager() { NodeTimelineCollectorManager collectorManager = spy(new NodeTimelineCollectorManager()); doReturn(new Configuration()).when(collectorManager).getConfig(); CollectorNodemanagerProtocol nmCollectorService = mock(CollectorNodemanagerProtocol.class); GetTimelineCollectorContextResponse response = GetTimelineCollectorContextResponse.newInstance(null, null, null, 0L); try { when(nmCollectorService.getTimelineCollectorContext(any( GetTimelineCollectorContextRequest.class))).thenReturn(response); } catch (YarnException | IOException e) { fail(); } doReturn(nmCollectorService).when(collectorManager).getNMCollectorService(); return collectorManager; }
private NodeTimelineCollectorManager createCollectorManager() { final NodeTimelineCollectorManager cm = spy(new NodeTimelineCollectorManager()); doReturn(new Configuration()).when(cm).getConfig(); CollectorNodemanagerProtocol nmCollectorService = mock(CollectorNodemanagerProtocol.class); GetTimelineCollectorContextResponse response = GetTimelineCollectorContextResponse.newInstance(null, null, null, 0L); try { when(nmCollectorService.getTimelineCollectorContext(any( GetTimelineCollectorContextRequest.class))).thenReturn(response); } catch (YarnException | IOException e) { fail(); } doReturn(nmCollectorService).when(cm).getNMCollectorService(); return cm; } }
@Override public GetTimelineCollectorContextResponse getTimelineCollectorContext( GetTimelineCollectorContextRequest request) throws YarnException, IOException { Application app = context.getApplications().get(request.getApplicationId()); if (app == null) { throw new YarnException("Application " + request.getApplicationId() + " doesn't exist on NM."); } return GetTimelineCollectorContextResponse.newInstance( app.getUser(), app.getFlowName(), app.getFlowVersion(), app.getFlowRunId()); } }
@Override public GetTimelineCollectorContextResponse getTimelineCollectorContext( GetTimelineCollectorContextRequest request) throws YarnException, IOException { if (request.getApplicationId().getId() == 1) { return GetTimelineCollectorContextResponse.newInstance( "test_user_id", "test_flow_name", "test_flow_version", 12345678L); } else { throw new YarnException("The application is not found."); } } }