Fix test_lms.PayAndVerifyTest.test_enrollment_upgrade bokchoy test.

This commit is contained in:
Alex Dusenbery
2019-05-01 11:48:47 -04:00
committed by Alex Dusenbery
parent 5db761ed5a
commit 4768edabca

View File

@@ -155,8 +155,11 @@ class FakePaymentPage(PageObject):
def is_browser_on_page(self):
"""Check if a step in the payment and verification flow has loaded."""
message = self.q(css='BODY').text[0]
match = re.search('Payment page', message)
message = self.q(css='BODY').text
if not message:
return False
match = re.search('Payment page', message[0])
return True if match else False
def submit_payment(self):
@@ -176,8 +179,11 @@ class FakeSoftwareSecureVerificationPage(PageObject):
def is_browser_on_page(self):
""" Determine if browser is on the page. """
message = self.q(css='BODY').text[0]
match = re.search('Fake Software Secure page', message)
message = self.q(css='BODY').text
if not message:
return False
match = re.search('Fake Software Secure page', message[0])
return True if match else False
def mark_approved(self):