included more test cases

Author: Priyanka-Microsoft

tests/e2e-test/pages/draftPage.py

@@ -541,4 +541,83 @@ def test_character_limit_restriction(self, section_index=0):
             logger.error(f"❌ Failed to verify character count label: {e}")
             raise
 
-        return actual_length
+        return actual_length
+
+    def enter_document_title(self, title):
+        """
+        Enter a title in the document title text box on the Draft page
+        
+        Args:
+            title: The title to enter in the document title field
+        """
+        try:
+            logger.info(f"🔹 Entering document title: '{title}'")
+            
+            # Primary locator: by placeholder text
+            title_input = self.page.locator("input[placeholder='Enter title here']")
+            
+            # Check if input field is visible
+            if not title_input.is_visible(timeout=5000):
+                # Try alternative locator: by class name
+                logger.warning("⚠️ Title input not found by placeholder, trying by class")
+                title_input = self.page.locator("input.ms-TextField-field")
+            
+            # Wait for the input to be editable
+            title_input.wait_for(state="visible", timeout=5000)
+            
+            # Scroll to the input field if needed
+            title_input.scroll_into_view_if_needed()
+            self.page.wait_for_timeout(500)
+            
+            # Clear any existing title
+            title_input.clear()
+            self.page.wait_for_timeout(300)
+            
+            # Enter new title
+            title_input.fill(title)
+            self.page.wait_for_timeout(500)
+            
+            # Verify the title was entered correctly
+            entered_value = title_input.input_value()
+            if entered_value == title:
+                logger.info(f"✅ Successfully entered document title: '{title}'")
+            else:
+                logger.warning(f"⚠️ Title mismatch - Expected: '{title}', Got: '{entered_value}'")
+            
+        except Exception as e:
+            logger.error(f"❌ Failed to enter document title: {e}")
+            # Take screenshot for debugging
+            try:
+                screenshot_path = "screenshots/title_input_error.png"
+                os.makedirs("screenshots", exist_ok=True)
+                self.page.screenshot(path=screenshot_path)
+                logger.error(f"📸 Screenshot saved: {screenshot_path}")
+            except:
+                pass
+            raise
+
+    def click_export_document_button(self):
+        """
+        Click the 'Export Document' button at the bottom of the Draft page
+        """
+        try:
+            # Locate the Export Document button using the class and text
+            # Button structure: <button class="ms-Button ms-Button--commandBar _exportDocumentIcon_1x53n_11 root-239" aria-label="export document">
+            export_button = self.page.locator("button[aria-label='export document']")
+            
+            if not export_button.is_visible(timeout=5000):
+                # Try alternative locator by text
+                export_button = self.page.locator("button:has-text('Export Document')")
+            
+            # Scroll to button if needed
+            export_button.scroll_into_view_if_needed()
+            self.page.wait_for_timeout(500)
+            
+            # Click the button
+            export_button.click()
+            
+            logger.info("✅ Clicked 'Export Document' button")
+            
+        except Exception as e:
+            logger.error(f"❌ Failed to click Export Document button: {e}")
+            raise

tests/e2e-test/pages/generatePage.py

@@ -73,6 +73,12 @@ def click_generate_draft_button(self):
         # Type a question in the text area
         self.page.locator(self.GENERATE_DRAFT).click()
         self.page.wait_for_timeout(15000)
+    
+    def click_browse_button(self):
+        # click on BROWSE
+        self.page.wait_for_timeout(3000)
+        self.page.locator(self.BROWSE_BUTTON).click()
+        self.page.wait_for_timeout(5000)
 
     def show_chat_history(self):
         """Click to show chat history if the button is visible."""
@@ -120,13 +126,20 @@ def delete_chat_history(self):
             self.page.wait_for_timeout(2000)
 
     def validate_draft_button_enabled(self):
+        """
+        Check if Generate Draft button is enabled.
+        Returns True if enabled, False if disabled.
+        """
         self.page.wait_for_timeout(5000)
         generate_draft_button = self.page.locator(self.GENERATE_DRAFT)
-        with check:
-            if not generate_draft_button.is_enabled():
-                logger.error("❌ 'Generate Draft' button is disabled.")
-            else:
-                logger.info("✅ 'Generate Draft' button is enabled.")
+        is_enabled = generate_draft_button.is_enabled()
+        
+        if not is_enabled:
+            logger.info("✅ 'Generate Draft' button is disabled (as expected on launch).")
+        else:
+            logger.info("✅ 'Generate Draft' button is enabled.")
+        
+        return is_enabled
 
     def select_history_thread(self, thread_index=0):
         """Select a history thread from the template history panel."""
@@ -157,29 +170,27 @@ def click_new_chat_button(self):
         logger.info("New Chat button clicked successfully")
 
     def verify_saved_chat(self, expected_text: str):
-        """Verify that the saved chat contains specific expected text."""
-        
-        # Locator for all chat messages (user + GPT)
-        chat_messages = self.page.locator(
-            '._chatMessageUserMessage_1dc7g_87, ._answerText_1qm4u_14'
-        )
 
-        # Fail if there are no messages at all
+        chat_messages_locator = '._chatMessageUserMessage_1dc7g_87, ._answerText_1qm4u_14'
+
+        # Wait for first message to load (up to 5 seconds)
+        try:
+            self.page.locator(chat_messages_locator).first.wait_for(timeout=5000)
+        except:
+            assert False, "Chat messages did not load within timeout."
+
+        chat_messages = self.page.locator(chat_messages_locator)
+
         assert chat_messages.count() > 0, "No chat messages found — saved chat did not load."
 
-        # Check if expected text exists in any message
-        found = False
-        count = chat_messages.count()
-        
-        for i in range(count):
-            message_text = chat_messages.nth(i).inner_text()
-            if expected_text in message_text:
-                found = True
-                break
+        # Search messages for expected text
+        found = any(
+            expected_text in chat_messages.nth(i).inner_text()
+            for i in range(chat_messages.count())
+        )
 
         assert found, f"Expected text '{expected_text}' not found in saved chat."
-        logger.info(f"Verified saved chat contains expected text: {expected_text}")
-                
+            
     def delete_thread_by_index(self, thread_index: int = 0):
         """
         Delete a session thread based on its index and verify it is removed.