From b2e08ddc72f6b46b46dfd86a7d4d4b15d8e221e6 Mon Sep 17 00:00:00 2001 From: Muhammad Ammar Date: Mon, 14 Jul 2014 11:18:51 +0000 Subject: [PATCH] Add promise to ensure unit visibility is changed --- common/test/acceptance/pages/studio/unit.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/common/test/acceptance/pages/studio/unit.py b/common/test/acceptance/pages/studio/unit.py index 998cd9b990..1639f8ca54 100644 --- a/common/test/acceptance/pages/studio/unit.py +++ b/common/test/acceptance/pages/studio/unit.py @@ -73,6 +73,14 @@ class UnitPage(PageObject): self.q(css='select[name="visibility-select"] option[value="{}"]'.format(visibility)).first.click() self.wait_for_ajax() + selector = '.edit-button' + if visibility == 'private': + check_func = lambda: self.q(css=selector).visible + elif visibility == 'public': + check_func = lambda: not self.q(css=selector).visible + + EmptyPromise(check_func, 'Unit Visibility is {}'.format(visibility)).fulfill() + COMPONENT_BUTTONS = { 'advanced_tab': '.editor-tabs li.inner_tab_wrap:nth-child(2) > a',