From 1549924ab0387b9c6bdb419369a6973910089d0d Mon Sep 17 00:00:00 2001
From: ArtemiiBezguzikov <artemii.bezguzikov@gmail.com>
Date: Fri, 5 May 2017 04:16:07 +0300
Subject: [PATCH] Code refactor

---
 .../controller/DiagramEditorController.ts     |  7 ++-
 .../editorCore/controller/SceneController.ts  |  6 ++-
 .../WEB-INF/views/editor/bpmn/toolbar.jsp     |  5 ++-
 .../com/qreal/wmp/selector/Application.java   |  6 +--
 .../com/qreal/wmp/selector/ConfigsMerger.java | 26 +++--------
 .../qreal/wmp/selector/RestController.java    | 14 +++---
 .../configs/editor/folders/folderArea.json    |  2 +-
 .../headerpanel/DiagramStoreService.java      |  6 +--
 .../headerpanel/EditorHeaderPanelImpl.java    | 18 ++++++--
 .../headerpanel/folderwindow/FileItem.java    | 44 ++++++++++++++-----
 .../folderwindow/FolderAreaImpl.java          | 33 ++++++++++----
 .../folderwindow/SaveDiagramConfirm.java      | 19 +++++---
 .../mousegestures/RobotCalibration.java       |  5 +++
 .../innertests/SaveOpenDiagramTest.java       |  2 -
 14 files changed, 125 insertions(+), 68 deletions(-)

diff --git a/editor-core/src/main/webapp/app/core/editorCore/controller/DiagramEditorController.ts b/editor-core/src/main/webapp/app/core/editorCore/controller/DiagramEditorController.ts
index 1f237839..f88bf1f7 100644
--- a/editor-core/src/main/webapp/app/core/editorCore/controller/DiagramEditorController.ts
+++ b/editor-core/src/main/webapp/app/core/editorCore/controller/DiagramEditorController.ts
@@ -35,7 +35,12 @@ export class DiagramEditorController {
             return this.getNodeType(type);
         };
         this.diagramEditor = new DiagramEditor();
-        this.sceneController = new SceneController(this, this.diagramEditor.getScene(), selectorService.scene);
+        this.sceneController = new SceneController(
+            this,
+            this.diagramEditor.getScene(),
+            selectorService.scene,
+            selectorService.gestures);
+
         this.elementsTypeLoader = new ElementsTypeLoader();
 
         $scope.undo = () => {
diff --git a/editor-core/src/main/webapp/app/core/editorCore/controller/SceneController.ts b/editor-core/src/main/webapp/app/core/editorCore/controller/SceneController.ts
index d208fa84..1e079a76 100644
--- a/editor-core/src/main/webapp/app/core/editorCore/controller/SceneController.ts
+++ b/editor-core/src/main/webapp/app/core/editorCore/controller/SceneController.ts
@@ -30,8 +30,9 @@ export class SceneController {
     private paperCommandFactory: SceneCommandFactory;
     private contextMenuId;
     private selectorService : any;
+    private gestures : any;
 
-    constructor(diagramEditorController: DiagramEditorController, paper: DiagramScene, selectorService: any) {
+    constructor(diagramEditorController: DiagramEditorController, paper: DiagramScene, selectorService: any, gestures: any) {
         this.diagramEditorController = diagramEditorController;
         this.undoRedoController = diagramEditorController.getUndoRedoController();
         this.scene = paper;
@@ -44,6 +45,7 @@ export class SceneController {
         this.lastCellScrollPosition = { x: 0, y: 0 };
         this.contextMenuId = selectorService.contextMenu.id;
         this.selectorService = selectorService;
+        this.gestures = gestures;
 
         this.scene.on('cell:pointerdown', (cellView, event, x, y): void => {
             this.cellPointerdownListener(cellView, event, x, y);
@@ -172,7 +174,7 @@ export class SceneController {
 
         var contextMenu = new ContextMenu();
         var menuDiv = document.createElement("div");
-        menuDiv.id = this.selectorService.gestures.gesturesMenu.id;
+        menuDiv.id = this.gestures.gesturesMenu.id;
         menuDiv.className = "gestures-menu";
         menuDiv.style.left = event.x + "px";
         menuDiv.style.top = event.y + "px";
diff --git a/editor-service/src/main/webapp/WEB-INF/views/editor/bpmn/toolbar.jsp b/editor-service/src/main/webapp/WEB-INF/views/editor/bpmn/toolbar.jsp
index 7e109170..3550ff7b 100644
--- a/editor-service/src/main/webapp/WEB-INF/views/editor/bpmn/toolbar.jsp
+++ b/editor-service/src/main/webapp/WEB-INF/views/editor/bpmn/toolbar.jsp
@@ -1,8 +1,9 @@
 <%JSONObject headerPanelSelector = selectors.getJSONObject("editorHeaderPanel"); %>
 
 <!-- Context menu in File->'Open diagram' window -->
-<ul id="<%=headerPanelSelector.getJSONObject("contextMenu").getString("id")%>" class='custom-menu'>
-    <li id="<%=headerPanelSelector.getJSONObject("contextMenu").getJSONObject("deleteItem").getString("id")%>"
+<%JSONObject contextMenuSelector = headerPanelSelector.getJSONObject("folderArea").getJSONObject("contextMenu"); %>
+<ul id="<%=contextMenuSelector.getString("id")%>" class='custom-menu'>
+    <li id="<%=contextMenuSelector.getJSONObject("deleteItem").getString("id")%>"
         data-action="delete">Delete</li>
     <li data-action="share" >Share</li>
 </ul>
diff --git a/selector-service/src/main/java/com/qreal/wmp/selector/Application.java b/selector-service/src/main/java/com/qreal/wmp/selector/Application.java
index 285a2bde..00040850 100644
--- a/selector-service/src/main/java/com/qreal/wmp/selector/Application.java
+++ b/selector-service/src/main/java/com/qreal/wmp/selector/Application.java
@@ -5,7 +5,7 @@
 
 @SpringBootApplication
 public class Application {
-	public static void main(String[] args) throws Exception {
-		SpringApplication.run(Application.class, args);
-	}
+    public static void main(String[] args) throws Exception {
+        SpringApplication.run(Application.class, args);
+    }
 }
\ No newline at end of file
diff --git a/selector-service/src/main/java/com/qreal/wmp/selector/ConfigsMerger.java b/selector-service/src/main/java/com/qreal/wmp/selector/ConfigsMerger.java
index 26f5f889..abb1a6e6 100644
--- a/selector-service/src/main/java/com/qreal/wmp/selector/ConfigsMerger.java
+++ b/selector-service/src/main/java/com/qreal/wmp/selector/ConfigsMerger.java
@@ -3,8 +3,6 @@
 import com.google.gson.JsonElement;
 import com.google.gson.JsonObject;
 import com.google.gson.JsonParser;
-import org.slf4j.Logger;
-import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.context.annotation.PropertySource;
 import org.springframework.stereotype.Component;
@@ -18,27 +16,20 @@
 @PropertySource("classpath:application.properties")
 public class ConfigsMerger {
     
-    private static final Logger logger = LoggerFactory.getLogger(ConfigsMerger.class);
-    
     private final JsonParser parser = new JsonParser();
     
     @Value("${selectorConfig}")
     private String configFile;
     
     /** Generates one config file from all which are used. */
-    public JsonObject generateCommonConfig() {
+    public JsonObject generateCommonConfig() throws FileNotFoundException {
         String path = getClass().getClassLoader().getResource(configFile).getPath();
-        try {
-            JsonElement jsonElement = parser.parse(new FileReader(path));
-            JsonObject initialConfig = jsonElement.getAsJsonObject();
-            return merge(initialConfig, path);
-        } catch (FileNotFoundException e) {
-            logger.error(e.getMessage());
-            throw new RuntimeException("There are no files on path: " + path);
-        }
+        JsonElement jsonElement = parser.parse(new FileReader(path));
+        JsonObject initialConfig = jsonElement.getAsJsonObject();
+        return merge(initialConfig, path);
     }
     
-    private JsonObject merge(JsonObject parent, String path) {
+    private JsonObject merge(JsonObject parent, String path) throws FileNotFoundException {
         JsonObject result = new JsonObject();
         for (Map.Entry<String, JsonElement> childEntry: parent.entrySet()) {
             if (childEntry.getValue().isJsonPrimitive()) {
@@ -51,12 +42,7 @@ private JsonObject merge(JsonObject parent, String path) {
                             + stringValue.substring(pathLink.length(), stringValue.length());
                     
                     JsonObject parsedChild;
-                    try {
-                        parsedChild = parser.parse(new FileReader(newPath)).getAsJsonObject();
-                    } catch (FileNotFoundException e) {
-                        logger.error(e.getMessage());
-                        throw new RuntimeException("There are no files on path: " + path);
-                    }
+                    parsedChild = parser.parse(new FileReader(newPath)).getAsJsonObject();
                     result.add(childEntry.getKey(), merge(parsedChild, newPath));
                 } else {
                     result.add(childEntry.getKey(), childEntry.getValue());
diff --git a/selector-service/src/main/java/com/qreal/wmp/selector/RestController.java b/selector-service/src/main/java/com/qreal/wmp/selector/RestController.java
index cfd9b04a..7d0dd5af 100644
--- a/selector-service/src/main/java/com/qreal/wmp/selector/RestController.java
+++ b/selector-service/src/main/java/com/qreal/wmp/selector/RestController.java
@@ -5,6 +5,8 @@
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.ResponseBody;
 
+import java.io.FileNotFoundException;
+
 @Controller
 public class RestController {
     
@@ -17,37 +19,37 @@ public RestController(ConfigsMerger merger) {
     
     @GetMapping("/all")
     @ResponseBody
-    public String getAllConfig() {
+    public String getAllConfig() throws FileNotFoundException {
         return merger.generateCommonConfig().toString();
     }
     
     @GetMapping("/editor/robots")
     @ResponseBody
-    public String getRobotEditor() {
+    public String getRobotEditor() throws FileNotFoundException {
         return merger.generateCommonConfig().get("robotEditor").toString();
     }
     
     @GetMapping("/editor/bpmn")
     @ResponseBody
-    public String getBPMNEditor() {
+    public String getBPMNEditor() throws FileNotFoundException {
         return merger.generateCommonConfig().get("bpmnEditor").toString();
     }
     
     @GetMapping("/dashboard")
     @ResponseBody
-    public String getDashboard() {
+    public String getDashboard() throws FileNotFoundException {
         return merger.generateCommonConfig().get("dashboard").toString();
     }
     
     @GetMapping("/authform")
     @ResponseBody
-    public String getAuthform() {
+    public String getAuthform() throws FileNotFoundException {
         return merger.generateCommonConfig().get("authform").toString();
     }
     
     @GetMapping("/auth")
     @ResponseBody
-    public String getAuth() {
+    public String getAuth() throws FileNotFoundException {
         return merger.generateCommonConfig().get("authService").toString();
     }
     
diff --git a/selector-service/src/main/resources/configs/editor/folders/folderArea.json b/selector-service/src/main/resources/configs/editor/folders/folderArea.json
index d400f446..9256dbfe 100644
--- a/selector-service/src/main/resources/configs/editor/folders/folderArea.json
+++ b/selector-service/src/main/resources/configs/editor/folders/folderArea.json
@@ -1,5 +1,5 @@
 {
-  "id" : "diagramArea",//"#diagrams .modal-content",
+  "id" : "diagramArea",
   "createItem" : {
     "id" : "creating-menu"
   },
diff --git a/ui-testing/src/main/java/com/qreal/wmp/uitesting/headerpanel/DiagramStoreService.java b/ui-testing/src/main/java/com/qreal/wmp/uitesting/headerpanel/DiagramStoreService.java
index ff67514c..c06d97e5 100644
--- a/ui-testing/src/main/java/com/qreal/wmp/uitesting/headerpanel/DiagramStoreService.java
+++ b/ui-testing/src/main/java/com/qreal/wmp/uitesting/headerpanel/DiagramStoreService.java
@@ -41,13 +41,11 @@ public DiagramStoreService(EditorPageFacade facade, SelectorService selectorServ
     
     /** Saves diagram. */
     public void saveDiagram(String key) {
-        SelenideElement element =
-                $(By.id(selectorService.getId("savingMenu.savingInput")));
+        SelenideElement element = $(By.id(selectorService.getId("savingMenu.savingInput")));
         element.setValue(getFilename(key));
         diagrams.put(key, prepareElement(Jsoup.parseBodyFragment($(sceneSelector).innerHtml()).body()));
         $(By.id(selectorService.getId("savingMenu.savingItem"))).click();
-        $(By.id(selectorService.getId("savingMenu.savingItem")))
-                .shouldBe(Condition.disappear);
+        $(By.id(selectorService.getId("savingMenu.savingItem"))).waitUntil(Condition.disappear, 5000);
         lastKnownKey = key;
     }
     
diff --git a/ui-testing/src/main/java/com/qreal/wmp/uitesting/headerpanel/EditorHeaderPanelImpl.java b/ui-testing/src/main/java/com/qreal/wmp/uitesting/headerpanel/EditorHeaderPanelImpl.java
index b6f86b12..29fd3115 100644
--- a/ui-testing/src/main/java/com/qreal/wmp/uitesting/headerpanel/EditorHeaderPanelImpl.java
+++ b/ui-testing/src/main/java/com/qreal/wmp/uitesting/headerpanel/EditorHeaderPanelImpl.java
@@ -1,6 +1,6 @@
 package com.qreal.wmp.uitesting.headerpanel;
 
-import com.codeborne.selenide.Condition;
+import com.google.common.base.Predicate;
 import com.qreal.wmp.uitesting.PageFactory;
 import com.qreal.wmp.uitesting.headerpanel.folderwindow.FileItem;
 import com.qreal.wmp.uitesting.headerpanel.folderwindow.FolderArea;
@@ -9,6 +9,8 @@
 import com.qreal.wmp.uitesting.services.SelectorService;
 import org.openqa.selenium.By;
 import org.openqa.selenium.WebDriver;
+import org.openqa.selenium.interactions.Actions;
+import org.openqa.selenium.support.ui.WebDriverWait;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -31,6 +33,8 @@ public class EditorHeaderPanelImpl implements EditorHeaderPanel {
     
     private final SelectorService selectorService;
     
+    private final WebDriver webDriver;
+    
     private EditorHeaderPanelImpl(
             PageFactory pageFactory,
             WebDriver webDriver,
@@ -42,6 +46,7 @@ private EditorHeaderPanelImpl(
         this.pageFactory = pageFactory;
         this.editorPageFacade = editorPageFacade;
         this.selectorService = selectorService;
+        this.webDriver = webDriver;
     }
 
     @Override
@@ -103,8 +108,15 @@ public static EditorHeaderPanel getEditorHeaderPanel(PageFactory pageFactory,
     }
     
     private FileItem clickFile() {
-        $(By.id(selectorService.getId("fileItem"))).waitUntil(Condition.visible, 5000);
-        $(By.id(selectorService.getId("fileItem"))).click();
+        try {
+            Thread.sleep(2000);
+        } catch (InterruptedException e) {
+            logger.error(e.getMessage());
+        }
+        new Actions(webDriver).click($(By.id(selectorService.getId("fileItem")))).perform();
+        (new WebDriverWait(webDriver, 10))
+                .until((Predicate<WebDriver>) webDriver ->
+                        $(By.id(selectorService.getId("fileItem"))).attr("class").contains("open"));
         return fileItem;
     }
     
diff --git a/ui-testing/src/main/java/com/qreal/wmp/uitesting/headerpanel/folderwindow/FileItem.java b/ui-testing/src/main/java/com/qreal/wmp/uitesting/headerpanel/folderwindow/FileItem.java
index c1ea0a81..13336363 100644
--- a/ui-testing/src/main/java/com/qreal/wmp/uitesting/headerpanel/folderwindow/FileItem.java
+++ b/ui-testing/src/main/java/com/qreal/wmp/uitesting/headerpanel/folderwindow/FileItem.java
@@ -1,15 +1,18 @@
 package com.qreal.wmp.uitesting.headerpanel.folderwindow;
 
-import com.codeborne.selenide.Condition;
 import com.qreal.wmp.uitesting.services.SelectorService;
 import org.openqa.selenium.By;
 import org.openqa.selenium.WebDriver;
+import org.openqa.selenium.interactions.Actions;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 
 import static com.codeborne.selenide.Selenide.$;
 
 /** Describes file item on the header menu. */
 public class FileItem {
-
+    private static final Logger logger = LoggerFactory.getLogger(FileItem.class);
+    
     private final FolderArea folderArea;
     
     private final WebDriver driver;
@@ -24,28 +27,49 @@ public FileItem(WebDriver driver, SelectorService selectorService) {
 
     /** Corresponds 'New' button. */
     public void newDiagram() {
-        $(By.id(selectorService.getId("fileItem.newDiagramItem"))).click();
-        $(By.id(selectorService.getId("saveDiagramConfirmWindow")))
-                .waitUntil(Condition.visible, 10000);
+        new Actions(driver).click($(By.id(selectorService.getId("fileItem.newDiagramItem")))).perform();
+        try {
+            Thread.sleep(3000);
+        } catch (InterruptedException e) {
+            logger.error(e.getMessage());
+        }
         SaveDiagramConfirm.getSaveDiagramConfirm(driver, selectorService.create("saveDiagramConfirmWindow")).notSave();
+        try {
+            Thread.sleep(3000);
+        } catch (InterruptedException e) {
+            logger.error(e.getMessage());
+        }
     }
     
     /** Returns folder window by clicking 'SaveAs'. */
     public FolderArea getSaveItem() {
-        $(By.id(selectorService.getId("fileItem.saveAsItem"))).click();
-        $(By.id(selectorService.getId("folderArea"))).waitUntil(Condition.visible, 10000);
+        new Actions(driver).click($(By.id(selectorService.getId("fileItem.saveAsItem")))).perform();
+        try {
+            Thread.sleep(3000);
+        } catch (InterruptedException e) {
+            logger.error(e.getMessage());
+        }
         return folderArea;
     }
     
     /** Returns folder window by clicking 'Open'. */
     public FolderArea getOpenItem() {
-        $(By.id(selectorService.getId("fileItem.OpenItem"))).click();
-        $(By.id(selectorService.getId("folderArea"))).waitUntil(Condition.visible, 10000);
+        new Actions(driver).click($(By.id(selectorService.getId("fileItem.openItem")))).perform();
+        try {
+            Thread.sleep(3000);
+        } catch (InterruptedException e) {
+            logger.error(e.getMessage());
+        }
         return folderArea;
     }
     
     /** Corresponds 'Save' button. */
     public void saveDiagram() {
-        $(By.id(selectorService.getId("fileItem.SaveItem"))).click();
+        new Actions(driver).click($(By.id(selectorService.getId("fileItem.saveItem")))).perform();
+        try {
+            Thread.sleep(3000);
+        } catch (InterruptedException e) {
+            logger.error(e.getMessage());
+        }
     }
 }
diff --git a/ui-testing/src/main/java/com/qreal/wmp/uitesting/headerpanel/folderwindow/FolderAreaImpl.java b/ui-testing/src/main/java/com/qreal/wmp/uitesting/headerpanel/folderwindow/FolderAreaImpl.java
index 60042bd5..9ff38630 100644
--- a/ui-testing/src/main/java/com/qreal/wmp/uitesting/headerpanel/folderwindow/FolderAreaImpl.java
+++ b/ui-testing/src/main/java/com/qreal/wmp/uitesting/headerpanel/folderwindow/FolderAreaImpl.java
@@ -6,6 +6,7 @@
 import com.qreal.wmp.uitesting.services.SelectorService;
 import org.openqa.selenium.By;
 import org.openqa.selenium.WebDriver;
+import org.openqa.selenium.interactions.Actions;
 import org.openqa.selenium.support.ui.WebDriverWait;
 
 import java.util.Arrays;
@@ -30,9 +31,14 @@ public FolderAreaImpl(WebDriver driver, SelectorService selectorService) {
     
     @Override
     public FolderArea createFolder(String folderName) {
-        $(By.id(selectorService.getId("createItem"))).click();
-        $(By.id(selectorService.getId("folderMenu.folderNameInput"))).setValue(folderName);
-        $(By.id(selectorService.getId("folderMenu.confirmItem"))).click();
+        ($(By.id(selectorService.getId("createItem")))).click();
+        (new WebDriverWait(driver, 10))
+                .until((Predicate<WebDriver>) webDriver ->
+                        $(By.id(selectorService.getId("folderMenu.folderNameInput"))).exists());
+        
+        new Actions(driver).sendKeys($(By.id(selectorService.getId("folderMenu.folderNameInput"))), folderName)
+                .click($(By.id(selectorService.getId("folderMenu.confirmItem")))).build().perform();
+        
         if ($(By.id(selectorService.getId("warningMessage"))).isDisplayed()) {
             throw new IllegalArgumentException("The folder with this name already exists");
         }
@@ -52,7 +58,9 @@ public FolderArea moveForward(String name) {
         }
         String oldPath = getCurrentPath();
         oldPath = "".equals(oldPath) ? name : oldPath + "/" + name;
-        $(By.cssSelector(selectorService.getSelector("folders"))).find(byText(name)).click();
+        new Actions(driver)
+                .click($(By.cssSelector(selectorService.getSelector("folders"))).find(byText(name)))
+                .perform();
         waitUntilEquals(oldPath, FolderArea::getCurrentPath);
         return this;
     }
@@ -69,6 +77,10 @@ public FolderArea moveBack() {
     
     @Override
     public String getCurrentPath() {
+        (new WebDriverWait(driver, 10))
+                .until((Predicate<WebDriver>) webDriver ->
+                        $(By.id(selectorService.getId("folderPath"))).exists());
+    
         String result = $(By.id(selectorService.getId("folderPath")))
                 .find(By.tagName("p"))
                 .getText();
@@ -96,10 +108,13 @@ public FolderArea deleteFolder(String name) {
         if (!isFolderExist(name)) {
             throw new IllegalArgumentException("Folder is not exist");
         }
-        $(By.cssSelector(selectorService.getSelector("folders")))
-                .find(byText(name)).contextClick();
-        $(By.id(selectorService.getId("contextMenu"))).shouldBe(Condition.visible);
-        $(By.id(selectorService.getId("contextMenu.deleteItem"))).click();
+        new Actions(driver)
+                .contextClick($(By.cssSelector(selectorService.getSelector("folders"))).find(byText(name)))
+                .perform();
+        $(By.id(selectorService.getId("contextMenu"))).shouldBe(Condition.appear);
+        new Actions(driver)
+                .click($(By.id(selectorService.getId("contextMenu.deleteItem"))))
+                .perform();
         return this;
     }
     
@@ -107,7 +122,7 @@ public FolderArea deleteFolder(String name) {
     public void close() {
         SelenideElement closeButton = $(By.id(selectorService.getId("closeItem")));
         if ($(By.id(selectorService.getId())).isDisplayed() && closeButton.isDisplayed()) {
-            closeButton.click();
+            new Actions(driver).moveToElement(closeButton).click().perform();
             $(By.id(selectorService.getId())).shouldBe(Condition.disappear);
         }
     }
diff --git a/ui-testing/src/main/java/com/qreal/wmp/uitesting/headerpanel/folderwindow/SaveDiagramConfirm.java b/ui-testing/src/main/java/com/qreal/wmp/uitesting/headerpanel/folderwindow/SaveDiagramConfirm.java
index 1856405a..30571200 100644
--- a/ui-testing/src/main/java/com/qreal/wmp/uitesting/headerpanel/folderwindow/SaveDiagramConfirm.java
+++ b/ui-testing/src/main/java/com/qreal/wmp/uitesting/headerpanel/folderwindow/SaveDiagramConfirm.java
@@ -1,9 +1,11 @@
 package com.qreal.wmp.uitesting.headerpanel.folderwindow;
 
-import com.codeborne.selenide.Condition;
+import com.google.common.base.Predicate;
 import com.qreal.wmp.uitesting.services.SelectorService;
 import org.openqa.selenium.By;
 import org.openqa.selenium.WebDriver;
+import org.openqa.selenium.interactions.Actions;
+import org.openqa.selenium.support.ui.WebDriverWait;
 
 import static com.codeborne.selenide.Selenide.$;
 
@@ -21,14 +23,21 @@ public SaveDiagramConfirm(WebDriver driver, SelectorService selectorService) {
     
     /** 'No' button. */
     public void notSave() {
-        $(By.id(selectorService.getId("cancel"))).click();
-        $(By.id(selectorService.getId())).shouldBe(Condition.disappear);
+        (new WebDriverWait(driver, 10))
+                .until((Predicate<WebDriver>) webDriver ->
+                        $(By.id(selectorService.getId())).attr("class").contains("in"));
+    
+        new Actions(driver).click($(By.id(selectorService.getId("cancel")))).perform();
     }
     
     /** 'Yes' button. */
     public FolderArea save() {
-        $(By.id(selectorService.getId("confirm"))).click();
-        $(By.id(selectorService.getId())).shouldBe(Condition.disappear);
+        (new WebDriverWait(driver, 10))
+                .until((Predicate<WebDriver>) webDriver ->
+                        $(By.id(selectorService.getId())).attr("class").contains("in"));
+    
+        new Actions(driver).click($(By.id(selectorService.getId("confirm")))).perform();
+    
         return new FolderAreaImpl(driver, selectorService);
     }
     
diff --git a/ui-testing/src/main/java/com/qreal/wmp/uitesting/mousegestures/RobotCalibration.java b/ui-testing/src/main/java/com/qreal/wmp/uitesting/mousegestures/RobotCalibration.java
index 0a2814ab..d77ede85 100644
--- a/ui-testing/src/main/java/com/qreal/wmp/uitesting/mousegestures/RobotCalibration.java
+++ b/ui-testing/src/main/java/com/qreal/wmp/uitesting/mousegestures/RobotCalibration.java
@@ -100,6 +100,11 @@ private RobotCalibration(WebDriver driver) {
     }
     
     private Point calibrate() {
+        try {
+            Thread.sleep(1000);
+        } catch (InterruptedException e) {
+            logger.error(e.getMessage());
+        }
         click(xPoint.begin, yPoint.begin);
         // find left border
         while (xPoint.begin < xPoint.end) {
diff --git a/ui-testing/src/test/java/com/qreal/wmp/uitesting/innertests/SaveOpenDiagramTest.java b/ui-testing/src/test/java/com/qreal/wmp/uitesting/innertests/SaveOpenDiagramTest.java
index 2bc5db5a..27bcf9a6 100644
--- a/ui-testing/src/test/java/com/qreal/wmp/uitesting/innertests/SaveOpenDiagramTest.java
+++ b/ui-testing/src/test/java/com/qreal/wmp/uitesting/innertests/SaveOpenDiagramTest.java
@@ -87,8 +87,6 @@ public void equalsFalseTest() {
     
     @Test
     public void saveAfterChangesTest() {
-        scene.clean();
-        addElements();
         scene.moveToCell(elements.get(0), 10, 10);
         headerPanel.saveDiagram();
         assert headerPanel.equalsDiagrams(diagram);