Close tab with generated test suite if all test cases were deleted (#331

) * check generatedTestsTabData.testCasePanelFactories.size * fix TopButtonsPanelBuilder.update
JetBrains-Research · Sep 30, 2024 · b4a2449 · b4a2449
1 parent b991154
commit b4a2449
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/.../kotlin/org/jetbrains/research/testspark/display/generatedTests/TopButtonsPanelBuilder.kt b/.../kotlin/org/jetbrains/research/testspark/display/generatedTests/TopButtonsPanelBuilder.kt
@@ -40,13 +40,15 @@ class TopButtonsPanelBuilder {
      * Updates the labels.
      */
     fun update(generatedTestsTabData: GeneratedTestsTabData) {
-        var numberOfPassedTests = 0
-        for (testCasePanelFactory in generatedTestsTabData.testCasePanelFactories) {
-            if (testCasePanelFactory.isRemoved()) continue
-            val error = testCasePanelFactory.getError()
-            if ((error is String) && error.isEmpty()) {
-                numberOfPassedTests++
-            }
+        val passedTestsCount = generatedTestsTabData.testCasePanelFactories
+            .filter { !it.isRemoved() }
+            .count { it.getError()?.isEmpty() == true }
+
+        val removedTestsCount = generatedTestsTabData.testCasePanelFactories.count { it.isRemoved() }
+
+        if (generatedTestsTabData.testCasePanelFactories.size == removedTestsCount) {
+            removeAllButton.doClick()
+            return
         }
         testsSelectedLabel.text = String.format(
             testsSelectedText,
@@ -56,7 +58,7 @@ class TopButtonsPanelBuilder {
         testsPassedLabel.text =
             String.format(
                 testsPassedText,
-                numberOfPassedTests,
+                passedTestsCount,
                 generatedTestsTabData.testCaseNameToPanel.size,
             )
         runAllButton.isEnabled = false