codemie-test-harness 0.1.206__py3-none-any.whl → 0.1.207__py3-none-any.whl

codemie_test_harness/tests/ui/pageobject/assistants/{create_assistant_page.py → create_edit_assistant_page.py}

@@ -7,11 +7,15 @@ from codemie_test_harness.tests.ui.pageobject.assistants.generate_with_ai_modal
     AIAssistantGeneratorPage,
 )
 from codemie_test_harness.tests.ui.pageobject.base_page import BasePage
+from tests.ui.pageobject.assistants.assistant_mcp_server import (
+    AssistantMCPIntegrationModal,
+)
+from tests.ui.pageobject.assistants.assistant_sidebar import AssistantSidebar
 
 
-class CreateAssistantPage(BasePage):
+class CreateEditAssistantPage(BasePage):
     """
-    Create Assistant page object following Page Object Model (POM) best practices.
+    Create/Edit Assistant page object following Page Object Model (POM) best practices.
 
     This class encapsulates all interactions with the Create Assistant page,
     providing a clean interface for test automation while hiding implementation details.
@@ -23,6 +27,8 @@ class CreateAssistantPage(BasePage):
     def __init__(self, page):
         """Initialize the Create Assistant page object."""
         super().__init__(page)
+        self.sidebar = AssistantSidebar(page)
+        self.mcp = AssistantMCPIntegrationModal(page)
         self.ai_generator_modal = AIAssistantGeneratorPage(
             page
         )  # AI Assistant Generator modal
@@ -39,34 +45,37 @@ class CreateAssistantPage(BasePage):
     @property
     def page_title(self) -> Locator:
         """Page title 'Create Assistant' element"""
-        return (
-            self.page.locator('.text-h3:has-text("Create Assistant")')
-            or self.page.locator(
-                'div.text-h3.text-white.font-semibold:has-text("Create Assistant")'
-            )
-            or self.page.locator('h1:has-text("Create Assistant")')
+        return self.page.locator(
+            'h1.text-h3.text-text-main.font-semibold:has-text("Create Assistant")'
         )
 
     @property
     def generate_with_ai_button(self) -> Locator:
         """Generate with AI button in header with magical styling"""
-        return (
-            self.page.locator(
-                'button.button.magical.medium:has-text("Generate with AI")'
-            )
-            or self.page.locator('button:has-text("Generate with AI")')
-            or self.page.locator("button:has(svg + text)").filter(
-                has_text="Generate with AI"
-            )
-        )
+        return self.page.locator(
+            'button.bg-magical-button:has-text("Generate with AI")'
+        ).first
 
     @property
     def create_button(self) -> Locator:
         """Create button with plus icon (primary button)"""
-        return (
-            self.page.locator("button#submit")
-            or self.page.locator('button.button.primary.medium:has-text("Create")')
-            or self.page.locator('button:has-text("Create"):has(svg)')
+        return self.page.locator('button.bg-button-primary-bg:has-text("Create")')
+
+    @property
+    def save_button(self) -> Locator:
+        """Save button"""
+        return self.page.locator('button.bg-button-primary-bg:has-text("Save")')
+
+    @property
+    def cancel_button(self) -> Locator:
+        """Cancel button"""
+        return self.page.locator('button.bg-button-secondary-bg:has-text("Cancel")')
+
+    @property
+    def back_button(self) -> Locator:
+        """Back button"""
+        return self.page.locator(
+            'button.bg-button-secondary-bg:has(svg[viewBox="0 0 18 18"])'
         )
 
     # =============================================================================
@@ -79,82 +88,103 @@ class CreateAssistantPage(BasePage):
         return self.page.locator('h4:has-text("Assistant Setup")')
 
     @property
-    def project_dropdown(self) -> Locator:
-        """Project selection multiselect dropdown"""
-        return (
-            self.page.locator("div.p-multiselect#project")
-            or self.page.locator('[name="project"].p-multiselect')
-            or self.page.locator(".p-multiselect-label-container")
-        )
+    def shared_toggle(self) -> Locator:
+        """'Shared with project' toggle switch - returns the label which is clickable"""
+        return self.page.locator('label:has-text("Shared with project")')
 
     @property
-    def shared_toggle(self) -> Locator:
-        """'Shared with project' toggle switch"""
-        return self.page.locator("label.switch-wrapper span.switch")
+    def shared_toggle_checkbox(self) -> Locator:
+        """'Shared with project' checkbox input (hidden) - use for checking state only"""
+        return self.page.locator(
+            'label:has-text("Shared with project") input[type="checkbox"]'
+        )
 
     @property
     def name_input(self) -> Locator:
         """Assistant name input field with data-testid validation"""
-        return (
-            self.page.locator('input#name[data-testid="validation"]')
-            or self.page.locator('input[placeholder="Name*"]')
-            or self.page.locator('input[name="name"]')
-        )
+        return self.page.locator('input[placeholder="Name*"]')
 
     @property
     def slug_input(self) -> Locator:
         """Assistant slug input field"""
-        return (
-            self.page.locator('input#slug[data-testid="validation"]')
-            or self.page.locator(
-                'input[placeholder="Unique human-readable identifier"]'
-            )
-            or self.page.locator('input[name="slug"]')
-        )
+        return self.page.locator('input[name="slug"]')
 
     @property
     def icon_url_input(self) -> Locator:
         """Assistant icon URL input field"""
-        return (
-            self.page.locator('input#icon_url[data-testid="validation"]')
-            or self.page.locator('input[placeholder="URL to the assistant\'s icon"]')
-            or self.page.locator('input[name="icon_url"]')
-        )
+        return self.page.locator('input[name="icon_url"]')
 
     @property
     def description_textarea(self) -> Locator:
         """Assistant description textarea with placeholder"""
-        return (
-            self.page.locator('textarea#description[name="description"]')
-            or self.page.locator('textarea[placeholder="Description*"]')
-            or self.page.locator(".textarea-wrapper textarea")
+        return self.page.locator(
+            'textarea[name="description"][placeholder="Description*"]'
+        )
+
+    @property
+    def categories_dropdown(self) -> Locator:
+        """Categories selection multiselect dropdown"""
+        return self.page.locator(
+            'div.p-multiselect:has(div.p-multiselect-label:has-text("Select categories"))'
         )
 
     @property
     def conversation_starters_input(self) -> Locator:
         """Conversation starters input field with InputGroup"""
-        return (
-            self.page.locator("input.p-inputtext#conversationStarters-0")
-            or self.page.locator('input[name="conversationStarters"]')
-            or self.page.locator(".p-inputgroup input")
-        )
+        return self.page.locator('input[name="conversation_starters"]')
 
     @property
     def add_conversation_starter_button(self) -> Locator:
         """Add conversation starter button with plus icon"""
-        return (
-            self.page.locator('button.button.secondary.medium:has-text("Add")').nth(0)
-            or self.page.locator('button:has-text("Add"):has(svg)').first
-            or self.page.locator('.flex.justify-between button:has-text("Add")')
+        return self.page.locator(
+            'label:has-text("Conversation starters") + button:has-text("Add")'
         )
 
     @property
     def delete_conversation_starter_button(self) -> Locator:
         """Delete conversation starter button (trash icon in InputGroup)"""
-        return (
-            self.page.locator(".p-inputgroup-addon button:has(svg)")
-            or self.page.locator('button:has(path[d*="M9.5 1.25a3.25"])')
-            or self.page.locator(".p-inputgroup button")
+        return self.page.locator(".p-inputgroup button.bg-button-secondary-bg:has(svg)")
+
+    @property
+    def name_error_message(self):
+        """Name field error message."""
+        return self.page.locator(
+            'label:has(input[name="name"]) div.text-sm.text-error-main.input-error-message'
+        )
+
+    @property
+    def icon_error_message(self):
+        """Icon field error message."""
+        return self.page.locator(
+            'label:has(input[name="icon_url"]) div.text-sm.text-error-main.input-error-message'
+        )
+
+    @property
+    def description_error_message(self):
+        """Description field error message."""
+        return self.page.locator(
+            'div:has(textarea[name="description"][placeholder="Description*"]) div.text-fire-50.text-sm'
+        )
+
+    @property
+    def system_prompt_error_message(self):
+        """System prompt field error message."""
+        return self.page.locator(
+            'div:has(textarea[placeholder="System Instructions*"]) div.text-fire-50.text-sm'
+        )
+
+    @property
+    def temperature_error_message(self):
+        """Temperature field error message."""
+        return self.page.locator(
+            'label:has(input[name="temperature"]) div.text-sm.text-error-main.input-error-message'
+        )
+
+    @property
+    def top_p_error_message(self):
+        """Top P field error message."""
+        return self.page.locator(
+            'label:has(input[name="top_p"]) div.text-sm.text-error-main.input-error-message'
         )
 
     # =============================================================================
@@ -170,56 +200,201 @@ class CreateAssistantPage(BasePage):
     def system_instructions_label(self) -> Locator:
         """System Instructions label"""
         return self.page.locator(
-            '.text-sm.font-semibold:has-text("System Instructions")'
+            'p.text-sm.font-semibold:has-text("System Instructions")'
         )
 
     @property
     def system_prompt_textarea(self) -> Locator:
         """System instructions textarea with full height"""
-        return (
-            self.page.locator('textarea#system_prompt[name="system_prompt"]')
-            or self.page.locator('textarea[placeholder="System Instructions*"]')
-            or self.page.locator(".textarea-wrapper.h-full textarea")
-        )
+        return self.page.locator('textarea[placeholder="System Instructions*"]')
+
+    @property
+    def generate_with_ai_system_instructions_button(self) -> Locator:
+        """Generate with AI button for system instructions"""
+        return self.page.locator(
+            'button.bg-magical-button:has-text("Generate with AI")'
+        ).nth(1)
 
     @property
     def expand_system_prompt_button(self) -> Locator:
         """Expand system prompt button"""
-        return (
-            self.page.locator('button.button.secondary.medium:has-text("Expand")')
-            or self.page.locator('button:has-text("Expand"):has(svg)')
-            or self.page.locator('.flex.gap-4 button:has-text("Expand")')
+        return self.page.locator('button.bg-button-secondary-bg:has-text("Expand")')
+
+    @property
+    def current_user_prompt_var_button(self) -> Locator:
+        """Current User prompt variable button"""
+        return self.page.locator('button:has-text("Current User")')
+
+    @property
+    def date_prompt_var_button(self) -> Locator:
+        """Date prompt variable button"""
+        return self.page.locator('button:has-text("Date")')
+
+    @property
+    def manage_prompt_vars_button(self) -> Locator:
+        """Manage Prompt Vars button"""
+        return self.page.locator(
+            'button.bg-button-primary-bg:has-text("Manage Prompt Vars")'
         )
 
     @property
-    def model_type_dropdown(self) -> Locator:
-        """LLM model type multiselect dropdown"""
-        return (
-            self.page.locator("div.p-multiselect#model_type")
-            or self.page.locator('[name="model_type"].p-multiselect')
-            or self.page.locator(
-                '.p-multiselect:has(.p-multiselect-label:has-text("Default LLM Model"))'
-            )
+    def llm_model_dropdown(self) -> Locator:
+        """LLM model selection dropdown"""
+        return self.page.locator(
+            'div.p-multiselect:has(div.p-multiselect-label:has-text("Default: GPT-4.1"))'
         )
 
     @property
     def temperature_input(self) -> Locator:
         """Temperature input field (0-2 range)"""
-        return (
-            self.page.locator('input#temperature[data-testid="validation"]')
-            or self.page.locator('input[placeholder="0-2"]')
-            or self.page.locator('input[name="temperature"]')
-        )
+        return self.page.locator('input[name="temperature"][placeholder="0-2"]')
 
     @property
     def top_p_input(self) -> Locator:
         """Top P input field (0-1 range)"""
+        return self.page.locator('input[name="top_p"][placeholder="0-1"]')
+
+    # ==================== DataSource Context ====================
+    @property
+    def datasource_context_label(self):
+        """Label 'Datasource Context' above the dropdown."""
+        return self.page.locator(
+            'div.text-xs.text-text-secondary:has-text("Datasource Context")'
+        )
+
+    @property
+    def datasource_context_add_button(self):
+        """'Add' button next to the Datasource Context label."""
+        return self.page.locator(
+            'div:has-text("Datasource Context") + button:has-text("Add")'
+        )
+
+    @property
+    def datasource_context_dropdown(self):
+        """Datasource Context dropdown (multiselect input)."""
+        return self.page.locator("div.p-multiselect#context-selector")
+
+    @property
+    def datasource_context_dropdown_label(self):
+        """The visible label text inside the datasource context dropdown."""
+        return self.datasource_context_dropdown.locator("div.p-multiselect-label")
+
+    # ==================== Sub-Assistant Context ====================
+    @property
+    def sub_assistants_label(self):
+        """Label 'Sub-Assistants' above the sub assistants dropdown."""
+        return self.page.get_by_text("Sub-Assistants", exact=True)
+
+    @property
+    def sub_assistants_dropdown(self):
+        """Sub Assistants dropdown (multiselect input)."""
+        return self.page.locator(
+            'div.p-multiselect:has(div.p-multiselect-label:has-text("Select Sub-Assistants"))'
+        )
+
+    @property
+    def sub_assistants_dropdown_label(self):
+        """The label inside the sub assistants dropdown."""
+        return self.sub_assistants_dropdown.locator("div.p-multiselect-label")
+
+    # ======= Tools Accordion Locators =======
+
+    @property
+    def available_tools_accordion(self) -> Locator:
+        """Available Tools accordion section"""
+        return self.page.locator(
+            'div.p-accordion-header:has(h1.font-bold:has-text("Available Tools"))'
+        )
+
+    @property
+    def external_tools_accordion(self) -> Locator:
+        """External Tools accordion section"""
+        return self.page.locator(
+            'div.p-accordion-header:has(h1.font-bold:has-text("External Tools"))'
+        )
+
+    @step
+    def section(self, name: str) -> Locator:
+        """Get main accordion section (Available Tools or External Tools) by name"""
+        return self.page.locator(
+            f'div.p-accordion-header:has(h1.font-bold.text-text-quaternary:has-text("{name}"))'
+        )
+
+    @step
+    def toolkit(self, section_name: str, toolkit_name: str) -> Locator:
+        """Get toolkit accordion header within a section by name
+        Handles both spaced and non-spaced toolkit names (e.g., 'Open API' vs 'OpenAPI')"""
+        # Try the exact name first, if not found try without spaces
+        toolkit_name_no_space = toolkit_name.replace(" ", "")
+        # Look for h2 with the toolkit name, which is inside the accordion header
+        return self.page.locator(
+            f'div.p-accordion-header:has(h2.font-medium:text-is("{toolkit_name}")),'
+            f'div.p-accordion-header:has(h2.font-medium:text-is("{toolkit_name_no_space}"))'
+        ).first
+
+    @step
+    def tool_rows_under_toolkit(self, toolkit_name: str) -> Locator:
+        """Returns all tool rows (each with a label and a checkbox) under an expanded toolkit panel."""
+        panel = self.page.locator(
+            "div.p-accordion-tab.p-accordion-tab-active div.p-accordion-content"
+        ).filter(has_text=toolkit_name)
+        return panel.locator(
+            "div.grid.grid-cols-[auto,1fr]"
+        )  # Adapt selector to match tool row
+
+    @step
+    def tool_label_spans_under_toolkit(self, tool: str) -> Locator:
+        tool_stripped = tool.strip()
+        if "mcp" in tool_stripped.lower():
+            return (
+                self.page.locator("div.p-accordion-tab-active div.p-accordion-content")
+                .locator('button, [id*="mcp"]')
+                .filter(has_text=tool_stripped)
+                .first
+            )
+        return self.page.locator(
+            "div.p-accordion-tab-active div.p-accordion-content label"
+        ).get_by_text(tool_stripped, exact=True)
+
+    @step
+    def tool_checkbox(self, tool_name: str) -> Locator:
         return (
-            self.page.locator('input#top_p[data-testid="validation"]')
-            or self.page.locator('input[placeholder="0-1"]')
-            or self.page.locator('input[name="top_p"]')
+            self.page.locator("span")
+            .filter(has_text=tool_name)
+            .locator("xpath=..")
+            .locator("input[type=checkbox]")
         )
 
+    @step
+    def tool_row(self, tool_name: str) -> Locator:
+        return (
+            self.page.locator("span")
+            .filter(has_text=tool_name)
+            .locator("xpath=ancestor::div[contains(@class,'grid')]")
+        )
+
+    @step
+    def select_section(self, section_name: str):
+        self.section(section_name).click()
+        return self
+
+    @step
+    def select_toolkit(self, section: str, toolkit: str):
+        self.toolkit(section, toolkit).click()
+        return self
+
+    @step
+    def should_be_visible_tool(self, tool: str):
+        expect(self.tool_label_spans_under_toolkit(tool)).to_be_visible()
+        return self
+
+    @step
+    def select_tool(self, tool: str):
+        self.tool_label_spans_under_toolkit(tool).click()
+        if self.mcp.is_pop_visible():
+            self.mcp.fill_mcp_server_base_form()
+        return self
+
     # ==================== NAVIGATION METHODS ====================
 
     @step
@@ -444,8 +619,8 @@ class CreateAssistantPage(BasePage):
         Returns:
             self: Returns the page object for method chaining
         """
-        # Check current state and toggle if needed
-        is_currently_checked = self.shared_toggle.is_checked()
+        # Check current state using the hidden checkbox and toggle if needed by clicking the label
+        is_currently_checked = self.shared_toggle_checkbox.is_checked()
         if (shared and not is_currently_checked) or (
             not shared and is_currently_checked
         ):
@@ -495,6 +670,17 @@ class CreateAssistantPage(BasePage):
         self.create_button.click()
         return self
 
+    @step
+    def click_save(self):
+        """
+        Click the Create button to create the assistant.
+
+        Returns:
+            self: Returns the page object for method chaining
+        """
+        self.save_button.click()
+        return self
+
     @step
     def click_cancel(self):
         """
@@ -528,6 +714,27 @@ class CreateAssistantPage(BasePage):
         self.generate_with_ai_button.click()
         return self
 
+    def select_tool_option(self, toolkit_name: str, label: str, checked: bool = True):
+        self.expand_toolkit(toolkit_name)
+        checkbox = self.toolkit_checkbox_by_label(toolkit_name, label)
+        if checked != checkbox.is_checked():
+            checkbox.click()
+        return self
+
+    def get_all_tools_in_toolkit(self, toolkit_name: str):
+        """
+        Returns a list of tuples (label, checked) for all tool features inside a toolkit accordion.
+        """
+        panel = self.toolkit_accordion_panel(toolkit_name)
+        labels = panel.locator(".checkbox-label span.text-sm").all_text_contents()
+        checkboxes = panel.locator('input[type="checkbox"]')
+        return [
+            (lbl, checkboxes.nth(idx).is_checked()) for idx, lbl in enumerate(labels)
+        ]
+
+    def toolkit_select_all_checkbox(self, toolkit_name: str):
+        return self.toolkit_checkbox_by_label(toolkit_name, "Select all")
+
     # ==================== COMPREHENSIVE ASSISTANT CREATION METHOD ====================
 
     @step
@@ -594,6 +801,7 @@ class CreateAssistantPage(BasePage):
     def should_have_all_form_fields_visible(self):
         """Verify that all essential form fields are visible."""
         expect(self.name_input).to_be_visible()
+        expect(self.slug_input).to_be_visible()
         expect(self.description_textarea).to_be_visible()
         expect(self.system_prompt_textarea).to_be_visible()
         return self
@@ -623,6 +831,12 @@ class CreateAssistantPage(BasePage):
         expect(self.system_prompt_textarea).to_have_value(expected_prompt)
         return self
 
+    @step
+    def should_have_categories_visible(self):
+        """Verify categories dropdown is visible.."""
+        expect(self.categories_dropdown).to_be_visible()
+        return self
+
     @step
     def should_have_icon_url_value(self, expected_url: str):
         """Verify icon URL field has expected value."""
@@ -632,13 +846,13 @@ class CreateAssistantPage(BasePage):
     @step
     def should_have_shared_checked(self):
         """Verify shared toggle is checked."""
-        expect(self.shared_toggle).to_be_checked()
+        expect(self.shared_toggle_checkbox).to_be_checked()
         return self
 
     @step
     def should_have_shared_unchecked(self):
         """Verify shared toggle is unchecked."""
-        expect(self.shared_toggle).not_to_be_checked()
+        expect(self.shared_toggle_checkbox).not_to_be_checked()
         return self
 
     @step
@@ -667,3 +881,73 @@ class CreateAssistantPage(BasePage):
         expect(self.system_prompt_textarea).to_have_value("")
         expect(self.icon_url_input).to_have_value("")
         return self
+
+    @step
+    def should_have_top_p_and_temperature(self):
+        """Verify top p and temperature fields are visible."""
+        expect(self.temperature_input).to_be_visible()
+        expect(self.top_p_input).to_be_visible()
+        return self
+
+    @step
+    def should_have_top_p_and_temperature_value(self, temperature: str, top_p: str):
+        """Verify top p and temperature field values is visible."""
+        expect(self.temperature_input).to_have_value(temperature)
+        expect(self.top_p_input).to_have_value(top_p)
+
+    @step
+    def should_have_datasource_context(self):
+        """Verify datasource context fields are visible."""
+        expect(self.datasource_context_label).to_be_visible()
+        expect(self.datasource_context_add_button).to_be_visible()
+        expect(self.datasource_context_dropdown).to_be_visible()
+        return self
+
+    @step
+    def should_have_sub_assistants_context(self):
+        """Verify sub assistants context fields are visible."""
+        expect(self.sub_assistants_label).to_be_visible()
+        expect(self.sub_assistants_dropdown).to_be_visible()
+        return self
+
+    @step
+    def should_have_name_error_textarea(self, error_message: str):
+        """Verify name error text field is visible."""
+        expect(self.name_error_message).to_be_visible()
+        expect(self.name_error_message).to_have_text(error_message)
+        return self
+
+    @step
+    def should_have_description_error_textarea(self, error_message: str):
+        """Verify description error text field is visible."""
+        expect(self.description_error_message).to_be_visible()
+        expect(self.description_error_message).to_have_text(error_message)
+        return self
+
+    @step
+    def should_have_system_prompt_error_textarea(self, error_message: str):
+        """Verify system prompt error text field is visible."""
+        expect(self.system_prompt_error_message).to_be_visible()
+        expect(self.system_prompt_error_message).to_have_text(error_message)
+        return self
+
+    @step
+    def should_have_icon_error_textarea(self, error_message: str):
+        """Verify icon error text field is visible."""
+        expect(self.icon_error_message).to_be_visible()
+        expect(self.icon_error_message).to_have_text(error_message)
+        return self
+
+    @step
+    def should_have_temperature_error_textarea(self, error_message: str):
+        """Verify temperature error text field is visible."""
+        expect(self.temperature_error_message).to_be_visible()
+        expect(self.temperature_error_message).to_have_text(error_message)
+        return self
+
+    @step
+    def should_have_top_p_error_textarea(self, error_message: str):
+        """Verify top p error text field is visible."""
+        expect(self.top_p_error_message).to_be_visible()
+        expect(self.top_p_error_message).to_have_text(error_message)
+        return self