1212import pyexcel .ext .xlsx # noqa
1313import pyexcel .ext .ods3 # noqa
1414from django_excel ._compact import urllib_quote
15+ from nose .tools import eq_
1516
1617PY2 = sys .version_info [0 ] == 2
1718if sys .version_info [0 ] == 2 and sys .version_info [1 ] < 7 :
@@ -178,7 +179,7 @@ def setUp(self):
178179 def testBook (self ):
179180 fp = open (self .testfile , "rb" )
180181 response = self .client .post ('/polls/import/' , data = {"file" : fp })
181- assert response .status_code == 200
182+ eq_ ( response .status_code , 302 )
182183 response2 = self .client .get ('/polls/export/book' )
183184 assert response2 .status_code == 200
184185 book = pe .get_book (file_type = 'xls' , file_content = response2 .content )
@@ -214,7 +215,7 @@ def testBook(self):
214215 def testSheet (self ):
215216 fp = open (self .testfile , "rb" )
216217 response = self .client .post ('/polls/import/' , data = {"file" : fp })
217- assert response .status_code == 200
218+ eq_ ( response .status_code , 302 )
218219 response2 = self .client .get ('/polls/export/sheet' )
219220 assert response2 .status_code == 200
220221 sheet = pe .get_sheet (file_type = 'xls' , file_content = response2 .content )
@@ -232,9 +233,9 @@ def testSheet(self):
232233 def testImportSheet (self ):
233234 fp = open ("sample-sheet.xls" , "rb" )
234235 response = self .client .post ('/polls/import_sheet/' , data = {"file" : fp })
235- assert response .status_code == 200
236+ eq_ ( response .status_code , 200 )
236237 response2 = self .client .get ('/polls/export/sheet' )
237- assert response2 .status_code == 200
238+ eq_ ( response2 .status_code , 200 )
238239 sheet = pe .get_sheet (file_type = 'xls' , file_content = response2 .content )
239240 content = dedent ("""
240241 question:
@@ -250,9 +251,9 @@ def testImportSheet(self):
250251 def testCustomExport (self ):
251252 fp = open (self .testfile , "rb" )
252253 response = self .client .post ('/polls/import/' , data = {"file" : fp })
253- assert response .status_code == 200
254+ eq_ ( response .status_code , 302 )
254255 response2 = self .client .get ('/polls/export/custom' )
255- assert response2 .status_code == 200
256+ eq_ ( response2 .status_code , 200 )
256257 sheet = pe .get_sheet (file_type = 'xls' , file_content = response2 .content )
257258 content = dedent ("""
258259 pyexcel_sheet1:
0 commit comments