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" )