from django.test import TestCase
from django.urls import reverse
from model_bakery import baker


class AuthenticationTestCase(TestCase):
    @classmethod
    def setUpTestData(cls):
        cls.project = baker.make("projects.Project")
        cls.project_key = cls.project.projectkey_set.first()
        cls.organization = cls.project.organization

    def setUp(self):
        self.url = (
            reverse("api:event_envelope", args=[self.project.id])
            + f"?sentry_key={self.project_key.public_key}"
        )

    def test_org_throttle(self):
        res = self.client.post(self.url, [{}], content_type="application/json")
        self.assertEqual(res.status_code, 200)
        self.organization.event_throttle_rate = 100
        self.organization.save()
        res = self.client.post(self.url, [{}], content_type="application/json")
        self.assertEqual(res.headers.get("Retry-After"), "600")
        self.assertEqual(res.status_code, 429)