1 files changed, 107 insertions, 0 deletions
diff --git a/test/buildtool/build_engine/expression/configuration.test.cpp b/test/buildtool/build_engine/expression/configuration.test.cpp
new file mode 100644
index 00000000..95145038
--- /dev/null
+++ b/test/buildtool/build_engine/expression/configuration.test.cpp
@@ -0,0 +1,107 @@
+#include <vector>
+
+#include "catch2/catch.hpp"
+#include "src/buildtool/build_engine/expression/configuration.hpp"
+#include "test/utils/container_matchers.hpp"
+
+TEST_CASE("Access", "[configuration]") {
+    auto env =
+        Configuration{Expression::FromJson(R"({"foo": 1, "bar": 2})"_json)};
+
+    CHECK(env["foo"] == Expression::FromJson("1"_json));
+    CHECK(env[Expression::FromJson(R"("bar")"_json)] ==
+          Expression::FromJson("2"_json));
+
+    CHECK(env["baz"] == Expression::FromJson(R"(null)"_json));
+    CHECK(env[Expression::FromJson(R"("baz")"_json)] ==
+          Expression::FromJson(R"(null)"_json));
+}
+
+TEST_CASE("Update", "[configuration]") {
+    SECTION("Append") {
+        auto env = Configuration{Expression::FromJson(R"({})"_json)};
+        env = env.Update(Expression::FromJson(R"({"foo": 1})"_json));
+        CHECK(env["foo"] == Expression::FromJson("1"_json));
+
+        env = env.Update("bar", Expression::number_t{2});
+        CHECK(env["bar"] == Expression::FromJson("2"_json));
+
+        env = env.Update(Expression::map_t::underlying_map_t{
+            {"baz", ExpressionPtr{Expression::number_t{3}}}});
+        CHECK(env["baz"] == Expression::FromJson("3"_json));
+    }
+
+    SECTION("Overwrite") {
+        auto env = Configuration{
+            Expression::FromJson(R"({"foo": 1, "bar": 2, "baz" : 3})"_json)};
+        CHECK(env["foo"] == Expression::FromJson("1"_json));
+        CHECK(env["bar"] == Expression::FromJson("2"_json));
+        CHECK(env["baz"] == Expression::FromJson("3"_json));
+
+        env = env.Update(Expression::FromJson(R"({"foo": 10})"_json));
+        CHECK(env["foo"] == Expression::FromJson("10"_json));
+        CHECK(env["bar"] == Expression::FromJson("2"_json));
+        CHECK(env["baz"] == Expression::FromJson("3"_json));
+
+        env = env.Update("bar", Expression::number_t{20});  // NOLINT
+        CHECK(env["foo"] == Expression::FromJson("10"_json));
+        CHECK(env["bar"] == Expression::FromJson("20"_json));
+        CHECK(env["baz"] == Expression::FromJson("3"_json));
+
+        env = env.Update(Expression::map_t::underlying_map_t{
+            {"baz", ExpressionPtr{Expression::number_t{30}}}});  // NOLINT
+        CHECK(env["foo"] == Expression::FromJson("10"_json));
+        CHECK(env["bar"] == Expression::FromJson("20"_json));
+        CHECK(env["baz"] == Expression::FromJson("30"_json));
+    }
+}
+
+TEST_CASE("Prune", "[configuration]") {
+    auto env =
+        Configuration{Expression::FromJson(R"({"foo": 1, "bar": 2})"_json)};
+    CHECK(env["foo"] == Expression::FromJson("1"_json));
+    CHECK(env["bar"] == Expression::FromJson("2"_json));
+
+    SECTION("Via string list") {
+        env = env.Prune(std::vector<std::string>{"foo", "bar", "baz"});
+        CHECK(env["foo"] == Expression::FromJson("1"_json));
+        CHECK(env["bar"] == Expression::FromJson("2"_json));
+
+        env = env.Prune(std::vector<std::string>{"foo", "bar"});
+        CHECK(env["foo"] == Expression::FromJson("1"_json));
+        CHECK(env["bar"] == Expression::FromJson("2"_json));
+
+        env = env.Prune(std::vector<std::string>{"foo"});
+        CHECK(env["foo"] == Expression::FromJson("1"_json));
+        CHECK(env["bar"] == Expression::FromJson(R"(null)"_json));
+
+        env = env.Prune(std::vector<std::string>{});
+        CHECK(env["foo"] == Expression::FromJson(R"(null)"_json));
+        CHECK(env["bar"] == Expression::FromJson(R"(null)"_json));
+    }
+
+    SECTION("Via expression") {
+        env = env.Prune(Expression::FromJson(R"(["foo", "bar", "baz"])"_json));
+        CHECK(env["foo"] == Expression::FromJson("1"_json));
+        CHECK(env["bar"] == Expression::FromJson("2"_json));
+
+        env = env.Prune(Expression::FromJson(R"(["foo", "bar"])"_json));
+        CHECK(env["foo"] == Expression::FromJson("1"_json));
+        CHECK(env["bar"] == Expression::FromJson("2"_json));
+
+        env = env.Prune(Expression::FromJson(R"(["foo"])"_json));
+        CHECK(env["foo"] == Expression::FromJson("1"_json));
+        CHECK(env["bar"] == Expression::FromJson(R"(null)"_json));
+
+        env = env.Prune(Expression::FromJson(R"([])"_json));
+        CHECK(env["foo"] == Expression::FromJson(R"(null)"_json));
+        CHECK(env["bar"] == Expression::FromJson(R"(null)"_json));
+
+        CHECK_THROWS_AS(env.Prune(Expression::FromJson(
+                            R"(["not_all_string", false])"_json)),
+                        Expression::ExpressionTypeError);
+
+        CHECK_THROWS_AS(env.Prune(Expression::FromJson(R"("not_a_list")"_json)),
+                        Expression::ExpressionTypeError);
+    }
+}