def prepare_bot(self, raise_on_connect=False):
self.log_stream = io.StringIO()
self.bot_id = 'test-bot'
src_name = "{}-input".format(self.bot_id)
dst_name = "{}-output".format(self.bot_id)
self.mocked_config = test.mocked_config(
self.bot_id, src_name, [dst_name],
{"raise_on_connect": raise_on_connect})
logger = logging.getLogger(self.bot_id)
logger.setLevel("DEBUG")
console_formatter = logging.Formatter(utils.LOG_FORMAT)
console_handler = logging.StreamHandler(self.log_stream)
console_handler.setFormatter(console_formatter)
logger.addHandler(console_handler)
self.mocked_log = test.mocked_logger(logger)
class Parameters(object):
source_queue = src_name
destination_queues = [dst_name]
parameters = Parameters()
pipe = pipeline.Pythonlist(parameters)
pipe.set_queues(parameters.source_queue, "source")
pipe.set_queues(parameters.destination_queues, "destination")
with mock.patch('intelmq.lib.utils.load_configuration',
new=self.mocked_config):
with mock.patch('intelmq.lib.utils.log', self.mocked_log):
self.bot = self.bot_reference(self.bot_id)
self.pipe = pipe
def prepare_bot(self, raise_on_connect=False):
self.log_stream = io.StringIO()
self.bot_id = 'test-bot'
src_name = "{}-input".format(self.bot_id)
dst_name = "{}-output".format(self.bot_id)
self.mocked_config = test.mocked_config(self.bot_id,
src_name,
[dst_name],
{"raise_on_connect": raise_on_connect})
logger = logging.getLogger(self.bot_id)
logger.setLevel("DEBUG")
console_formatter = logging.Formatter(utils.LOG_FORMAT)
console_handler = logging.StreamHandler(self.log_stream)
console_handler.setFormatter(console_formatter)
logger.addHandler(console_handler)
self.mocked_log = test.mocked_logger(logger)
class Parameters(object):
source_queue = src_name
destination_queues = [dst_name]
parameters = Parameters()
pipe = pipeline.Pythonlist(parameters)
pipe.set_queues(parameters.source_queue, "source")
pipe.set_queues(parameters.destination_queues, "destination")
with mock.patch('intelmq.lib.utils.load_configuration',
new=self.mocked_config):
with mock.patch('intelmq.lib.utils.log', self.mocked_log):
self.bot = self.bot_reference(self.bot_id)
self.pipe = pipe
# -*- coding: utf-8 -*-
"""
Tests the Bot class itself.
"""
import io
import logging
import unittest
import unittest.mock as mock
import intelmq.lib.pipeline as pipeline
import intelmq.lib.utils as utils
import intelmq.lib.test as test
with mock.patch('intelmq.lib.utils.load_configuration',
new=test.mocked_config()):
from intelmq.tests.lib import test_parser_bot
class TestBot(unittest.TestCase):
""" Testing generic funtionalties of Bot base class. """
def prepare_bot(self, raise_on_connect=False):
self.log_stream = io.StringIO()
self.bot_id = 'test-bot'
src_name = "{}-input".format(self.bot_id)
dst_name = "{}-output".format(self.bot_id)
self.mocked_config = test.mocked_config(
self.bot_id, src_name, [dst_name],
{"raise_on_connect": raise_on_connect})
"""
Tests the Bot class itself.
"""
import io
import logging
import unittest
import unittest.mock as mock
import intelmq.lib.pipeline as pipeline
import intelmq.lib.utils as utils
import intelmq.lib.test as test
with mock.patch('intelmq.lib.utils.load_configuration', new=test.mocked_config()):
from intelmq.tests.lib import test_parser_bot
class TestBot(unittest.TestCase):
""" Testing generic funtionalties of Bot base class. """
def prepare_bot(self, raise_on_connect=False):
self.log_stream = io.StringIO()
self.bot_id = 'test-bot'
src_name = "{}-input".format(self.bot_id)
dst_name = "{}-output".format(self.bot_id)
self.mocked_config = test.mocked_config(self.bot_id,
src_name,
