test: Added ABACData

Signed-off-by: Yash Pandey (YP) <[email protected]>

Author: EmperorYP7

casbin/abac_data.cpp

@@ -36,7 +36,7 @@ namespace casbin {
  * Key's type is std::string and value's type can be one of std::string, int32_t, and float only
  * @return Pointer to casbin::ABACData entity
  */
-static const std::shared_ptr<ABACData> GetData(const ABACData::VariantMap& attribs) {
+const std::shared_ptr<ABACData> GetData(const ABACData::VariantMap& attribs) {
     return std::make_shared<ABACData>(attribs);
 }
 

casbin/abac_data.h

@@ -104,6 +104,7 @@ class ABACData {
 // Casbin ABAC entity type
 typedef ABACData ABACData;
 
+const std::shared_ptr<ABACData> GetData(const ABACData::VariantMap& attribs);
 }
 
 #endif

casbin/casbin.h

@@ -25,3 +25,4 @@
 #include "util.h"
 #include "exception.h"
 #include "rbac.h"
+#include "abac_data.h"

tests/built_in_functions_test.cpp

@@ -164,4 +164,4 @@ TEST(TestBuiltInFunctions, TestIPMatch) {
     TestIPMatchFn("11.0.0.123", "10.0.0.0/8", false);
 }
 
-}
+} // namespace

tests/config_test.cpp

@@ -42,4 +42,4 @@ TEST(TestConfig, TestRedis) {
     ASSERT_EQ(std::string("push2"), values[1]);
 }
 
-}
+} // namespace

tests/enforcer_cached_test.cpp

@@ -19,6 +19,8 @@
 #include <gtest/gtest.h>
 #include <casbin/casbin.h>
 
+namespace {
+
 TEST(TestEnforcerCached, TestCache) {
     std::string model = "../../examples/basic_model.conf";
     std::string policy = "../../examples/basic_policy.csv";
@@ -43,4 +45,6 @@ TEST(TestEnforcerCached, TestCache) {
     ASSERT_EQ(e.Enforce({ "alice", "data1", "write" }), false);
     ASSERT_EQ(e.Enforce({ "alice", "data2", "read" }), false);
     ASSERT_EQ(e.Enforce({ "alice", "data2", "write" }), false);
-}
+}
+
+} // namespace

tests/enforcer_synced_test.cpp

@@ -19,6 +19,8 @@
 #include <gtest/gtest.h>
 #include <casbin/casbin.h>
 
+namespace {
+
 // void TestSyncFn(casbin::SyncedEnforcer& e, const std::string& sub, const std::string& obj, const std::string& act, bool control) {
 //     bool response = e.Enforce({ sub, obj, act });
 //     ASSERT_EQ(response, control);
@@ -72,3 +74,5 @@
 
 //     EXPECT_EQ(e.IsAutoLoadingRunning(), false);
 // }
+
+} // namespace

tests/enforcer_test.cpp

@@ -19,6 +19,8 @@
 #include <gtest/gtest.h>
 #include <casbin/casbin.h>
 
+namespace {
+
 TEST(TestEnforcer, TestFourParams) {
     std::string model = "../../examples/rbac_with_domains_model.conf";
     std::string policy = "../../examples/rbac_with_domains_policy.csv";
@@ -93,3 +95,31 @@ TEST(TestEnforcer, TestMapParams) {
     params = { {"sub","bob"},{"obj","data2"},{"act","write"} };
     ASSERT_EQ(e.Enforce(params), true);
 }
+
+TEST(TestEnforcer, ABACData) {
+    casbin::ABACData::VariantMap params = {
+        { "Name", "Yash" },
+        { "Grade", 8.6f },
+        { "Age", 18 },
+    };
+
+    auto data = casbin::GetData(params);
+    EXPECT_EQ(params, data->GetAttributes());
+
+    data->DeleteAttribute("Name");
+    params = {
+        { "Grade", 8.6f },
+        { "Age", 18 },
+    };
+    EXPECT_EQ(params, data->GetAttributes());
+
+    data->AddAttribute("ID", 156);
+    params["ID"] = 156;
+    EXPECT_EQ(params, data->GetAttributes());
+
+    data->UpdateAttribute("ID", 152);
+    params["ID"] = 152;
+    EXPECT_EQ(params, data->GetAttributes());
+}
+
+} // namespace

tests/management_api_test.cpp

@@ -284,4 +284,4 @@ TEST(TestManagementAPI, TestModifyGroupingPolicyAPI) {
     ASSERT_TRUE(casbin::ArrayEquals({"data4_admin"}, e.GetRolesForUser("admin")));
 }
 
-}
+} // namespace

tests/model_enforcer_test.cpp

@@ -756,4 +756,4 @@ func TestCommentModel(t* testing.T) {
 }
 */
 
-}
+} // namespace