matthias/mayan-edms
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add API views to download documents and document versions.

Browse Source
This commit is contained in:
Roberto Rosario
2015-10-07 02:42:36 -04:00
parent 41ede494db
commit 8bdd4ba452
4 changed files with 128 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
mayan/apps/documents/tests/test_api.py
View File

@@ -2,6 +2,7 @@
from __future__ import unicode_literals
import io
import time
from json import loads
@@ -17,9 +18,9 @@ from rest_framework.test import APITestCase
from .literals import (
TEST_ADMIN_EMAIL, TEST_ADMIN_PASSWORD, TEST_ADMIN_USERNAME,
TEST_DOCUMENT_FILENAME, TEST_DOCUMENT_PATH, TEST_DOCUMENT_TYPE,
TEST_SMALL_DOCUMENT_PATH,
TEST_SMALL_DOCUMENT_CHECKSUM, TEST_SMALL_DOCUMENT_PATH,
)
from ..models import Document, DocumentType
from ..models import Document, DocumentType, HASH_FUNCTION
class DocumentTypeAPITestCase(APITestCase):
@@ -231,5 +232,46 @@ class DocumentAPITestCase(APITestCase):
self.assertEqual(document_version, document.latest_version)
def test_document_download(self):
with open(TEST_SMALL_DOCUMENT_PATH) as file_object:
document = self.document_type.new_document(
file_object=File(file_object),
)
response = self.client.get(
reverse(
'rest_api:document-download', args=(document.pk,)
)
)
buf = io.BytesIO()
buf.write(response.content)
self.assertEqual(
HASH_FUNCTION(buf.getvalue()), TEST_SMALL_DOCUMENT_CHECKSUM
)
del(buf)
def test_document_version_download(self):
with open(TEST_SMALL_DOCUMENT_PATH) as file_object:
document = self.document_type.new_document(
file_object=File(file_object),
)
response = self.client.get(
reverse(
'rest_api:documentversion-download',
args=(document.latest_version.pk,)
)
)
buf = io.BytesIO()
buf.write(response.content)
self.assertEqual(
HASH_FUNCTION(buf.getvalue()), TEST_SMALL_DOCUMENT_CHECKSUM
)
del(buf)
#def test_document_set_document_type(self):
# pass