From 2f8e95dace6847845acf12591c237f12a2d70fb9 Mon Sep 17 00:00:00 2001
From: Dmitriy Novozhilov <dmitriy.novozhilov@jetbrains.com>
Date: Fri, 5 Jun 2020 12:16:34 +0300
Subject: [PATCH] [FIR-TEST] Add validation for completed graph in diagnostic
 tests

---
 .../tests/org/jetbrains/kotlin/fir/AbstractFirDiagnosticTest.kt  | 1 +
 1 file changed, 1 insertion(+)

diff --git a/compiler/tests-common/tests/org/jetbrains/kotlin/fir/AbstractFirDiagnosticTest.kt b/compiler/tests-common/tests/org/jetbrains/kotlin/fir/AbstractFirDiagnosticTest.kt
index 229e0806d08..bf8ca2070da 100644
--- a/compiler/tests-common/tests/org/jetbrains/kotlin/fir/AbstractFirDiagnosticTest.kt
+++ b/compiler/tests-common/tests/org/jetbrains/kotlin/fir/AbstractFirDiagnosticTest.kt
@@ -283,6 +283,7 @@ abstract class AbstractFirDiagnosticsTest : AbstractFirBaseDiagnosticsTest() {
 
         override fun visitControlFlowGraphReference(controlFlowGraphReference: FirControlFlowGraphReference) {
             val graph = (controlFlowGraphReference as? FirControlFlowGraphReferenceImpl)?.controlFlowGraph ?: return
+            assertEquals(ControlFlowGraph.State.Completed, graph.state)
             checkConsistency(graph)
         }