Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

fix: Missing permissions for LaunchDarkly view #4531

Merged
merged 3 commits into from
Aug 22, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
28 changes: 19 additions & 9 deletions api/integrations/launch_darkly/views.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@
from drf_yasg.utils import swagger_auto_schema
from rest_framework import mixins, status, viewsets
from rest_framework.exceptions import ValidationError
from rest_framework.permissions import BasePermission
from rest_framework.request import Request
from rest_framework.response import Response

Expand All @@ -16,7 +17,12 @@
from integrations.launch_darkly.tasks import (
process_launch_darkly_import_request,
)
from projects.permissions import CREATE_ENVIRONMENT, VIEW_PROJECT
from projects.models import Project
from projects.permissions import (
CREATE_ENVIRONMENT,
VIEW_PROJECT,
NestedProjectPermissions,
)


class LaunchDarklyImportRequestViewSet(
Expand All @@ -29,14 +35,21 @@ class LaunchDarklyImportRequestViewSet(
pagination_class = None # set here to ensure documentation is correct
model_class = LaunchDarklyImportRequest

def get_permissions(self) -> list[BasePermission]:
return [
NestedProjectPermissions(
action_permission_map={
"retrieve": VIEW_PROJECT,
"create": CREATE_ENVIRONMENT,
}
),
]

def get_queryset(self) -> QuerySet[LaunchDarklyImportRequest]:
if getattr(self, "swagger_fake_view", False):
return self.model_class.objects.none()

project = get_object_or_404(
self.request.user.get_permitted_projects(VIEW_PROJECT),
pk=self.kwargs["project_pk"],
)
project = get_object_or_404(Project, pk=self.kwargs["project_pk"])
return self.model_class.objects.filter(project=project)

@swagger_auto_schema(
Expand All @@ -49,10 +62,7 @@ def create(self, request: Request, *args, **kwargs) -> Response:
)
request_serializer.is_valid(raise_exception=True)

project = get_object_or_404(
self.request.user.get_permitted_projects(CREATE_ENVIRONMENT),
pk=self.kwargs["project_pk"],
)
project = get_object_or_404(Project, pk=self.kwargs["project_pk"])

try:
instance = create_import_request(
Expand Down
34 changes: 33 additions & 1 deletion api/tests/unit/integrations/launch_darkly/test_views.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,39 @@ def test_launch_darkly_import_request_view__list__wrong_project__return_expected
response = api_client.get(url)

# Then
assert response.status_code == status.HTTP_404_NOT_FOUND
assert response.status_code == status.HTTP_403_FORBIDDEN


def test_launch_darkly_import_request_view__list__return_expected(
import_request: LaunchDarklyImportRequest,
project: Project,
admin_client: APIClient,
mocker: MockerFixture,
) -> None:
# Given
url = reverse("api-v1:projects:imports-launch-darkly-list", args=[project.id])

# When
response = admin_client.get(url)

# Then
assert response.status_code == status.HTTP_200_OK
assert response.json() == [
{
"completed_at": None,
"created_at": mocker.ANY,
"created_by": "[email protected]",
"id": import_request.id,
"project": project.id,
"status": {
"error_messages": [],
"requested_environment_count": 2,
"requested_flag_count": 9,
"result": None,
},
"updated_at": mocker.ANY,
},
]


def test_launch_darkly_import_request_view__create__return_expected(
Expand Down
Loading