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)