def test_del_batch_admin(self):
admins = user_api.find_admins_by_query_terms({'query_box': 'aaa'})
admin_ids = [admin.id for admin in admins]
user_api.del_batch_dtu_user(admin_ids)
admins = user_api.find_admins_by_query_terms({'query_box': 'aaa'})
self.assertEqual(len(admins), 0)
def test_del_admin(self):
admins = user_api.find_admins_by_query_terms({'query_box': 'aaa0'})
admin = admins[0]
user_api.del_admin(admin.id)
admins = user_api.find_admins_by_query_terms({'query_box': 'aaa0'})
self.assertEqual(len(admins), 0)
def test_edit_admin(self):
admins = user_api.find_admins_by_query_terms({'query_box': 'aaa0'})
admin = admins[0]
admin_dict = admin.get_dict()
admin_dict['name'] = 'new' + admin.name
admin_dict.pop('remark')
admin_dict.pop('create_time')
user_api.edit_admin(admin_dict)
admins = user_api.find_admins_by_query_terms({'query_box': 'aaa0'})
admin = admins[0]
self.assertEqual(admin.name, 'newaaa0')
admin_dict['phone'] = '23850592088'
error = False
try:
user_api.edit_admin(admin_dict)
except NotUniqueException:
error = True
self.assertTrue(error)
error = False
admin_dict['id'] = 1000
try:
user_api.edit_admin(admin_dict)
except DoesNotExistException:
error = True
self.assertTrue(error)
def test_find_admins_by_query_terms(self):
# 查询全部管理员
admins = user_api.find_admins_by_query_terms({})
self.assertEqual(len(admins), 10)
admins = user_api.find_admins_by_query_terms({'query_box': 'aaa'})
self.assertEqual(len(admins), 10)
admins = user_api.find_admins_by_query_terms({'query_box': 'bbb'})
self.assertEqual(len(admins), 0)
admins = user_api.find_admins_by_query_terms({'query_box': '13477'})
self.assertEqual(len(admins), 1)
admins = user_api.find_admins_by_query_terms({'query_box': '238'})
self.assertEqual(len(admins), 10)
admins = user_api.find_admins_by_query_terms(
{'end_time': int(time.mktime(datetime.datetime(2020, 2, 3).timetuple()))})
self.assertEqual(len(admins), 3)
admins = user_api.find_admins_by_query_terms(
{'begin_time': int(time.mktime(datetime.datetime(2020, 2, 3).timetuple()))})
self.assertEqual(len(admins), 8)
admins = user_api.find_admins_by_query_terms(
{'begin_time': int(time.mktime(datetime.datetime(2020, 2, 3).timetuple())),
'end_time': int(time.mktime(datetime.datetime(2020, 2, 5).timetuple()))})
self.assertEqual(len(admins), 3)
def query_admin(self, request):
param = request_api.get_param(request)
page = request_api.get_page(request)
admins, num = app_user_api.find_admins_by_query_terms(param, page)
return Result.success(data=admins, count=num)
def test_admin_import_and_export(self):
admins = user_api.find_admins_by_query_terms({'query_box': 'aaa'})
# 开始导出
user_api.admin_export({'query_box': 'aaa'}, '/admin.xls')
admin_ids = [admin.id for admin in admins]
user_api.del_batch_admin(admin_ids)
# 测试是否被删除成功
admins = user_api.find_admins_by_query_terms({'query_box': 'aaa'})
self.assertEqual(len(admins), 0)
# 开始导入
user_api.admin_import('/admin.xls')
# 测试是否导入成功
admins = user_api.find_admins_by_query_terms({'query_box': 'aaa'})
self.assertEqual(len(admins), 10)
# 清理生成的额外文件
del_file('/admin.xls')