mirror of https://github.com/doccano/doccano.git
pythondatasetsactive-learningtext-annotationdatasetnatural-language-processingdata-labelingmachine-learningannotation-tool
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
118 lines
4.3 KiB
118 lines
4.3 KiB
from rest_framework import status
|
|
from rest_framework.reverse import reverse
|
|
|
|
from api.tests.api.utils import CRUDMixin, prepare_project
|
|
from roles.tests.utils import create_default_roles
|
|
from users.tests.utils import make_user
|
|
|
|
|
|
class TestProjectList(CRUDMixin):
|
|
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
cls.project = prepare_project()
|
|
cls.non_member = make_user()
|
|
cls.url = reverse(viewname='project_list')
|
|
|
|
def test_return_projects_to_member(self):
|
|
for member in self.project.members:
|
|
response = self.assert_fetch(member, status.HTTP_200_OK)
|
|
project = response.data['results'][0]
|
|
self.assertEqual(response.data['count'], 1)
|
|
self.assertEqual(project['id'], self.project.item.id)
|
|
|
|
def test_does_not_return_project_to_non_member(self):
|
|
response = self.assert_fetch(self.non_member, status.HTTP_200_OK)
|
|
self.assertEqual(response.data['count'], 0)
|
|
|
|
|
|
class TestProjectCreate(CRUDMixin):
|
|
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
create_default_roles()
|
|
cls.user = make_user()
|
|
cls.url = reverse(viewname='project_list')
|
|
cls.data = {
|
|
'name': 'example',
|
|
'project_type': 'DocumentClassification',
|
|
'description': 'example',
|
|
'guideline': 'example',
|
|
'resourcetype': 'TextClassificationProject'
|
|
}
|
|
|
|
def test_allows_staff_user_to_create_project(self):
|
|
self.user.is_staff = True
|
|
self.user.save()
|
|
response = self.assert_create(self.user, status.HTTP_201_CREATED)
|
|
self.assertEqual(response.data['name'], self.data['name'])
|
|
|
|
def test_denies_non_staff_user_to_create_project(self):
|
|
self.assert_create(self.user, status.HTTP_403_FORBIDDEN)
|
|
|
|
def test_denies_unauthenticated_user_to_create_project(self):
|
|
self.assert_create(expected=status.HTTP_403_FORBIDDEN)
|
|
|
|
|
|
class TestSequenceLabelingProjectCreation(CRUDMixin):
|
|
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
create_default_roles()
|
|
cls.user = make_user()
|
|
cls.url = reverse(viewname='project_list')
|
|
cls.data = {
|
|
'name': 'example',
|
|
'project_type': 'SequenceLabeling',
|
|
'description': 'example',
|
|
'guideline': 'example',
|
|
'allow_overlapping': True,
|
|
'grapheme_mode': True,
|
|
'resourcetype': 'SequenceLabelingProject'
|
|
}
|
|
|
|
def test_allows_staff_user_to_create_project(self):
|
|
self.user.is_staff = True
|
|
self.user.save()
|
|
response = self.assert_create(self.user, status.HTTP_201_CREATED)
|
|
self.assertEqual(response.data['allow_overlapping'], self.data['allow_overlapping'])
|
|
self.assertEqual(response.data['grapheme_mode'], self.data['grapheme_mode'])
|
|
|
|
|
|
class TestProjectDetailAPI(CRUDMixin):
|
|
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
cls.project = prepare_project('SequenceLabeling')
|
|
cls.non_member = make_user()
|
|
cls.url = reverse(viewname='project_detail', args=[cls.project.item.id])
|
|
cls.data = {'description': 'lorem', 'resourcetype': 'SequenceLabelingProject'}
|
|
|
|
def test_return_project_to_member(self):
|
|
for member in self.project.members:
|
|
response = self.assert_fetch(member, status.HTTP_200_OK)
|
|
self.assertEqual(response.data['id'], self.project.item.id)
|
|
|
|
def test_does_not_return_project_to_non_member(self):
|
|
self.assert_fetch(self.non_member, status.HTTP_403_FORBIDDEN)
|
|
|
|
def test_allows_admin_to_update_project(self):
|
|
response = self.assert_update(self.project.admin, status.HTTP_200_OK)
|
|
self.assertEqual(response.data['description'], self.data['description'])
|
|
|
|
def test_denies_project_staff_to_update_project(self):
|
|
for member in self.project.staffs:
|
|
self.assert_update(member, status.HTTP_403_FORBIDDEN)
|
|
|
|
def test_denies_non_member_to_update_project(self):
|
|
self.assert_update(self.non_member, status.HTTP_403_FORBIDDEN)
|
|
|
|
def test_allows_admin_to_delete_project(self):
|
|
self.assert_delete(self.project.admin, status.HTTP_204_NO_CONTENT)
|
|
|
|
def test_denies_project_staff_to_delete_project(self):
|
|
for member in self.project.staffs:
|
|
self.assert_delete(member, status.HTTP_403_FORBIDDEN)
|
|
|
|
def test_denies_non_member_to_delete_project(self):
|
|
self.assert_delete(self.non_member, status.HTTP_403_FORBIDDEN)
|