codemie-test-harness 0.1.176__py3-none-any.whl → 0.1.178__py3-none-any.whl

codemie_test_harness/tests/assistant/tools/mcp/test_single_assistant_dual_time_plugins.py

@@ -115,10 +115,16 @@ def dual_time_plugins(integration_utils):
         credential_type=CredentialTypes.PLUGIN, credential_values=credential_values
     )
 
-    # Download codemie-tools
-    download_process = subprocess.Popen(["uvx", "codemie-tools"])
+    # Download codemie-plugins and wait for completion
+    download_process = subprocess.Popen(["uvx", "codemie-plugins"])
     processes.append(download_process)
 
+    return_code = download_process.wait()
+    if return_code != 0:
+        raise RuntimeError(
+            f"Failed to download codemie-plugins, exit code: {return_code}"
+        )
+
     # Find paths and configure servers
     servers_json_path, codemie_plugins_path = find_cache_paths()
     configure_mcp_servers(servers_json_path)

codemie_test_harness/tests/conftest.py

@@ -253,7 +253,7 @@ def conversation_utils(client):
     return ConversationUtils(client)
 
 
-@pytest.fixture(scope="function")
+@pytest.fixture(scope="session")
 def git_integration(integration_utils):
     integration = integration_utils.create_integration(
         credential_type=CredentialTypes.GIT,
@@ -264,7 +264,7 @@ def git_integration(integration_utils):
         integration_utils.delete_integration(integration)
 
 
-@pytest.fixture(scope="function")
+@pytest.fixture(scope="session")
 def github_integration(integration_utils):
     integration = integration_utils.create_integration(
         credential_type=CredentialTypes.GIT,
@@ -275,7 +275,7 @@ def github_integration(integration_utils):
         integration_utils.delete_integration(integration)
 
 
-@pytest.fixture(scope="function")
+@pytest.fixture(scope="session")
 def gitlab_integration(integration_utils):
     integration = integration_utils.create_integration(
         credential_type=CredentialTypes.GIT,
@@ -438,7 +438,7 @@ def general_integration(integration_utils):
         pass
 
 
-@pytest.fixture(scope="function")
+@pytest.fixture(scope="session")
 def code_datasource(
     datasource_utils, gitlab_integration, github_integration, default_embedding_llm
 ):
@@ -458,7 +458,7 @@ def code_datasource(
         datasource_utils.delete_datasource(datasource)
 
 
-@pytest.fixture(scope="function")
+@pytest.fixture(scope="session")
 def gitlab_datasource(datasource_utils, gitlab_integration, default_embedding_llm):
     datasource = datasource_utils.create_gitlab_datasource(
         setting_id=gitlab_integration.id,
@@ -469,7 +469,7 @@ def gitlab_datasource(datasource_utils, gitlab_integration, default_embedding_ll
         datasource_utils.delete_datasource(datasource)
 
 
-@pytest.fixture(scope="function")
+@pytest.fixture(scope="session")
 def github_datasource(datasource_utils, github_integration, default_embedding_llm):
     datasource = datasource_utils.create_github_datasource(
         setting_id=github_integration.id,

codemie_test_harness/tests/llm/assistants/test_lite_llm.py

@@ -13,6 +13,11 @@ from codemie_test_harness.tests.utils.env_resolver import (
 )
 from codemie_test_harness.tests.utils.pytest_utils import check_mark
 
+pytestmark = pytest.mark.skipif(
+    EnvironmentResolver.is_sandbox() or EnvironmentResolver.is_localhost(),
+    reason="Skipping this tests on sandbox environments",
+)
+
 
 @pytest.fixture(scope="module")
 def lite_llm_integration(integration_utils):
@@ -66,10 +71,6 @@ def pytest_generate_tests(metafunc):
 @pytest.mark.assistant
 @pytest.mark.lite_llm
 @pytest.mark.regression
-@pytest.mark.skipif(
-    EnvironmentResolver.is_sandbox(),
-    reason="Skipping this tests on sandbox environments",
-)
 def test_assistant_with_different_models_in_lite_llm(
     llm_utils,
     lite_llm_integration,

codemie_test_harness/tests/test_data/google_datasource_test_data.py

@@ -1,6 +1,9 @@
 """Test data constants for Google Datasource tests"""
 
-GOOGLE_DOC_URL = "https://docs.google.com/document/d/19EXgnFCgJontz0ToCAH6zMGwBTdhi5X97P9JIby4wHs/edit#heading=h.b01c2ig0adfg"
+GOOGLE_DOC_URL = "https://docs.google.com/document/d/19EXgnFCgJontz0ToCAH6zMGwBTdhi5X97P9JIby4wHs/edit?tab=t.0#heading=h.b01c2ig0adfg"
+GOOGLE_GUIDE_URL = (
+    "https://docs.google.com/document/d/1ZNWwxN8ukpJZyTbYjWQPym3Bc1-bQvBxBpTL0yfVF-w"
+)
 
 USER_PROMPT = 'Tell context of the What is "Platform A"? section?'
 

codemie_test_harness/tests/ui/_test_data/datasource_test_data.py

@@ -0,0 +1,125 @@
+"""
+Test Data for Data Source UI – used for parametrized testing and test validation
+"""
+
+from enum import Enum
+
+
+# ==================== ENUMS ====================
+class DataSourceFilterType(str, Enum):
+    CODE = "Code"
+    CONFLUENCE = "Confluence"
+    JIRA = "Jira"
+    FILE = "File"
+    GOOGLE = "Google"
+    PROVIDER = "Provider"
+
+    def __str__(self):
+        return self.value
+
+
+class DataSourceType(str, Enum):
+    GIT = "Git"
+    CONFLUENCE = "Confluence"
+    JIRA = "Jira"
+    FILE = "File"
+    GOOGLE = "Google"
+
+    def __str__(self):
+        return self.value
+
+
+class DataSourceFilterStatus(str, Enum):
+    COMPLETED = "Completed"
+    FAILED = "Failed"
+    IN_PROGRESS = "In Progress"
+
+    def __str__(self):
+        return self.value
+
+
+class DataSourceStatus(str, Enum):
+    COMPLETED = "Completed"
+    ERROR = "Error"
+    FETCHING = "Fetching"
+
+    def __str__(self):
+        return self.value
+
+
+class SummarizationMethod(str, Enum):
+    WHOLE_CODEBASE = "Whole Codebase"
+    PER_FILE = "Summarization per file"
+    PER_CHUNKS = "Summarization per chunks"
+
+    def __str__(self):
+        return self.value
+
+
+class EmbeddingModel(str, Enum):
+    ADA = "Text Embedding Ada"
+    GECKO = "Text Embedding Gecko"
+    TITAN_V2 = "Titan Embed Text v2.0"
+
+    def __str__(self):
+        return self.value
+
+
+class DataSourceColumnName(str, Enum):
+    NAME = "Name"
+    PROJECT = "Project"
+    TYPE = "Type"
+    CREATED_BY = "Created By"
+    CREATED = "Created"
+    UPDATED = "Updated"
+    SHARED = "Shared"
+    STATUS = "Status"
+    ACTIONS = "Actions"
+
+    def __str__(self):
+        return self.value
+
+
+# ==================== CONSTANTS ====================
+
+DATA_SOURCE_FILTER_TYPES_LIST = [t for t in DataSourceFilterType]
+DATA_SOURCE_TYPES_LIST = [t for t in DataSourceType]
+DATA_SOURCE_FILTER_STATUSES_LIST = [s for s in DataSourceFilterStatus]
+DATA_SOURCE_STATUSES_LIST = [s for s in DataSourceStatus]
+DATAS_SOURCE_COLUMN_LIST = [c for c in DataSourceColumnName]
+SUMMARIZATION_METHODS_LIST = [m for m in SummarizationMethod]
+EMBEDDING_MODELS_LIST = [m for m in EmbeddingModel]
+
+PROJECT_LABEL = "Project"
+STATUS_LABEL = "Status"
+
+MAIN_TITLE_DATASOURCE = "Data Sources"
+MAIN_SUBTITLE_DATASOURCE = "Connect and manage your data sources in one place"
+TITLE_CREATE_DATASOURCE = "New DataSource"
+SUBTITLE_CREATE_DATASOURCE = "Start indexing your data source"
+TITLE_VIEW_DATASOURCE = "View DataSource"
+SUBTITLE_VIEW_DATASOURCE = "View your data source datails"
+UPDATE_TITLE_DATASOURCE = "Update DataSource"
+UPDATE_SUBTITLE_DATASOURCE = "Update your data source and start re-indexing"
+
+FILE_INSTRUCTIONS = (
+    "Max size: 100Mb. Formats: .yml, .yaml, .json, .pptx, .csv, .txt, "
+    ".pdf, .docx, .xlsx, .xml"
+)
+GOOGLE_INSTRUCTIONS = (
+    "Please ensure your Google document is properly formatted and shared with the"
+    " service account. For detailed instructions, refer to the Guide"
+)
+GOOGLE_EXAMPLE = "Google documents must follow a specific format for LLM routing: View Format Example"
+
+EMPTY_NAME_ERROR = "Data source name is required"
+EMPTY_DESCRIPTION_ERROR = "description is a required field"
+EMPTY_REPO_LINK_ERROR = "Repo Link is required"
+EMPTY_BRANCH_ERROR = "Branch is required"
+EMPTY_CQL_ERROR = "cql is a required field"
+EMPTY_JQL_ERROR = "jql is a required field"
+EMPTY_FILE_ERROR = (
+    "Invalid file type found, only .yml, .yaml, .json, .pptx, .csv, .txt, .pdf, "
+    ".docx, .xlsx, .xml are allowed"
+)
+EMPTY_GOOGLE_LINK_ERROR = "Google Docs link is required"

codemie_test_harness/tests/ui/datasource/test_create_datasource.py

@@ -0,0 +1,188 @@
+import pytest
+from tests import CredentialsManager, PROJECT, TEST_USER
+from tests.test_data.google_datasource_test_data import GOOGLE_DOC_URL
+from tests.ui._test_data.datasource_test_data import (
+    DataSourceType,
+    DataSourceFilterType,
+    DataSourceStatus,
+)
+from tests.ui._test_data.datasource_test_data import (
+    TITLE_CREATE_DATASOURCE,
+    SUBTITLE_CREATE_DATASOURCE,
+)
+from tests.ui.pageobject.datasources.create_edit_datasource_page import (
+    CreateEditDatasourcePage,
+)
+from tests.ui.pageobject.datasources.datasource_page import DataSourcePage
+from tests.utils.constants import FILES_PATH
+
+
+@pytest.mark.datasource_ui
+@pytest.mark.ui
+def test_open_create_datasource_page_and_elements(page):
+    """Open Create Datasource page and verify critical elements."""
+    create_page = CreateEditDatasourcePage(page)
+
+    create_page.navigate_to()
+    create_page.sidebar.should_see_title_subtitle(
+        TITLE_CREATE_DATASOURCE, SUBTITLE_CREATE_DATASOURCE
+    )
+    create_page.should_see_main_fields()
+
+    create_page.select_datasource_type(DataSourceType.GIT)
+    create_page.should_see_git_fields()
+
+    create_page.select_datasource_type(DataSourceFilterType.CONFLUENCE)
+    create_page.should_see_confluence_cql_field()
+    create_page.should_see_integration_input_or_button()
+
+    create_page.select_datasource_type(DataSourceFilterType.JIRA)
+    create_page.should_see_jira_jql_field()
+    create_page.should_see_integration_input_or_button()
+
+    create_page.select_datasource_type(DataSourceFilterType.FILE)
+    create_page.should_see_file_fields()
+    create_page.select_datasource_type(DataSourceFilterType.GOOGLE)
+    create_page.should_see_google_fields()
+
+
+@pytest.mark.datasource_ui
+@pytest.mark.ui
+def test_datasource_creation_validation(page):
+    """Test create datasource without data and observing errors."""
+    create_page = CreateEditDatasourcePage(page)
+
+    create_page.navigate_to()
+    create_page.select_datasource_type(DataSourceType.GIT).click_create()
+    create_page.should_see_error_for_empty_main_fields()
+    create_page.should_see_error_for_empty_git_fields()
+
+    create_page.select_datasource_type(DataSourceFilterType.CONFLUENCE).click_create()
+    create_page.should_see_error_for_empty_main_fields()
+    create_page.should_see_error_for_empty_confluence_fields()
+
+    create_page.select_datasource_type(DataSourceFilterType.JIRA).click_create()
+    create_page.should_see_error_for_empty_main_fields()
+    create_page.should_see_error_for_empty_jira_fields()
+
+    create_page.select_datasource_type(DataSourceFilterType.FILE).click_create()
+    create_page.should_see_error_for_empty_main_fields()
+    create_page.should_see_error_for_empty_file_fields()
+
+    create_page.select_datasource_type(DataSourceFilterType.GOOGLE).click_create()
+    create_page.should_see_error_for_empty_main_fields()
+    create_page.should_see_error_for_empty_google_fields()
+
+
+@pytest.mark.datasource_ui
+@pytest.mark.ui
+def test_create_git_datasource(page, git_integration):
+    """Test creating a new datasource with all required fields."""
+    datasource_page = DataSourcePage(page)
+    create_page = CreateEditDatasourcePage(page)
+
+    create_page.navigate_to()
+    datasource = create_page.create_git_datasource(
+        project_name=PROJECT,
+        repo_link=CredentialsManager.get_parameter("GITLAB_PROJECT"),
+        branch="main",
+        integration=git_integration.alias,
+    )
+
+    datasource_page.should_see_table_row_with_values(
+        name=datasource,
+        status=DataSourceStatus.COMPLETED,
+        project=PROJECT,
+        type_=DataSourceFilterType.CODE,
+        created_by=TEST_USER,
+    )
+
+
+@pytest.mark.datasource_ui
+@pytest.mark.ui
+def test_create_confluence_datasource(page, confluence_integration):
+    """Test creating a new datasource with all required fields."""
+    datasource_page = DataSourcePage(page)
+    create_page = CreateEditDatasourcePage(page)
+
+    create_page.navigate_to()
+    datasource = create_page.create_confluence_datasource(
+        project_name=PROJECT,
+        cql_query=CredentialsManager.confluence_cql(),
+        integration=confluence_integration.alias,
+    )
+
+    datasource_page.should_see_table_row_with_values(
+        name=datasource,
+        status=DataSourceStatus.COMPLETED,
+        project=PROJECT,
+        type_=DataSourceFilterType.CONFLUENCE,
+        created_by=TEST_USER,
+    )
+
+
+@pytest.mark.datasource_ui
+@pytest.mark.ui
+def test_create_jira_datasource(page, jira_integration):
+    """Test creating a new datasource with all required fields."""
+    datasource_page = DataSourcePage(page)
+    create_page = CreateEditDatasourcePage(page)
+
+    create_page.navigate_to()
+    datasource = create_page.create_jira_datasource(
+        project_name=PROJECT,
+        jql_query=CredentialsManager.jira_jql(),
+        integration=jira_integration.alias,
+    )
+
+    datasource_page.should_see_table_row_with_values(
+        name=datasource,
+        status=DataSourceStatus.COMPLETED,
+        project=PROJECT,
+        type_=DataSourceFilterType.JIRA,
+        created_by=TEST_USER,
+    )
+
+
+@pytest.mark.datasource_ui
+@pytest.mark.ui
+def test_create_file_datasource(page):
+    """Test creating a new datasource with all required fields."""
+    datasource_page = DataSourcePage(page)
+    create_page = CreateEditDatasourcePage(page)
+
+    create_page.navigate_to()
+    datasource = create_page.create_file_datasource(
+        project_name=PROJECT,
+        file_path=[str(FILES_PATH / "test.txt")],
+    )
+
+    datasource_page.should_see_table_row_with_values(
+        name=datasource,
+        status=DataSourceStatus.COMPLETED,
+        project=PROJECT,
+        type_=DataSourceFilterType.FILE,
+        created_by=TEST_USER,
+    )
+
+
+@pytest.mark.datasource_ui
+@pytest.mark.ui
+def test_create_google_datasource(page):
+    """Test creating a new datasource with all required fields."""
+    datasource_page = DataSourcePage(page)
+    create_page = CreateEditDatasourcePage(page)
+
+    create_page.navigate_to()
+    datasource = create_page.create_google_datasource(
+        project_name=PROJECT,
+        google_doc_link=GOOGLE_DOC_URL,
+    )
+
+    datasource_page.should_see_table_row_with_values(
+        name=datasource,
+        status=DataSourceStatus.COMPLETED,
+        project=PROJECT,
+        type_=DataSourceFilterType.GOOGLE,
+        created_by=TEST_USER,
+    )

codemie_test_harness/tests/ui/datasource/test_datasource_page.py

@@ -0,0 +1,61 @@
+import pytest
+
+from codemie_test_harness.tests.ui.pageobject.datasources.datasource_page import (
+    DataSourcePage,
+)
+from codemie_test_harness.tests.utils.base_utils import get_random_name
+from tests import TEST_USER
+from tests.ui._test_data.datasource_test_data import (
+    MAIN_TITLE_DATASOURCE,
+    MAIN_SUBTITLE_DATASOURCE,
+)
+
+
+@pytest.mark.datasource_ui
+@pytest.mark.ui
+def test_datasource_sidebar_interaction(page):
+    """Test sidebar filters interactions."""
+    search_input = get_random_name()
+    datasource_page = DataSourcePage(page)
+
+    datasource_page.navigate_to()
+    datasource_page.sidebar.should_see_title_subtitle(
+        MAIN_TITLE_DATASOURCE, MAIN_SUBTITLE_DATASOURCE
+    )
+    datasource_page.sidebar.input_data_search(search_input).should_see_search_input(
+        search_input
+    )
+    datasource_page.sidebar.should_see_selected_checkboxes()
+    datasource_page.sidebar.toggle_created_by_me().should_see_created_by_me_value(
+        TEST_USER
+    )
+    datasource_page.sidebar.should_see_project_filter()
+    datasource_page.sidebar.should_see_status_dropdown()
+    datasource_page.sidebar.should_select_status_dropdown()
+    datasource_page.sidebar.click_clear_all_button().should_see_cleared_filters()
+
+
+@pytest.mark.datasource_ui
+@pytest.mark.ui
+def test_datasource_sidebar_hiding_elements(page):
+    """Test sidebar filters hiding."""
+    datasource_page = DataSourcePage(page)
+
+    datasource_page.navigate_to()
+    datasource_page.sidebar.click_type_filter_hide_button().should_not_see_type_filters()
+    datasource_page.sidebar.click_project_filter_hide_button().should_not_see_project_filters()
+    datasource_page.sidebar.click_created_by_filter_hide_button().should_not_see_created_by_filters()
+    datasource_page.sidebar.click_status_filter_hide_button().should_not_see_status_filters()
+
+
+@pytest.mark.datasource_ui
+@pytest.mark.ui
+def test_datasource_page_elements_visibility(page):
+    """Test DataSource index page and sidebar elements are visible."""
+    datasource_page = DataSourcePage(page)
+
+    datasource_page.navigate_to()
+    datasource_page.should_see_create_datasource_button()
+    datasource_page.should_see_table_rows(minimum_count=10)
+    datasource_page.should_see_table_column_names()
+    datasource_page.should_see_pagination()

codemie_test_harness/tests/ui/datasource/test_edit_datasource.py

@@ -0,0 +1,180 @@
+import pytest
+from tests import PROJECT, TEST_USER
+from tests.test_data.google_datasource_test_data import GOOGLE_DOC_URL
+from tests.ui._test_data.datasource_test_data import (
+    DataSourceStatus,
+    DataSourceFilterType,
+    UPDATE_SUBTITLE_DATASOURCE,
+    UPDATE_TITLE_DATASOURCE,
+)
+from tests.ui.pageobject.datasources.create_edit_datasource_page import (
+    CreateEditDatasourcePage,
+)
+from tests.ui.pageobject.datasources.datasource_page import DataSourcePage
+from tests.utils.constants import FILES_PATH
+
+
+@pytest.mark.datasource_ui
+@pytest.mark.ui
+def test_edit_git_datasource(
+    page, datasource_utils, git_integration, default_embedding_llm
+):
+    """Test that all main Edit Datasource page elements are visible."""
+    datasource_page = DataSourcePage(page)
+    edit_page = CreateEditDatasourcePage(page)
+
+    datasource = datasource_utils.create_gitlab_datasource(
+        setting_id=git_integration.id,
+        embeddings_model=default_embedding_llm.base_name,
+    )
+
+    datasource_page.navigate_to()
+    datasource_page.should_see_table_row_with_values(
+        name=datasource.name,
+        status=DataSourceStatus.COMPLETED,
+        project=PROJECT,
+        type_=DataSourceFilterType.CODE,
+        created_by=TEST_USER,
+    )
+
+    datasource_page.open_row_action_menu(datasource.name)
+    datasource_page.should_see_edit_dropdown_values()
+    datasource_page.should_see_edit_dropdown_full_reindex_value()
+    datasource_page.click_row_action("Edit")
+
+    edit_page.sidebar.should_see_title_subtitle(
+        UPDATE_TITLE_DATASOURCE, UPDATE_SUBTITLE_DATASOURCE
+    )
+    edit_page.selector.should_see_disabled_multiselect(PROJECT)
+    edit_page.should_see_disabled_name_input(datasource.name)
+    edit_page.should_see_save_reindex_button()
+
+
+@pytest.mark.datasource_ui
+@pytest.mark.ui
+def test_edit_confluence_datasource(page, datasource_utils, confluence_integration):
+    """Test that all main Edit Datasource page elements are visible."""
+    datasource_page = DataSourcePage(page)
+    edit_page = CreateEditDatasourcePage(page)
+
+    datasource = datasource_utils.create_confluence_datasource(
+        setting_id=confluence_integration.id,
+    )
+
+    datasource_page.navigate_to()
+    datasource_page.should_see_table_row_with_values(
+        name=datasource.name,
+        status=DataSourceStatus.COMPLETED,
+        project=PROJECT,
+        type_=DataSourceFilterType.CONFLUENCE,
+        created_by=TEST_USER,
+    )
+
+    datasource_page.open_row_action_menu(datasource.name)
+    datasource_page.should_see_edit_dropdown_values()
+    datasource_page.should_see_edit_dropdown_full_reindex_value()
+    datasource_page.click_row_action("Edit")
+
+    edit_page.sidebar.should_see_title_subtitle(
+        UPDATE_TITLE_DATASOURCE, UPDATE_SUBTITLE_DATASOURCE
+    )
+    edit_page.selector.should_see_disabled_multiselect(PROJECT)
+    edit_page.should_see_disabled_name_input(datasource.name)
+    edit_page.should_see_save_reindex_button()
+
+
+@pytest.mark.datasource_ui
+@pytest.mark.ui
+def test_edit_jira_datasource(page, datasource_utils, jira_integration):
+    """Test that all main Edit Datasource page elements are visible."""
+    datasource_page = DataSourcePage(page)
+    edit_page = CreateEditDatasourcePage(page)
+
+    datasource = datasource_utils.create_jira_datasource(
+        setting_id=jira_integration.id,
+    )
+
+    datasource_page.navigate_to()
+    datasource_page.should_see_table_row_with_values(
+        name=datasource.name,
+        status=DataSourceStatus.COMPLETED,
+        project=PROJECT,
+        type_=DataSourceFilterType.JIRA,
+        created_by=TEST_USER,
+    )
+
+    datasource_page.open_row_action_menu(datasource.name)
+    datasource_page.should_see_edit_dropdown_values()
+    datasource_page.should_see_edit_dropdown_index_values()
+    datasource_page.click_row_action("Edit")
+
+    edit_page.sidebar.should_see_title_subtitle(
+        UPDATE_TITLE_DATASOURCE, UPDATE_SUBTITLE_DATASOURCE
+    )
+    edit_page.selector.should_see_disabled_multiselect(PROJECT)
+    edit_page.should_see_disabled_name_input(datasource.name)
+    edit_page.should_see_save_reindex_button()
+
+
+@pytest.mark.datasource_ui
+@pytest.mark.ui
+def test_edit_file_datasource(page, datasource_utils):
+    """Test that all main Edit Datasource page elements are visible."""
+    datasource_page = DataSourcePage(page)
+    edit_page = CreateEditDatasourcePage(page)
+
+    datasource = datasource_utils.create_file_datasource(
+        files=[str(FILES_PATH / "test.txt")],
+    )
+
+    datasource_page.navigate_to()
+    datasource_page.should_see_table_row_with_values(
+        name=datasource.name,
+        status=DataSourceStatus.COMPLETED,
+        project=PROJECT,
+        type_=DataSourceFilterType.FILE,
+        created_by=TEST_USER,
+    )
+
+    datasource_page.open_row_action_menu(datasource.name)
+    datasource_page.should_see_edit_dropdown_values()
+    datasource_page.click_row_action("Edit")
+
+    edit_page.sidebar.should_see_title_subtitle(
+        UPDATE_TITLE_DATASOURCE, UPDATE_SUBTITLE_DATASOURCE
+    )
+    edit_page.selector.should_see_disabled_multiselect(PROJECT)
+    edit_page.should_see_disabled_name_input(datasource.name)
+
+
+@pytest.mark.datasource_ui
+@pytest.mark.ui
+def test_edit_google_datasource(page, datasource_utils):
+    """Test that all main Edit Datasource page elements are visible."""
+    datasource_page = DataSourcePage(page)
+    edit_page = CreateEditDatasourcePage(page)
+
+    datasource = datasource_utils.create_google_doc_datasource(
+        google_doc=GOOGLE_DOC_URL
+    )
+
+    datasource_page.navigate_to()
+    datasource_page.should_see_table_row_with_values(
+        name=datasource.name,
+        status=DataSourceStatus.COMPLETED,
+        project=PROJECT,
+        type_=DataSourceFilterType.GOOGLE,
+        created_by=TEST_USER,
+    )
+
+    datasource_page.open_row_action_menu(datasource.name)
+    datasource_page.should_see_edit_dropdown_values()
+    datasource_page.should_see_edit_dropdown_full_reindex_value()
+    datasource_page.click_row_action("Edit")
+
+    edit_page.sidebar.should_see_title_subtitle(
+        UPDATE_TITLE_DATASOURCE, UPDATE_SUBTITLE_DATASOURCE
+    )
+    edit_page.selector.should_see_disabled_multiselect(PROJECT)
+    edit_page.should_see_disabled_name_input(datasource.name)
+    edit_page.should_see_save_reindex_button()