第一步：先实现新增用户的单元测试

import unittest
import reqclass TestUserAddRoles(unittest.TestCase):def assertApiSuccess(self, resp):"""断言API接口的响应是成功的"""self.assertEqual(200, resp.status_code)data = resp.json()self.assertEqual(10000, data["code"])self.assertEqual(True, data["status"])return data@classmethoddef setUpClass(cls):cls.base_url = "http://127.0.0.1:8888/auth"def test_user_add_roles(self):"""测试批量绑定角色"""# 1、添加用户target_url = f"{self.base_url}/user/register"username = "zhangdapeng1"data = {"username": username,"password": username,"re_password": username,}resp = req.post(target_url, json=data)self.assertApiSuccess(resp)# 2、添加3个角色# 3、获取用户idtarget_url = f"{self.base_url}/user"resp = req.get(target_url)data = self.assertApiSuccess(resp)users = data.get("data")self.assertTrue(isinstance(users, list))user_id = Nonefor user in users:if user.get("username") == username:user_id = user.get("id")breakself.assertIsNotNone(user_id)# 4、获取角色id# 5、给用户绑定两个角色# 6、获取用户已经绑定的所有角色# 7、删除用户target_url = f"{self.base_url}/user/{user_id}"resp = req.delete(target_url)self.assertApiSuccess(resp)# 8、删除角色# 9、批量解绑用户关联的所有角色

第二步：实现新增角色的单元测试

import unittest
import reqclass TestUserAddRoles(unittest.TestCase):def assertApiSuccess(self, resp):"""断言API接口的响应是成功的"""self.assertEqual(200, resp.status_code)data = resp.json()self.assertEqual(10000, data["code"])self.assertEqual(True, data["status"])return data@classmethoddef setUpClass(cls):cls.base_url = "http://127.0.0.1:8888/auth"def test_user_add_roles(self):"""测试批量绑定角色"""# 1、添加用户target_url = f"{self.base_url}/user/register"username = "zhangdapeng1"data = {"username": username,"password": username,"re_password": username,}resp = req.post(target_url, json=data)self.assertApiSuccess(resp)# 2、添加3个角色target_url = f"{self.base_url}/role"rolename1, rolename2, rolename3 = "testrole1", "testrole2", "testrole3"rolenames = (rolename1, rolename2, rolename3)for rolename in rolenames:data = {"name": rolename,"nickname": rolename,}resp = req.post(target_url, json=data)self.assertApiSuccess(resp)# 3、获取用户idtarget_url = f"{self.base_url}/user"resp = req.get(target_url)data = self.assertApiSuccess(resp)users = data.get("data")self.assertTrue(isinstance(users, list))user_id = Nonefor user in users:if user.get("username") == username:user_id = user.get("id")breakself.assertIsNotNone(user_id)# 4、获取角色idtarget_url = f"{self.base_url}/role"resp = req.get(target_url)data = self.assertApiSuccess(resp)roles = data.get("data")self.assertTrue(isinstance(roles, list))roleid1, roleid2, roleid3 = None, None, Nonefor role in roles:if role.get("name") == rolename1:roleid1 = role.get("id")elif role.get("name") == rolename2:roleid2 = role.get("id")elif role.get("name") == rolename3:roleid3 = role.get("id")self.assertIsNotNone(roleid1)self.assertIsNotNone(roleid2)self.assertIsNotNone(roleid3)# 5、给用户绑定两个角色# 6、获取用户已经绑定的所有角色# 7、删除用户target_url = f"{self.base_url}/user/{user_id}"resp = req.delete(target_url)self.assertApiSuccess(resp)# 8、删除角色roleids = [roleid1, roleid2, roleid3]for roleid in roleids:target_url = f"{self.base_url}/role/{roleid}"resp = req.delete(target_url)self.assertApiSuccess(resp)# 9、批量解绑用户关联的所有角色

第三步：测试批量绑定角色和解绑角色

import unittest
import reqclass TestUserAddRoles(unittest.TestCase):def assertApiSuccess(self, resp):"""断言API接口的响应是成功的"""self.assertEqual(200, resp.status_code)data = resp.json()self.assertEqual(10000, data["code"])self.assertEqual(True, data["status"])return data@classmethoddef setUpClass(cls):cls.base_url = "http://127.0.0.1:8888/auth"def test_user_add_roles(self):"""测试批量绑定角色"""# 1、添加用户target_url = f"{self.base_url}/user/register"username = "zhangdapeng1"data = {"username": username,"password": username,"re_password": username,}resp = req.post(target_url, json=data)self.assertApiSuccess(resp)# 2、添加3个角色target_url = f"{self.base_url}/role"rolename1, rolename2, rolename3 = "testrole1", "testrole2", "testrole3"rolenames = (rolename1, rolename2, rolename3)for rolename in rolenames:data = {"name": rolename,"nickname": rolename,}resp = req.post(target_url, json=data)self.assertApiSuccess(resp)# 3、获取用户idtarget_url = f"{self.base_url}/user"resp = req.get(target_url)data = self.assertApiSuccess(resp)users = data.get("data")self.assertTrue(isinstance(users, list))user_id = Nonefor user in users:if user.get("username") == username:user_id = user.get("id")breakself.assertIsNotNone(user_id)# 4、获取角色idtarget_url = f"{self.base_url}/role"resp = req.get(target_url)data = self.assertApiSuccess(resp)roles = data.get("data")self.assertTrue(isinstance(roles, list))roleid1, roleid2, roleid3 = None, None, Nonefor role in roles:if role.get("name") == rolename1:roleid1 = role.get("id")elif role.get("name") == rolename2:roleid2 = role.get("id")elif role.get("name") == rolename3:roleid3 = role.get("id")self.assertIsNotNone(roleid1)self.assertIsNotNone(roleid2)self.assertIsNotNone(roleid3)# 5、给用户绑定两个角色target_url = f"{self.base_url}/user/{user_id}/role"resp = req.post(target_url, json={"ids": [roleid1, roleid2]})self.assertApiSuccess(resp)# 6、获取用户已经绑定的所有角色target_url = f"{self.base_url}/user/{user_id}/role"resp = req.get(target_url)data = self.assertApiSuccess(resp)roles = data.get("data")self.assertTrue(isinstance(roles, list))self.assertTrue(len(roles) == 2)# 7、删除用户target_url = f"{self.base_url}/user/{user_id}"resp = req.delete(target_url)self.assertApiSuccess(resp)# 8、删除角色roleids = [roleid1, roleid2, roleid3]for roleid in roleids:target_url = f"{self.base_url}/role/{roleid}"resp = req.delete(target_url)self.assertApiSuccess(resp)# 9、批量解绑用户关联的所有角色target_url = f"{self.base_url}/user/{user_id}/role"resp = req.delete(target_url)self.assertApiSuccess(resp)