zdppy_amauth 实现给角色批量绑定权限

新增接口

api.resp.post("/auth/role_auth", amauth.role.add_auths)

如何测试

如何测试能不能给指定的角色批量的添加权限呢？

• 1、需要新建一个角色
• 2、需要拿到这个角色的ID
• 3、需要新增三个权限
• 4、需要拿到新增的三个权限的ID
• 5、拿着角色ID和权限ID列表，调用批量新增角色接口进行测试
• 6、校验测试结果
• 7、调用根据角色ID批量获取权限的接口
• 8、校验查询结果是否符合预期

测试流程分析

需要新建一个角色

调用 POST:/auth/role 新增角色

需要拿到这个角色的ID

调用 GET:/auth/role 拿到所有的角色，然后根据角色名筛选ID。

需要新增三个权限

调用 POST:/auth/auth 新增三个权限。

需要拿到新增的三个权限的ID

调用 GET:/auth/auth 根据权限名筛选ID。

拿着角色ID和权限ID列表，调用批量新增角色接口进行测试

调用 POST:/auth/role_auth 进行新增。

校验测试结果

status_cod = 200
data[“code”] = 10000
data[“status”] = True

调用根据角色ID批量获取权限的接口

调用 GET:/auth/role_auth/{id} 进行查询

校验查询结果是否符合预期

status_cod = 200
data[“code”] = 10000
data[“status”] = True

data里面有很多个对象，每个对象都有ID。将所有的ID取出来，必须和我们新增时的ID列表相同。

怎么判断两个列表的元素完全相同？

有两种方案可以搞定：

• 比较排序后的两个列表是否相同
• 取交集，如果交集的元素个数和原本的两个集合的元素个数都相同，说明两个集合本身就是相同的。
  

代码实现

import unittest
import req


class TestRoleAuthAdd(unittest.TestCase):

    def test_add_role_auth(self):
        # 新建角色
        target_url = "http://127.0.0.1:8888/auth/role"
        role_name = "testrole"
        data = {
            "name": role_name,
            "nickname": role_name
        }
        resp = req.post(target_url, json=data)
        self.assertEqual(200, resp.status_code)
        data = resp.json()
        self.assertEqual(10000, data.get("code"))
        self.assertTrue(data.get("status"))

        # 拿到这个角色的ID
        target_url = "http://127.0.0.1:8888/auth/role"
        resp = req.get(target_url)
        self.assertEqual(200, resp.status_code)
        data = resp.json()
        self.assertEqual(10000, data.get("code"))
        self.assertTrue(data.get("status"))
        data = data.get("data")
        role_id = None
        for role in data:
            if role.get("name") == role_name:
                role_id = role.get("id")
                break
        self.assertIsNotNone(role_id)

        # 新增三个权限
        target_url = "http://127.0.0.1:8888/auth/auth"
        auth_name1 = "GET:/model"
        auth_name2 = "POST:/model"
        auth_name3 = "DELETE:/model"
        resp = req.post(target_url, json={"name": auth_name1, "nickname": auth_name1})
        self.assertEqual(200, resp.status_code)
        data = resp.json()
        self.assertEqual(10000, data.get("code"))
        self.assertTrue(data.get("status"))
        resp = req.post(target_url, json={"name": auth_name2, "nickname": auth_name2})
        self.assertEqual(200, resp.status_code)
        data = resp.json()
        self.assertEqual(10000, data.get("code"))
        self.assertTrue(data.get("status"))
        resp = req.post(target_url, json={"name": auth_name3, "nickname": auth_name3})
        self.assertEqual(200, resp.status_code)
        data = resp.json()
        self.assertEqual(10000, data.get("code"))
        self.assertTrue(data.get("status"))

        # 获取权限ID列表
        target_url = "http://127.0.0.1:8888/auth/auth"
        auth_ids = []
        resp = req.get(target_url)
        self.assertEqual(200, resp.status_code)
        data = resp.json()
        self.assertEqual(10000, data.get("code"))
        self.assertTrue(data.get("status"))
        data = data.get("data")
        for auth in data:
            if auth.get('name') in (auth_name1, auth_name2, auth_name3):
                auth_ids.append(auth.get("id"))
        self.assertEqual(3, len(auth_ids))

        # 核心：批量绑定
        target_url = "http://127.0.0.1:8888/auth/role_auth/" + role_id
        resp = req.post(target_url, json={"ids": auth_ids})
        self.assertEqual(200, resp.status_code)
        data = resp.json()
        self.assertEqual(10000, data.get("code"))
        self.assertTrue(data.get("status"))

        # 批量获取权限
        resp = req.get(target_url)
        self.assertEqual(200, resp.status_code)
        data = resp.json()
        self.assertEqual(10000, data.get("code"))
        self.assertTrue(data.get("status"))

        query_ids = []
        data = data.get("data")
        for auth in data:
            if auth.get('name') in (auth_name1, auth_name2, auth_name3):
                query_ids.append(auth.get("id"))
        self.assertEqual(3, len(auth_ids))

        auth_ids.sort()
        query_ids.sort()
        self.assertEqual(auth_ids, query_ids)

        # 删除权限
        target_url = "http://127.0.0.1:8888/auth/auth"
        resp = req.delete(target_url, json={"ids": auth_ids})
        self.assertEqual(200, resp.status_code)
        data = resp.json()
        self.assertEqual(10000, data.get("code"))
        self.assertTrue(data.get("status"))

        # 删除角色
        target_url = "http://127.0.0.1:8888/auth/role/" + role_id
        resp = req.delete(target_url)
        self.assertEqual(200, resp.status_code)
        data = resp.json()
        self.assertEqual(10000, data.get("code"))
        self.assertTrue(data.get("status"))