From 632dba82282907260435dc87294ac8940340ffaa Mon Sep 17 00:00:00 2001
From: Ilya Chernikov <ilya.chernikov@jetbrains.com>
Date: Mon, 12 Dec 2016 10:08:36 +0100
Subject: [PATCH] minor: improve daemon test stability

---
 .../kotlin/daemon/CompilerDaemonTest.kt       | 24 +++++++++++--------
 1 file changed, 14 insertions(+), 10 deletions(-)

diff --git a/compiler/tests/org/jetbrains/kotlin/daemon/CompilerDaemonTest.kt b/compiler/tests/org/jetbrains/kotlin/daemon/CompilerDaemonTest.kt
index 74e79178a2c..53265810e59 100644
--- a/compiler/tests/org/jetbrains/kotlin/daemon/CompilerDaemonTest.kt
+++ b/compiler/tests/org/jetbrains/kotlin/daemon/CompilerDaemonTest.kt
@@ -413,16 +413,20 @@ class CompilerDaemonTest : KotlinIntegrationTestBase() {
             KotlinCompilerClient.shutdownCompileService(compilerId, daemonOptions)
             Thread.sleep(200)
 
-            val exception: Exception? = try {
-                daemon!!.getUsedMemory()
-                null
-            }
-            catch (e: java.rmi.ConnectException) {
-                e
-            }
-            catch (e: java.rmi.UnmarshalException) {
-                e
-            }
+            val exception: Exception? =
+                    try {
+                        daemon!!.getUsedMemory()
+                        null
+                    }
+                    catch (e: java.rmi.ConnectException) {
+                        e
+                    }
+                    catch (e: java.rmi.UnmarshalException) {
+                        e
+                    }
+                    catch (e: java.rmi.ConnectIOException) {
+                        e
+                    }
             assertNotNull(exception)
         }
     }