From 892972229c3952b5ba9a77bef2ce2e0cece5677f Mon Sep 17 00:00:00 2001
From: adityacp
Date: Tue, 15 Dec 2020 22:22:13 +0530
Subject: Fix tests and quiz template

---
 yaksh/live_server_tests/selenium_test.py | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

(limited to 'yaksh/live_server_tests')

diff --git a/yaksh/live_server_tests/selenium_test.py b/yaksh/live_server_tests/selenium_test.py
index 6d9117d..0c9cfcc 100644
--- a/yaksh/live_server_tests/selenium_test.py
+++ b/yaksh/live_server_tests/selenium_test.py
@@ -126,9 +126,16 @@ class SeleniumTest():
 
     def open_quiz(self):
         # open module link
-        self.driver.find_elements_by_partial_link_text(
-            'START')[0].click()
-        self.driver.find_element_by_link_text('START').click()
+        try:
+            self.driver.find_elements_by_partial_link_text(
+                'Start')[0].click()
+        except IndexError:
+            self.driver.find_elements_by_partial_link_text(
+                'Continue')[0].click()
+        try:
+            self.driver.find_element_by_link_text('Start').click()
+        except Exception:
+            self.driver.find_element_by_link_text('Continue').click()
         # open quiz link
         self.driver.find_element_by_link_text(self.quiz_name).click()
 
-- 
cgit