Heri | #1012 | Fix FT, debugging Qre not preselected

mangroveorg · Jan 23, 2019 · 26a32a0 · 26a32a0
1 parent 0e89519
commit 26a32a0
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/func_tests/pages/edituserpage/edit_user_page.py b/func_tests/pages/edituserpage/edit_user_page.py
@@ -63,10 +63,14 @@ def is_role_project_manager(self):
         project_manager_checked = self.driver.find(by_css("input[id=option_project_manager]")).get_attribute('checked')
         return True if project_manager_checked is not None else False
 
-    def are_questionnaires_preselected(self, questionnaires):
+    def are_questionnaires_preselected(self, questionnaires, debug=False):
         elements = self.driver.find_elements_by_css_selector('.questionnaire-list ul li input:checked + span')
         preselected_questionnaires = [element.text for element in elements]
-        return True if set(questionnaires) == set(preselected_questionnaires) else False
+        if set(questionnaires) == set(preselected_questionnaires):
+            return True
+        if debug:
+            self.driver.create_screenshot("debug-ft-qre-not-preselected")
+        return False
 
     def _uncheck_all_questionnaires(self):
         selected_questionnaires = self.driver.find_elements_by_css_selector('.questionnaire-list ul li input:checked')

diff --git a/func_tests/tests/editusertests/edit_user_tests.py b/func_tests/tests/editusertests/edit_user_tests.py
@@ -68,7 +68,7 @@ def test_should_edit_a_project_manager_as_ngo_admin(self):
         self.create_user_and_open_it_on_edit_page(False, [2,4])
         self.assertTrue(self.edit_user_page.is_user_name_is_prefetched(self.username))
         self.assertTrue(self.edit_user_page.is_role_project_manager())
-        self.assertTrue(self.edit_user_page.are_questionnaires_preselected(self.user_questionnaire_list))
+        self.assertTrue(self.edit_user_page.are_questionnaires_preselected(self.user_questionnaire_list, True))
         selected_questionnaires = self.edit_user_page.select_questionnaires(3)
         self.edit_user_page.save_changes({
             "mobile_phone": random_number(9),