mirror of https://github.com/doccano/doccano.git
pythonannotation-tooldatasetsactive-learningtext-annotationdatasetnatural-language-processingdata-labelingmachine-learning
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.
97 lines
3.3 KiB
97 lines
3.3 KiB
from django.urls import reverse
|
|
from rest_framework import status
|
|
from rest_framework.test import APITestCase
|
|
from mixer.backend.django import mixer
|
|
from ..models import *
|
|
|
|
|
|
class TestDocAPI(APITestCase):
|
|
def setUp(self):
|
|
self.username, self.password = 'user', 'pass'
|
|
|
|
def create_user(self):
|
|
user = User.objects.create_user(username=self.username, password=self.password)
|
|
|
|
return user
|
|
|
|
def create_superuser(self):
|
|
user = User.objects.create_superuser(username=self.username,
|
|
password=self.password,
|
|
email='hoge@example.com')
|
|
return user
|
|
|
|
def create_project(self):
|
|
project = mixer.blend('server.Project')
|
|
|
|
return project
|
|
|
|
def create_doc(self):
|
|
doc = mixer.blend('server.Document')
|
|
|
|
return doc
|
|
|
|
def test_get_docs_by_project_user(self):
|
|
"""
|
|
Ensure we can get document objects by project user.
|
|
"""
|
|
user = self.create_user()
|
|
project = self.create_project()
|
|
project.users.add(user)
|
|
doc = self.create_doc()
|
|
project.documents.add(doc)
|
|
url = reverse('docs', args=[project.id])
|
|
|
|
self.client.login(username=self.username, password=self.password)
|
|
response = self.client.get(url, format='json')
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
self.assertIsInstance(response.data, dict)
|
|
self.assertEqual(response.data['count'], 1)
|
|
self.assertEqual(response.data['results'][0]['id'], doc.id)
|
|
|
|
def test_get_docs_by_other(self):
|
|
"""
|
|
Ensure we cannot get label objects by other.
|
|
"""
|
|
user = self.create_user()
|
|
project = self.create_project()
|
|
doc = self.create_doc()
|
|
project.documents.add(doc)
|
|
url = reverse('docs', args=[project.id])
|
|
|
|
self.client.login(username=self.username, password=self.password)
|
|
response = self.client.get(url, format='json')
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN)
|
|
|
|
def test_create_doc_by_admin(self):
|
|
"""
|
|
Ensure we can create a new document object by admin.
|
|
"""
|
|
# user = self.create_superuser()
|
|
# project = self.create_project()
|
|
# project.users.add(user)
|
|
#
|
|
# data = {'text': 'doc', 'labels': []}
|
|
# url = reverse('docs', args=[project.id])
|
|
# self.client.login(username=self.username, password=self.password)
|
|
# response = self.client.post(url, data, format='json')
|
|
#
|
|
# self.assertEqual(response.status_code, status.HTTP_201_CREATED)
|
|
# self.assertEqual(Document.objects.count(), 1)
|
|
# self.assertEqual(Document.objects.get().text, 'doc')
|
|
|
|
def test_create_doc_by_user(self):
|
|
"""
|
|
Ensure we cannot create a new project object by user.
|
|
"""
|
|
user = self.create_user()
|
|
project = self.create_project()
|
|
project.users.add(user)
|
|
|
|
data = {'text': 'doc', 'labels': []}
|
|
url = reverse('docs', args=[project.id])
|
|
self.client.login(username=self.username, password=self.password)
|
|
response = self.client.post(url, data, format='json')
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN)
|