def get_object_params_to_get( desired_cfg: Optional[EnodebConfiguration], device_cfg: EnodebConfiguration, data_model: DataModel, request_all_params: bool = False, ) -> List[ParameterName]: """ Returns a list of parameter names for object parameters we don't know the current value of If `request_all_params` is set to True, the function will return a list of all device model param names, including already known ones. """ names = [] # TODO: This might a string for some strange reason, investigate why num_plmns = \ int(device_cfg.get_parameter(ParameterName.NUM_PLMNS)) for i in range(1, num_plmns + 1): obj_name = ParameterName.PLMN_N % i if not device_cfg.has_object(obj_name): device_cfg.add_object(obj_name) obj_to_params = data_model.get_numbered_param_names() desired = obj_to_params[obj_name] if request_all_params: names += desired else: current = [] if desired_cfg is not None: current = desired_cfg.get_parameter_names_for_object(obj_name) names_to_add = list(set(desired) - set(current)) names += names_to_add return names
def get_object_params_to_get( desired_cfg: Optional[EnodebConfiguration], device_cfg: EnodebConfiguration, data_model: DataModel, ) -> List[ParameterName]: """ Returns a list of parameter names for object parameters we don't know the current value of """ names = [] # TODO: This might a string for some strange reason, investigate why num_plmns = \ int(device_cfg.get_parameter(ParameterName.NUM_PLMNS)) for i in range(1, num_plmns + 1): obj_name = ParameterName.PLMN_N % i if not device_cfg.has_object(obj_name): device_cfg.add_object(obj_name) obj_to_params = data_model.get_numbered_param_names() desired = obj_to_params[obj_name] current = [] if desired_cfg is not None: current = desired_cfg.get_parameter_names_for_object(obj_name) names_to_add = list(set(desired) - set(current)) names = names + names_to_add return names
def _set_plmnids_tac( cfg: EnodebConfiguration, plmnids: Union[int, str], tac: Any, ) -> None: """ Set the following parameters: - PLMNID list (including all child parameters) Input 'plmnids' is comma-separated list of PLMNIDs """ # Convert int PLMNID to string if type(plmnids) == int: plmnid_str = str(plmnids) else: config_assert(type(plmnids) == str, 'PLMNID must be string') plmnid_str = plmnids # Multiple PLMNIDs will be supported using comma-separated list. # Currently, just one supported for char in plmnid_str: config_assert(char in '0123456789, ', 'Unhandled character (%s) in PLMNID' % char) plmnid_list = plmnid_str.split(',') # TODO - add support for multiple PLMNIDs config_assert(len(plmnid_list) == 1, 'Exactly one PLMNID must be configured') # Validate PLMNIDs plmnid_list[0] = plmnid_list[0].strip() config_assert(len(plmnid_list[0]) <= 6, 'PLMNID must be length <=6 (%s)' % plmnid_list[0]) # We just need one PLMN element in the config. Delete all others. for i in range(1, 2):#data_model.get_num_plmns() + 1): object_name = ParameterName.PLMN_N % i enable_plmn = i == 1 cfg.add_object(object_name) cfg.set_parameter_for_object(ParameterName.PLMN_N_ENABLE % i, enable_plmn, object_name) if enable_plmn: cfg.set_parameter_for_object( ParameterName.PLMN_N_CELL_RESERVED % i, False, object_name) cfg.set_parameter_for_object(ParameterName.PLMN_N_PRIMARY % i, enable_plmn, object_name) cfg.set_parameter_for_object(ParameterName.PLMN_N_PLMNID % i, plmnid_list[i - 1], object_name) cfg.set_parameter(ParameterName.TAC, tac)
class EnodebConfigurationTest(TestCase): def setUp(self): self.config = EnodebConfiguration(CaviumTrDataModel) def tearDown(self): self.config = None def test_data_model(self) -> None: data_model = self.config.data_model expected = CaviumTrDataModel self.assertEqual(data_model, expected, 'Data model fetch incorrect') def test_get_has_set_parameter(self) -> None: param = ParameterName.ADMIN_STATE self.config.set_parameter(param, True) self.assertTrue(self.config.has_parameter(param), 'Expected to have parameter') param_value = self.config.get_parameter(param) expected = True self.assertEqual(param_value, expected, 'Parameter value does not match what was set') def test_add_has_delete_object(self) -> None: object_name = ParameterName.PLMN_N % 1 self.assertFalse(self.config.has_object(object_name)) self.config.add_object(object_name) self.assertTrue(self.config.has_object(object_name)) self.config.delete_object(object_name) self.assertFalse(self.config.has_object(object_name)) def test_get_parameter_names(self) -> None: # Should start off as an empty list names_list = self.config.get_parameter_names() self.assertEqual(len(names_list), 0, 'Expected 0 names') # Should grow as we set parameters self.config.set_parameter(ParameterName.ADMIN_STATE, True) names_list = self.config.get_parameter_names() self.assertEqual(len(names_list), 1, 'Expected 1 name') # Parameter names should not include objects self.config.add_object(ParameterName.PLMN) names_list = self.config.get_parameter_names() self.assertEqual(len(names_list), 1, 'Expected 1 name') def test_get_object_names(self) -> None: # Should start off as an empty list obj_list = self.config.get_object_names() self.assertEqual(len(obj_list), 0, 'Expected 0 names') # Should grow as we set parameters self.config.add_object(ParameterName.PLMN) obj_list = self.config.get_object_names() self.assertEqual(len(obj_list), 1, 'Expected 1 names') def test_get_set_parameter_for_object(self) -> None: self.config.add_object(ParameterName.PLMN_N % 1) self.config.set_parameter_for_object( ParameterName.PLMN_N_CELL_RESERVED % 1, True, ParameterName.PLMN_N % 1) param_value = self.config.get_parameter_for_object( ParameterName.PLMN_N_CELL_RESERVED % 1, ParameterName.PLMN_N % 1) self.assertTrue( param_value, 'Expected that the param for object was set correctly') def test_get_parameter_names_for_object(self) -> None: # Should start off empty self.config.add_object(ParameterName.PLMN_N % 1) param_list = self.config.get_parameter_names_for_object( ParameterName.PLMN_N % 1) self.assertEqual(len(param_list), 0, 'Should be an empty param list') # Should increment as we set parameters self.config.set_parameter_for_object( ParameterName.PLMN_N_CELL_RESERVED % 1, True, ParameterName.PLMN_N % 1) param_list = self.config.get_parameter_names_for_object( ParameterName.PLMN_N % 1) self.assertEqual(len(param_list), 1, 'Should not be an empty list')