sentinel/tests/sentry/api/serializers/test_saved_search.py

from sentry.api.serializers import serialize
from sentry.models.savedsearch import SavedSearch, Visibility
from sentry.testutils.cases import TestCase


class SavedSearchSerializerTest(TestCase):
    def test_simple(self):
        search = SavedSearch.objects.create(name="Something", query="some query")
        result = serialize(search)

        assert result["id"] == str(search.id)
        assert result["type"] == search.type
        assert result["name"] == search.name
        assert result["query"] == search.query
        assert result["visibility"] == Visibility.OWNER
        assert result["dateCreated"] == search.date_added
        assert not result["isGlobal"]
        assert not result["isPinned"]

    def test_global(self):
        search = SavedSearch(
            name="Unresolved Issues",
            query="is:unresolved",
            is_global=True,
            visibility=Visibility.ORGANIZATION,
        )
        result = serialize(search)

        assert result["id"] == str(search.id)
        assert result["type"] == search.type
        assert result["name"] == search.name
        assert result["query"] == search.query
        assert result["visibility"] == Visibility.ORGANIZATION
        assert result["dateCreated"] == search.date_added
        assert result["isGlobal"]
        assert not result["isPinned"]

    def test_organization(self):
        search = SavedSearch.objects.create(
            organization=self.organization,
            name="Something",
            query="some query",
            visibility=Visibility.ORGANIZATION,
        )
        result = serialize(search)

        assert result["id"] == str(search.id)
        assert result["type"] == search.type
        assert result["name"] == search.name
        assert result["query"] == search.query
        assert result["visibility"] == Visibility.ORGANIZATION
        assert result["dateCreated"] == search.date_added
        assert not result["isGlobal"]
        assert not result["isPinned"]

    def test_pinned(self):
        search = SavedSearch.objects.create(
            organization=self.organization,
            owner_id=self.user.id,
            name="Something",
            query="some query",
            visibility=Visibility.OWNER_PINNED,
        )
        result = serialize(search)

        assert result["id"] == str(search.id)
        assert result["type"] == search.type
        assert result["name"] == search.name
        assert result["query"] == search.query
        assert result["visibility"] == Visibility.OWNER_PINNED
        assert result["dateCreated"] == search.date_added
        assert not result["isGlobal"]
        assert result["isPinned"]