From 6f804671cd3d5fb0035c70fa327eb9212e7bc8e5 Mon Sep 17 00:00:00 2001
From: Max Medvedev <maxim.medvedev@jetbrains.com>
Date: Mon, 1 Feb 2016 14:31:40 +0300
Subject: [PATCH] build memory snaphost if a leak found

---
 platform/testFramework/src/_LastInSuiteTest.java | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/platform/testFramework/src/_LastInSuiteTest.java b/platform/testFramework/src/_LastInSuiteTest.java
index 85c790a0f8aa..8774a8e51627 100644
--- a/platform/testFramework/src/_LastInSuiteTest.java
+++ b/platform/testFramework/src/_LastInSuiteTest.java
@@ -74,6 +74,10 @@ public class _LastInSuiteTest extends TestCase {
       LeakHunter.checkProjectLeak();
       Disposer.assertIsEmpty(true);
     }
+    catch (AssertionError e) {
+      captureMemorySnapshot();
+      throw e;
+    }
     catch (Exception e) {
       captureMemorySnapshot();
       throw e;