Module (final Location<? extends PsiMethod> methodLocation) { final PsiMethod method = methodLocation.getPsiElement(); METHOD_NAME = getMethodPresentation(method); TEST_OBJECT = TEST_METHOD; return setMainClass(methodLocation instanceof MethodLocation ? ((MethodLocation)methodLocation).getContainingClass() : method.getContainingClass()); }