unit test #5

Author: chfw

django_excel/__init__.py

@@ -116,7 +116,7 @@ def new_file(self, file_name, *args, **kwargs):
             self.content_type_extra)
 
 
-def make_response(content, content_type, status, file_name=None):
+def _make_response(content, content_type, status, file_name=None):
     """
     Custom response function that is called by pyexcel-webio
     """
@@ -126,7 +126,7 @@ def make_response(content, content_type, status, file_name=None):
     return response
 
 
-webio.ExcelResponse = make_response
+webio.ExcelResponse = _make_response
 
 
 from pyexcel_webio import (

polls/urls.py

@@ -6,6 +6,8 @@
     '',
     url(r'^$', views.upload, name='uplink'),
     url(r'^download/(.*)', views.download, name="download"),
+    url(r'^download_attachment/(.*)/(.*)', views.download_as_attachment,
+        name="download_attachment"),
     url(r'^exchange/(.*)', views.exchange, name="exchange"),
     url(r'^parse/(.*)', views.parse, name="parse"),
     url(r'^import/', views.import_data, name="import"),

polls/views.py

@@ -42,6 +42,10 @@ def download(request, file_type):
     sheet = excel.pe.Sheet(data)
     return excel.make_response(sheet, file_type)
 
+    
+def download_as_attachment(request, file_type, file_name):
+    return excel.make_response_from_array(data, file_type, file_name=file_name)
+
 
 def export_data(request, atype):
     if atype == "sheet":

testResponse.py

@@ -59,6 +59,18 @@ def test_download(self):
             array = sheet.to_array()
             assert array == self.data
 
+    def test_download_attachment(self):
+        test_file_name = "test"
+        for file_type in FILE_TYPE_MIME_TABLE.keys():
+            print(file_type)
+            response = self.client.get("/polls/download_attachment/"+file_type+"/"+test_file_name)
+            assert response['Content-Type'] == FILE_TYPE_MIME_TABLE[file_type]
+            assert response['Content-Disposition'] == "attachment; filename=%s.%s" % (test_file_name, file_type)
+            sheet = pe.get_sheet(file_type=file_type, file_content=response.content)
+            sheet.format(int)
+            array = sheet.to_array()
+            assert array == self.data
+
     def test_parse_single_sheet(self):
         test_sample = {
             "array": {u'result': [[u'X', u'Y', u'Z'], [1.0, 2.0, 3.0], [4.0, 5.0, 6.0]]},