From 6aa5b4fadd2138f38cd3f17c15d2a5a51d40d325 Mon Sep 17 00:00:00 2001 From: zhengxie Date: Tue, 24 Nov 2015 12:15:14 +0800 Subject: [PATCH] [tests] Update user export excel --- tests/seahub/views/test_sysadmin.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/tests/seahub/views/test_sysadmin.py b/tests/seahub/views/test_sysadmin.py index e8c1616ad2..66c2a79c32 100644 --- a/tests/seahub/views/test_sysadmin.py +++ b/tests/seahub/views/test_sysadmin.py @@ -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']