void restoreTransformedClasses(@Nonnull Set<ClassIdentification> previousTransformedClasses) {
if (!transformedClasses.isEmpty()) {
Set<ClassIdentification> classesToRestore;
if (previousTransformedClasses.isEmpty()) {
classesToRestore = transformedClasses.keySet();
} else {
classesToRestore = getTransformedClasses();
classesToRestore.removeAll(previousTransformedClasses);
}
if (!classesToRestore.isEmpty()) {
restoreAndRemoveTransformedClasses(classesToRestore);
}
}
}
private void restoreDefinition(@Nonnull Class<?> redefinedClass) {
if (redefinedClassesWithNativeMethods.contains(redefinedClass.getName())) {
reregisterNativeMethodsForRestoredClass(redefinedClass);
}
removeMockedClass(redefinedClass);
}
public void addRedefinedClassWithNativeMethods(@Nonnull String redefinedClassInternalName) {
redefinedClassesWithNativeMethods.add(redefinedClassInternalName.replace('/', '.'));
}