org.mockito.internal.util.reflection.LenientCopyTool Maven / Gradle / Ivy
/*
* Copyright (c) 2007 Mockito contributors
* This program is made available under the terms of the MIT License.
*/
package org.mockito.internal.util.reflection;
import java.lang.reflect.Field;
import java.lang.reflect.Modifier;
@SuppressWarnings("unchecked")
public class LenientCopyTool {
FieldCopier fieldCopier = new FieldCopier();
public void copyToMock(T from, T mock) {
copy(from, mock, from.getClass(), mock.getClass().getSuperclass());
}
public void copyToRealObject(T from, T to) {
copy(from, to, from.getClass(), to.getClass());
}
private void copy(T from, T to, Class fromClazz, Class toClass) {
while (fromClazz != Object.class) {
copyValues(from, to, fromClazz);
fromClazz = fromClazz.getSuperclass();
}
}
private void copyValues(T from, T mock, Class classFrom) {
Field[] fields = classFrom.getDeclaredFields();
for (int i = 0; i < fields.length; i++) {
// ignore static fields
Field field = fields[i];
if (Modifier.isStatic(field.getModifiers())) {
continue;
}
AccessibilityChanger accessibilityChanger = new AccessibilityChanger();
try {
accessibilityChanger.enableAccess(field);
fieldCopier.copyValue(from, mock, field);
} catch (Throwable t) {
//Ignore - be lenient - if some field cannot be copied then let's be it
} finally {
accessibilityChanger.safelyDisableAccess(field);
}
}
}
}