From 2ea0e69a565e513bd273ba90a32860ef3e14f713 Mon Sep 17 00:00:00 2001
From: Simon Ogorodnik <simon.ogorodnik@jetbrains.com>
Date: Wed, 7 Oct 2020 04:57:09 +0300
Subject: [PATCH] [FIR-Test] Fix incorrect file count

---
 .../kotlin/fir/NonFirResolveModularizedTotalKotlinTest.kt   | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/compiler/fir/modularized-tests/tests/org/jetbrains/kotlin/fir/NonFirResolveModularizedTotalKotlinTest.kt b/compiler/fir/modularized-tests/tests/org/jetbrains/kotlin/fir/NonFirResolveModularizedTotalKotlinTest.kt
index 00e5739d56c..8239839ca1c 100644
--- a/compiler/fir/modularized-tests/tests/org/jetbrains/kotlin/fir/NonFirResolveModularizedTotalKotlinTest.kt
+++ b/compiler/fir/modularized-tests/tests/org/jetbrains/kotlin/fir/NonFirResolveModularizedTotalKotlinTest.kt
@@ -46,11 +46,13 @@ class NonFirResolveModularizedTotalKotlinTest : AbstractModularizedTest() {
         }
         val vmAfter = vmStateSnapshot()
 
-        files += environment.getSourceFiles().size
-        lines += environment.getSourceFiles().sumBy { StringUtil.countNewLines(it.text) }
+        val psiFiles = environment.getSourceFiles()
+        files += psiFiles.size
+        lines += psiFiles.sumBy { StringUtil.countNewLines(it.text) }
         totalTime += time
         measure.time += time
         measure.vmCounters += vmAfter - vmBefore
+        measure.files += psiFiles.size
 
         println("Time is ${time * 1e-6} ms")
     }