def test_connection_explicit(self):
amqp_conn = AmqpConnection(self.amqp_url)
amqp_conn.connect()
self.assertIsNotNone(amqp_conn._connection)
self.assertIsNotNone(amqp_conn._channel)
amqp_conn.disconnect()
self.assertIsNone(amqp_conn._connection)
self.assertIsNone(amqp_conn._channel)
def __init__(self, logger):
self.logger = logger
self.db_store = MongoStore.get_default()
self.cache_store = RedisStore.get_default()
self.amqp_url, self.amqp_exchange = AmqpConnection.parse_conf()
self.ts_client = InfluxDBClient.from_configuration(self.logger)
def test_bad_canopsis_event_raises(self):
event = {}
with AmqpConnection(self.amqp_url) as ac:
amqp_pub = AmqpPublisher(ac, Mock())
with self.assertRaises(KeyError):
amqp_pub.canopsis_event(event, self.amqp_exname)
def __init__(self, logger):
self.logger = logger
self.db_store = MongoStore.get_default()
self.cache_store = RedisStore.get_default()
self.amqp_url, self.amqp_exchange = AmqpConnection.parse_conf()
self.ts_client = InfluxDBClient.from_configuration(self.logger)
def setUpClass(cls):
cls.amqp_url, cls.amqp_exname = AmqpConnection.parse_conf()
cls.event = {
'connector': 'test_amqp',
'connector_name': 'test_amqp',
'source_type': 'resource',
'event_type': 'check',
'component': 'test',
'resource': 'test'
}
def setUpClass(cls):
cls.amqp_url, cls.amqp_exname = AmqpConnection.parse_conf()
cls.event = {
'connector': 'test_amqp',
'connector_name': 'test_amqp',
'source_type': 'resource',
'event_type': 'check',
'component': 'test',
'resource': 'test'
}
def test_connection_explicit(self):
amqp_conn = AmqpConnection(self.amqp_url)
amqp_conn.connect()
self.assertIsNotNone(amqp_conn._connection)
self.assertIsNotNone(amqp_conn._channel)
amqp_conn.disconnect()
self.assertIsNone(amqp_conn._connection)
self.assertIsNone(amqp_conn._channel)
def __init__(self, logger):
self.logger = logger
self.db_store = MongoStore.get_default()
self.cache_store = RedisStore.get_default()
self.amqp_url, self.amqp_exchange = AmqpConnection.parse_conf()
self.ts_client = InfluxDBClient.from_configuration(self.logger)
parser = Configuration.load(CONF_PATH, Ini)
section = parser.get(ConfName.SECT_HC)
self.check_amqp_limit_size = int(
section.get(ConfName.CHECK_AMQP_LIMIT_SIZE, ""))
self.check_amqp_queues = section.get(ConfName.CHECK_AMQP_QUEUES,
"").split(",")
self.check_collections = section.get(ConfName.CHECK_COLLECTIONS,
"").split(",")
self.check_engines_list = section.get(ConfName.CHECK_ENGINES,
"").split(",")
self.check_ts_db = section.get(ConfName.CHECK_TS_DB, "")
self.check_webserver = section.get(ConfName.CHECK_WEBSERVER, "")
self.systemctl_engine_prefix = section.get(
ConfName.SYSTEMCTL_ENGINE_PREFIX, "")
def check_amqp(self):
"""
Check if amqp service is available.
:rtype: ServiceState
"""
try:
with AmqpConnection(self.amqp_url) as amqp_conn:
channel = amqp_conn._channel
try:
channel.basic_publish(self.amqp_exchange, '', 'test')
except Exception as exc:
msg = 'Failed to publish: {}'.format(exc)
return ServiceState(message=msg)
if not amqp_conn._connection.is_open:
return ServiceState(message='Connection is not opened')
if not channel.is_open:
return ServiceState(message='Channel is not opened')
except ConnectionClosed as exc:
return ServiceState(message='Failed to connect: {}'.format(exc))
return self._check_rabbitmq_state()
def _amqp_setup(self):
logger = logging.getLogger("test_base")
self.amqp_conn = AmqpConnection(self.AMQP_URL)
self.amqp_pub = AmqpPublisher(self.amqp_conn, logger)
def _amqp_setup(self):
self.amqp_conn = AmqpConnection(self.AMQP_URL)
self.amqp_pub = AmqpPublisher(self.amqp_conn)
def test_json_document(self):
jdoc = {'bla': 'bla'}
with AmqpConnection(self.amqp_url) as ac:
amqp_pub = AmqpPublisher(ac, Mock())
amqp_pub.json_document(jdoc, self.amqp_exname, '#')
def test_canopsis_event(self):
with AmqpConnection(self.amqp_url) as ac:
amqp_pub = AmqpPublisher(ac, Mock())
amqp_pub.canopsis_event(self.event, self.amqp_exname)
def test_connection_with_statement(self):
with AmqpConnection(self.amqp_url) as amqp_conn:
self.assertIsNotNone(amqp_conn.connection)
self.assertIsNotNone(amqp_conn.channel)