intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 98 insertions, 0 deletions

diff --git a/library/cpp/logger/composite_ut.cpp b/library/cpp/logger/composite_ut.cpp
new file mode 100644
index 0000000000..f330419271
--- /dev/null
+++ b/library/cpp/logger/composite_ut.cpp
@@ -0,0 +1,98 @@
+#include "log.h"
+#include <library/cpp/logger/init_context/config.h>
+#include <library/cpp/logger/init_context/yconf.h>
+#include <library/cpp/testing/unittest/registar.h>
+#include <library/cpp/yconf/patcher/unstrict_config.h>
+#include <util/stream/file.h>
+#include <util/system/fs.h>
+
+Y_UNIT_TEST_SUITE(TCompositeLogTest)
+{
+    TVector<TString> ReadLines(const TString & filename) {
+        TVector<TString> lines;
+        TIFStream fin(filename);
+        TString line;
+        while (fin.ReadLine(line)) {
+            lines.push_back(std::move(line));
+        }
+        return lines;
+    }
+
+    void Clear(const TString & filename) {
+        NFs::Remove(filename + "1");
+        NFs::Remove(filename + "2");
+    }
+
+    void DoTestComposite(const ILogBackendCreator::IInitContext& ctx, const TString & filename) {
+        Clear(filename);
+        {
+            TLog log;
+            {
+                auto creator = ILogBackendCreator::Create(ctx);
+                log.ResetBackend(creator->CreateLogBackend());
+                log.ReopenLog();
+            }
+            log.Write(TLOG_ERR, "first\n");
+            log.Write(TLOG_DEBUG, "second\n");
+        }
+        auto data1 = ReadLines(filename + "1");
+        auto data2 = ReadLines(filename + "2");
+        UNIT_ASSERT_VALUES_EQUAL(data1.size(), 2);
+        UNIT_ASSERT(data1[0] == "first");
+        UNIT_ASSERT(data1[1] == "second");
+
+        UNIT_ASSERT_VALUES_EQUAL(data2.size(), 1);
+        UNIT_ASSERT(data2[0] == "first");
+        Clear(filename);
+    }
+
+    Y_UNIT_TEST(TestCompositeConfig) {
+        TString s(R"(
+{
+    "LoggerType": "composite",
+    "SubLogger":[
+        {
+            "LoggerType": "file",
+            "Path": "config_log_1"
+        }, {
+            "LoggerType": "config_log_2",
+            "LogLevel": "INFO"
+        }
+    ]
+})");
+        TStringInput si(s);
+        NConfig::TConfig cfg = NConfig::TConfig::FromJson(si);
+        //Прогоняем конфигурацию через серализацию и десериализацию
+        TLogBackendCreatorInitContextConfig ctx(cfg);
+        TString newCfg = ILogBackendCreator::Create(ctx)->AsJson().GetStringRobust();
+        TStringInput si2(newCfg);
+        DoTestComposite(TLogBackendCreatorInitContextConfig(NConfig::TConfig::FromJson(si2)), "config_log_");
+
+    }
+    Y_UNIT_TEST(TestCompositeYConf) {
+        constexpr const char* CONFIG = R"(
+<Logger>
+    LoggerType: composite
+    <SubLogger>
+        LoggerType: file
+        Path: yconf_log_1
+    </SubLogger>
+    <SubLogger>
+        LoggerType: yconf_log_2
+        LogLevel: INFO
+    </SubLogger>
+</Logger>
+)";
+        TUnstrictConfig cfg;
+        if (!cfg.ParseMemory(CONFIG)) {
+            TString errors;
+            cfg.PrintErrors(errors);
+            UNIT_ASSERT_C(false, errors);
+        }
+        TLogBackendCreatorInitContextYConf ctx(*cfg.GetFirstChild("Logger"));
+        //Прогоняем конфигурацию через серализацию и десериализацию
+        TUnstrictConfig newCfg;
+        UNIT_ASSERT(newCfg.ParseJson(ILogBackendCreator::Create(ctx)->AsJson()));
+        DoTestComposite(TLogBackendCreatorInitContextYConf(*newCfg.GetRootSection()), "yconf_log_");
+    }
+};