def test_map_to_api(self): excpected_result = { "id": 1, "username": '******', "device_slot": 1, "provisioning_extension": 123456 } data_dict = { 'id': 1, 'username': '******', 'num': 1, 'provisioningid': 123456 } mapping_model_to_api = { 'id': 'id', 'username': '******', 'num': 'device_slot', 'provisioningid': 'provisioning_extension', } result = mapper.map_to_api(mapping_model_to_api, data_dict) self.assertEqual(excpected_result, result)
def _prepare_model_to_api(self, model): model_dict = model.to_user_data() mapped_dict = mapper.map_to_api(self._mapper.MAPPING, model_dict) self._mapper.add_links_to_dict(mapped_dict, model) return mapped_dict