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.
79 lines
2.8 KiB
79 lines
2.8 KiB
import os
|
|
from django.urls import reverse
|
|
from django.test import TestCase, Client
|
|
from rest_framework import status
|
|
from mixer.backend.django import mixer
|
|
from ..models import *
|
|
|
|
|
|
class TestUpload(TestCase):
|
|
|
|
def setUp(self):
|
|
self.username, self.password = 'user', 'pass'
|
|
self.client = Client()
|
|
self.csv_path = os.path.join(os.path.dirname(__file__), 'data/test.csv')
|
|
self.json_path = os.path.join(os.path.dirname(__file__), 'data/test.jsonl')
|
|
|
|
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 test_upload_csv_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)
|
|
|
|
with open(self.csv_path) as f:
|
|
url = reverse('upload', args=[project.id])
|
|
self.client.login(username=self.username, password=self.password)
|
|
response = self.client.post(url, {'file': f, 'format': 'csv'})
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_302_FOUND)
|
|
self.assertEqual(Document.objects.count(), 3)
|
|
|
|
def test_upload_csv_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)
|
|
|
|
with open(self.csv_path) as f:
|
|
url = reverse('upload', args=[project.id])
|
|
self.client.login(username=self.username, password=self.password)
|
|
response = self.client.post(url, {'file': f, 'format': 'csv'})
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN)
|
|
self.assertEqual(Document.objects.count(), 0)
|
|
|
|
def test_upload_json_by_admin(self):
|
|
"""
|
|
Ensure we can create a new document object from json by admin.
|
|
"""
|
|
user = self.create_superuser()
|
|
project = self.create_project()
|
|
project.users.add(user)
|
|
|
|
with open(self.json_path, 'r') as f:
|
|
url = reverse('upload', args=[project.id])
|
|
self.client.login(username=self.username, password=self.password)
|
|
response = self.client.post(url, {'file': f, 'format': 'json'})
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_302_FOUND)
|
|
self.assertEqual(Document.objects.count(), 18)
|