Improve test framework for diagnostic tests with NI

Mark with `NI;` or `OI;` only diagnostics that are specific for concrete inference
This commit is contained in:
Mikhail Zarechenskiy
2017-11-20 09:16:09 +03:00
parent bd4d847943
commit 81b3fefa58
3 changed files with 193 additions and 144 deletions
@@ -17,7 +17,6 @@
package org.jetbrains.kotlin.checkers;
import com.google.common.collect.LinkedListMultimap;
import com.google.common.collect.Lists;
import com.intellij.openapi.util.TextRange;
import com.intellij.openapi.util.text.StringUtil;
import com.intellij.psi.PsiElement;
@@ -30,6 +29,7 @@ import com.intellij.util.containers.ContainerUtil;
import com.intellij.util.containers.Stack;
import kotlin.Pair;
import kotlin.TuplesKt;
import kotlin.collections.ArraysKt;
import kotlin.collections.CollectionsKt;
import kotlin.text.StringsKt;
import org.jetbrains.annotations.NotNull;
@@ -88,6 +88,7 @@ public class CheckerTestUtil {
private static final Pattern INDIVIDUAL_PARAMETER_PATTERN = Pattern.compile(DIAGNOSTIC_PARAMETER);
private static final String NEW_INFERENCE_PREFIX = "NI";
private static final String OLD_INFERENCE_PREFIX = "OI";
@NotNull
public static List<ActualDiagnostic> getDiagnosticsIncludingSyntaxErrors(
@@ -213,39 +214,25 @@ public class CheckerTestUtil {
void wrongParametersDiagnostic(TextDiagnostic expectedDiagnostic, TextDiagnostic actualDiagnostic, int start, int end);
void unexpectedDiagnostic(TextDiagnostic diagnostic, int actualStart, int actualEnd);
void uncheckedDiagnostic(TextDiagnostic diagnostic, int expectedStart, int expectedEnd);
boolean shouldUseDiagnosticsForNI();
boolean isWithNewInferenceDirective();
}
public static Map<ActualDiagnostic, TextDiagnostic> diagnosticsDiff(
public static Map<AbstractTestDiagnostic, TextDiagnostic> diagnosticsDiff(
List<DiagnosedRange> expected,
Collection<ActualDiagnostic> actual,
DiagnosticDiffCallbacks callbacks
) {
Map<ActualDiagnostic, TextDiagnostic> diagnosticToExpectedDiagnostic = new HashMap<>();
Map<AbstractTestDiagnostic, TextDiagnostic> diagnosticToExpectedDiagnostic = new HashMap<>();
assertSameFile(actual);
Iterator<DiagnosedRange> expectedDiagnostics = expected.iterator();
List<ActualDiagnosticDescriptor> sortedDiagnosticDescriptors =
getActualSortedDiagnosticDescriptors(actual, !callbacks.isWithNewInferenceDirective());
List<ActualDiagnosticDescriptor> sortedDiagnosticDescriptors = getActualSortedDiagnosticDescriptors(actual);
Iterator<ActualDiagnosticDescriptor> actualDiagnostics = sortedDiagnosticDescriptors.iterator();
DiagnosedRange currentExpected = safeAdvance(expectedDiagnostics);
ActualDiagnosticDescriptor currentActual = safeAdvance(actualDiagnostics);
while (currentExpected != null || currentActual != null) {
if (currentExpected != null) {
boolean currentDiagnosticsForNI = CollectionsKt.any(currentExpected.diagnostics, diagnostic -> diagnostic.withNewInference);
if (callbacks.shouldUseDiagnosticsForNI() != currentDiagnosticsForNI) {
uncheckedDiagnostics(callbacks, currentExpected);
currentExpected = safeAdvance(expectedDiagnostics);
continue;
}
if (currentActual == null) {
missingDiagnostics(callbacks, currentExpected);
currentExpected = safeAdvance(expectedDiagnostics);
@@ -296,7 +283,7 @@ public class CheckerTestUtil {
@NotNull DiagnosticDiffCallbacks callbacks,
@NotNull DiagnosedRange currentExpected,
@NotNull ActualDiagnosticDescriptor currentActual,
@NotNull Map<ActualDiagnostic, TextDiagnostic> diagnosticToInput
@NotNull Map<AbstractTestDiagnostic, TextDiagnostic> diagnosticToInput
) {
int expectedStart = currentExpected.getStart();
int expectedEnd = currentExpected.getEnd();
@@ -305,16 +292,20 @@ public class CheckerTestUtil {
int actualEnd = currentActual.getEnd();
assert expectedStart == actualStart && expectedEnd == actualEnd;
Map<ActualDiagnostic, TextDiagnostic> actualDiagnostics = currentActual.getTextDiagnosticsMap();
Map<AbstractTestDiagnostic, TextDiagnostic> actualDiagnostics = currentActual.getTextDiagnosticsMap();
List<TextDiagnostic> expectedDiagnostics = currentExpected.getDiagnostics();
for (TextDiagnostic expectedDiagnostic : expectedDiagnostics) {
Map.Entry<ActualDiagnostic, TextDiagnostic> actualDiagnosticEntry = CollectionsKt.firstOrNull(
actualDiagnostics.entrySet(), entry -> expectedDiagnostic.getDescription().equals(entry.getValue().getDescription())
Map.Entry<AbstractTestDiagnostic, TextDiagnostic> actualDiagnosticEntry = CollectionsKt.firstOrNull(
actualDiagnostics.entrySet(), entry -> {
TextDiagnostic actualDiagnostic = entry.getValue();
return expectedDiagnostic.getDescription().equals(actualDiagnostic.getDescription()) &&
expectedDiagnostic.inferenceCompatibility.isCompatible(actualDiagnostic.inferenceCompatibility);
}
);
if (actualDiagnosticEntry != null) {
ActualDiagnostic actualDiagnostic = actualDiagnosticEntry.getKey();
AbstractTestDiagnostic actualDiagnostic = actualDiagnosticEntry.getKey();
TextDiagnostic actualTextDiagnostic = actualDiagnosticEntry.getValue();
if (!compareTextDiagnostic(expectedDiagnostic, actualTextDiagnostic)) {
@@ -322,6 +313,8 @@ public class CheckerTestUtil {
}
actualDiagnostics.remove(actualDiagnostic);
actualDiagnostic.enhanceInferenceCompatibility(expectedDiagnostic.inferenceCompatibility);
diagnosticToInput.put(actualDiagnostic, expectedDiagnostic);
}
else {
@@ -360,7 +353,7 @@ public class CheckerTestUtil {
}
private static void unexpectedDiagnostics(ActualDiagnosticDescriptor descriptor, DiagnosticDiffCallbacks callbacks) {
for (ActualDiagnostic diagnostic : descriptor.diagnostics) {
for (AbstractTestDiagnostic diagnostic : descriptor.diagnostics) {
callbacks.unexpectedDiagnostic(TextDiagnostic.asTextDiagnostic(diagnostic), descriptor.getStart(), descriptor.getEnd());
}
}
@@ -371,12 +364,6 @@ public class CheckerTestUtil {
}
}
private static void uncheckedDiagnostics(DiagnosticDiffCallbacks callbacks, DiagnosedRange currentExpected) {
for (TextDiagnostic diagnostic : currentExpected.getDiagnostics()) {
callbacks.uncheckedDiagnostic(diagnostic, currentExpected.getStart(), currentExpected.getEnd());
}
}
private static <T> T safeAdvance(Iterator<T> iterator) {
return iterator.hasNext() ? iterator.next() : null;
}
@@ -419,7 +406,7 @@ public class CheckerTestUtil {
public static StringBuffer addDiagnosticMarkersToText(
@NotNull PsiFile psiFile,
@NotNull Collection<ActualDiagnostic> diagnostics,
@NotNull Map<ActualDiagnostic, TextDiagnostic> diagnosticToExpectedDiagnostic,
@NotNull Map<AbstractTestDiagnostic, TextDiagnostic> diagnosticToExpectedDiagnostic,
@NotNull Function<PsiFile, String> getFileText,
@NotNull Collection<PositionalTextDiagnostic> uncheckedDiagnostics,
boolean withNewInferenceDirective
@@ -432,7 +419,7 @@ public class CheckerTestUtil {
return result;
}
List<AbstractDiagnosticDescriptor> diagnosticDescriptors = getSortedDiagnosticDescriptors(diagnostics, uncheckedDiagnostics, !withNewInferenceDirective);
List<AbstractDiagnosticDescriptor> diagnosticDescriptors = getSortedDiagnosticDescriptors(diagnostics, uncheckedDiagnostics);
Stack<AbstractDiagnosticDescriptor> opened = new Stack<>();
ListIterator<AbstractDiagnosticDescriptor> iterator = diagnosticDescriptors.listIterator();
@@ -445,7 +432,7 @@ public class CheckerTestUtil {
opened.pop();
}
while (currentDescriptor != null && i == currentDescriptor.start) {
openDiagnosticsString(result, currentDescriptor, diagnosticToExpectedDiagnostic);
openDiagnosticsString(result, currentDescriptor, diagnosticToExpectedDiagnostic, withNewInferenceDirective);
if (currentDescriptor.getEnd() == i) {
closeDiagnosticString(result);
}
@@ -465,7 +452,7 @@ public class CheckerTestUtil {
if (currentDescriptor != null) {
assert currentDescriptor.start == text.length();
assert currentDescriptor.end == text.length();
openDiagnosticsString(result, currentDescriptor, diagnosticToExpectedDiagnostic);
openDiagnosticsString(result, currentDescriptor, diagnosticToExpectedDiagnostic, withNewInferenceDirective);
opened.push(currentDescriptor);
}
@@ -482,7 +469,8 @@ public class CheckerTestUtil {
private static void openDiagnosticsString(
StringBuffer result,
AbstractDiagnosticDescriptor currentDescriptor,
Map<ActualDiagnostic, TextDiagnostic> diagnosticToExpectedDiagnostic
Map<AbstractTestDiagnostic, TextDiagnostic> diagnosticToExpectedDiagnostic,
boolean withNewInferenceDirective
) {
result.append("<!");
if (currentDescriptor instanceof TextDiagnosticDescriptor) {
@@ -490,9 +478,9 @@ public class CheckerTestUtil {
result.append(diagnostic.asString());
}
else if (currentDescriptor instanceof ActualDiagnosticDescriptor) {
List<ActualDiagnostic> diagnostics = ((ActualDiagnosticDescriptor) currentDescriptor).getDiagnostics();
for (Iterator<ActualDiagnostic> iterator = diagnostics.iterator(); iterator.hasNext(); ) {
ActualDiagnostic diagnostic = iterator.next();
List<AbstractTestDiagnostic> diagnostics = ((ActualDiagnosticDescriptor) currentDescriptor).getDiagnostics();
for (Iterator<AbstractTestDiagnostic> iterator = diagnostics.iterator(); iterator.hasNext(); ) {
AbstractTestDiagnostic diagnostic = iterator.next();
TextDiagnostic expectedDiagnostic = diagnosticToExpectedDiagnostic.get(diagnostic);
if (expectedDiagnostic != null) {
TextDiagnostic actualTextDiagnostic = TextDiagnostic.asTextDiagnostic(diagnostic);
@@ -504,12 +492,12 @@ public class CheckerTestUtil {
}
}
else {
if (diagnostic.withNewInference) {
result.append(NEW_INFERENCE_PREFIX);
if (withNewInferenceDirective && diagnostic.getInferenceCompatibility().abbreviation != null) {
result.append(diagnostic.getInferenceCompatibility().abbreviation);
result.append(";");
}
if (diagnostic.platform != null) {
result.append(diagnostic.platform);
if (diagnostic.getPlatform() != null) {
result.append(diagnostic.getPlatform());
result.append(":");
}
result.append(diagnostic.getName());
@@ -630,11 +618,10 @@ public class CheckerTestUtil {
}
private static List<ActualDiagnosticDescriptor> getActualSortedDiagnosticDescriptors(
@NotNull Collection<ActualDiagnostic> diagnostics,
boolean groupDiagnosticsByRange
@NotNull Collection<ActualDiagnostic> diagnostics
) {
return CollectionsKt.filterIsInstance(
getSortedDiagnosticDescriptors(diagnostics, Collections.emptyList(), groupDiagnosticsByRange),
getSortedDiagnosticDescriptors(diagnostics, Collections.emptyList()),
ActualDiagnosticDescriptor.class
);
}
@@ -642,45 +629,20 @@ public class CheckerTestUtil {
@NotNull
private static List<AbstractDiagnosticDescriptor> getSortedDiagnosticDescriptors(
@NotNull Collection<ActualDiagnostic> diagnostics,
@NotNull Collection<PositionalTextDiagnostic> uncheckedDiagnostics,
boolean groupDiagnosticsByRange
@NotNull Collection<PositionalTextDiagnostic> uncheckedDiagnostics
) {
List<ActualDiagnostic> validDiagnostics = CollectionsKt.filter(diagnostics, actualDiagnostic -> actualDiagnostic.diagnostic.isValid());
List<AbstractDiagnosticDescriptor> diagnosticDescriptors = groupDiagnosticsByRange ?
groupDiagnosticsByTextRange(validDiagnostics) :
asPlainDiagnosticDescriptors(validDiagnostics);
for (PositionalTextDiagnostic diagnostic : uncheckedDiagnostics) {
diagnosticDescriptors.add(new TextDiagnosticDescriptor(diagnostic));
}
diagnosticDescriptors.sort((d1, d2) -> {
if (d1.start != d2.start) return d1.start - d2.start;
if (d1.end != d2.end) return d2.end - d1.end;
if (d1.isWithNewInference() && !d2.isWithNewInference()) return -1;
if (!d1.isWithNewInference() && d2.isWithNewInference()) return 1;
return 0;
});
return diagnosticDescriptors;
}
private static List<AbstractDiagnosticDescriptor> asPlainDiagnosticDescriptors(@NotNull Collection<ActualDiagnostic> diagnostics) {
List<AbstractDiagnosticDescriptor> diagnosticDescriptors = Lists.newArrayList();
for (ActualDiagnostic actualDiagnostic : diagnostics) {
for (TextRange range : actualDiagnostic.diagnostic.getTextRanges()) {
diagnosticDescriptors.add(
new ActualDiagnosticDescriptor(
range.getStartOffset(),
range.getEndOffset(),
Collections.singletonList(actualDiagnostic))
);
}
}
List<AbstractDiagnosticDescriptor> diagnosticDescriptors = groupDiagnosticsByTextRange(validDiagnostics, uncheckedDiagnostics);
diagnosticDescriptors.sort((d1, d2) -> (d1.start != d2.start) ? d1.start - d2.start : d2.end - d1.end);
return diagnosticDescriptors;
}
@NotNull
private static List<AbstractDiagnosticDescriptor> groupDiagnosticsByTextRange(@NotNull Collection<ActualDiagnostic> diagnostics) {
LinkedListMultimap<TextRange, ActualDiagnostic> diagnosticsGroupedByRanges = LinkedListMultimap.create();
private static List<AbstractDiagnosticDescriptor> groupDiagnosticsByTextRange(
@NotNull Collection<ActualDiagnostic> diagnostics,
@NotNull Collection<PositionalTextDiagnostic> uncheckedDiagnostics
) {
LinkedListMultimap<TextRange, AbstractTestDiagnostic> diagnosticsGroupedByRanges = LinkedListMultimap.create();
for (ActualDiagnostic actualDiagnostic : diagnostics) {
Diagnostic diagnostic = actualDiagnostic.diagnostic;
for (TextRange textRange : diagnostic.getTextRanges()) {
@@ -688,12 +650,27 @@ public class CheckerTestUtil {
}
}
return CollectionsKt.map(diagnosticsGroupedByRanges.keySet(), range ->
new ActualDiagnosticDescriptor(
range.getStartOffset(),
range.getEndOffset(),
diagnosticsGroupedByRanges.get(range))
);
for (PositionalTextDiagnostic uncheckedDiagnostic : uncheckedDiagnostics) {
TextRange range = new TextRange(uncheckedDiagnostic.getStart(), uncheckedDiagnostic.getEnd());
diagnosticsGroupedByRanges.put(range, uncheckedDiagnostic.getDiagnostic());
}
return CollectionsKt.map(diagnosticsGroupedByRanges.keySet(), range -> {
List<AbstractTestDiagnostic> abstractDiagnostics = diagnosticsGroupedByRanges.get(range);
Comparator<AbstractTestDiagnostic> comparator = Comparator.comparing(AbstractTestDiagnostic::getInferenceCompatibility);
boolean needSortingByName = CollectionsKt.any(
abstractDiagnostics,
diagnostic -> diagnostic.getInferenceCompatibility() != TextDiagnostic.InferenceCompatibility.ALL
);
if (needSortingByName) {
comparator = comparator.thenComparing(Comparator.comparing(AbstractTestDiagnostic::getName));
}
abstractDiagnostics.sort(comparator);
return new ActualDiagnosticDescriptor(range.getStartOffset(), range.getEndOffset(), abstractDiagnostics);
});
}
private static abstract class AbstractDiagnosticDescriptor {
@@ -716,34 +693,27 @@ public class CheckerTestUtil {
public TextRange getTextRange() {
return new TextRange(start, end);
}
public abstract boolean isWithNewInference();
}
private static class ActualDiagnosticDescriptor extends AbstractDiagnosticDescriptor {
private final List<ActualDiagnostic> diagnostics;
private final List<AbstractTestDiagnostic> diagnostics;
ActualDiagnosticDescriptor(int start, int end, List<ActualDiagnostic> diagnostics) {
ActualDiagnosticDescriptor(int start, int end, List<AbstractTestDiagnostic> diagnostics) {
super(start, end);
this.diagnostics = diagnostics;
}
public List<ActualDiagnostic> getDiagnostics() {
public List<AbstractTestDiagnostic> getDiagnostics() {
return diagnostics;
}
public Map<ActualDiagnostic, TextDiagnostic> getTextDiagnosticsMap() {
Map<ActualDiagnostic, TextDiagnostic> diagnosticMap = new HashMap<>();
for (ActualDiagnostic diagnostic : diagnostics) {
public Map<AbstractTestDiagnostic, TextDiagnostic> getTextDiagnosticsMap() {
Map<AbstractTestDiagnostic, TextDiagnostic> diagnosticMap = new HashMap<>();
for (AbstractTestDiagnostic diagnostic : diagnostics) {
diagnosticMap.put(diagnostic, TextDiagnostic.asTextDiagnostic(diagnostic));
}
return diagnosticMap;
}
@Override
public boolean isWithNewInference() {
return false;
}
}
private static class TextDiagnosticDescriptor extends AbstractDiagnosticDescriptor {
@@ -757,34 +727,57 @@ public class CheckerTestUtil {
public TextDiagnostic getTextDiagnostic() {
return positionalTextDiagnostic.getDiagnostic();
}
@Override
public boolean isWithNewInference() {
return positionalTextDiagnostic.getDiagnostic().withNewInference;
}
}
public static class ActualDiagnostic {
public interface AbstractTestDiagnostic {
String getName();
String getPlatform();
TextDiagnostic.InferenceCompatibility getInferenceCompatibility();
void enhanceInferenceCompatibility(TextDiagnostic.InferenceCompatibility inferenceCompatibility);
}
public static class ActualDiagnostic implements AbstractTestDiagnostic {
public final Diagnostic diagnostic;
public final String platform;
public final boolean withNewInference;
public TextDiagnostic.InferenceCompatibility inferenceCompatibility;
ActualDiagnostic(@NotNull Diagnostic diagnostic, @Nullable String platform, boolean withNewInference) {
this.diagnostic = diagnostic;
this.platform = platform;
this.withNewInference = withNewInference;
this.inferenceCompatibility = withNewInference ?
TextDiagnostic.InferenceCompatibility.NEW :
TextDiagnostic.InferenceCompatibility.OLD;
}
@Override
@NotNull
public String getName() {
return diagnostic.getFactory().getName();
}
@Override
public String getPlatform() {
return platform;
}
@NotNull
public PsiFile getFile() {
return diagnostic.getPsiFile();
}
@Override
public TextDiagnostic.InferenceCompatibility getInferenceCompatibility() {
return inferenceCompatibility;
}
@Override
public void enhanceInferenceCompatibility(TextDiagnostic.InferenceCompatibility inferenceCompatibility) {
this.inferenceCompatibility = inferenceCompatibility;
}
@Override
public boolean equals(Object obj) {
if (!(obj instanceof ActualDiagnostic)) return false;
@@ -793,44 +786,66 @@ public class CheckerTestUtil {
// '==' on diagnostics is intentional here
return other.diagnostic == diagnostic &&
(other.platform == null ? platform == null : other.platform.equals(platform)) &&
(other.withNewInference == withNewInference);
(other.inferenceCompatibility == inferenceCompatibility);
}
@Override
public int hashCode() {
int result = System.identityHashCode(diagnostic);
result = 31 * result + (platform != null ? platform.hashCode() : 0);
result = 31 * result + (withNewInference ? 0 : 1);
result = 31 * result + inferenceCompatibility.hashCode();
return result;
}
@Override
public String toString() {
return (withNewInference ? NEW_INFERENCE_PREFIX + ";" : "") + (platform != null ? platform + ":" : "") + diagnostic.toString();
String inferenceAbbreviation = inferenceCompatibility.abbreviation;
return (inferenceAbbreviation != null ? inferenceAbbreviation + ";" : "") +
(platform != null ? platform + ":" : "") +
diagnostic.toString();
}
}
public static class TextDiagnostic {
public static class TextDiagnostic implements AbstractTestDiagnostic {
public enum InferenceCompatibility {
NEW(NEW_INFERENCE_PREFIX), OLD(OLD_INFERENCE_PREFIX), ALL(null);
@Nullable String abbreviation;
InferenceCompatibility(@Nullable String abbreviation) {
this.abbreviation = abbreviation;
}
public boolean isCompatible(InferenceCompatibility other) {
return this == other || this == ALL || other == ALL;
}
}
@NotNull
private static TextDiagnostic parseDiagnostic(String text) {
Matcher matcher = INDIVIDUAL_DIAGNOSTIC_PATTERN.matcher(text);
if (!matcher.find())
throw new IllegalArgumentException("Could not parse diagnostic: " + text);
boolean withNewInference = NEW_INFERENCE_PREFIX.equals(extractDataBefore(matcher.group(1), ";"));
InferenceCompatibility inference = computeInferenceCompatibility(extractDataBefore(matcher.group(1), ";"));
String platform = extractDataBefore(matcher.group(2), ":");
String name = matcher.group(3);
String parameters = matcher.group(4);
if (parameters == null) {
return new TextDiagnostic(name, platform, null, withNewInference);
return new TextDiagnostic(name, platform, null, inference);
}
List<String> parsedParameters = new SmartList<>();
Matcher parametersMatcher = INDIVIDUAL_PARAMETER_PATTERN.matcher(parameters);
while (parametersMatcher.find())
parsedParameters.add(unescape(parametersMatcher.group().trim()));
return new TextDiagnostic(name, platform, parsedParameters, withNewInference);
return new TextDiagnostic(name, platform, parsedParameters, inference);
}
private static InferenceCompatibility computeInferenceCompatibility(@Nullable String abbreviation) {
if (abbreviation == null) return InferenceCompatibility.ALL;
return ArraysKt.single(InferenceCompatibility.values(), inference -> abbreviation.equals(inference.abbreviation));
}
private static String extractDataBefore(@Nullable String prefix, @NotNull String anchor) {
@@ -846,6 +861,14 @@ public class CheckerTestUtil {
return s.replaceAll("\\\\([" + SHOULD_BE_ESCAPED + "])", "$1");
}
public static TextDiagnostic asTextDiagnostic(@NotNull AbstractTestDiagnostic abstractTestDiagnostic) {
if (abstractTestDiagnostic instanceof ActualDiagnostic) {
return asTextDiagnostic((ActualDiagnostic) abstractTestDiagnostic);
}
return (TextDiagnostic) abstractTestDiagnostic;
}
@NotNull
public static TextDiagnostic asTextDiagnostic(@NotNull ActualDiagnostic actualDiagnostic) {
Diagnostic diagnostic = actualDiagnostic.diagnostic;
@@ -856,9 +879,9 @@ public class CheckerTestUtil {
//noinspection unchecked
Object[] renderParameters = ((AbstractDiagnosticWithParametersRenderer) renderer).renderParameters(diagnostic);
List<String> parameters = ContainerUtil.map(renderParameters, Object::toString);
return new TextDiagnostic(diagnosticName, actualDiagnostic.platform, parameters, actualDiagnostic.withNewInference);
return new TextDiagnostic(diagnosticName, actualDiagnostic.platform, parameters, actualDiagnostic.inferenceCompatibility);
}
return new TextDiagnostic(diagnosticName, actualDiagnostic.platform, null, actualDiagnostic.withNewInference);
return new TextDiagnostic(diagnosticName, actualDiagnostic.platform, null, actualDiagnostic.inferenceCompatibility);
}
@NotNull
@@ -867,15 +890,28 @@ public class CheckerTestUtil {
private final String platform;
@Nullable
private final List<String> parameters;
private final boolean withNewInference;
@NotNull
private InferenceCompatibility inferenceCompatibility;
public TextDiagnostic(@NotNull String name, @Nullable String platform, @Nullable List<String> parameters, boolean withNewInference) {
public TextDiagnostic(
@NotNull String name,
@Nullable String platform,
@Nullable List<String> parameters,
@Nullable InferenceCompatibility inference
) {
this.name = name;
this.platform = platform;
this.parameters = parameters;
this.withNewInference = withNewInference;
this.inferenceCompatibility = inference != null ? inference : InferenceCompatibility.ALL;
}
@NotNull
@Override
public String getName() {
return name;
}
@Override
@Nullable
public String getPlatform() {
return platform;
@@ -891,6 +927,17 @@ public class CheckerTestUtil {
return parameters;
}
@NotNull
@Override
public InferenceCompatibility getInferenceCompatibility() {
return inferenceCompatibility;
}
@Override
public void enhanceInferenceCompatibility(InferenceCompatibility inferenceCompatibility) {
this.inferenceCompatibility = inferenceCompatibility;
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
@@ -901,7 +948,7 @@ public class CheckerTestUtil {
if (!name.equals(that.name)) return false;
if (platform != null ? !platform.equals(that.platform) : that.platform != null) return false;
if (parameters != null ? !parameters.equals(that.parameters) : that.parameters != null) return false;
if (withNewInference != that.withNewInference) return false;
if (inferenceCompatibility != that.inferenceCompatibility) return false;
return true;
}
@@ -911,15 +958,15 @@ public class CheckerTestUtil {
int result = name.hashCode();
result = 31 * result + (platform != null ? platform.hashCode() : 0);
result = 31 * result + (parameters != null ? parameters.hashCode() : 0);
result = 31 * result + (withNewInference ? 0 : 1);
result = 31 * result + inferenceCompatibility.hashCode();
return result;
}
@NotNull
public String asString() {
StringBuilder result = new StringBuilder();
if (withNewInference) {
result.append(NEW_INFERENCE_PREFIX);
if (inferenceCompatibility.abbreviation != null) {
result.append(inferenceCompatibility.abbreviation);
result.append(";");
}
if (platform != null) {
@@ -225,15 +225,22 @@ abstract class BaseDiagnosticsTest : KotlinMultiFileTestWithJava<TestModule, Tes
val withNewInference = newInferenceEnabled && withNewInferenceDirective && !USE_OLD_INFERENCE_DIAGNOSTICS_FOR_NI
val diagnostics = ContainerUtil.filter(
CheckerTestUtil.getDiagnosticsIncludingSyntaxErrors(
bindingContext, implementingModulesBindings, ktFile, markDynamicCalls, dynamicCallDescriptors, withNewInference
bindingContext, implementingModulesBindings, ktFile, markDynamicCalls, dynamicCallDescriptors, newInferenceEnabled
) + jvmSignatureDiagnostics,
{ whatDiagnosticsToConsider.value(it.diagnostic) }
)
val uncheckedDiagnostics = mutableListOf<PositionalTextDiagnostic>()
val inferenceCompatibilityOfTest = asInferenceCompatibility(withNewInference)
val invertedInferenceCompatibilityOfTest = asInferenceCompatibility(!withNewInference)
val diagnosticToExpectedDiagnostic = CheckerTestUtil.diagnosticsDiff(diagnosedRanges, diagnostics, object : CheckerTestUtil.DiagnosticDiffCallbacks {
override fun missingDiagnostic(diagnostic: CheckerTestUtil.TextDiagnostic, expectedStart: Int, expectedEnd: Int) {
if (withNewInferenceDirective && diagnostic.inferenceCompatibility != inferenceCompatibilityOfTest) {
updateUncheckedDiagnostics(diagnostic, expectedStart, expectedEnd)
return
}
val message = "Missing " + diagnostic.description + DiagnosticUtils.atLocation(ktFile, TextRange(expectedStart, expectedEnd))
System.err.println(message)
ok[0] = false
@@ -253,18 +260,20 @@ abstract class BaseDiagnosticsTest : KotlinMultiFileTestWithJava<TestModule, Tes
}
override fun unexpectedDiagnostic(diagnostic: CheckerTestUtil.TextDiagnostic, actualStart: Int, actualEnd: Int) {
if (withNewInferenceDirective && diagnostic.inferenceCompatibility != inferenceCompatibilityOfTest) {
updateUncheckedDiagnostics(diagnostic, actualStart, actualEnd)
return
}
val message = "Unexpected ${diagnostic.description}${DiagnosticUtils.atLocation(ktFile, TextRange(actualStart, actualEnd))}"
System.err.println(message)
ok[0] = false
}
override fun uncheckedDiagnostic(diagnostic: CheckerTestUtil.TextDiagnostic, expectedStart: Int, expectedEnd: Int) {
uncheckedDiagnostics.add(PositionalTextDiagnostic(diagnostic, expectedStart, expectedEnd))
fun updateUncheckedDiagnostics(diagnostic: CheckerTestUtil.TextDiagnostic, start: Int, end: Int) {
diagnostic.enhanceInferenceCompatibility(invertedInferenceCompatibilityOfTest)
uncheckedDiagnostics.add(PositionalTextDiagnostic(diagnostic, start, end))
}
override fun shouldUseDiagnosticsForNI(): Boolean = withNewInference
override fun isWithNewInferenceDirective(): Boolean = withNewInferenceDirective
})
actualText.append(CheckerTestUtil.addDiagnosticMarkersToText(
@@ -276,6 +285,13 @@ abstract class BaseDiagnosticsTest : KotlinMultiFileTestWithJava<TestModule, Tes
return ok[0]
}
private fun asInferenceCompatibility(isNewInference: Boolean): CheckerTestUtil.TextDiagnostic.InferenceCompatibility {
return if (isNewInference)
CheckerTestUtil.TextDiagnostic.InferenceCompatibility.NEW
else
CheckerTestUtil.TextDiagnostic.InferenceCompatibility.OLD
}
private fun computeJvmSignatureDiagnostics(bindingContext: BindingContext): Set<ActualDiagnostic> {
val jvmSignatureDiagnostics = HashSet<ActualDiagnostic>()
val declarations = PsiTreeUtil.findChildrenOfType(ktFile, KtDeclaration::class.java)
@@ -105,7 +105,7 @@ public class CheckerTestUtilTest extends KotlinTestWithEnvironment {
DiagnosticData unused = diagnostics.get(2);
String unusedDiagnostic = asTextDiagnostic(unused, "i");
DiagnosedRange range = asDiagnosticRange(unused, unusedDiagnostic);
doTest(new TheTest(wrongParameters(unusedDiagnostic, "UNUSED_VARIABLE(a)", unused.startOffset, unused.endOffset)) {
doTest(new TheTest(wrongParameters(unusedDiagnostic, "OI;UNUSED_VARIABLE(a)", unused.startOffset, unused.endOffset)) {
@Override
protected void makeTestData(List<ActualDiagnostic> diagnostics, List<DiagnosedRange> diagnosedRanges) {
diagnosedRanges.set(unused.rangeIndex, range);
@@ -118,7 +118,7 @@ public class CheckerTestUtilTest extends KotlinTestWithEnvironment {
String unusedDiagnostic = asTextDiagnostic(unresolvedReference, "i");
String toManyArguments = asTextDiagnostic(diagnostics.get(7));
DiagnosedRange range = asDiagnosticRange(unresolvedReference, unusedDiagnostic, toManyArguments);
doTest(new TheTest(wrongParameters(unusedDiagnostic, "UNRESOLVED_REFERENCE(xx)", unresolvedReference.startOffset, unresolvedReference.endOffset)) {
doTest(new TheTest(wrongParameters(unusedDiagnostic, "OI;UNRESOLVED_REFERENCE(xx)", unresolvedReference.startOffset, unresolvedReference.endOffset)) {
@Override
protected void makeTestData(List<ActualDiagnostic> diagnostics, List<DiagnosedRange> diagnosedRanges) {
diagnosedRanges.set(unresolvedReference.rangeIndex, range);
@@ -181,20 +181,6 @@ public class CheckerTestUtilTest extends KotlinTestWithEnvironment {
public void unexpectedDiagnostic(CheckerTestUtil.TextDiagnostic diagnostic, int actualStart, int actualEnd) {
actualMessages.add(unexpected(diagnostic.getDescription(), actualStart, actualEnd));
}
@Override
public void uncheckedDiagnostic(CheckerTestUtil.TextDiagnostic diagnostic, int expectedStart, int expectedEnd) {
}
@Override
public boolean shouldUseDiagnosticsForNI() {
return false;
}
@Override
public boolean isWithNewInferenceDirective() {
return false;
}
});
assertEquals(listToString(expectedMessages), listToString(actualMessages));