def test_successful_outbound_stream_creation(self, create_handler_mock,
server_init_mock,
broker_class_mock):
""" Tests that all types of outbound streams are successfully created """
# Testing creation of outbound stream with ZMQ input/output
server = Server()
server.broker = ait.core.server.broker.Broker()
config = {
'name': 'some_stream',
'handlers': [{
'name': 'some-handler'
}]
}
created_stream = server._create_outbound_stream(config)
assert type(created_stream) == ait.core.server.stream.ZMQStream
assert created_stream.name == 'some_stream'
assert type(created_stream.handlers) == list
# Testing creation of outbound stream with port output
config = cfg.AitConfig(config={'name': 'some_stream', 'output': 3333})
created_stream = server._create_outbound_stream(config)
assert type(created_stream) == ait.core.server.stream.PortOutputStream
assert created_stream.name == 'some_stream'
assert created_stream.out_port == 3333
assert created_stream.handlers == []
def test_successful_outbound_stream_creation(
self, create_handler_mock, server_stream_plugin_mock_mock,
broker_class_mock):
"""Tests that all types of outbound streams are successfully created"""
# Testing creation of outbound stream with ZMQ input/output
server = Server()
server.broker = ait.core.server.broker.Broker()
config = {
"name": "some_stream",
"handlers": [{
"name": "some-handler"
}]
}
created_stream = server._create_outbound_stream(config)
assert type(created_stream) == ait.core.server.stream.ZMQStream
assert created_stream.name == "some_stream"
assert type(created_stream.handlers) == list
# Testing creation of outbound stream with port output
config = cfg.AitConfig(config={"name": "some_stream", "output": 3333})
created_stream = server._create_outbound_stream(config)
assert type(created_stream) == ait.core.server.stream.PortOutputStream
assert created_stream.name == "some_stream"
assert created_stream.out_port == 3333
assert created_stream.handlers == []
def test_plugin_missing_outputs(self, log_warn_mock, server_init_mock,
broker_mock):
""" Tests that warning logged if plugin has no inputs and
plugin created anyways """
server = Server()
server.broker = ait.core.server.broker.Broker()
config = {
'name': 'ait.core.server.plugins.TelemetryLimitMonitor',
'inputs': 'some_stream'
}
server._create_plugin(config)
log_warn_mock.assert_called_with(
'No plugin outputs specified for ait.core.server.plugins.TelemetryLimitMonitor'
)
def test_plugin_missing_inputs(self, log_warn_mock,
server_stream_plugin_mock_mock,
broker_mock):
"""Tests that warning logged if plugin has no inputs and
plugin created anyways"""
server = Server()
server.broker = ait.core.server.broker.Broker()
config = {
"name": "ait.core.server.plugins.TelemetryLimitMonitor",
"outputs": "some_stream",
}
server._create_plugin(config)
log_warn_mock.assert_called_with(
"No plugin inputs specified for ait.core.server.plugins.TelemetryLimitMonitor"
)