KaneAI - World’s First E2E Software Testing Agent.Signup for Private Beta Join TestMu Conference

How to use restoreOriginalConstructorsAccesses method of org.powermock.core.transformers.javassist.testclass.JavaAssistTestClassTransformer class

Best Powermock code snippet using org.powermock.core.transformers.javassist.testclass.JavaAssistTestClassTransformer.restoreOriginalConstructorsAccesses

Source:JavaAssistTestClassTransformer.java

...55        if (isTestClass(clazz)) {56            removeTestAnnotationsForTestMethodsThatRunOnOtherClassLoader(clazz);57            addLifeCycleNotifications(clazz);58            makeDeferConstructorNonPublic(clazz);59            restoreOriginalConstructorsAccesses(clazz);60            61        } else if (isNestedWithinTestClass(clazz)) {62            makeDeferConstructorNonPublic(clazz);63            restoreOriginalConstructorsAccesses(clazz);64        }65    66    }67    68    private boolean isTestClass(CtClass clazz) {69        try {70            return Class.forName(clazz.getName(), false, getTestClass().getClassLoader())71                        .isAssignableFrom(getTestClass());72        } catch (ClassNotFoundException ex) {73            return false;74        }75    }76    77    private boolean isNestedWithinTestClass(CtClass clazz) {78        String clazzName = clazz.getName();79        return clazzName.startsWith(getTestClass().getName())80                   && '$' == clazzName.charAt(getTestClass().getName().length());81    }82    83    private Class<?> asOriginalClass(CtClass type) throws Exception {84        try {85            return type.isArray()86                       ? Array.newInstance(asOriginalClass(type.getComponentType()), 0).getClass()87                       : type.isPrimitive()88                             ? Primitives.getClassFor((CtPrimitiveType) type)89                             : Class.forName(type.getName(), true, getTestClass().getClassLoader());90        } catch (Exception ex) {91            throw new RuntimeException("Cannot resolve type: " + type, ex);92        }93    }94    95    private Class<?>[] asOriginalClassParams(CtClass[] parameterTypes)96        throws Exception {97        final Class<?>[] classParams = new Class[parameterTypes.length];98        for (int i = 0; i < classParams.length; ++i) {99            classParams[i] = asOriginalClass(parameterTypes[i]);100        }101        return classParams;102    }103    104    private void removeTestMethodAnnotationFrom(CtMethod m) {105        final AnnotationsAttribute attr = (AnnotationsAttribute)106                                              m.getMethodInfo().getAttribute(AnnotationsAttribute.visibleTag);107        javassist.bytecode.annotation.Annotation[] newAnnotations =108            new javassist.bytecode.annotation.Annotation[attr.numAnnotations() - 1];109        int i = -1;110        for (javassist.bytecode.annotation.Annotation a : attr.getAnnotations()) {111            if (a.getTypeName().equals(getTestMethodAnnotationType().getName())) {112                continue;113            }114            newAnnotations[++i] = a;115        }116        attr.setAnnotations(newAnnotations);117    }118    119    private void removeTestAnnotationsForTestMethodsThatRunOnOtherClassLoader(CtClass clazz)120        throws Exception {121        for (CtMethod m : clazz.getDeclaredMethods()) {122            if (m.hasAnnotation(getTestMethodAnnotationType()) && mustHaveTestAnnotationRemoved(m)) {123                removeTestMethodAnnotationFrom(m);124            }125        }126    }127    128    private void addLifeCycleNotifications(CtClass clazz) {129        try {130            addClassInitializerNotification(clazz);131            addConstructorNotification(clazz);132        } catch (CannotCompileException ex) {133            throw new Error("Powermock error: " + ex.getMessage(), ex);134        }135    }136    137    private void addClassInitializerNotification(CtClass clazz)138        throws CannotCompileException {139        if (null == clazz.getClassInitializer()) {140            clazz.makeClassInitializer();141        }142        clazz.getClassInitializer().insertBefore(143            GlobalNotificationBuildSupport.class.getName()144                + ".testClassInitiated(" + clazz.getName() + ".class);");145    }146    147    private static boolean hasSuperClass(CtClass clazz) {148        try {149            CtClass superClazz = clazz.getSuperclass();150            /*151             * Being extra careful here - and backup in case the152             * work-in-progress clazz doesn't cause NotFoundException ...153             */154            return null != superClazz155                       && !"java.lang.Object".equals(superClazz.getName());156        } catch (NotFoundException noWasSuperClassFound) {157            return false;158        }159    }160    161    private void addConstructorNotification(final CtClass clazz)162        throws CannotCompileException {163        final String notificationCode =164            GlobalNotificationBuildSupport.class.getName()165                + ".testInstanceCreated(this);";166        final boolean asFinally = !hasSuperClass(clazz);167        for (final CtConstructor constr : clazz.getDeclaredConstructors()) {168            constr.insertAfter(169                notificationCode,170                asFinally/* unless there is a super-class, because of this171                              * problem: https://community.jboss.org/thread/94194*/);172        }173    }174    175    private void restoreOriginalConstructorsAccesses(CtClass clazz) throws Exception {176        Class<?> originalClass = getTestClass().getName().equals(clazz.getName())177                                     ? getTestClass()178                                     : Class.forName(clazz.getName(), true, getTestClass().getClassLoader());179        for (final CtConstructor ctConstr : clazz.getConstructors()) {180            int ctModifiers = ctConstr.getModifiers();181            if (!Modifier.isPublic(ctModifiers)) {182                /* Probably a defer-constructor */183                continue;184            }185            int desiredAccessModifiers = originalClass.getDeclaredConstructor(186                asOriginalClassParams(ctConstr.getParameterTypes())).getModifiers();187            188            if (Modifier.isPrivate(desiredAccessModifiers)) {189                ctConstr.setModifiers(Modifier.setPrivate(ctModifiers));...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.