fix: adjust test to async

Author: setiawanisasi

tests/test_adapter.py

@@ -4,7 +4,7 @@
 from casbin_databases_adapter.adapter import Filter
 
 
-def test_load_policy(db: Database, enforcer: Enforcer):
+async def test_load_policy(db: Database, enforcer: Enforcer):
 
     assert enforcer.enforce("alice", "data1", "read") == True
     assert enforcer.enforce("bob", "data2", "write") == True
@@ -16,60 +16,60 @@ def test_load_policy(db: Database, enforcer: Enforcer):
     assert enforcer.enforce("bob", "data2", "read") == False
 
 
-def test_add_policy(db: Database, enforcer: Enforcer):
+async def test_add_policy(db: Database, enforcer: Enforcer):
     assert not enforcer.enforce("eve", "data3", "read")
-    result = enforcer.add_permission_for_user("eve", "data3", "read")
+    result = await enforcer.add_permission_for_user("eve", "data3", "read")
     assert result
     assert enforcer.enforce("eve", "data3", "read")
 
 
-def test_save_policy(db: Database, enforcer: Enforcer):
+async def test_save_policy(db: Database, enforcer: Enforcer):
     assert not enforcer.enforce("alice", "data4", "read")
 
     model: Model = enforcer.get_model()
     model.clear_policy()
 
     model.add_policy("p", "p", ["alice", "data4", "read"])
     adapter: Adapter = enforcer.get_adapter()
-    adapter.save_policy(model)
+    await adapter.save_policy(model)
     assert enforcer.enforce("alice", "data4", "read")
 
 
-def test_remove_policy(db: Database, enforcer: Enforcer):
+async def test_remove_policy(db: Database, enforcer: Enforcer):
     assert not (enforcer.enforce("alice", "data5", "read"))
-    enforcer.add_permission_for_user("alice", "data5", "read")
+    await enforcer.add_permission_for_user("alice", "data5", "read")
     assert enforcer.enforce("alice", "data5", "read")
-    enforcer.delete_permission_for_user("alice", "data5", "read")
+    await enforcer.delete_permission_for_user("alice", "data5", "read")
     assert not (enforcer.enforce("alice", "data5", "read"))
 
 
-def test_remove_filtered_policy(db: Database, enforcer: Enforcer):
+async def test_remove_filtered_policy(db: Database, enforcer: Enforcer):
 
     assert enforcer.enforce("alice", "data1", "read")
-    enforcer.remove_filtered_policy(1, "data1")
+    await enforcer.remove_filtered_policy(1, "data1")
     assert not (enforcer.enforce("alice", "data1", "read"))
 
     assert enforcer.enforce("bob", "data2", "write")
     assert enforcer.enforce("alice", "data2", "read")
     assert enforcer.enforce("alice", "data2", "write")
 
-    enforcer.remove_filtered_policy(1, "data2", "read")
+    await enforcer.remove_filtered_policy(1, "data2", "read")
 
     assert enforcer.enforce("bob", "data2", "write")
     assert not (enforcer.enforce("alice", "data2", "read"))
     assert enforcer.enforce("alice", "data2", "write")
 
-    enforcer.remove_filtered_policy(2, "write")
+    await enforcer.remove_filtered_policy(2, "write")
 
     assert not (enforcer.enforce("bob", "data2", "write"))
     assert not (enforcer.enforce("alice", "data2", "write"))
 
 
-def test_filtered_policy(db: Database, enforcer: Enforcer):
+async def test_filtered_policy(db: Database, enforcer: Enforcer):
     filter = Filter()
 
     filter.ptype = ["p"]
-    enforcer.load_filtered_policy(filter)
+    await enforcer.load_filtered_policy(filter)
     assert enforcer.enforce("alice", "data1", "read")
     assert not (enforcer.enforce("alice", "data1", "write"))
     assert not (enforcer.enforce("alice", "data2", "read"))
@@ -81,7 +81,7 @@ def test_filtered_policy(db: Database, enforcer: Enforcer):
 
     filter.ptype = []
     filter.v0 = ["alice"]
-    enforcer.load_filtered_policy(filter)
+    await enforcer.load_filtered_policy(filter)
     assert enforcer.enforce("alice", "data1", "read")
     assert not (enforcer.enforce("alice", "data1", "write"))
     assert not (enforcer.enforce("alice", "data2", "read"))
@@ -94,7 +94,7 @@ def test_filtered_policy(db: Database, enforcer: Enforcer):
     assert not (enforcer.enforce("data2_admin", "data2", "write"))
 
     filter.v0 = ["bob"]
-    enforcer.load_filtered_policy(filter)
+    await enforcer.load_filtered_policy(filter)
     assert not (enforcer.enforce("alice", "data1", "read"))
     assert not (enforcer.enforce("alice", "data1", "write"))
     assert not (enforcer.enforce("alice", "data2", "read"))
@@ -107,7 +107,7 @@ def test_filtered_policy(db: Database, enforcer: Enforcer):
     assert not (enforcer.enforce("data2_admin", "data2", "write"))
 
     filter.v0 = ["data2_admin"]
-    enforcer.load_filtered_policy(filter)
+    await enforcer.load_filtered_policy(filter)
     assert enforcer.enforce("data2_admin", "data2", "read")
     assert enforcer.enforce("data2_admin", "data2", "read")
     assert not (enforcer.enforce("alice", "data1", "read"))
@@ -120,7 +120,7 @@ def test_filtered_policy(db: Database, enforcer: Enforcer):
     assert not (enforcer.enforce("bob", "data2", "write"))
 
     filter.v0 = ["alice", "bob"]
-    enforcer.load_filtered_policy(filter)
+    await enforcer.load_filtered_policy(filter)
     assert enforcer.enforce("alice", "data1", "read")
     assert not (enforcer.enforce("alice", "data1", "write"))
     assert not (enforcer.enforce("alice", "data2", "read"))
@@ -134,7 +134,7 @@ def test_filtered_policy(db: Database, enforcer: Enforcer):
 
     filter.v0 = []
     filter.v1 = ["data1"]
-    enforcer.load_filtered_policy(filter)
+    await enforcer.load_filtered_policy(filter)
     assert enforcer.enforce("alice", "data1", "read")
     assert not (enforcer.enforce("alice", "data1", "write"))
     assert not (enforcer.enforce("alice", "data2", "read"))
@@ -147,7 +147,7 @@ def test_filtered_policy(db: Database, enforcer: Enforcer):
     assert not (enforcer.enforce("data2_admin", "data2", "write"))
 
     filter.v1 = ["data2"]
-    enforcer.load_filtered_policy(filter)
+    await enforcer.load_filtered_policy(filter)
     assert not (enforcer.enforce("alice", "data1", "read"))
     assert not (enforcer.enforce("alice", "data1", "write"))
     assert not (enforcer.enforce("alice", "data2", "read"))
@@ -161,7 +161,7 @@ def test_filtered_policy(db: Database, enforcer: Enforcer):
 
     filter.v1 = []
     filter.v2 = ["read"]
-    enforcer.load_filtered_policy(filter)
+    await enforcer.load_filtered_policy(filter)
     assert enforcer.enforce("alice", "data1", "read")
     assert not (enforcer.enforce("alice", "data1", "write"))
     assert not (enforcer.enforce("alice", "data2", "read"))
@@ -174,7 +174,7 @@ def test_filtered_policy(db: Database, enforcer: Enforcer):
     assert not (enforcer.enforce("data2_admin", "data2", "write"))
 
     filter.v2 = ["write"]
-    enforcer.load_filtered_policy(filter)
+    await enforcer.load_filtered_policy(filter)
     assert not (enforcer.enforce("alice", "data1", "read"))
     assert not (enforcer.enforce("alice", "data1", "write"))
     assert not (enforcer.enforce("alice", "data2", "read"))