From ab3e48445f6e42f3f036615f597241bebf4a8652 Mon Sep 17 00:00:00 2001
From: Hironsan <light.tree.1.13@gmail.com>
Date: Wed, 1 Aug 2018 13:55:30 +0900
Subject: [PATCH] Add test case for document api

---
 app/server/tests/test_doc_api.py | 97 ++++++++++++++++++++++++++++++++
 1 file changed, 97 insertions(+)
 create mode 100644 app/server/tests/test_doc_api.py

diff --git a/app/server/tests/test_doc_api.py b/app/server/tests/test_doc_api.py
new file mode 100644
index 00000000..02ed58f8
--- /dev/null
+++ b/app/server/tests/test_doc_api.py
@@ -0,0 +1,97 @@
+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)