[AA] AbstractSymbolTest: fix race

^KT-54311

analysis/analysis-api-impl-base/tests/org/jetbrains/kotlin/analysis/api/impl/base/test/cases/symbols/AbstractSymbolTest.kt

@@ -57,14 +57,16 @@ abstract class AbstractSymbolTest : AbstractAnalysisApiSingleFileTest() {
         }
 
         fun KtSymbol.safePointer(): PointerWrapper? {
-            val regularPointer = runCatching(::createPointer).let {
+            val regularPointer = runCatching {
+                KtPsiBasedSymbolPointer.withDisabledPsiBasedPointers(disable = false, action = ::createPointer)
+            }.let {
                 if (directiveToIgnoreSymbolRestore == null) it.getOrThrow() else it.getOrNull()
             } ?: return null
 
             val nonPsiPointer = kotlin.runCatching {
                 if (this is KtFileSymbol) return@runCatching null
 
-                KtPsiBasedSymbolPointer.withDisabledPsiBasedPointers(::createPointer)
+                KtPsiBasedSymbolPointer.withDisabledPsiBasedPointers(disable = true, action = ::createPointer)
             }
 
             return PointerWrapper(

analysis/analysis-api/src/org/jetbrains/kotlin/analysis/api/symbols/pointers/KtPsiBasedSymbolPointer.kt

@@ -56,13 +56,11 @@ public class KtPsiBasedSymbolPointer<S : KtSymbol> private constructor(private v
 
         @TestOnly
         @Synchronized
-        public fun <T> withDisabledPsiBasedPointers(action: () -> T): T {
+        public fun <T> withDisabledPsiBasedPointers(disable: Boolean, action: () -> T): T = try {
-            disablePsiPointer = true
+            disablePsiPointer = disable
-            return try {
+            action()
-                action()
+        } finally {
-            } finally {
+            disablePsiPointer = false
-                disablePsiPointer = false
-            }
         }
 
         @Volatile