Jedd Morgan
c1f27b78f9
* Add model permission checks * test_public * This is the real fix * mistake * public api resource
143 lines
5.1 KiB
Python
143 lines
5.1 KiB
Python
import pytest
|
|
|
|
from specklepy.api.client import SpeckleClient
|
|
from specklepy.core.api.enums import ProjectVisibility
|
|
from specklepy.core.api.inputs.project_inputs import (
|
|
ProjectCreateInput,
|
|
ProjectUpdateInput,
|
|
)
|
|
from specklepy.core.api.models import Project
|
|
from specklepy.core.api.models.current import ProjectPermissionChecks
|
|
from specklepy.logging.exceptions import GraphQLException
|
|
|
|
|
|
@pytest.mark.run()
|
|
class TestProjectResource:
|
|
@pytest.fixture()
|
|
def test_project(self, client: SpeckleClient) -> Project:
|
|
project = client.project.create(
|
|
ProjectCreateInput(
|
|
name="test project123",
|
|
description="desc",
|
|
visibility=ProjectVisibility.PRIVATE,
|
|
)
|
|
)
|
|
return project
|
|
|
|
@pytest.fixture()
|
|
def test_public_project(self, client: SpeckleClient) -> Project:
|
|
project = client.project.create(
|
|
ProjectCreateInput(
|
|
name="test project123",
|
|
description="desc",
|
|
visibility=ProjectVisibility.PUBLIC,
|
|
)
|
|
)
|
|
return project
|
|
|
|
@pytest.mark.parametrize(
|
|
"name, description, visibility",
|
|
[
|
|
("Very private project", "My secret project", ProjectVisibility.PRIVATE),
|
|
("Very discoverable project", None, ProjectVisibility.UNLISTED),
|
|
("Very public project", None, ProjectVisibility.PUBLIC),
|
|
],
|
|
)
|
|
def test_project_create(
|
|
self,
|
|
client: SpeckleClient,
|
|
name: str,
|
|
description: str,
|
|
visibility: ProjectVisibility,
|
|
):
|
|
input = ProjectCreateInput(
|
|
name=name,
|
|
description=description,
|
|
visibility=visibility,
|
|
)
|
|
result = client.project.create(input)
|
|
|
|
assert isinstance(result, Project)
|
|
assert result.id is not None
|
|
assert result.name == name
|
|
assert result.description == (description or "")
|
|
# we've disabled creation of unlisted projects for now, they fall back to public
|
|
if visibility == ProjectVisibility.UNLISTED:
|
|
assert result.visibility == ProjectVisibility.PUBLIC
|
|
else:
|
|
assert result.visibility == visibility
|
|
|
|
def test_project_get(self, client: SpeckleClient, test_project: Project):
|
|
result = client.project.get(test_project.id)
|
|
|
|
assert isinstance(result, Project)
|
|
assert result.id == test_project.id
|
|
assert result.name == test_project.name
|
|
assert result.description == test_project.description
|
|
assert result.visibility == test_project.visibility
|
|
assert result.created_at == test_project.created_at
|
|
|
|
def test_project_get_permissions(
|
|
self,
|
|
client: SpeckleClient,
|
|
second_client: SpeckleClient,
|
|
test_project: Project,
|
|
test_public_project: Project,
|
|
):
|
|
result_private = client.project.get_permissions(test_project.id)
|
|
|
|
assert isinstance(result_private, ProjectPermissionChecks)
|
|
assert result_private.can_create_model.authorized is True
|
|
assert result_private.can_delete.authorized is True
|
|
assert result_private.can_load.authorized is True
|
|
|
|
result = client.project.get_permissions(test_public_project.id)
|
|
|
|
assert isinstance(result, ProjectPermissionChecks)
|
|
assert result.can_create_model.authorized is True
|
|
assert result.can_delete.authorized is True
|
|
assert result.can_load.authorized is True
|
|
|
|
guest = second_client.project.get_permissions(test_public_project.id)
|
|
|
|
assert isinstance(result, ProjectPermissionChecks)
|
|
assert guest.can_create_model.authorized is False
|
|
assert guest.can_delete.authorized is False
|
|
assert guest.can_load.authorized is False
|
|
|
|
def test_project_update(self, client: SpeckleClient, test_project: Project):
|
|
new_name = "MY new name"
|
|
new_description = "MY new desc"
|
|
new_visibility = ProjectVisibility.UNLISTED
|
|
|
|
update_data = ProjectUpdateInput(
|
|
id=test_project.id,
|
|
name=new_name,
|
|
description=new_description,
|
|
visibility=new_visibility,
|
|
)
|
|
|
|
updated_project = client.project.update(update_data)
|
|
|
|
assert isinstance(updated_project, Project)
|
|
assert updated_project.id == test_project.id
|
|
assert updated_project.name == new_name
|
|
assert updated_project.description == new_description
|
|
# we've disabled creation of public projects for now, they fall back to unlisted
|
|
if new_visibility == ProjectVisibility.UNLISTED:
|
|
assert updated_project.visibility == ProjectVisibility.PUBLIC
|
|
else:
|
|
assert updated_project.visibility == new_visibility
|
|
|
|
def test_project_delete(self, client: SpeckleClient):
|
|
"""Test deleting a project."""
|
|
project_to_delete = client.project.create(
|
|
ProjectCreateInput(name="Delete me", description=None, visibility=None)
|
|
)
|
|
|
|
response = client.project.delete(project_to_delete.id)
|
|
assert response is True
|
|
|
|
with pytest.raises(GraphQLException):
|
|
client.project.get(project_to_delete.id)
|