Fix test cases failing issue #183
Merged
amalka.subasinghe
merged 4 commits from tcdlpds/device-mgt-core:sync-master
into apim420
1 year ago
@ -1,122 +0,0 @@
|
|||||||
/*
|
|
||||||
* Copyright (c) 2018 - 2023, Entgra (Pvt) Ltd. (http://www.entgra.io) All Rights Reserved.
|
|
||||||
*
|
|
||||||
* Entgra (Pvt) Ltd. licenses this file to you under the Apache License,
|
|
||||||
* Version 2.0 (the "License"); you may not use this file except
|
|
||||||
* in compliance with the License.
|
|
||||||
* You may obtain a copy of the License at
|
|
||||||
*
|
|
||||||
* http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
*
|
|
||||||
* Unless required by applicable law or agreed to in writing,
|
|
||||||
* software distributed under the License is distributed on an
|
|
||||||
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
||||||
* KIND, either express or implied. See the License for the
|
|
||||||
* specific language governing permissions and limitations
|
|
||||||
* under the License.
|
|
||||||
*/
|
|
||||||
package io.entgra.device.mgt.core.device.mgt.core.search.util;
|
|
||||||
|
|
||||||
import io.entgra.device.mgt.core.device.mgt.common.search.Condition;
|
|
||||||
import sun.reflect.ConstructorAccessor;
|
|
||||||
import sun.reflect.FieldAccessor;
|
|
||||||
import sun.reflect.ReflectionFactory;
|
|
||||||
|
|
||||||
import java.lang.reflect.AccessibleObject;
|
|
||||||
import java.lang.reflect.Array;
|
|
||||||
import java.lang.reflect.Field;
|
|
||||||
import java.lang.reflect.Modifier;
|
|
||||||
import java.util.ArrayList;
|
|
||||||
import java.util.Arrays;
|
|
||||||
import java.util.List;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Changes the Enum values of a given class.
|
|
||||||
*/
|
|
||||||
public class ChangeEnumValues {
|
|
||||||
|
|
||||||
private static ReflectionFactory reflectionFactory = ReflectionFactory.getReflectionFactory();
|
|
||||||
|
|
||||||
public ChangeEnumValues() {}
|
|
||||||
|
|
||||||
private static void setFailSafeFieldValue(Field field, Object target, Object value) throws NoSuchFieldException,
|
|
||||||
IllegalAccessException {
|
|
||||||
field.setAccessible(true);
|
|
||||||
Field modifiersField = Field.class.getDeclaredField("modifiers");
|
|
||||||
modifiersField.setAccessible(true);
|
|
||||||
int modifiers = modifiersField.getInt(field);
|
|
||||||
modifiers &= ~Modifier.FINAL;
|
|
||||||
modifiersField.setInt(field, modifiers);
|
|
||||||
FieldAccessor fa = reflectionFactory.newFieldAccessor(field, false);
|
|
||||||
fa.set(target, value);
|
|
||||||
}
|
|
||||||
|
|
||||||
private static void blankField(Class<?> enumClass, String fieldName) throws NoSuchFieldException,
|
|
||||||
IllegalAccessException {
|
|
||||||
for (Field field : Class.class.getDeclaredFields()) {
|
|
||||||
if (field.getName().contains(fieldName)) {
|
|
||||||
AccessibleObject.setAccessible(new Field[]{field}, true);
|
|
||||||
setFailSafeFieldValue(field, enumClass, null);
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private static void cleanEnumCache(Class<?> enumClass) throws NoSuchFieldException, IllegalAccessException {
|
|
||||||
blankField(enumClass, "enumConstantDirectory");
|
|
||||||
blankField(enumClass, "enumConstants");
|
|
||||||
}
|
|
||||||
|
|
||||||
private static ConstructorAccessor getConstructorAccessor(Class<?> enumClass, Class<?>[] additionalParameterTypes)
|
|
||||||
throws NoSuchMethodException {
|
|
||||||
Class<?>[] parameterTypes = new Class[additionalParameterTypes.length + 2];
|
|
||||||
parameterTypes[0] = String.class;
|
|
||||||
parameterTypes[1] = int.class;
|
|
||||||
System.arraycopy(additionalParameterTypes, 0, parameterTypes, 2, additionalParameterTypes.length);
|
|
||||||
return reflectionFactory.newConstructorAccessor(enumClass.getDeclaredConstructor(parameterTypes));
|
|
||||||
}
|
|
||||||
|
|
||||||
private static Object makeEnum(Class<?> enumClass, String value, int ordinal, Class<?>[] additionalTypes,
|
|
||||||
Object[] additionalValues) throws Exception {
|
|
||||||
Object[] parms = new Object[additionalValues.length + 2];
|
|
||||||
parms[0] = value;
|
|
||||||
parms[1] = ordinal;
|
|
||||||
System.arraycopy(additionalValues, 0, parms, 2, additionalValues.length);
|
|
||||||
return enumClass.cast(getConstructorAccessor(enumClass, additionalTypes).newInstance(parms));
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Add an enum instance to the enum class given as argument
|
|
||||||
*
|
|
||||||
* @param <T> the type of the enum
|
|
||||||
* @param enumType the class of the enum to be modified
|
|
||||||
* @param enumName the name of the new enum instance to be added to the class.
|
|
||||||
*/
|
|
||||||
public static <T extends Enum<?>> void addEnum(Class<T> enumType, String enumName) {
|
|
||||||
|
|
||||||
if (!Enum.class.isAssignableFrom(enumType)) {
|
|
||||||
throw new RuntimeException("class " + enumType + " is not an instance of Enum");
|
|
||||||
}
|
|
||||||
|
|
||||||
Field valuesField = null;
|
|
||||||
Field[] fields = Condition.State.class.getDeclaredFields();
|
|
||||||
for (Field field : fields) {
|
|
||||||
if (field.getName().contains("$VALUES")) {
|
|
||||||
valuesField = field;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
AccessibleObject.setAccessible(new Field[]{valuesField}, true);
|
|
||||||
|
|
||||||
try {
|
|
||||||
T[] previousValues = (T[]) valuesField.get(enumType);
|
|
||||||
List<T> values = new ArrayList<T>(Arrays.asList(previousValues));
|
|
||||||
T newValue = (T) makeEnum(enumType, enumName, values.size(), new Class<?>[]{}, new Object[]{});
|
|
||||||
values.add(newValue);
|
|
||||||
setFailSafeFieldValue(valuesField, null, values.toArray((T[]) Array.newInstance(enumType, 0)));
|
|
||||||
cleanEnumCache(enumType);
|
|
||||||
} catch (Exception e) {
|
|
||||||
throw new RuntimeException(e.getMessage(), e);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
Loading…
Reference in new issue