Skip to content

Commit

Permalink
Add with_environment_permissions
Browse files Browse the repository at this point in the history
  • Loading branch information
zachaysan committed Nov 6, 2023
1 parent 5ad077f commit c6d00ec
Showing 1 changed file with 18 additions and 0 deletions.
18 changes: 18 additions & 0 deletions api/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -176,6 +176,24 @@ def environment(project):
return Environment.objects.create(name="Test Environment", project=project)


@pytest.fixture()
def with_environment_permissions(
environment: Environment, staff_user: FFAdminUser
) -> typing.Callable:
def _with_environment_permissions(
permission_keys: list[str], environment_id: typing.Optional[int] = None
) -> UserEnvironmentPermission:
environment_id = environment_id or environment.id
uep, __ = UserEnvironmentPermission.objects.get_or_create(
environment_id=environment_id, user=staff_user
)
uep.permissions.add(*permission_keys)

return uep

return _with_environment_permissions


@pytest.fixture()
def identity(environment):
return Identity.objects.create(identifier="test_identity", environment=environment)
Expand Down

0 comments on commit c6d00ec

Please sign in to comment.