sentry/tests/acceptance/test_issue_saved_searches.py

# from datetime import datetime
# from unittest.mock import patch

from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.wait import WebDriverWait

from fixtures.page_objects.issue_list import IssueListPage
from sentry.models import Visibility
from sentry.models.savedsearch import SavedSearch, SortOptions
from sentry.testutils import AcceptanceTestCase, SnubaTestCase
from sentry.testutils.silo import region_silo_test

SAVED_SEARCH_FEATURES = {"organizations:issue-list-saved-searches-v2": True}


@region_silo_test
class OrganizationGroupIndexTest(AcceptanceTestCase, SnubaTestCase):
    def setUp(self):
        super().setUp()
        self.user = self.create_user("foo@example.com")
        self.org = self.create_organization(owner=self.user, name="Rowdy Tiger")
        self.team = self.create_team(
            organization=self.org, name="Mariachi Band", members=[self.user]
        )
        self.project = self.create_project(organization=self.org, teams=[self.team], name="Bengal")
        self.other_project = self.create_project(
            organization=self.org, teams=[self.team], name="Sumatra"
        )
        self.login_as(self.user)
        self.page = IssueListPage(self.browser, self.client)
        self.dismiss_assistant()

        # Create recommended saved searches
        self.create_saved_search(
            name="Assigned to Me",
            query="is:unresolved assigned:me",
            visibility=Visibility.ORGANIZATION,
            is_global=True,
        )
        self.create_saved_search(
            name="Errors Only",
            query="is:unresolved evel:error",
            visibility=Visibility.ORGANIZATION,
            is_global=True,
        )

    def test_click_saved_search(self):
        with self.feature(SAVED_SEARCH_FEATURES):
            self.page.visit_issue_list(self.org.slug)
            self.browser.click_when_visible('button[aria-label="Saved Searches"]')

            # Navigate to a recommended saved search
            self.browser.click('button[aria-label="Errors Only"]')
            self.page.wait_until_loaded()

            self.browser.snapshot("issue list after navigating to saved search")

    def test_create_saved_search(self):
        with self.feature(SAVED_SEARCH_FEATURES):
            self.page.visit_issue_list(self.org.slug)
            self.browser.click_when_visible('button[aria-label="Saved Searches"]')
            self.browser.snapshot("issue list with no saved searches")

            self.browser.click('[aria-label="Create a new saved search"]')

            self.browser.wait_until('[role="dialog"]')
            self.browser.snapshot("create saved search modal open")

            self.browser.find_element(by=By.NAME, value="name").send_keys("My Saved Search")
            query_input = self.browser.find_element(
                by=By.CSS_SELECTOR, value='[role="dialog"] textarea'
            )
            self.browser.click('[role="dialog"] button[aria-label="Clear search"]')
            query_input.send_keys("browser.name:Firefox", Keys.ENTER)
            self.browser.click('[role="dialog"] button[aria-label="Save"]')
            self.browser.wait_until_not('[data-test-id="loading-indicator"]')

            # The saved search should have been created with the correct options
            created_search = SavedSearch.objects.get(name="My Saved Search")
            assert created_search
            assert created_search.query == "browser.name:Firefox"
            assert created_search.sort == SortOptions.DATE
            assert created_search.visibility == Visibility.OWNER
            assert not created_search.is_global
            assert created_search.owner == self.user

            # And the sidebar should have been updated with the new search item
            assert self.browser.find_element(
                by=By.CSS_SELECTOR, value='button[aria-label="My Saved Search"]'
            )

    def test_edit_saved_search(self):
        with self.feature(SAVED_SEARCH_FEATURES):
            # Start with a user saved search
            self.create_saved_search(
                organization=self.org,
                name="My Saved Search",
                query="browser.name:Firefox",
                visibility=Visibility.OWNER,
                owner=self.user,
            )

            self.page.visit_issue_list(self.org.slug)
            self.browser.click_when_visible('button[aria-label="Saved Searches"]')

            self.browser.move_to('button[aria-label="My Saved Search"]')
            self.browser.wait_until_clickable('button[aria-label="Saved search options"]')
            self.browser.click('button[aria-label="Saved search options"]')
            self.browser.click('[data-test-id="edit"]')

            self.browser.wait_until('[role="dialog"]')
            self.browser.snapshot("edit saved search modal open")

            self.browser.find_element(by=By.NAME, value="name").clear()
            self.browser.find_element(by=By.NAME, value="name").send_keys("New Saved Search Name")
            self.browser.click('[role="dialog"] button[aria-label="Save"]')
            self.browser.wait_until_not('[data-test-id="loading-indicator"]')

            # The saved search should have been updated with the correct options
            created_search = SavedSearch.objects.get(name="New Saved Search Name")
            assert created_search
            assert created_search.query == "browser.name:Firefox"
            assert created_search.sort == SortOptions.DATE
            assert created_search.visibility == Visibility.OWNER
            assert not created_search.is_global
            assert created_search.owner == self.user

            # And the sidebar should have been updated
            assert self.browser.find_element(
                by=By.CSS_SELECTOR, value='button[aria-label="New Saved Search Name"]'
            )

    def test_delete_saved_search(self):
        with self.feature(SAVED_SEARCH_FEATURES):
            # Start with a user saved search
            self.create_saved_search(
                organization=self.org,
                name="My Saved Search",
                query="browser.name:Firefox",
                visibility=Visibility.OWNER,
                owner=self.user,
            )

            self.page.visit_issue_list(self.org.slug)
            self.browser.click_when_visible('button[aria-label="Saved Searches"]')

            self.browser.move_to('button[aria-label="My Saved Search"]')
            self.browser.wait_until_clickable('button[aria-label="Saved search options"]')
            self.browser.click('button[aria-label="Saved search options"]')
            self.browser.click('[data-test-id="delete"]')

            self.browser.wait_until('[role="dialog"]')
            self.browser.click('[role="dialog"] button[aria-label="Confirm"]')

            # Search is immediately removed from the UI
            assert not self.browser.element_exists('button[aria-label="My Saved Search"]')

            # The saved search should be removed from the db
            # Since this is an optimistic update there is nothing to wait for in the UI
            wait = WebDriverWait(self.browser.driver, 10)
            wait.until(lambda _: not SavedSearch.objects.filter(name="My Saved Search").exists())