haiwen/seahub
1
0
Fork 0
mirror of https://github.com/haiwen/seahub.git synced 2025-10-21 02:42:26 +00:00
Code Issues Releases Wiki Activity

[tests] Update user export excel

Browse Source
This commit is contained in:
zhengxie
2015-11-24 12:15:14 +08:00
parent 4bccc4abac
commit 6aa5b4fadd
1 changed files with 17 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
tests/seahub/views/test_sysadmin.py
View File

@@ -1,7 +1,9 @@
from mock import patch
from django.core.urlresolvers import reverse
from django.http.cookie import parse_cookie
from seahub.base.accounts import User
from seahub.utils.ms_excel import write_xls as real_write_xls
from seahub.test_utils import BaseTestCase
from seaserv import ccnet_threaded_rpc
@@ -152,3 +154,18 @@ class SysUserAdminExportExcelTest(BaseTestCase):
resp = self.client.get(reverse('sys_useradmin_export_excel'))
self.assertEqual(200, resp.status_code)
assert 'application/ms-excel' in resp._headers['content-type']
def write_xls(self, sheet_name, head, data_list):
assert 'Role' in head
return real_write_xls(sheet_name, head, data_list)
@patch('seahub.views.sysadmin.write_xls')
@patch('seahub.views.sysadmin.is_pro_version')
def test_can_export_excel_in_pro(self, mock_is_pro_version, mock_write_xls):
mock_is_pro_version.return_value = True
mock_write_xls.side_effect = self.write_xls
mock_write_xls.assert_called_once()
resp = self.client.get(reverse('sys_useradmin_export_excel'))
self.assertEqual(200, resp.status_code)
assert 'application/ms-excel' in resp._headers['content-type']