test TemporaryExcelFileUploadHandler

Author: chfw

django_excel/__init__.py

@@ -27,6 +27,8 @@ def load_book(self):
 
 class ExcelMemoryFileUploadHandler(MemoryFileUploadHandler):
     def file_complete(self, file_size):
+        if not self.activated:
+            return
         self.file.seek(0)
         return ExcelMemoryFile(
             file=self.file,

testResponse.py

@@ -1,5 +1,5 @@
-from unittest import TestCase
-from django.test import Client
+from django.test import Client, TestCase
+from django.test.utils import override_settings
 import pyexcel as pe
 import pyexcel.ext.xls
 import pyexcel.ext.xlsx
@@ -109,4 +109,9 @@ def test_exchange(self):
                     sheet.format(int)
                     array = sheet.to_array()
                     assert array == self.data
-            os.unlink(tmp_filename)
+            os.unlink(tmp_filename)
+
+
+@override_settings(FILE_UPLOAD_MAX_MEMORY_SIZE=1)
+class ExcelResponseUsingFileTestCase(ExcelResponseTestCase):
+    pass