codemie-test-harness 0.1.168__py3-none-any.whl → 0.1.170__py3-none-any.whl

codemie_test_harness/cli/commands/workflow_cmd.py

@@ -0,0 +1,64 @@
+"""Workflow CLI commands for executing and managing workflows."""
+
+from __future__ import annotations
+
+
+import click
+
+from codemie_test_harness.tests.utils.client_factory import get_client
+from ..constants import CONSOLE
+
+
+@click.group(name="workflow")
+def workflow_cmd():
+    """Interact with CodeMie workflows."""
+    pass
+
+
+@workflow_cmd.command(name="execute")
+@click.option(
+    "--workflow-id",
+    required=True,
+    help="Workflow ID to execute",
+)
+@click.option(
+    "--user-input",
+    "-i",
+    default="",
+    help="User input for the workflow execution (optional)",
+)
+def execute_cmd(
+    workflow_id: str,
+    user_input: str,
+):
+    """Execute a workflow.
+
+    Example:
+        codemie-test-harness workflow execute --workflow-id "wf_123"
+        codemie-test-harness workflow execute --workflow-id "wf_123" --user-input "process this data"
+    """
+    try:
+        client = get_client()
+
+        CONSOLE.print(f"[yellow]Executing workflow {workflow_id}...[/yellow]")
+        if user_input:
+            CONSOLE.print(f"[blue]User input: {user_input}[/blue]")
+
+        # Execute workflow
+        response = client.workflows.run(workflow_id=workflow_id, user_input=user_input)
+
+        # Display response
+        CONSOLE.print("[green]Workflow execution started successfully![/green]")
+
+        if hasattr(response, "execution_id"):
+            CONSOLE.print(f"[cyan]Execution ID: {response.execution_id}[/cyan]")
+
+        if hasattr(response, "status"):
+            CONSOLE.print(f"[cyan]Status: {response.status}[/cyan]")
+
+        # Display the full response for debugging
+        CONSOLE.print(f"[dim]Response: {response}[/dim]")
+
+    except Exception as e:
+        CONSOLE.print(f"[red]Error executing workflow: {e}[/red]")
+        raise click.ClickException(str(e))

codemie_test_harness/cli/constants.py

@@ -12,13 +12,16 @@ CONFIG_FILE = CONFIG_DIR / "test-harness.json"
 KEY_AUTH_SERVER_URL = "AUTH_SERVER_URL"
 KEY_AUTH_CLIENT_ID = "AUTH_CLIENT_ID"
 KEY_AUTH_CLIENT_SECRET = "AUTH_CLIENT_SECRET"
+KEY_AUTH_PASSWORD = "AUTH_PASSWORD"
+KEY_AUTH_USERNAME = "AUTH_USERNAME"
 KEY_AUTH_REALM_NAME = "AUTH_REALM_NAME"
 KEY_CODEMIE_API_DOMAIN = "CODEMIE_API_DOMAIN"
 KEY_MARKS = "PYTEST_MARKS"
 KEY_XDIST_N = "PYTEST_N"
 KEY_RERUNS = "PYTEST_RERUNS"
 
-# Integrations / external systems
+# === COMPLETE INTEGRATION CREDENTIALS KEYS ===
+# Version Control Systems (GitLab, GitHub)
 KEY_GIT_ENV = "GIT_ENV"  # e.g. gitlab or github
 # GitLab
 KEY_GITLAB_URL = "GITLAB_URL"
@@ -29,14 +32,115 @@ KEY_GITLAB_PROJECT_ID = "GITLAB_PROJECT_ID"
 KEY_GITHUB_URL = "GITHUB_URL"
 KEY_GITHUB_TOKEN = "GITHUB_TOKEN"
 KEY_GITHUB_PROJECT = "GITHUB_PROJECT"
-# JIRA
+
+# Project Management (JIRA, Confluence)
+# JIRA Server
 KEY_JIRA_URL = "JIRA_URL"
 KEY_JIRA_TOKEN = "JIRA_TOKEN"
-KEY_JQL = "JQL"
-# Confluence
+KEY_JQL = "JIRA_JQL"
+# JIRA Cloud
+KEY_JIRA_CLOUD_URL = "JIRA_CLOUD_URL"
+KEY_JIRA_CLOUD_EMAIL = "JIRA_CLOUD_EMAIL"
+KEY_JIRA_CLOUD_TOKEN = "JIRA_CLOUD_TOKEN"
+KEY_JIRA_CLOUD_JQL = "JIRA_CLOUD_JQL"
+# Confluence Server
 KEY_CONFLUENCE_URL = "CONFLUENCE_URL"
 KEY_CONFLUENCE_TOKEN = "CONFLUENCE_TOKEN"
-KEY_CQL = "CQL"
+KEY_CQL = "CONFLUENCE_CQL"
+# Confluence Cloud
+KEY_CONFLUENCE_CLOUD_URL = "CONFLUENCE_CLOUD_URL"
+KEY_CONFLUENCE_CLOUD_EMAIL = "CONFLUENCE_CLOUD_EMAIL"
+KEY_CONFLUENCE_CLOUD_TOKEN = "CONFLUENCE_CLOUD_TOKEN"
+KEY_CONFLUENCE_CLOUD_CQL = "CONFLUENCE_CLOUD_CQL"
+
+# Cloud Providers (AWS, Azure, GCP)
+# AWS
+KEY_AWS_ACCESS_KEY_ID = "AWS_ACCESS_KEY_ID"
+KEY_AWS_SECRET_ACCESS_KEY = "AWS_SECRET_ACCESS_KEY"
+KEY_AWS_REGION = "AWS_REGION"
+KEY_AWS_SESSION_TOKEN = "AWS_SESSION_TOKEN"
+# Azure
+KEY_AZURE_CLIENT_ID = "AZURE_CLIENT_ID"
+KEY_AZURE_CLIENT_SECRET = "AZURE_CLIENT_SECRET"
+KEY_AZURE_TENANT_ID = "AZURE_TENANT_ID"
+KEY_AZURE_SUBSCRIPTION_ID = "AZURE_SUBSCRIPTION_ID"
+# GCP
+KEY_GCP_SA_KEY_BASE64 = "GCP_SA_KEY_BASE64"
+
+# Development Tools (Azure DevOps, ServiceNow, Keycloak, SonarQube)
+# Azure DevOps
+KEY_AZURE_DEVOPS_URL = "AZURE_DEVOPS_URL"
+KEY_AZURE_DEVOPS_TOKEN = "AZURE_DEVOPS_TOKEN"
+KEY_AZURE_DEVOPS_PROJECT_NAME = "AZURE_DEVOPS_PROJECT_NAME"
+KEY_AZURE_DEVOPS_ORGANIZATION_NAME = "AZURE_DEVOPS_ORGANIZATION_NAME"
+# ServiceNow
+KEY_SERVICENOW_URL = "SERVICENOW_URL"
+KEY_SERVICENOW_TOKEN = "SERVICENOW_TOKEN"
+# Keycloak
+KEY_KEYCLOAK_URL = "KEYCLOAK_URL"
+KEY_KEYCLOAK_REALM = "KEYCLOAK_REALM"
+KEY_KEYCLOAK_CLIENT_ID = "KEYCLOAK_CLIENT_ID"
+KEY_KEYCLOAK_CLIENT_SECRET = "KEYCLOAK_CLIENT_SECRET"
+# SonarQube Server
+KEY_SONAR_URL = "SONAR_URL"
+KEY_SONAR_TOKEN = "SONAR_TOKEN"
+KEY_SONAR_PROJECT_KEY = "SONAR_PROJECT_KEY"
+# SonarCloud
+KEY_SONAR_CLOUD_URL = "SONAR_CLOUD_URL"
+KEY_SONAR_CLOUD_TOKEN = "SONAR_CLOUD_TOKEN"
+KEY_SONAR_CLOUD_PROJECT_KEY = "SONAR_CLOUD_PROJECT_KEY"
+
+# Notifications (Email/Gmail, OAuth, Telegram)
+# Email/Gmail
+KEY_GMAIL_URL = "GMAIL_URL"
+KEY_SMTP_USERNAME = "SMTP_USERNAME"
+KEY_SMTP_PASSWORD = "SMTP_PASSWORD"
+# OAuth
+KEY_OAUTH_URL = "OAUTH_URL"
+KEY_OAUTH_CLIENT_ID = "OAUTH_CLIENT_ID"
+KEY_OAUTH_CLIENT_SECRET = "OAUTH_CLIENT_SECRET"
+KEY_OAUTH_REFRESH_TOKEN = "OAUTH_REFRESH_TOKEN"
+# Telegram
+KEY_TELEGRAM_TOKEN = "TELEGRAM_TOKEN"
+KEY_TELEGRAM_CHAT_ID = "TELEGRAM_CHAT_ID"
+
+# Research Tools (Kubernetes, Report Portal, Elasticsearch)
+# Kubernetes
+KEY_KUBERNETES_URL = "KUBERNETES_URL"
+KEY_KUBERNETES_TOKEN = "KUBERNETES_TOKEN"
+# Report Portal
+KEY_REPORT_PORTAL_URL = "REPORT_PORTAL_URL"
+KEY_REPORT_PORTAL_API_KEY = "REPORT_PORTAL_API_KEY"
+KEY_REPORT_PORTAL_PROJECT = "REPORT_PORTAL_PROJECT"
+# Elasticsearch
+KEY_ELASTICSEARCH_URL = "ELASTICSEARCH_URL"
+KEY_ELASTICSEARCH_API_KEY_ID = "ELASTICSEARCH_API_KEY_ID"
+KEY_ELASTICSEARCH_API_KEY = "ELASTICSEARCH_API_KEY"
+
+# Data Management (SQL databases, LiteLLM)
+# LiteLLM
+KEY_LITE_LLM_API_KEY = "LITE_LLM_API_KEY"
+# MySQL
+KEY_MYSQL_DIALECT = "MYSQL_DIALECT"
+KEY_MYSQL_URL = "MYSQL_URL"
+KEY_MYSQL_PORT = "MYSQL_PORT"
+KEY_MYSQL_DATABASE_NAME = "MYSQL_DATABASE_NAME"
+KEY_MYSQL_USERNAME = "MYSQL_USERNAME"
+KEY_MYSQL_PASSWORD = "MYSQL_PASSWORD"
+# PostgreSQL
+KEY_POSTGRES_DIALECT = "POSTGRES_DIALECT"
+KEY_POSTGRES_URL = "POSTGRES_URL"
+KEY_POSTGRES_PORT = "POSTGRES_PORT"
+KEY_POSTGRES_DATABASE_NAME = "POSTGRES_DATABASE_NAME"
+KEY_POSTGRES_USERNAME = "POSTGRES_USERNAME"
+KEY_POSTGRES_PASSWORD = "POSTGRES_PASSWORD"
+# MSSQL
+KEY_MSSQL_DIALECT = "MSSQL_DIALECT"
+KEY_MSSQL_URL = "MSSQL_URL"
+KEY_MSSQL_PORT = "MSSQL_PORT"
+KEY_MSSQL_DATABASE_NAME = "MSSQL_DATABASE_NAME"
+KEY_MSSQL_USERNAME = "MSSQL_USERNAME"
+KEY_MSSQL_PASSWORD = "MSSQL_PASSWORD"
 
 DEFAULT_MARKS = "smoke"
 DEFAULT_XDIST_N = 8
@@ -46,11 +150,14 @@ AUTH_KEYS = [
     KEY_AUTH_SERVER_URL,
     KEY_AUTH_CLIENT_ID,
     KEY_AUTH_CLIENT_SECRET,
+    KEY_AUTH_USERNAME,
+    KEY_AUTH_PASSWORD,
     KEY_AUTH_REALM_NAME,
     KEY_CODEMIE_API_DOMAIN,
 ]
 
-INTEGRATION_KEYS = [
+# === CREDENTIAL CATEGORIES ===
+VERSION_CONTROL_KEYS = [
     KEY_GIT_ENV,
     KEY_GITLAB_URL,
     KEY_GITLAB_TOKEN,
@@ -59,10 +166,282 @@ INTEGRATION_KEYS = [
     KEY_GITHUB_URL,
     KEY_GITHUB_TOKEN,
     KEY_GITHUB_PROJECT,
+]
+
+PROJECT_MANAGEMENT_KEYS = [
+    # JIRA Server
     KEY_JIRA_URL,
     KEY_JIRA_TOKEN,
     KEY_JQL,
+    # JIRA Cloud
+    KEY_JIRA_CLOUD_URL,
+    KEY_JIRA_CLOUD_EMAIL,
+    KEY_JIRA_CLOUD_TOKEN,
+    KEY_JIRA_CLOUD_JQL,
+    # Confluence Server
     KEY_CONFLUENCE_URL,
     KEY_CONFLUENCE_TOKEN,
     KEY_CQL,
+    # Confluence Cloud
+    KEY_CONFLUENCE_CLOUD_URL,
+    KEY_CONFLUENCE_CLOUD_EMAIL,
+    KEY_CONFLUENCE_CLOUD_TOKEN,
+    KEY_CONFLUENCE_CLOUD_CQL,
+]
+
+CLOUD_PROVIDERS_KEYS = [
+    # AWS
+    KEY_AWS_ACCESS_KEY_ID,
+    KEY_AWS_SECRET_ACCESS_KEY,
+    KEY_AWS_REGION,
+    KEY_AWS_SESSION_TOKEN,
+    # Azure
+    KEY_AZURE_CLIENT_ID,
+    KEY_AZURE_CLIENT_SECRET,
+    KEY_AZURE_TENANT_ID,
+    KEY_AZURE_SUBSCRIPTION_ID,
+    # GCP
+    KEY_GCP_SA_KEY_BASE64,
+]
+
+DEVELOPMENT_TOOLS_KEYS = [
+    # Azure DevOps
+    KEY_AZURE_DEVOPS_URL,
+    KEY_AZURE_DEVOPS_TOKEN,
+    KEY_AZURE_DEVOPS_PROJECT_NAME,
+    KEY_AZURE_DEVOPS_ORGANIZATION_NAME,
+    # ServiceNow
+    KEY_SERVICENOW_URL,
+    KEY_SERVICENOW_TOKEN,
+    # Keycloak
+    KEY_KEYCLOAK_URL,
+    KEY_KEYCLOAK_REALM,
+    KEY_KEYCLOAK_CLIENT_ID,
+    KEY_KEYCLOAK_CLIENT_SECRET,
+    # SonarQube Server
+    KEY_SONAR_URL,
+    KEY_SONAR_TOKEN,
+    KEY_SONAR_PROJECT_KEY,
+    # SonarCloud
+    KEY_SONAR_CLOUD_URL,
+    KEY_SONAR_CLOUD_TOKEN,
+    KEY_SONAR_CLOUD_PROJECT_KEY,
+]
+
+NOTIFICATIONS_KEYS = [
+    # Email/Gmail
+    KEY_GMAIL_URL,
+    KEY_SMTP_USERNAME,
+    KEY_SMTP_PASSWORD,
+    # OAuth
+    KEY_OAUTH_URL,
+    KEY_OAUTH_CLIENT_ID,
+    KEY_OAUTH_CLIENT_SECRET,
+    KEY_OAUTH_REFRESH_TOKEN,
+    # Telegram
+    KEY_TELEGRAM_TOKEN,
+    KEY_TELEGRAM_CHAT_ID,
+]
+
+RESEARCH_TOOLS_KEYS = [
+    # Kubernetes
+    KEY_KUBERNETES_URL,
+    KEY_KUBERNETES_TOKEN,
+    # Report Portal
+    KEY_REPORT_PORTAL_URL,
+    KEY_REPORT_PORTAL_API_KEY,
+    KEY_REPORT_PORTAL_PROJECT,
+    # Elasticsearch
+    KEY_ELASTICSEARCH_URL,
+    KEY_ELASTICSEARCH_API_KEY_ID,
+    KEY_ELASTICSEARCH_API_KEY,
 ]
+
+DATA_MANAGEMENT_KEYS = [
+    # LiteLLM
+    KEY_LITE_LLM_API_KEY,
+    # MySQL
+    KEY_MYSQL_DIALECT,
+    KEY_MYSQL_URL,
+    KEY_MYSQL_PORT,
+    KEY_MYSQL_DATABASE_NAME,
+    KEY_MYSQL_USERNAME,
+    KEY_MYSQL_PASSWORD,
+    # PostgreSQL
+    KEY_POSTGRES_DIALECT,
+    KEY_POSTGRES_URL,
+    KEY_POSTGRES_PORT,
+    KEY_POSTGRES_DATABASE_NAME,
+    KEY_POSTGRES_USERNAME,
+    KEY_POSTGRES_PASSWORD,
+    # MSSQL
+    KEY_MSSQL_DIALECT,
+    KEY_MSSQL_URL,
+    KEY_MSSQL_PORT,
+    KEY_MSSQL_DATABASE_NAME,
+    KEY_MSSQL_USERNAME,
+    KEY_MSSQL_PASSWORD,
+]
+
+# Combined lists for backwards compatibility
+INTEGRATION_KEYS = (
+    VERSION_CONTROL_KEYS
+    + PROJECT_MANAGEMENT_KEYS
+    + CLOUD_PROVIDERS_KEYS
+    + DEVELOPMENT_TOOLS_KEYS
+    + NOTIFICATIONS_KEYS
+    + RESEARCH_TOOLS_KEYS
+    + DATA_MANAGEMENT_KEYS
+)
+
+# Category mapping for CLI
+CREDENTIAL_CATEGORIES = {
+    "version-control": {
+        "name": "Version Control Systems",
+        "description": "GitLab, GitHub, and other version control platforms",
+        "keys": VERSION_CONTROL_KEYS,
+    },
+    "code-base": {
+        "name": "Code Base",
+        "description": "SonarQube and code analysis tools",
+        "keys": [
+            KEY_SONAR_URL,
+            KEY_SONAR_TOKEN,
+            KEY_SONAR_PROJECT_KEY,
+            KEY_SONAR_CLOUD_URL,
+            KEY_SONAR_CLOUD_TOKEN,
+            KEY_SONAR_CLOUD_PROJECT_KEY,
+        ],
+    },
+    "project-management": {
+        "name": "Project Management",
+        "description": "JIRA, Confluence, and project management tools",
+        "keys": PROJECT_MANAGEMENT_KEYS,
+    },
+    "cloud": {
+        "name": "Cloud",
+        "description": "AWS, GCP cloud services and Kubernetes configurations",
+        "keys": [
+            # AWS
+            KEY_AWS_ACCESS_KEY_ID,
+            KEY_AWS_SECRET_ACCESS_KEY,
+            KEY_AWS_REGION,
+            KEY_AWS_SESSION_TOKEN,
+            # GCP
+            KEY_GCP_SA_KEY_BASE64,
+            # Kubernetes
+            KEY_KUBERNETES_URL,
+            KEY_KUBERNETES_TOKEN,
+            # Azure Cloud
+            KEY_AZURE_CLIENT_ID,
+            KEY_AZURE_CLIENT_SECRET,
+            KEY_AZURE_TENANT_ID,
+            KEY_AZURE_SUBSCRIPTION_ID,
+        ],
+    },
+    "azure-devops": {
+        "name": "Azure DevOps",
+        "description": "Azure DevOps and Azure cloud platform services",
+        "keys": [
+            # Azure DevOps
+            KEY_AZURE_DEVOPS_URL,
+            KEY_AZURE_DEVOPS_TOKEN,
+            KEY_AZURE_DEVOPS_PROJECT_NAME,
+            KEY_AZURE_DEVOPS_ORGANIZATION_NAME,
+        ],
+    },
+    "access-management": {
+        "name": "Access Management",
+        "description": "Keycloak and identity management systems",
+        "keys": [
+            KEY_KEYCLOAK_URL,
+            KEY_KEYCLOAK_REALM,
+            KEY_KEYCLOAK_CLIENT_ID,
+            KEY_KEYCLOAK_CLIENT_SECRET,
+        ],
+    },
+    "notification-systems": {
+        "name": "Notification Systems",
+        "description": "Email, OAuth, Telegram, and communication platforms",
+        "keys": NOTIFICATIONS_KEYS,
+    },
+    "data-management": {
+        "name": "Data Management",
+        "description": "SQL databases, Elasticsearch, and data storage systems",
+        "keys": DATA_MANAGEMENT_KEYS
+        + [
+            KEY_ELASTICSEARCH_URL,
+            KEY_ELASTICSEARCH_API_KEY_ID,
+            KEY_ELASTICSEARCH_API_KEY,
+        ],
+    },
+    "it-service-management": {
+        "name": "IT Service Management",
+        "description": "ServiceNow and IT service management platforms",
+        "keys": [
+            KEY_SERVICENOW_URL,
+            KEY_SERVICENOW_TOKEN,
+        ],
+    },
+    "quality-assurance": {
+        "name": "Quality Assurance",
+        "description": "Report Portal and quality assurance tools",
+        "keys": [
+            KEY_REPORT_PORTAL_URL,
+            KEY_REPORT_PORTAL_API_KEY,
+            KEY_REPORT_PORTAL_PROJECT,
+        ],
+    },
+}
+
+# Sensitive fields that should be masked in input/output
+SENSITIVE_KEYS = {
+    KEY_AUTH_CLIENT_SECRET,
+    KEY_AUTH_PASSWORD,
+    KEY_GITLAB_TOKEN,
+    KEY_GITHUB_TOKEN,
+    KEY_JIRA_TOKEN,
+    KEY_JIRA_CLOUD_TOKEN,
+    KEY_CONFLUENCE_TOKEN,
+    KEY_CONFLUENCE_CLOUD_TOKEN,
+    KEY_AWS_SECRET_ACCESS_KEY,
+    KEY_AWS_SESSION_TOKEN,
+    KEY_AZURE_CLIENT_SECRET,
+    KEY_GCP_SA_KEY_BASE64,
+    KEY_AZURE_DEVOPS_TOKEN,
+    KEY_SERVICENOW_TOKEN,
+    KEY_KEYCLOAK_CLIENT_SECRET,
+    KEY_SONAR_TOKEN,
+    KEY_SONAR_CLOUD_TOKEN,
+    KEY_SMTP_PASSWORD,
+    KEY_OAUTH_CLIENT_SECRET,
+    KEY_OAUTH_REFRESH_TOKEN,
+    KEY_TELEGRAM_TOKEN,
+    KEY_KUBERNETES_TOKEN,
+    KEY_REPORT_PORTAL_API_KEY,
+    KEY_ELASTICSEARCH_API_KEY,
+    KEY_LITE_LLM_API_KEY,
+    KEY_MYSQL_PASSWORD,
+    KEY_POSTGRES_PASSWORD,
+    KEY_MSSQL_PASSWORD,
+}
+
+
+def is_sensitive_key(key: str) -> bool:
+    """Check if a credential key contains sensitive information."""
+    return key in SENSITIVE_KEYS
+
+
+def mask_sensitive_value(value: str, show_real: bool = False) -> str:
+    """Mask sensitive credential values for display.
+
+    Args:
+        value: The value to potentially mask
+        show_real: If True, return the real value instead of masking
+    """
+    if show_real:
+        return value
+
+    if not value or len(value) < 4:
+        return "*" * len(value) if value else ""
+    return value[:2] + "*" * (len(value) - 4) + value[-2:]

codemie_test_harness/cli/utils.py

@@ -43,3 +43,12 @@ def ensure_env_from_config(keys: list[str]) -> None:
     for k in keys:
         if k not in os.environ and k in cfg:
             os.environ[k] = str(cfg[k])
+
+
+def unset_config_key(key: str) -> bool:
+    cfg = load_config()
+    if key in cfg:
+        del cfg[key]
+        save_config(cfg)
+        return True
+    return False

codemie_test_harness/tests/test_data/assistant_test_data.py

@@ -0,0 +1,197 @@
+"""
+Test Data for Assistant UI Tests
+
+This module provides test data generation and management for assistant-related UI tests.
+Following best practices by separating test data from test logic and providing
+reusable data factories for consistent testing.
+"""
+
+from dataclasses import dataclass
+from typing import Optional, List
+
+from codemie_test_harness.tests.utils.base_utils import get_random_name
+
+
+@dataclass
+class AssistantTestData:
+    """
+    Data class for assistant test data.
+
+    This class encapsulates all the data needed for assistant creation tests,
+    providing a clean and type-safe way to manage test data.
+    """
+
+    name: str
+    description: str
+    system_prompt: str
+    icon_url: Optional[str] = None
+    shared: bool = False
+
+
+class AssistantTestDataFactory:
+    """
+    Factory class for generating assistant test data.
+
+    This factory provides various methods to create different types of
+    assistant test data for different testing scenarios.
+    """
+
+    @staticmethod
+    def create_minimal_assistant_data() -> AssistantTestData:
+        """
+        Create minimal assistant data with only required fields.
+
+        This represents the most basic assistant creation scenario
+        with minimal required information.
+
+        Returns:
+            AssistantTestData: Minimal assistant test data
+        """
+        return AssistantTestData(
+            name=f"QA Test Assistant {get_random_name()}",
+            description="Minimal test assistant for QA automation.",
+            system_prompt=(
+                "You are a test assistant created for QA validation purposes. "
+                "Provide helpful and accurate responses to user queries."
+            ),
+            shared=False,
+            icon_url=ICON_URL,
+        )
+
+    @staticmethod
+    def create_shared_assistant_data() -> AssistantTestData:
+        """
+        Create shared assistant data for public/shared testing scenarios.
+
+        Returns:
+            AssistantTestData: Shared assistant test data
+        """
+        return AssistantTestData(
+            name=f"QA Shared Assistant {get_random_name()}",
+            description="Shared QA assistant available to all team members",
+            system_prompt=(
+                "You are a shared QA assistant available to the entire team. "
+                "Provide collaborative testing support, knowledge sharing, and "
+                "help maintain consistent quality standards across projects."
+            ),
+            icon_url=ICON_URL,
+            shared=True,
+        )
+
+    @staticmethod
+    def create_validation_test_data() -> List[AssistantTestData]:
+        """
+        Create a list of assistant data for validation testing scenarios.
+
+        This includes data for testing various validation scenarios,
+        edge cases in form validation, and error handling.
+
+        Returns:
+            List[AssistantTestData]: List of validation test data
+        """
+        return [
+            # Empty name scenario
+            AssistantTestData(
+                name="",
+                description="Test description",
+                system_prompt="Test prompt",
+            ),
+            # Long name scenario
+            AssistantTestData(
+                name="A" * 100,  # Very long name
+                description="Test description for long name validation",
+                system_prompt="Test prompt for long name scenario",
+            ),
+            # Empty description scenario
+            AssistantTestData(
+                name="Test Assistant",
+                description="",
+                system_prompt="Test prompt",
+            ),
+            # Empty system prompt scenario
+            AssistantTestData(
+                name="Test Assistant",
+                description="Test description",
+                system_prompt="",
+            ),
+        ]
+
+
+class AssistantValidationRules:
+    """
+    Validation rules and constraints for assistant data.
+
+    This class defines the validation rules that should be applied
+    to assistant data during testing.
+    """
+
+    # Field length constraints
+    MAX_NAME_LENGTH = 100
+    MAX_DESCRIPTION_LENGTH = 1000
+    MAX_ICON_URL_LENGTH = 500
+
+    # Required fields
+    REQUIRED_FIELDS = ["name", "description", "system_prompt"]
+
+    # Validation error messages (expected messages for testing)
+    ERROR_MESSAGES = {
+        "name_required": "Name is required",
+        "name_too_long": f"Name must be less than {MAX_NAME_LENGTH} characters",
+        "description_required": "Description is required",
+        "description_too_long": f"Description must be less than {MAX_DESCRIPTION_LENGTH} characters",
+        "system_prompt_required": "System prompt is required",
+        "invalid_url": "Please enter a valid URL",
+        "invalid_type": "Please select a valid assistant type",
+    }
+
+
+# ==================== CONVENIENCE FUNCTIONS ====================
+
+
+def get_minimal_assistant_data() -> AssistantTestData:
+    """Convenience function to get minimal assistant data."""
+    return AssistantTestDataFactory.create_minimal_assistant_data()
+
+
+def get_shared_assistant_data() -> AssistantTestData:
+    """Convenience function to get shared assistant data."""
+    return AssistantTestDataFactory.create_shared_assistant_data()
+
+
+def get_validation_test_data() -> List[AssistantTestData]:
+    """Convenience function to get validation test data."""
+    return AssistantTestDataFactory.create_validation_test_data()
+
+
+# ==================== TEST DATA CONSTANTS ====================
+
+# Common test values for reuse
+COMMON_TEST_PROMPTS = {
+    "qa_assistant": (
+        "You are a QA testing assistant. Your primary role is to help with "
+        "quality assurance tasks, test automation, and ensuring software quality. "
+        "Provide detailed and actionable guidance."
+    ),
+    "general_assistant": (
+        "You are a helpful assistant. Provide clear, accurate, and helpful "
+        "responses to user queries. Always be polite and professional."
+    ),
+    "specialist_assistant": (
+        "You are a specialist assistant with deep expertise in your domain. "
+        "Provide expert-level guidance and detailed technical solutions."
+    ),
+}
+
+COMMON_TEST_DESCRIPTIONS = {
+    "qa_assistant": "QA testing assistant for automation and quality assurance tasks",
+    "general_assistant": "General purpose assistant for various tasks and queries",
+    "specialist_assistant": "Specialist assistant with domain-specific expertise",
+}
+
+COMMON_ICON_URLS = {
+    "qa_icon": "https://example.com/qa-assistant-icon.png",
+    "general_icon": "https://example.com/general-assistant-icon.png",
+    "specialist_icon": "https://example.com/specialist-assistant-icon.png",
+}
+
+ICON_URL = "https://raw.githubusercontent.com/epam-gen-ai-run/ai-run-install/main/docs/assets/ai/AQAUiTestGenerator.png"