feat: Add function hasPolicies (#85)

* feat: Add function hasPolicies Signed-off-by: Edmond <[email protected]> * feat: Add hasPolicies unit tests Signed-off-by: Edmond <[email protected]>
casbin · Jul 14, 2021 · 2b48145 · 2b48145
1 parent 3f7e220
commit 2b48145
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 3 deletions.
diff --git a/src/model/Policy.lua b/src/model/Policy.lua
@@ -138,7 +138,7 @@ function Policy:getFilteredPolicy(sec, ptype, fieldIndex, ...)
 
     if not self.model[sec] then return res end
     if not self.model[sec][ptype] then return res end
-    
+
     for _, rule in pairs(self.model[sec][ptype].policy) do
         local matched = true
         for i, v in ipairs(fieldValues) do
@@ -172,6 +172,24 @@ function Policy:hasPolicy(sec, ptype, rule)
     return false
 end
 
+--[[
+     * hasPolicies determines whether a model has any of the specified policies.
+     *
+     * @param sec the section, "p" or "g".
+     * @param ptype the policy type, "p", "p2", .. or "g", "g2", ..
+     * @param rules the policies rules.
+     * @return whether one is found.
+]]
+function Policy:hasPolicies(sec, ptype, rules)
+    for i = 1, #rules do
+        if self:hasPolicy(sec, ptype, rules[i]) then
+            return true
+        end
+    end
+    return false
+end
+
+
 --[[
      * addPolicy adds a policy rule to the model.
      *
@@ -205,7 +223,7 @@ function Policy:addPolicies(sec, ptype, rules)
 
     if size < #self.model[sec][ptype].policy then
         return true
-    else 
+    else
         return false
     end
 end
@@ -281,7 +299,7 @@ function Policy:removePolicies(sec, ptype, rules)
 
     if size > #self.model[sec][ptype].policy then
         return true
-    else 
+    else
         return false
     end
 end

diff --git a/tests/model/model_spec.lua b/tests/model/model_spec.lua
@@ -41,6 +41,25 @@ describe("model tests", function()
         assert.is.True(m:hasPolicy("p", "p", rule))
     end)
 
+    it("test hasPolicies", function ()
+        local m = Model:new()
+        m:loadModel(basic_path)
+        local rule = {'admin', 'domain1', 'data1', 'read'}
+        m:addPolicy("p", "p", rule)
+        local rule1={'admin', 'domain2', 'data2', 'read'}
+        m:addPolicy("p", "p", rule1)
+        local rule2={'admin', 'domain3', 'data3', 'read'}
+        m:addPolicy("p", "p", rule2)
+        local rule3={'admin', 'domain4', 'data4', 'read'}
+        local rulesallmatched={rule,rule1,rule2}
+        local rulesonematched={rule,rule3}
+        local rulesnotmatched={rule3}
+        assert.is.True(m:hasPolicies("p", "p", rulesallmatched))
+        assert.is.True(m:hasPolicies("p", "p", rulesonematched))
+        assert.is.False(m:hasPolicies("p", "p", rulesnotmatched))
+    end)
+
+
     it("test addPolicy", function ()
         local m = Model:new()
         m:loadModel(basic_path)