Skip to content

Commit c2d77dc

Browse files
committed
update test cases
1 parent 8987a3d commit c2d77dc

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

testResponse.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
import pyexcel.ext.xlsx # noqa
1313
import pyexcel.ext.ods3 # noqa
1414
from django_excel._compact import urllib_quote
15+
from nose.tools import eq_
1516

1617
PY2 = sys.version_info[0] == 2
1718
if 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

Comments
 (0)