From 38543ce2a3d7b2a5d27b950054a73b2c239339f2 Mon Sep 17 00:00:00 2001
From: "vladislavf7@gmail.com" <BVfp46jJS783vfX>
Date: Thu, 16 Jul 2020 15:00:58 +0300
Subject: [PATCH] [FIR] add way to configure session in Diagnostic Tests

---
 .../org/jetbrains/kotlin/fir/AbstractFirBaseDiagnosticsTest.kt | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/compiler/tests-common/tests/org/jetbrains/kotlin/fir/AbstractFirBaseDiagnosticsTest.kt b/compiler/tests-common/tests/org/jetbrains/kotlin/fir/AbstractFirBaseDiagnosticsTest.kt
index 256f4ce975e..509422e5152 100644
--- a/compiler/tests-common/tests/org/jetbrains/kotlin/fir/AbstractFirBaseDiagnosticsTest.kt
+++ b/compiler/tests-common/tests/org/jetbrains/kotlin/fir/AbstractFirBaseDiagnosticsTest.kt
@@ -90,6 +90,7 @@ abstract class AbstractFirBaseDiagnosticsTest : BaseDiagnosticsTest() {
                 moduleFiles.mapNotNull { it.ktFile })
             FirJavaModuleBasedSession.create(info, sessionProvider, scope).also {
                 registerFirExtensions(it.extensionService)
+                configureSession(it)
             }
         }
 
@@ -331,4 +332,6 @@ abstract class AbstractFirBaseDiagnosticsTest : BaseDiagnosticsTest() {
         }
         return result
     }
+
+    protected open fun configureSession(session: FirSession) {}
 }
\ No newline at end of file