/** * Test that the agent filtering is correct. */ @Test public void agentFilterDataProcessor() { AbstractDataProcessor abstractDataProcessor = mock(AbstractDataProcessor.class); AgentFilterDataProcessor dataProcessor = new AgentFilterDataProcessor(Collections.singletonList(abstractDataProcessor), Collections.singleton(10L)); DefaultData data1 = mock(DefaultData.class); DefaultData data2 = mock(DefaultData.class); when(data1.getPlatformIdent()).thenReturn(10L); when(data2.getPlatformIdent()).thenReturn(20L); dataProcessor.process(data1); dataProcessor.process(data2); assertThat(dataProcessor.canBeProcessed(data1), is(true)); assertThat(dataProcessor.canBeProcessed(data2), is(true)); verify(abstractDataProcessor, times(1)).process(data1); verify(abstractDataProcessor, times(0)).process(data2); }
if (!selectAgentsWizardPage.isAllAgents()) { Set<Long> agentsIds = new HashSet<>(selectAgentsWizardPage.getSelectedAgents()); AgentFilterDataProcessor agentFilterDataProcessor = new AgentFilterDataProcessor(recordingProcessors, agentsIds); recordingProcessors = new ArrayList<>(1); recordingProcessors.add(agentFilterDataProcessor);