def test_parse_command_request(self):
mock_klass = MagicMock()
mock_broker = MagicMock()
self.request_cmds['testklass'] = mock_klass
self.broker = mock_broker
mock_klass.process_arguments.return_value = 'fake_request_dict'
mock_klass.return_value = 'fake_request'
mock_broker.send.return_value = (8129, 'fake_response')
_parse_command(self.broker, self.request_classes, self.request_cmds, 'testklass', ['someargs'])
mock_klass.process_arguments.assert_called_once_with(['someargs'])
mock_klass.assert_called_once_with('fake_request_dict')
mock_broker.send.assert_called_once_with('fake_request')
def test_parse_command_errors(self, mock_errors):
_parse_command(self.broker, self.request_classes, self.request_cmds, 'errors', [])
mock_errors.assert_called_once_with()
def test_parse_command_unknown(self):
# Should just not raise an error, as it prints out
_parse_command(self.broker, self.request_classes, self.request_cmds, 'unknown_command', [])
assert True
def test_parse_command_help_pass_args(self, mock_help):
_parse_command(self.broker, self.request_classes, self.request_cmds, 'help', ['foo'])
mock_help.assert_called_once_with(self.request_classes, self.request_cmds, ['foo'])
