class zones(unittest.TestCase):
def setUp(self):
s = requests.session()
self.WDMS = WDMS(s)
#create zone
def test1_create_zone(self):
'''创建区域'''
self.WDMS.login('admin', 'admin')
r = self.WDMS.create_zone(122, 'test')
self.assertEqual(r['message'], 'Succeed')
self.assertEqual(r['code'], 200)
#update zone
def test2_update_zone(self):
'''更新区域名称'''
self.WDMS.login('admin', 'admin')
r = self.WDMS.create_zone(122, 'test2')
self.assertEqual(r['message'], 'Succeed')
self.assertEqual(r['code'], 200)
def test3_zone_info(self):
'''获取区域信息'''
self.WDMS.login('admin', 'admin')
r = self.WDMS.get_zone_info()
self.assertEqual(r['code'], 200)
self.assertEqual(r['message'], 'Succeed')
#判断返回信息值是否正确
result_zoneName = r['data'][0]['zoneName']
result_zoneNum = r['data'][0]['zoneNumber']
self.assertEqual(result_zoneNum, 122)
self.assertEqual(result_zoneName, 'test2')
class Test_login(unittest.TestCase):
def setUp(self):
s = requests.session()
self.WDMS = WDMS(s)
def test_login_successful(self):
'''账号正确，登录成功'''
r = self.WDMS.login('admin','admin')
self.assertEqual(r['message'],'Login Successful')
self.assertEqual(r['code'],200)
def test_login_defeat(self):
'''账号正确，密码错误'''
r = self.WDMS.login('admin','admin2')
self.assertEqual(r['message'],'Username Or Password Is Incorrect')
self.assertEqual(r['code'],1002)
class Test_logout(unittest.TestCase):
def setUp(self):
s = requests.session()
self.WDMS = WDMS(s)
def test1_logout_successful(self):
'''成功退出'''
self.WDMS.login('admin', 'admin')
r = self.WDMS.logout('admin', 'admin')
self.assertEqual(r['message'], 'Logout Successful')
self.assertEqual(r['code'], 200)
def test2_logout_defeat(self):
'''未登录,退出'''
r = self.WDMS.logout('admin', 'admin')
self.assertEqual(r['message'], 'Login Required')
self.assertEqual(r['code'], 1001)
def setUp(self): s = requests.session() self.WDMS = WDMS(s)
class Department(unittest.TestCase):
def setUp(self):
s = requests.session()
self.WDMS = WDMS(s)
#create zone
def test_create_department1(self):
'''在不存在的区域下创建部门'''
self.WDMS.login('admin','admin')
r = self.WDMS.create_department(1,'testDepartments',1)
self.assertEqual(r['message'],"Zone Doesn't Exist")
self.assertEqual(r['code'],4002)
def test_create_department2(self):
'''在某区域下创建部门'''
self.WDMS.login('admin','admin')
self.WDMS.create_zone('999','testZone999')
r = self.WDMS.create_department(1,'test2',999)
self.assertEqual(r['message'],'Succeed')
self.assertEqual(r['code'],200)
#get the infomation of departments
def test_get_all_departments_info(self):
'''获取所有部门信息'''
self.WDMS.login('admin','admin')
r = self.WDMS.get_department_info()
self.assertEqual(r['code'],200)
self.assertEqual(r['message'],'Succeed')
def test_get_department_info1(self):
'''获取非法区域编号下所有部门信息'''
self.WDMS.login('admin','admin')
data = {'zoneNumber':'sss'}
r = self.WDMS.get_department_info(data = data)
self.assertEqual(r['code'],3006)
self.assertEqual(r['message']['zoneNumber'][0],'Enter a whole number.')
def test_get_department_info2(self):
'''获取某个不存在区域下所有部门信息'''
self.WDMS.login('admin','admin')
data = {'zoneNumber':123}
r = self.WDMS.get_department_info(data = data)
self.assertEqual(r['code'],3008)
self.assertEqual(r['message'],'Zone 123 Not Found')
def test_get_department_info3(self):
'''正确获取某个区域下所有部门的信息'''
self.WDMS.login('admin','admin')
data = {'zoneNumber':999}
r = self.WDMS.get_department_info(data = data)
self.assertEqual(r['code'],200)
self.assertEqual(r['message'],'Succeed')
#delete departments
def test_delete_departments(self):
'''删除某区域下部门'''
self.WDMS.login('admin','admin')
r = self.WDMS.delete_department(1,999)
self.assertEqual(r['code'],200)
self.assertEqual(r['message'],'Succeed')