Add mkmap erasure test

DataDog · Jun 21, 2024 · 30abc1f · 30abc1f
1 parent 15e0e09
commit 30abc1f
Showing 1 changed file with 158 additions and 0 deletions.
diff --git a/tests/mkmap_test.cpp b/tests/mkmap_test.cpp
@@ -150,4 +150,162 @@ TEST(TestMultiKeyMap, Multifind)
     }
 }
 
+TEST(TestMultiKeyMap, Erase)
+{
+    rule_tag_map ruledb;
+
+    struct rule_spec {
+        std::string id;
+        std::string type;
+        std::string category;
+        std::unordered_map<std::string, std::string> tags;
+    };
+
+    std::vector<rule_spec> specs{{"id0", "type0", "category0", {{"key", "value0"}}},
+        {"id1", "type0", "category0", {{"key", "value1"}}},
+        {"id2", "type0", "category1", {{"key", "value0"}}},
+        {"id3", "type0", "category1", {{"key", "value1"}}},
+        {"id4", "type1", "category0", {{"key", "value0"}}},
+        {"id5", "type1", "category0", {{"key", "value1"}}},
+        {"id6", "type1", "category1", {{"key", "value0"}}},
+        {"id7", "type1", "category1", {{"key", "value1"}}}};
+
+    std::vector<std::shared_ptr<rule>> rules;
+    for (const auto &spec : specs) {
+        std::unordered_map<std::string, std::string> tags = spec.tags;
+        tags.emplace("type", spec.type);
+        tags.emplace("category", spec.category);
+
+        auto rule_ptr = std::make_shared<ddwaf::rule>(
+            std::string(spec.id), "name", decltype(tags)(tags), std::make_shared<expression>());
+        rules.emplace_back(rule_ptr);
+        ruledb.insert(rule_ptr->get_tags(), rule_ptr.get());
+    }
+
+    using sv_pair = std::pair<std::string_view, std::string_view>;
+    {
+        auto rules = ruledb.find(sv_pair{"type"sv, "type0"});
+        EXPECT_EQ(rules.size(), 4);
+    }
+
+    {
+        auto rules = ruledb.find(sv_pair{"category", "category0"});
+        EXPECT_EQ(rules.size(), 4);
+    }
+
+    {
+        auto rules = ruledb.find(sv_pair{"key"sv, "value0"sv});
+        EXPECT_EQ(rules.size(), 4);
+    }
+
+    using s_pair = std::pair<std::string, std::string>;
+    {
+        auto rules = ruledb.find(s_pair{"type"sv, "type1"});
+        EXPECT_EQ(rules.size(), 4);
+    }
+
+    {
+        auto rules = ruledb.find(s_pair{"category", "category1"});
+        EXPECT_EQ(rules.size(), 4);
+    }
+
+    {
+        auto rules = ruledb.find(s_pair{"key"sv, "value1"sv});
+        EXPECT_EQ(rules.size(), 4);
+    }
+
+    ruledb.erase(rules[3]->get_tags(), rules[3].get());
+    {
+        auto rules = ruledb.find(sv_pair{"type"sv, "type0"});
+        EXPECT_EQ(rules.size(), 3);
+    }
+
+    {
+        auto rules = ruledb.find(sv_pair{"category", "category0"});
+        EXPECT_EQ(rules.size(), 4);
+    }
+
+    {
+        auto rules = ruledb.find(sv_pair{"key"sv, "value0"sv});
+        EXPECT_EQ(rules.size(), 4);
+    }
+
+    {
+        auto rules = ruledb.find(sv_pair{"type"sv, "type1"});
+        EXPECT_EQ(rules.size(), 4);
+    }
+
+    {
+        auto rules = ruledb.find(sv_pair{"category", "category1"});
+        EXPECT_EQ(rules.size(), 3);
+    }
+
+    {
+        auto rules = ruledb.find(sv_pair{"key"sv, "value1"sv});
+        EXPECT_EQ(rules.size(), 3);
+    }
+
+    ruledb.erase(rules[0]->get_tags(), rules[0].get());
+    {
+        auto rules = ruledb.find(sv_pair{"type"sv, "type0"});
+        EXPECT_EQ(rules.size(), 2);
+    }
+
+    {
+        auto rules = ruledb.find(sv_pair{"category", "category0"});
+        EXPECT_EQ(rules.size(), 3);
+    }
+
+    {
+        auto rules = ruledb.find(sv_pair{"key"sv, "value0"sv});
+        EXPECT_EQ(rules.size(), 3);
+    }
+
+    {
+        auto rules = ruledb.find(sv_pair{"type"sv, "type1"});
+        EXPECT_EQ(rules.size(), 4);
+    }
+
+    {
+        auto rules = ruledb.find(sv_pair{"category", "category1"});
+        EXPECT_EQ(rules.size(), 3);
+    }
+
+    {
+        auto rules = ruledb.find(sv_pair{"key"sv, "value1"sv});
+        EXPECT_EQ(rules.size(), 3);
+    }
+
+    ruledb.erase(rules[7]->get_tags(), rules[7].get());
+    {
+        auto rules = ruledb.find(sv_pair{"type"sv, "type0"});
+        EXPECT_EQ(rules.size(), 2);
+    }
+
+    {
+        auto rules = ruledb.find(sv_pair{"category", "category0"});
+        EXPECT_EQ(rules.size(), 3);
+    }
+
+    {
+        auto rules = ruledb.find(sv_pair{"key"sv, "value0"sv});
+        EXPECT_EQ(rules.size(), 3);
+    }
+
+    {
+        auto rules = ruledb.find(sv_pair{"type"sv, "type1"});
+        EXPECT_EQ(rules.size(), 3);
+    }
+
+    {
+        auto rules = ruledb.find(sv_pair{"category", "category1"});
+        EXPECT_EQ(rules.size(), 2);
+    }
+
+    {
+        auto rules = ruledb.find(sv_pair{"key"sv, "value1"sv});
+        EXPECT_EQ(rules.size(), 2);
+    }
+}
+
 } // namespace