9cf2f9f298
This will remove imports from __future__ that are no longer needed. https://docs.python.org/3.5/library/2to3.html#2to3fixer-future
66 lines
1.7 KiB
Python
66 lines
1.7 KiB
Python
"""
|
|
Conditional Pages
|
|
"""
|
|
|
|
|
|
from bok_choy.page_object import PageObject
|
|
from bok_choy.promise import BrokenPromise, EmptyPromise
|
|
|
|
POLL_ANSWER = 'Yes, of course'
|
|
|
|
|
|
class ConditionalPage(PageObject):
|
|
"""
|
|
View of conditional page.
|
|
"""
|
|
|
|
url = None
|
|
|
|
def is_browser_on_page(self):
|
|
"""
|
|
Returns True if the browser is currently on the right page.
|
|
"""
|
|
# This is all a hack to work around the fact that there's no way to adjust the
|
|
# timeout parameters for self.q
|
|
def check_fn():
|
|
return self.q(css='.conditional-wrapper').visible
|
|
try:
|
|
EmptyPromise(
|
|
check_fn,
|
|
"On conditional page",
|
|
).fulfill()
|
|
return True
|
|
except BrokenPromise:
|
|
return False
|
|
|
|
def is_content_visible(self):
|
|
"""
|
|
Returns True if the conditional's content has been revealed,
|
|
False otherwise
|
|
"""
|
|
def check_fn():
|
|
return self.q(css='.hidden-contents').visible
|
|
try:
|
|
EmptyPromise(
|
|
check_fn,
|
|
"Conditional is visible",
|
|
).fulfill()
|
|
return True
|
|
except BrokenPromise:
|
|
return False
|
|
|
|
def fill_in_poll(self):
|
|
"""
|
|
Fills in a poll on the same page as the conditional
|
|
with the answer that matches POLL_ANSWER
|
|
"""
|
|
text_selector = '.poll_answer .text'
|
|
|
|
text_options = self.q(css=text_selector).text
|
|
|
|
# Out of the possible poll answers, we want
|
|
# to select the one that matches POLL_ANSWER and click it.
|
|
for idx, text in enumerate(text_options):
|
|
if text == POLL_ANSWER:
|
|
self.q(css=text_selector).nth(idx).click()
|