From 236a76f779d208cadd4462fb99cb2b6d2336eee2 Mon Sep 17 00:00:00 2001
From: Roman Golyshev <roman.golyshev@jetbrains.com>
Date: Mon, 30 Oct 2023 21:25:50 +0100
Subject: [PATCH] KT-62676 [AA] Small refactoring of
 `AbstractReferenceShortener*Test`

---
 .../references/AbstractReferenceShortenerForWholeFileTest.kt  | 4 ++--
 .../test/cases/references/AbstractReferenceShortenerTest.kt   | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/analysis/analysis-api-impl-base/tests/org/jetbrains/kotlin/analysis/api/impl/base/test/cases/references/AbstractReferenceShortenerForWholeFileTest.kt b/analysis/analysis-api-impl-base/tests/org/jetbrains/kotlin/analysis/api/impl/base/test/cases/references/AbstractReferenceShortenerForWholeFileTest.kt
index a466f76d7cd..35815cbd77d 100644
--- a/analysis/analysis-api-impl-base/tests/org/jetbrains/kotlin/analysis/api/impl/base/test/cases/references/AbstractReferenceShortenerForWholeFileTest.kt
+++ b/analysis/analysis-api-impl-base/tests/org/jetbrains/kotlin/analysis/api/impl/base/test/cases/references/AbstractReferenceShortenerForWholeFileTest.kt
@@ -20,7 +20,7 @@ abstract class AbstractReferenceShortenerForWholeFileTest : AbstractAnalysisApiB
 
         val shortenings = executeOnPooledThreadInReadAction {
             analyseForTest(file) {
-                ShortenStrategy.values().map { option ->
+                ShortenStrategy.entries.associateWith { option ->
                     val shorteningsForOption = collectPossibleReferenceShortenings(
                         file,
                         file.textRange,
@@ -28,7 +28,7 @@ abstract class AbstractReferenceShortenerForWholeFileTest : AbstractAnalysisApiB
                         callableShortenStrategy = { option }
                     )
 
-                    Pair(option.name, shorteningsForOption)
+                    shorteningsForOption
                 }
             }
         }
diff --git a/analysis/analysis-api-impl-base/tests/org/jetbrains/kotlin/analysis/api/impl/base/test/cases/references/AbstractReferenceShortenerTest.kt b/analysis/analysis-api-impl-base/tests/org/jetbrains/kotlin/analysis/api/impl/base/test/cases/references/AbstractReferenceShortenerTest.kt
index beeb240fa44..dde180d48f2 100644
--- a/analysis/analysis-api-impl-base/tests/org/jetbrains/kotlin/analysis/api/impl/base/test/cases/references/AbstractReferenceShortenerTest.kt
+++ b/analysis/analysis-api-impl-base/tests/org/jetbrains/kotlin/analysis/api/impl/base/test/cases/references/AbstractReferenceShortenerTest.kt
@@ -28,7 +28,7 @@ abstract class AbstractReferenceShortenerTest : AbstractAnalysisApiBasedSingleMo
 
         val shortenings = executeOnPooledThreadInReadAction {
             analyseForTest(element) {
-                ShortenStrategy.values().map { option ->
+                ShortenStrategy.entries.associateWith { option ->
                     val shorteningsForOption = collectPossibleReferenceShorteningsInElement(
                         element,
                         shortenOptions = ShortenOptions.ALL_ENABLED,
@@ -36,7 +36,7 @@ abstract class AbstractReferenceShortenerTest : AbstractAnalysisApiBasedSingleMo
                         callableShortenStrategy = { option }
                     )
 
-                    Pair(option.name, shorteningsForOption)
+                    shorteningsForOption
                 }
             }
         }