sentinel/tests/sentry/api/endpoints/test_project_rule_task_details.py

from unittest.mock import patch
from uuid import uuid4

from sentry.testutils.cases import APITestCase
from sentry.testutils.silo import region_silo_test


@region_silo_test
class ProjectRuleTaskDetailsTest(APITestCase):
    endpoint = "sentry-api-0-project-rule-task-details"

    def setUp(self):
        super().setUp()
        self.login_as(user=self.user)

        self.rule = self.project.rule_set.all()[0]
        self.uuid = uuid4().hex

    @patch("sentry.integrations.slack.utils.RedisRuleStatus.get_value")
    def test_status_pending(self, mock_get_value):
        mock_get_value.return_value = {"status": "pending"}
        response = self.get_success_response(self.organization.slug, self.project.slug, self.uuid)
        assert response.data["status"] == "pending"
        assert response.data["rule"] is None

    @patch("sentry.integrations.slack.utils.RedisRuleStatus.get_value")
    def test_status_failed(self, mock_get_value):
        mock_get_value.return_value = {"status": "failed", "error": "This failed"}
        response = self.get_success_response(self.organization.slug, self.project.slug, self.uuid)
        assert response.data["status"] == "failed"
        assert response.data["rule"] is None
        assert response.data["error"] == "This failed"

    @patch("sentry.integrations.slack.utils.RedisRuleStatus.get_value")
    def test_status_success(self, mock_get_value):
        mock_get_value.return_value = {"status": "success", "rule_id": self.rule.id}
        response = self.get_success_response(self.organization.slug, self.project.slug, self.uuid)
        assert response.data["status"] == "success"

        rule_data = response.data["rule"]
        # TODO(meredith): should I check every attribute?
        assert rule_data["id"] == str(self.rule.id)
        assert rule_data["name"] == self.rule.label