1 files changed, 2 insertions, 23 deletions
diff --git a/tests/test.integration.ui/src/main/java/org/apache/directory/studio/test/integration/ui/bots/DeleteDialogBot.java b/tests/test.integration.ui/src/main/java/org/apache/directory/studio/test/integration/ui/bots/DeleteDialogBot.java
index 986fa5316..7829fedfd 100644
--- a/tests/test.integration.ui/src/main/java/org/apache/directory/studio/test/integration/ui/bots/DeleteDialogBot.java
+++ b/tests/test.integration.ui/src/main/java/org/apache/directory/studio/test/integration/ui/bots/DeleteDialogBot.java
@@ -21,7 +21,6 @@ package org.apache.directory.studio.test.integration.ui.bots;
 
 
 import org.apache.directory.studio.ldapbrowser.core.BrowserCoreMessages;
-import org.apache.directory.studio.test.integration.ui.utils.JobWatcher;
 
 
 public class DeleteDialogBot extends DialogBot
@@ -34,26 +33,11 @@ public class DeleteDialogBot extends DialogBot
     public static final String DELETE_SERVER = "Delete Server";
     public static final String DELETE_PROJECT = "Delete Project";
 
-
     public DeleteDialogBot( String title )
     {
         super( title );
-    }
-
-
-    public void clickOkButton()
-    {
-        String jobWatcherTitle = getJobWatcherTitle( title );
-        if ( jobWatcherTitle != null )
-        {
-            JobWatcher watcher = new JobWatcher( jobWatcherTitle );
-            super.clickOkButton();
-            watcher.waitUntilDone();
-        }
-        else
-        {
-            super.clickOkButton();
-        }
+        String jobName = getJobWatcherTitle( title );
+        super.setWaitAfterClickOkButton( jobName != null, jobName );
     }
 
 
@@ -72,9 +56,4 @@ public class DeleteDialogBot extends DialogBot
         }
     }
 
-
-    public void clickCancelButton()
-    {
-        super.clickButton( "Cancel" );
-    }
 }