6b40f95033
Co-authored-by: Eve832 <eve.meelan@grafana.com>
Co-authored-by: Francisco Montes de Oca <nevermind89x@gmail.com>
Co-authored-by: Ildar Iskhakov <ildar.iskhakov@grafana.com>
Co-authored-by: Innokentii Konstantinov <innokenty.konstantinov@grafana.com>
Co-authored-by: Julia <ferril.darkdiver@gmail.com>
Co-authored-by: maskin25 <kengurek@gmail.com>
Co-authored-by: Matias Bordese <mbordese@gmail.com>
Co-authored-by: Matvey Kukuy <motakuk@gmail.com>
Co-authored-by: Michael Derynck <michael.derynck@grafana.com>
Co-authored-by: Richard Hartmann <richih@richih.org>
Co-authored-by: Robby Milo <robbymilo@fastmail.com>
Co-authored-by: Timur Olzhabayev <timur.olzhabayev@grafana.com>
Co-authored-by: Vadim Stepanov <vadimkerr@gmail.com>
Co-authored-by: Yulia Shanyrova <yulia.shanyrova@grafana.com>
33 lines
1,014 B
Python
33 lines
1,014 B
Python
from unittest.mock import patch
|
|
|
|
import pytest
|
|
from django.core.cache import cache
|
|
from django.urls import reverse
|
|
from rest_framework import status
|
|
from rest_framework.test import APIClient
|
|
|
|
|
|
@patch("apps.public_api.throttlers.user_throttle.UserThrottle.get_throttle_limits")
|
|
@pytest.mark.django_db
|
|
def test_throttling(mocked_throttle_limits, make_organization_and_user_with_token):
|
|
MAX_REQUESTS = 1
|
|
PERIOD = 360
|
|
|
|
_, _, token = make_organization_and_user_with_token()
|
|
cache.clear()
|
|
|
|
client = APIClient()
|
|
|
|
mocked_throttle_limits.return_value = MAX_REQUESTS, PERIOD
|
|
url = reverse("api-public:alert_groups-list")
|
|
|
|
response = client.get(url, format="json", HTTP_AUTHORIZATION=f"{token}")
|
|
|
|
assert response.status_code == status.HTTP_200_OK
|
|
|
|
response = client.get(url, format="json", HTTP_AUTHORIZATION=f"{token}")
|
|
|
|
assert response.status_code == status.HTTP_429_TOO_MANY_REQUESTS
|
|
|
|
# make sure RateLimitHeadersMixin used
|
|
assert response.has_header("RateLimit-Reset")
|