test: add RFP and Contract Compliance team selection, Add new smoke test cases for V4

Author: Avijit-Microsoft

tests/e2e-test/pages/HomePage.py

@@ -22,6 +22,8 @@ class BIABPage(BasePage):
     RETAIL_CUSTOMER_SUCCESS_SELECTED = "//span[.='Retail Customer Success Team']"
     PRODUCT_MARKETING = "//div[normalize-space()='Product Marketing Team']"
     HR_TEAM = "//div[normalize-space()='Human Resources Team']"
+    RFP_TEAM = "//div[normalize-space()='RFP Team']"
+    CONTRACT_COMPLIANCE_TEAM = "//div[normalize-space()='Contract Compliance Review Team']"
     CONTINUE_BTN = "//button[normalize-space()='Continue']"
     CREATING_PLAN = "//span[normalize-space()='Creating a plan']"
     CUSTOMER_DATA_AGENT = "//span[normalize-space()='Customer Data Agent']"
@@ -503,6 +505,22 @@ def input_clarification_and_send(self, clarification_text):
         self.page.locator(self.PROCESSING_PLAN).wait_for(state="hidden", timeout=200000)
         logger.info("✓ Plan processing completed")
 
+    def input_rai_clarification_and_send(self, clarification_text):
+        """Input RAI clarification text and click send button (for RAI testing)."""
+        logger.info("Starting RAI clarification input process...")
+        
+        logger.info(f"Typing RAI clarification: {clarification_text}")
+        self.page.locator(self.INPUT_CLARIFICATION).fill(clarification_text)
+        self.page.wait_for_timeout(1000)
+        logger.info("✓ RAI clarification text entered")
+        
+        logger.info("Clicking Send button for RAI clarification...")
+        self.page.locator(self.SEND_BUTTON_CLARIFICATION).click()
+        self.page.wait_for_timeout(2000)
+        logger.info("✓ RAI clarification send button clicked")
+        
+        logger.info("RAI clarification input and send completed successfully!")
+
     def validate_source_text_not_visible(self):
         """Validate that the source text element is not visible."""
         logger.info("Validating that source text is not visible...")
@@ -542,8 +560,95 @@ def click_cancel_button(self):
         self.page.wait_for_timeout(2000)
         logger.info("✓ 'Cancel' button clicked")
 
-    
-
-    
+    def select_rfp_team(self):
+        """Select RFP team and continue."""
+        logger.info("Starting RFP team selection process...")
+        
+        logger.info("Clicking on 'Current Team' button...")
+        self.page.locator(self.CURRENT_TEAM).click()
+        self.page.wait_for_timeout(2000)
+        logger.info("✓ 'Current Team' button clicked")
+        
+        logger.info("Selecting 'RFP Team' radio button...")
+        self.page.locator(self.RFP_TEAM).click()
+        self.page.wait_for_timeout(1000)
+        logger.info("✓ 'RFP Team' radio button selected")
+        
+        logger.info("Clicking 'Continue' button...")
+        self.page.locator(self.CONTINUE_BTN).click()
+        self.page.wait_for_timeout(2000)
+        logger.info("✓ 'Continue' button clicked")
+        
+        logger.info("RFP team selection completed successfully!")
 
-    
+    def select_contract_compliance_team(self):
+        """Select Contract Compliance Review team and continue."""
+        logger.info("Starting Contract Compliance team selection process...")
+        
+        logger.info("Clicking on 'Current Team' button...")
+        self.page.locator(self.CURRENT_TEAM).click()
+        self.page.wait_for_timeout(2000)
+        logger.info("✓ 'Current Team' button clicked")
+        
+        logger.info("Selecting 'Contract Compliance Review Team' radio button...")
+        self.page.locator(self.CONTRACT_COMPLIANCE_TEAM).click()
+        self.page.wait_for_timeout(1000)
+        logger.info("✓ 'Contract Compliance Review Team' radio button selected")
+        
+        logger.info("Clicking 'Continue' button...")
+        self.page.locator(self.CONTINUE_BTN).click()
+        self.page.wait_for_timeout(2000)
+        logger.info("✓ 'Continue' button clicked")
+        
+        logger.info("Contract Compliance team selection completed successfully!")
+
+    def validate_home_input_visible(self):
+        """Validate that user is redirected to home screen with input visible."""
+        logger.info("Validating home input is visible...")
+        expect(self.page.locator(self.HOME_INPUT_TITLE_WRAPPER)).to_be_visible(timeout=10000)
+        logger.info("✓ Home input is visible - user redirected to home screen")
+
+    def validate_send_button_disabled(self):
+        """Validate that send button is disabled for empty/space inputs."""
+        logger.info("Validating send button is disabled...")
+        send_button = self.page.locator(self.SEND_BUTTON)
+        is_disabled = send_button.is_disabled()
+        if is_disabled:
+            logger.info("✓ Send button is disabled as expected")
+        else:
+            logger.warning("⚠ Send button is enabled but should be disabled")
+            # Check if clicking does nothing
+            send_button.click()
+            self.page.wait_for_timeout(2000)
+            # Verify no plan creation started
+            try:
+                self.page.locator(self.CREATING_PLAN).wait_for(state="visible", timeout=3000)
+                logger.error("❌ Plan creation started unexpectedly")
+                raise AssertionError("System accepted empty/space query - test failed")
+            except Exception as e:
+                if "Timeout" in str(e) or "timeout" in str(e):
+                    logger.info("✓ No plan creation started - system correctly rejected query")
+                else:
+                    raise
+
+    def input_text_only(self, text):
+        """Input text without sending."""
+        logger.info(f"Typing text: {text}")
+        self.page.locator(self.PROMPT_INPUT).fill(text)
+        self.page.wait_for_timeout(1000)
+        logger.info("✓ Text entered")
+
+    def get_team_list_count(self, team_name):
+        """Get the count of a specific team in the team selection list."""
+        logger.info(f"Counting '{team_name}' entries in team list...")
+        team_locator = f"//div[normalize-space()='{team_name}']"
+        count = self.page.locator(team_locator).count()
+        logger.info(f"✓ Found {count} entries for '{team_name}'")
+        return count
+
+    def open_team_selection(self):
+        """Open the team selection dropdown."""
+        logger.info("Opening team selection dropdown...")
+        self.page.locator(self.CURRENT_TEAM).click()
+        self.page.wait_for_timeout(2000)
+        logger.info("✓ Team selection dropdown opened")