def test_should_exit_when_no_target_configured(self, mock_exit, mock_log):
receiver = Receiver()
configuration = ConfigurationDict(allowed_targets=set(),
broadcaster_host='broadcaster_host',
broadcaster_port=1234)
receiver.set_configuration(configuration)
mock_broadcaster_client = Mock()
receiver.broadcaster = mock_broadcaster_client
receiver.onConnect()
self.assertEquals(call(1), mock_exit.call_args)
def test_should_subscribe_to_target_from_configuration_when_connected(self):
receiver = Receiver()
mock_broadcaster_client = Mock()
receiver.broadcaster = mock_broadcaster_client
configuration = ConfigurationDict(allowed_targets=set(['devabc123']),
broadcaster_host='broadcaster_host',
broadcaster_port=1234)
receiver.set_configuration(configuration)
receiver.onConnect()
self.assertEquals(call(receiver.onEvent, 'devabc123'),
mock_broadcaster_client.client.subscribe.call_args)
def test_should_subscribe_to_targets_from_configuration_in_alphabetical_order_when_connected(self):
receiver = Receiver()
mock_broadcaster_client = Mock()
receiver.broadcaster = mock_broadcaster_client
configuration = ConfigurationDict(allowed_targets=set(['dev01', 'dev02', 'dev03']),
broadcaster_host='broadcaster_host',
broadcaster_port=1234)
receiver.set_configuration(configuration)
receiver.onConnect()
self.assertEquals(
[call(receiver.onEvent, 'dev01'),
call(receiver.onEvent, 'dev02'),
call(receiver.onEvent, 'dev03')],
mock_broadcaster_client.client.subscribe.call_args_list)
