oncall-engine/engine/apps/public_api/tests/test_user_groups.py
Joey Orlando 767c5352fa
augment API response pagination attributes (#2471)
# What this PR does

This PR:
- adds a few attributes to paginated API responses
- removes channel filter "send demo alert" internal API endpoint + tests
(this endpoint was marked as deprecated + not consumed by the web UI)

With the new paginated API response schema, the web UI will no longer
need to:
- hardcode `ITEMS_PER_PAGE` for each table
- manually calculate total number of pages

(these two things ☝️ will be done in
https://github.com/grafana/oncall/issues/2476)

For `GET /api/internal/v1/alertgroups` the response will now look like
this:
```diff
{
    "next": <url> | None,
    "previous": <url> | None,
    "results": [],
++  "page_size": <int>
}
```

For all other paginated API responses, the response will now look like:
```diff
{
    "count": <int>,
    "next": <url> | None,
    "previous": <url> | None,
    "results": [],
++  "page_size": <int>,
++  "current_page_number": <int>,
++  "total_pages": <int>
}
```

## TODO
- [x] update public API docs to include these new attributes

## Checklist

- [x] Unit, integration, and e2e (if applicable) tests updated
- [x] Documentation added (or `pr:no public docs` PR label added if not
required)
- [x] `CHANGELOG.md` updated (or `pr:no changelog` PR label added if not
required)
2023-07-14 11:19:40 -04:00

118 lines
3.5 KiB
Python

import pytest
from django.urls import reverse
from rest_framework import status
from rest_framework.test import APIClient
@pytest.fixture()
def slack_user_group_public_api_setup(
make_organization_and_user_with_slack_identities,
make_public_api_token,
make_slack_user_group,
):
organization, user, slack_team_identity, slack_user_identity = make_organization_and_user_with_slack_identities()
_, token = make_public_api_token(user, organization)
slack_user_group = make_slack_user_group(slack_team_identity, slack_id="SLACK_GROUP_ID")
return organization, user, token, slack_team_identity, slack_user_identity, slack_user_group
@pytest.mark.django_db
def test_get_user_groups(
slack_user_group_public_api_setup,
):
_, _, token, _, _, slack_user_group = slack_user_group_public_api_setup
client = APIClient()
url = reverse("api-public:user_groups-list")
response = client.get(url, format="json", HTTP_AUTHORIZATION=f"{token}")
expected_payload = {
"count": 1,
"next": None,
"previous": None,
"results": [
{
"id": slack_user_group.public_primary_key,
"type": "slack_based",
"slack": {
"id": slack_user_group.slack_id,
"name": slack_user_group.name,
"handle": slack_user_group.handle,
},
}
],
"current_page_number": 1,
"page_size": 50,
"total_pages": 1,
}
assert response.status_code == status.HTTP_200_OK
assert response.json() == expected_payload
@pytest.mark.django_db
def test_get_user_groups_filter_by_handle(
slack_user_group_public_api_setup,
make_slack_user_group,
):
_, _, token, slack_team_identity, slack_user_identity, slack_user_group_1 = slack_user_group_public_api_setup
client = APIClient()
make_slack_user_group(slack_team_identity, slack_id="SLACK_GROUP_ID_2")
url = reverse("api-public:user_groups-list")
response = client.get(
f"{url}?slack_handle={slack_user_group_1.handle}", format="json", HTTP_AUTHORIZATION=f"{token}"
)
expected_payload = {
"count": 1,
"next": None,
"previous": None,
"results": [
{
"id": slack_user_group_1.public_primary_key,
"type": "slack_based",
"slack": {
"id": slack_user_group_1.slack_id,
"name": slack_user_group_1.name,
"handle": slack_user_group_1.handle,
},
}
],
"current_page_number": 1,
"page_size": 50,
"total_pages": 1,
}
assert response.status_code == status.HTTP_200_OK
assert response.json() == expected_payload
@pytest.mark.django_db
def test_get_user_groups_filter_by_handle_empty_result(
slack_user_group_public_api_setup,
):
_, _, token, slack_team_identity, _, slack_user_group = slack_user_group_public_api_setup
client = APIClient()
url = reverse("api-public:user_groups-list")
response = client.get(f"{url}?slack_handle=NonExistentSlackHandle", format="json", HTTP_AUTHORIZATION=f"{token}")
expected_payload = {
"count": 0,
"next": None,
"previous": None,
"results": [],
"current_page_number": 1,
"page_size": 50,
"total_pages": 1,
}
assert response.status_code == status.HTTP_200_OK
assert response.data == expected_payload