123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298 |
- from __future__ import absolute_import
- from sentry.testutils import APITestCase, SnubaTestCase
- from django.core.urlresolvers import reverse
- from sentry.discover.models import DiscoverSavedQuery
- from sentry.testutils.helpers.datetime import before_now
- class DiscoverSavedQueryBase(APITestCase, SnubaTestCase):
- def setUp(self):
- super(DiscoverSavedQueryBase, self).setUp()
- self.login_as(user=self.user)
- self.org = self.create_organization(owner=self.user)
- self.project_ids = [
- self.create_project(organization=self.org).id,
- self.create_project(organization=self.org).id,
- ]
- self.project_ids_without_access = [self.create_project().id]
- query = {"fields": ["test"], "conditions": [], "limit": 10}
- model = DiscoverSavedQuery.objects.create(
- organization=self.org, created_by=self.user, name="Test query", query=query, version=1
- )
- model.set_projects(self.project_ids)
- class DiscoverSavedQueriesTest(DiscoverSavedQueryBase):
- feature_name = "organizations:discover"
- def test_get(self):
- url = reverse("sentry-api-0-discover-saved-queries", args=[self.org.slug])
- with self.feature(self.feature_name):
- response = self.client.get(url)
- assert response.status_code == 200, response.content
- assert len(response.data) == 1
- assert response.data[0]["name"] == "Test query"
- assert response.data[0]["projects"] == self.project_ids
- assert response.data[0]["fields"] == ["test"]
- assert response.data[0]["conditions"] == []
- assert response.data[0]["limit"] == 10
- assert response.data[0]["version"] == 1
- def test_get_version_filter(self):
- url = reverse("sentry-api-0-discover-saved-queries", args=[self.org.slug])
- with self.feature(self.feature_name):
- response = self.client.get(url, format="json", data={"query": "version:1"})
- assert response.status_code == 200, response.content
- assert len(response.data) == 1
- assert response.data[0]["name"] == "Test query"
- with self.feature(self.feature_name):
- response = self.client.get(url, format="json", data={"query": "version:2"})
- assert response.status_code == 200, response.content
- assert len(response.data) == 0
- def test_get_name_filter(self):
- url = reverse("sentry-api-0-discover-saved-queries", args=[self.org.slug])
- with self.feature(self.feature_name):
- response = self.client.get(url, format="json", data={"query": "Test"})
- assert response.status_code == 200, response.content
- assert len(response.data) == 1
- assert response.data[0]["name"] == "Test query"
- with self.feature(self.feature_name):
- # Also available as the name: filter.
- response = self.client.get(url, format="json", data={"query": "name:Test"})
- assert response.status_code == 200, response.content
- assert len(response.data) == 1
- assert response.data[0]["name"] == "Test query"
- with self.feature(self.feature_name):
- response = self.client.get(url, format="json", data={"query": "name:Nope"})
- assert response.status_code == 200, response.content
- assert len(response.data) == 0
- def test_get_all_paginated(self):
- for i in range(0, 10):
- query = {"fields": ["test"], "conditions": [], "limit": 10}
- model = DiscoverSavedQuery.objects.create(
- organization=self.org,
- created_by=self.user,
- name="My query {}".format(i),
- query=query,
- version=1,
- )
- model.set_projects(self.project_ids)
- url = reverse("sentry-api-0-discover-saved-queries", args=[self.org.slug])
- with self.feature(self.feature_name):
- response = self.client.get(url, data={"per_page": 1})
- assert response.status_code == 200, response.content
- assert len(response.data) == 1
- url = reverse("sentry-api-0-discover-saved-queries", args=[self.org.slug])
- with self.feature(self.feature_name):
- # The all parameter ignores pagination and returns all values.
- response = self.client.get(url, data={"per_page": 1, "all": 1})
- assert response.status_code == 200, response.content
- assert len(response.data) == 11
- def test_get_sortby(self):
- query = {"fields": ["message"], "query": "", "limit": 10}
- model = DiscoverSavedQuery.objects.create(
- organization=self.org,
- created_by=self.user,
- name="My query",
- query=query,
- version=2,
- date_created=before_now(minutes=10),
- date_updated=before_now(minutes=10),
- )
- model.set_projects(self.project_ids)
- url = reverse("sentry-api-0-discover-saved-queries", args=[self.org.slug])
- sort_options = {
- "dateCreated": True,
- "-dateCreated": False,
- "dateUpdated": True,
- "-dateUpdated": False,
- "name": True,
- "-name": False,
- }
- for sorting, forward_sort in sort_options.items():
- with self.feature(self.feature_name):
- response = self.client.get(url, data={"sortBy": sorting})
- assert response.status_code == 200
- values = [row[sorting.strip("-")] for row in response.data]
- if not forward_sort:
- values = list(reversed(values))
- assert list(sorted(values)) == values
- def test_post(self):
- with self.feature(self.feature_name):
- url = reverse("sentry-api-0-discover-saved-queries", args=[self.org.slug])
- response = self.client.post(
- url,
- {
- "name": "New query",
- "projects": self.project_ids,
- "fields": [],
- "range": "24h",
- "limit": 20,
- "conditions": [],
- "aggregations": [],
- "orderby": "-time",
- },
- )
- assert response.status_code == 201, response.content
- assert response.data["name"] == "New query"
- assert response.data["projects"] == self.project_ids
- assert response.data["range"] == "24h"
- assert not hasattr(response.data, "start")
- assert not hasattr(response.data, "end")
- def test_post_invalid_projects(self):
- with self.feature(self.feature_name):
- url = reverse("sentry-api-0-discover-saved-queries", args=[self.org.slug])
- response = self.client.post(
- url,
- {
- "name": "New query",
- "projects": self.project_ids_without_access,
- "fields": [],
- "range": "24h",
- "limit": 20,
- "conditions": [],
- "aggregations": [],
- "orderby": "-time",
- },
- )
- assert response.status_code == 403, response.content
- def test_post_all_projects(self):
- with self.feature(self.feature_name):
- url = reverse("sentry-api-0-discover-saved-queries", args=[self.org.slug])
- response = self.client.post(
- url,
- {
- "name": "All projects",
- "projects": [-1],
- "conditions": [],
- "fields": ["title", "count()"],
- "range": "24h",
- "orderby": "time",
- },
- )
- assert response.status_code == 201, response.content
- assert response.data["projects"] == [-1]
- assert response.data["name"] == "All projects"
- def test_post_cannot_use_version_two_fields(self):
- with self.feature(self.feature_name):
- url = reverse("sentry-api-0-discover-saved-queries", args=[self.org.slug])
- response = self.client.post(
- url,
- {
- "name": "New query",
- "projects": self.project_ids,
- "fields": ["id"],
- "range": "24h",
- "limit": 20,
- "environment": ["dev"],
- "yAxis": "count(id)",
- "aggregations": [],
- "orderby": "-time",
- },
- )
- assert response.status_code == 400, response.content
- assert "cannot use the environment, yAxis attribute(s)" in response.content
- class DiscoverSavedQueriesVersion2Test(DiscoverSavedQueryBase):
- feature_name = "organizations:events-v2"
- def test_post_invalid_conditions(self):
- with self.feature(self.feature_name):
- url = reverse("sentry-api-0-discover-saved-queries", args=[self.org.slug])
- response = self.client.post(
- url,
- {
- "name": "New query",
- "projects": self.project_ids,
- "fields": ["title", "count()"],
- "range": "24h",
- "version": 2,
- "conditions": [["field", "=", "value"]],
- },
- )
- assert response.status_code == 400, response.content
- assert "cannot use the conditions attribute(s)" in response.content
- def test_post_require_selected_fields(self):
- with self.feature(self.feature_name):
- url = reverse("sentry-api-0-discover-saved-queries", args=[self.org.slug])
- response = self.client.post(
- url,
- {
- "name": "New query",
- "projects": self.project_ids,
- "fields": [],
- "range": "24h",
- "version": 2,
- },
- )
- assert response.status_code == 400, response.content
- assert "include at least one field" in response.content
- def test_post_success(self):
- with self.feature(self.feature_name):
- url = reverse("sentry-api-0-discover-saved-queries", args=[self.org.slug])
- response = self.client.post(
- url,
- {
- "name": "new query",
- "projects": self.project_ids,
- "fields": ["title", "count()", "project"],
- "environment": ["dev"],
- "query": "event.type:error browser.name:Firefox",
- "range": "24h",
- "tags": ["release", "environment"],
- "yAxis": "count(id)",
- "version": 2,
- },
- )
- assert response.status_code == 201, response.content
- data = response.data
- assert data["fields"] == ["title", "count()", "project"]
- assert data["range"] == "24h"
- assert data["environment"] == ["dev"]
- assert data["query"] == "event.type:error browser.name:Firefox"
- assert data["tags"] == ["release", "environment"]
- assert data["yAxis"] == "count(id)"
- assert data["version"] == 2
- def test_post_all_projects(self):
- with self.feature(self.feature_name):
- url = reverse("sentry-api-0-discover-saved-queries", args=[self.org.slug])
- response = self.client.post(
- url,
- {
- "name": "New query",
- "projects": [-1],
- "fields": ["title", "count()"],
- "range": "24h",
- "version": 2,
- },
- )
- assert response.status_code == 201, response.content
- assert response.data["projects"] == [-1]
|