From e24044f3aa8b9e4925b653d712903f26ee736d91 Mon Sep 17 00:00:00 2001
From: anna <Anna.Kozlova@jetbrains.com>
Date: Wed, 7 Sep 2011 11:17:48 +0200
Subject: [PATCH] testng: connect out of AWT thread

---
 .../model/IDEARemoteTestRunnerClient.java     | 113 +++++++++---------
 1 file changed, 58 insertions(+), 55 deletions(-)

diff --git a/plugins/testng/src/com/theoryinpractice/testng/model/IDEARemoteTestRunnerClient.java b/plugins/testng/src/com/theoryinpractice/testng/model/IDEARemoteTestRunnerClient.java
index 1c8e2f23b4e5..71d5c60ddd80 100644
--- a/plugins/testng/src/com/theoryinpractice/testng/model/IDEARemoteTestRunnerClient.java
+++ b/plugins/testng/src/com/theoryinpractice/testng/model/IDEARemoteTestRunnerClient.java
@@ -43,87 +43,90 @@ public class IDEARemoteTestRunnerClient extends AbstractRemoteTestRunnerClient
         protected void handleThrowable(Throwable cause) {
           cause.printStackTrace();
         }
+
+        @Override
+        public void run() {
+          try {
+            messageSender.initReceiver();
+          }
+          catch (SocketTimeoutException e) {
+            e.printStackTrace();
+          }
+          super.run();
+        }
       };
-      try {
-        messageSender.initReceiver();
-      }
-      catch (SocketTimeoutException e) {
-        e.printStackTrace();
-      }
-      startListening(new IRemoteSuiteListener[]{myListener},
+    startListening(new IRemoteSuiteListener[]{myListener},
                      new IRemoteTestListener[]{myListener},
                      srvConnection);
     }
 
     @Override
     protected void notifyStart(final GenericMessage genericMessage) {
-        ApplicationManager.getApplication().invokeLater(new Runnable()
-        {
-            public void run() {
-                for (final IRemoteSuiteListener listener : m_suiteListeners) {
-                    listener.onInitialization(genericMessage);
-                }
+        ApplicationManager.getApplication().invokeLater(new Runnable() {
+          public void run() {
+            for (final IRemoteSuiteListener listener : m_suiteListeners) {
+              listener.onInitialization(genericMessage);
             }
+          }
         }, ModalityState.NON_MODAL);
     }
 
     @Override
     protected void notifySuiteEvents(final SuiteMessage suiteMessage) {
-        ApplicationManager.getApplication().invokeLater(new Runnable()
-        {
-            public void run() {
-                for (final IRemoteSuiteListener listener : m_suiteListeners) {
-                    if (suiteMessage.isMessageOnStart()) {
-                        listener.onStart(suiteMessage);
-                    } else {
-                        listener.onFinish(suiteMessage);
-                    }
-                }
+        ApplicationManager.getApplication().invokeLater(new Runnable() {
+          public void run() {
+            for (final IRemoteSuiteListener listener : m_suiteListeners) {
+              if (suiteMessage.isMessageOnStart()) {
+                listener.onStart(suiteMessage);
+              }
+              else {
+                listener.onFinish(suiteMessage);
+              }
             }
+          }
         }, ModalityState.NON_MODAL);
     }
 
     @Override
     protected void notifyTestEvents(final TestMessage testMessage) {
-        ApplicationManager.getApplication().invokeLater(new Runnable()
-        {
-            public void run() {
-                for (final IRemoteTestListener listener : m_testListeners) {
-                    if (testMessage.isMessageOnStart()) {
-                        listener.onStart(testMessage);
-                    } else {
-                        listener.onFinish(testMessage);
-                    }
-                }
+        ApplicationManager.getApplication().invokeLater(new Runnable() {
+          public void run() {
+            for (final IRemoteTestListener listener : m_testListeners) {
+              if (testMessage.isMessageOnStart()) {
+                listener.onStart(testMessage);
+              }
+              else {
+                listener.onFinish(testMessage);
+              }
             }
+          }
         }, ModalityState.NON_MODAL);
     }
 
     @Override
     protected void notifyResultEvents(final TestResultMessage testResultMessage) {
-        ApplicationManager.getApplication().invokeLater(new Runnable()
-        {
-            public void run() {
-                for (final IRemoteTestListener listener : m_testListeners) {
-                    switch (testResultMessage.getResult()) {
-                        case MessageHelper.TEST_STARTED:
-                            listener.onTestStart(testResultMessage);
-                            break;
-                        case MessageHelper.PASSED_TEST:
-                            listener.onTestSuccess(testResultMessage);
-                            break;
-                        case MessageHelper.FAILED_TEST:
-                            listener.onTestFailure(testResultMessage);
-                            break;
-                        case MessageHelper.SKIPPED_TEST:
-                            listener.onTestSkipped(testResultMessage);
-                            break;
-                        case MessageHelper.FAILED_ON_PERCENTAGE_TEST:
-                            listener.onTestFailedButWithinSuccessPercentage(testResultMessage);
-                            break;
-                    }
-                }
+        ApplicationManager.getApplication().invokeLater(new Runnable() {
+          public void run() {
+            for (final IRemoteTestListener listener : m_testListeners) {
+              switch (testResultMessage.getResult()) {
+                case MessageHelper.TEST_STARTED:
+                  listener.onTestStart(testResultMessage);
+                  break;
+                case MessageHelper.PASSED_TEST:
+                  listener.onTestSuccess(testResultMessage);
+                  break;
+                case MessageHelper.FAILED_TEST:
+                  listener.onTestFailure(testResultMessage);
+                  break;
+                case MessageHelper.SKIPPED_TEST:
+                  listener.onTestSkipped(testResultMessage);
+                  break;
+                case MessageHelper.FAILED_ON_PERCENTAGE_TEST:
+                  listener.onTestFailedButWithinSuccessPercentage(testResultMessage);
+                  break;
+              }
             }
+          }
         }, ModalityState.NON_MODAL);
     }