1 files changed, 7 insertions, 2 deletions
diff --git a/tests/UI/specs/SegmentSelectorEditor_spec.js b/tests/UI/specs/SegmentSelectorEditor_spec.js
index 16e42a6017..b260f3abe6 100644
--- a/tests/UI/specs/SegmentSelectorEditor_spec.js
+++ b/tests/UI/specs/SegmentSelectorEditor_spec.js
@@ -141,8 +141,8 @@ describe("SegmentSelectorEditorTest", function () {
         }, done);
     });
 
-    it("should correctly update the segment when its details are changed and saved", function (done) {
-        expect.screenshot("updated").to.be.captureSelector(selectorsToCapture, function (page) {
+    it("should correctly should show a confirmation when changing segment definition", function (done) {
+        expect.screenshot("update_confirmation").to.be.captureSelector('.ui-dialog', function (page) {
             page.click('.segmentEditorPanel .editSegmentName');
             page.evaluate(function () {
                 $('input.edit_segment_name').val('');
@@ -163,7 +163,12 @@ describe("SegmentSelectorEditorTest", function () {
             page.evaluate(function () {
                 $('button.saveAndApply').click();
             });
+        }, done);
+    });
 
+    it("should correctly update the segment when saving confirmed", function (done) {
+        expect.screenshot("updated").to.be.captureSelector(selectorsToCapture, function (page) {
+            page.click('.ui-dialog button:contains(Yes)');
             page.click('.segmentationContainer');
         }, done);
     });