def testUniqueDomainExtraction(self):
"""Tests for the unique domains plugin."""
event_queue = single_process.SingleProcessQueue()
knowledge_base = self._SetUpKnowledgeBase()
# Fill the incoming queue with events.
test_queue_producer = queue.ItemQueueProducer(event_queue)
events = [
self._CreateTestEventObject(test_event)
for test_event in self.TEST_EVENTS
]
test_queue_producer.ProduceItems(events)
# Set up the plugin.
analysis_plugin = unique_domains_visited.UniqueDomainsVisitedPlugin(
event_queue)
analysis_report_queue_consumer = self._RunAnalysisPlugin(
analysis_plugin, knowledge_base)
analysis_reports = self._GetAnalysisReportsFromQueue(
analysis_report_queue_consumer)
self.assertEqual(len(analysis_reports), 1)
report_text = analysis_reports[0].GetString()
for event_object in self.TEST_EVENTS:
self.assertIn(event_object.get(u'domain', u''), report_text)
def testExamineEventAndCompileReport(self):
"""Tests the ExamineEvent and CompileReport functions."""
plugin = unique_domains_visited.UniqueDomainsVisitedPlugin()
storage_writer = self._AnalyzeEvents(self._TEST_EVENTS, plugin)
self.assertEqual(len(storage_writer.analysis_reports), 1)
analysis_report = storage_writer.analysis_reports[0]
report_text = analysis_report.GetString()
for event_dictionary in self._TEST_EVENTS:
domain = event_dictionary.get('domain', '')
self.assertIn(domain, report_text)
def testExamineEventAndCompileReport(self):
"""Tests the ExamineEvent and CompileReport functions."""
plugin = unique_domains_visited.UniqueDomainsVisitedPlugin()
storage_writer = self._AnalyzeEvents(self._TEST_EVENTS, plugin)
self.assertEqual(len(storage_writer.analysis_reports), 1)
analysis_report = storage_writer.analysis_reports[0]
self.assertEqual(analysis_report.analysis_counter['firstevent.com'], 1)
self.assertEqual(analysis_report.analysis_counter['secondevent.net'],
1)
self.assertEqual(analysis_report.analysis_counter['thirdevent.org'], 1)
self.assertEqual(analysis_report.analysis_counter['fourthevent.co'], 1)
def testExamineEventAndCompileReport(self):
"""Tests the ExamineEvent and CompileReport functions."""
plugin = unique_domains_visited.UniqueDomainsVisitedPlugin()
storage_writer = self._AnalyzeEvents(self._TEST_EVENTS, plugin)
self.assertEqual(storage_writer.number_of_analysis_reports, 1)
analysis_report = storage_writer.GetAttributeContainerByIndex(
reports.AnalysisReport.CONTAINER_TYPE, 0)
self.assertEqual(analysis_report.analysis_counter['firstevent.com'], 1)
self.assertEqual(analysis_report.analysis_counter['secondevent.net'],
1)
self.assertEqual(analysis_report.analysis_counter['thirdevent.org'], 1)
self.assertEqual(analysis_report.analysis_counter['fourthevent.co'], 1)
def testExamineEventAndCompileReport(self):
"""Tests the ExamineEvent and CompileReport functions."""
events = []
for event_dictionary in self._TEST_EVENTS:
event_dictionary['url'] = 'https://{0:s}/{1:s}'.format(
event_dictionary['domain'], event_dictionary['path'])
event = self._CreateTestEventObject(event_dictionary)
events.append(event)
plugin = unique_domains_visited.UniqueDomainsVisitedPlugin()
storage_writer = self._AnalyzeEvents(events, plugin)
self.assertEqual(len(storage_writer.analysis_reports), 1)
analysis_report = storage_writer.analysis_reports[0]
report_text = analysis_report.GetString()
for event_dictionary in self._TEST_EVENTS:
domain = event_dictionary.get('domain', '')
self.assertIn(domain, report_text)