diff --git a/compiler/tests/org/jetbrains/jet/checkers/AbstractJetDiagnosticsTest.java b/compiler/tests/org/jetbrains/jet/checkers/AbstractJetDiagnosticsTest.java index 205f236ff69..d9f696f19c0 100644 --- a/compiler/tests/org/jetbrains/jet/checkers/AbstractJetDiagnosticsTest.java +++ b/compiler/tests/org/jetbrains/jet/checkers/AbstractJetDiagnosticsTest.java @@ -95,7 +95,7 @@ public abstract class AbstractJetDiagnosticsTest extends BaseDiagnosticsTest { StringBuilder actualText = new StringBuilder(); for (TestFile testFile : testFiles) { - ok &= testFile.getActualText(moduleBindings.get(testFile.getModule()), actualText); + ok &= testFile.getActualText(moduleBindings.get(testFile.getModule()), actualText, groupedByModule.size() > 1); } JetTestUtils.assertEqualsToFile(testDataFile, actualText.toString()); diff --git a/compiler/tests/org/jetbrains/jet/checkers/BaseDiagnosticsTest.java b/compiler/tests/org/jetbrains/jet/checkers/BaseDiagnosticsTest.java index 0a391c2be64..5d18b63cce4 100644 --- a/compiler/tests/org/jetbrains/jet/checkers/BaseDiagnosticsTest.java +++ b/compiler/tests/org/jetbrains/jet/checkers/BaseDiagnosticsTest.java @@ -307,20 +307,16 @@ public abstract class BaseDiagnosticsTest extends JetLiteFixture { return jetFile; } - public boolean getActualText(BindingContext bindingContext, StringBuilder actualText) { + public boolean getActualText(BindingContext bindingContext, StringBuilder actualText, boolean skipJvmSignatureDiagnostics) { if (this.jetFile == null) { // TODO: check java files too actualText.append(this.clearText); return true; } - Set jvmSignatureDiagnostics = new HashSet(); - Collection declarations = PsiTreeUtil.findChildrenOfType(jetFile, JetDeclaration.class); - for (JetDeclaration declaration : declarations) { - Diagnostics diagnostics = AsJavaPackage.getJvmSignatureDiagnostics(declaration, bindingContext.getDiagnostics()); - if (diagnostics == null) continue; - jvmSignatureDiagnostics.addAll(diagnostics.forElement(declaration)); - } + Set jvmSignatureDiagnostics = skipJvmSignatureDiagnostics + ? Collections.emptySet() + : computeJvmSignatureDiagnostics(bindingContext); final boolean[] ok = { true }; List diagnostics = ContainerUtil.filter( @@ -354,5 +350,16 @@ public abstract class BaseDiagnosticsTest extends JetLiteFixture { })); return ok[0]; } + + private Set computeJvmSignatureDiagnostics(BindingContext bindingContext) { + Set jvmSignatureDiagnostics = new HashSet(); + Collection declarations = PsiTreeUtil.findChildrenOfType(jetFile, JetDeclaration.class); + for (JetDeclaration declaration : declarations) { + Diagnostics diagnostics = AsJavaPackage.getJvmSignatureDiagnostics(declaration, bindingContext.getDiagnostics()); + if (diagnostics == null) continue; + jvmSignatureDiagnostics.addAll(diagnostics.forElement(declaration)); + } + return jvmSignatureDiagnostics; + } } }