def test_main_update(self, list_nodes, robot_update_mobility): """Run the parser.robot.main function for update commands.""" robot_update_mobility.return_value = {'result': 'test'} list_nodes.return_value = [] args = ['update', '-n', 'traj'] robot_parser.main(args) list_nodes.assert_called_with(self.api, 123, None, None) robot_update_mobility.assert_called_with(self.api, 123, 'traj', [])
def test_main_mobility(self, mobility_command): """Run the parser.robot.main function for mobility commands.""" mobility_command.return_value = {'result': 'test'} # List mobility args = ['get', '--list'] robot_parser.main(args) mobility_command.assert_called_with(self.api, 'list') # Get mobility args = ['get', '-n', 'traj_name,site_name'] robot_parser.main(args) mobility_command.assert_called_with(self.api, 'get', ('traj_name', 'site_name'))
def test_main_status(self, list_nodes, robot_command): """Run the parser.robot.main function for status commands.""" robot_command.return_value = {'result': 'test'} list_nodes.return_value = [] args = ['status'] robot_parser.main(args) list_nodes.assert_called_with(self.api, 123, None, None) robot_command.assert_called_with(self.api, 'status', 123, []) args = ['status', '-l', 'grenoble,m3,1-2', '-l', 'grenoble,m3,3'] list_nodes.return_value = ['m3-1', 'm3-2', 'm3-3'] # simplify robot_parser.main(args) robot_command.assert_called_with(self.api, 'status', 123, ['m3-1', 'm3-2', 'm3-3'])
def test_main_update(self, list_nodes, robot_update_mobility): """Run the parser.robot.main function for update commands.""" robot_update_mobility.return_value = {'result': 'test'} list_nodes.return_value = [] args = ['update', 'traj,grenoble'] robot_parser.main(args) list_nodes.assert_called_with(self.api, 123, None, None) robot_update_mobility.assert_called_with(self.api, 123, 'traj', 'grenoble', []) # Invalid mobility format robot_update_mobility.reset_mock() args = [ 'update', 'traj', '-l', 'grenoble,m3,1-2', '-l', 'grenoble,m3,3' ] self.assertRaises(SystemExit, robot_parser.main, args) self.assertFalse(robot_update_mobility.called)
def test_main_update(self, list_nodes, robot_update_mobility): """Run the parser.robot.main function for update commands.""" robot_update_mobility.return_value = {'result': 'test'} list_nodes.return_value = [] args = ['update', 'traj,grenoble'] robot_parser.main(args) list_nodes.assert_called_with(self.api, 123, None, None) robot_update_mobility.assert_called_with( self.api, 123, 'traj', 'grenoble', []) # Invalid mobility format robot_update_mobility.reset_mock() args = ['update', 'traj', '-l', 'grenoble,m3,1-2', '-l', 'grenoble,m3,3'] self.assertRaises(SystemExit, robot_parser.main, args) self.assertFalse(robot_update_mobility.called)
def test_main_mobility(self, circuit_command): """Run the parser.robot.main function for circuit commands.""" circuit_command.return_value = {'result': 'test'} # List circuits args = ['get', '--list'] robot_parser.main(args) circuit_command.assert_called_with(self.api, 'list') # List mobility args = ['get', '--list', '--site', 'grenoble', '--type', 'predefined'] robot_parser.main(args) circuit_command.assert_called_with(self.api, 'list', site='grenoble', type='predefined') # Get mobility args = ['get', '-n', 'site_name'] robot_parser.main(args) circuit_command.assert_called_with(self.api, 'get', 'site_name')