Use Django's six library for Py3 compatibility.
This commit is contained in:
@@ -2,7 +2,6 @@
|
|||||||
|
|
||||||
from __future__ import unicode_literals
|
from __future__ import unicode_literals
|
||||||
|
|
||||||
import io
|
|
||||||
import time
|
import time
|
||||||
|
|
||||||
from json import loads
|
from json import loads
|
||||||
@@ -11,6 +10,7 @@ from django.contrib.auth.models import User
|
|||||||
from django.core.files import File
|
from django.core.files import File
|
||||||
from django.core.urlresolvers import reverse
|
from django.core.urlresolvers import reverse
|
||||||
from django.test import override_settings
|
from django.test import override_settings
|
||||||
|
from django.utils.six import BytesIO
|
||||||
|
|
||||||
from rest_framework import status
|
from rest_framework import status
|
||||||
from rest_framework.test import APITestCase
|
from rest_framework.test import APITestCase
|
||||||
@@ -259,7 +259,7 @@ class DocumentAPITestCase(APITestCase):
|
|||||||
'rest_api:document-download', args=(document.pk,)
|
'rest_api:document-download', args=(document.pk,)
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
buf = io.BytesIO()
|
buf = BytesIO()
|
||||||
buf.write(response.content)
|
buf.write(response.content)
|
||||||
|
|
||||||
self.assertEqual(
|
self.assertEqual(
|
||||||
@@ -280,7 +280,7 @@ class DocumentAPITestCase(APITestCase):
|
|||||||
args=(document.latest_version.pk,)
|
args=(document.latest_version.pk,)
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
buf = io.BytesIO()
|
buf = BytesIO()
|
||||||
buf.write(response.content)
|
buf.write(response.content)
|
||||||
|
|
||||||
self.assertEqual(
|
self.assertEqual(
|
||||||
|
|||||||
Reference in New Issue
Block a user