Module (Location<PsiMethod> location) { final PsiMethod method = location.getPsiElement(); METHOD_NAME = method.getName(); TEST_OBJECT = TestType.METHOD.getType(); return setMainClass(location instanceof MethodLocation ? ((MethodLocation)location).getContainingClass() : method.getContainingClass()); }