def test_parameter_type_cache_works(self):
replies = [
{'payload': {'TestControllee': {'test_param_1': 'float', 'test_param_2': 'int'}}},
{'payload': {'TestControllee': {'test_param_1': 4.5, 'test_param_2': 34}}}
]
with patch.object(MockParameterControlClient, 'send', side_effect=replies) as mock_method:
controller = GeneralParameterController(MockParameterControlClient())
self.assertEqual(controller._convert_to_cached_type('TestControllee', 'test_param_1', '0.6'), 0.6)
self.assertEqual(controller._convert_to_cached_type('TestControllee', 'test_param_2', '443'), 443)
expected_calls = [call('control', 'send_parameters'),
call('get_values', {'TestControllee': {'test_param_1': 'float', 'test_param_2': 'int'}})]
mock_method.assert_has_calls(expected_calls)
def test_set_parameter_value(self):
replies = [
{'payload': {'TestControllee': {'test_param_1': 'float', 'test_param_2': 'int'}}},
{'payload': {'TestControllee': {'test_param_1': 4.5, 'test_param_2': 34}}},
{'payload': 'Ok.'}, {'payload': 'Ok.'}
]
with patch.object(MockParameterControlClient, 'send', side_effect=replies) as mock_method:
controller = GeneralParameterController(MockParameterControlClient())
controller.set_parameter_value('TestControllee', 'test_param_1', '34')
controller.set_parameter_value('TestControllee', 'test_param_2', '45')
expected_calls = [call('control', 'send_parameters'),
call('get_values', {'TestControllee': {'test_param_1': 'float', 'test_param_2': 'int'}}),
call('set_parameters', {'TestControllee': {'test_param_1': 34.0}}),
call('set_parameters', {'TestControllee': {'test_param_2': 45.0}})]
mock_method.assert_has_calls(expected_calls)
from parameter_control_client import ParameterControlClient
from general_parameter_controller import GeneralParameterController
parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument('-l', '--list-parameters', action='store_true',
help='Show a list of available parameters and their types.')
parser.add_argument('-c', '--controllee', type=str, default='BraggPeakEventGenerator', help='Specify controllee.')
parser.add_argument('-H', '--host', type=str, default='localhost', help='Host to connect to.')
parser.add_argument('-p', '--port', type=str, default='10002', help='Port to connect to.')
parser.add_argument('parameter_name', nargs='?', help='Name of parameter to set.')
parser.add_argument('parameter_value', nargs='?', help='New parameter value.')
args = parser.parse_args()
control_client = ParameterControlClient(args.host, args.port)
controller = GeneralParameterController(control_client)
if args.list_parameters:
controller.print_available_parameters()
if args.parameter_name:
if args.parameter_value:
print controller.set_parameter_value(args.controllee, args.parameter_name, args.parameter_value)
else:
print controller.get_parameter_value(args.controllee, args.parameter_name)
else:
controller.print_current_values()