intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 136 insertions, 0 deletions

diff --git a/library/cpp/monlib/encode/format_ut.cpp b/library/cpp/monlib/encode/format_ut.cpp
new file mode 100644
index 0000000000..22a0e30c03
--- /dev/null
+++ b/library/cpp/monlib/encode/format_ut.cpp
@@ -0,0 +1,136 @@
+#include "format.h"
+
+#include <library/cpp/testing/unittest/registar.h>
+
+#include <util/generic/string.h>
+#include <util/string/builder.h>
+
+#include <array>
+
+using namespace NMonitoring;
+
+Y_UNIT_TEST_SUITE(TFormatTest) {
+    Y_UNIT_TEST(ContentTypeHeader) {
+        UNIT_ASSERT_EQUAL(FormatFromContentType(""), EFormat::UNKNOWN);
+        UNIT_ASSERT_EQUAL(FormatFromContentType("application/json;some=stuff"), EFormat::JSON);
+        UNIT_ASSERT_EQUAL(FormatFromContentType("application/x-solomon-spack"), EFormat::SPACK);
+        UNIT_ASSERT_EQUAL(FormatFromContentType("application/xml"), EFormat::UNKNOWN);
+        UNIT_ASSERT_EQUAL(FormatFromContentType(";application/xml"), EFormat::UNKNOWN);
+    }
+
+    Y_UNIT_TEST(AcceptHeader) {
+        UNIT_ASSERT_EQUAL(FormatFromAcceptHeader(""), EFormat::UNKNOWN);
+        UNIT_ASSERT_EQUAL(FormatFromAcceptHeader("*/*"), EFormat::UNKNOWN);
+
+        UNIT_ASSERT_EQUAL(
+            FormatFromAcceptHeader("application/xml"),
+            EFormat::UNKNOWN);
+
+        UNIT_ASSERT_EQUAL(
+            FormatFromAcceptHeader("application/json"),
+            EFormat::JSON);
+
+        UNIT_ASSERT_EQUAL(
+            FormatFromAcceptHeader("application/x-solomon-spack"),
+            EFormat::SPACK);
+
+        UNIT_ASSERT_EQUAL(
+            FormatFromAcceptHeader("application/x-solomon-pb"),
+            EFormat::PROTOBUF);
+
+        UNIT_ASSERT_EQUAL(
+            FormatFromAcceptHeader("application/x-solomon-txt"),
+            EFormat::TEXT);
+
+        UNIT_ASSERT_EQUAL(
+            FormatFromAcceptHeader("application/json, text/plain"),
+            EFormat::JSON);
+
+        UNIT_ASSERT_EQUAL(
+            FormatFromAcceptHeader("application/x-solomon-spack, application/json, text/plain"),
+            EFormat::SPACK);
+
+        UNIT_ASSERT_EQUAL(
+            FormatFromAcceptHeader(" , application/x-solomon-spack ,, application/json , text/plain"),
+            EFormat::SPACK);
+
+        UNIT_ASSERT_EQUAL(
+            FormatFromAcceptHeader("application/xml, application/x-solomon-spack, text/plain"),
+            EFormat::SPACK);
+
+        UNIT_ASSERT_EQUAL(
+            FormatFromAcceptHeader("text/plain"),
+            EFormat::PROMETHEUS);
+    }
+
+    Y_UNIT_TEST(FormatToStrFromStr) {
+        const std::array<EFormat, 6> formats = {{
+            EFormat::UNKNOWN,
+            EFormat::SPACK,
+            EFormat::JSON,
+            EFormat::PROTOBUF,
+            EFormat::TEXT,
+            EFormat::PROMETHEUS,
+        }};
+
+        for (EFormat f : formats) {
+            TString str = (TStringBuilder() << f);
+            EFormat g = FromString<EFormat>(str);
+            UNIT_ASSERT_EQUAL(f, g);
+        }
+    }
+
+    Y_UNIT_TEST(AcceptEncodingHeader) {
+        UNIT_ASSERT_EQUAL(
+            CompressionFromAcceptEncodingHeader(""),
+            ECompression::UNKNOWN);
+
+        UNIT_ASSERT_EQUAL(
+            CompressionFromAcceptEncodingHeader("br"),
+            ECompression::UNKNOWN);
+
+        UNIT_ASSERT_EQUAL(
+            CompressionFromAcceptEncodingHeader("identity"),
+            ECompression::IDENTITY);
+
+        UNIT_ASSERT_EQUAL(
+            CompressionFromAcceptEncodingHeader("zlib"),
+            ECompression::ZLIB);
+
+        UNIT_ASSERT_EQUAL(
+            CompressionFromAcceptEncodingHeader("lz4"),
+            ECompression::LZ4);
+
+        UNIT_ASSERT_EQUAL(
+            CompressionFromAcceptEncodingHeader("zstd"),
+            ECompression::ZSTD);
+
+        UNIT_ASSERT_EQUAL(
+            CompressionFromAcceptEncodingHeader("zstd, zlib"),
+            ECompression::ZSTD);
+
+        UNIT_ASSERT_EQUAL(
+            CompressionFromAcceptEncodingHeader(" ,, , zstd , zlib"),
+            ECompression::ZSTD);
+
+        UNIT_ASSERT_EQUAL(
+            CompressionFromAcceptEncodingHeader("br, deflate,lz4, zlib"),
+            ECompression::LZ4);
+    }
+
+    Y_UNIT_TEST(CompressionToStrFromStr) {
+        const std::array<ECompression, 5> algs = {{
+            ECompression::UNKNOWN,
+            ECompression::IDENTITY,
+            ECompression::ZLIB,
+            ECompression::LZ4,
+            ECompression::ZSTD,
+        }};
+
+        for (ECompression a : algs) {
+            TString str = (TStringBuilder() << a);
+            ECompression b = FromString<ECompression>(str);
+            UNIT_ASSERT_EQUAL(a, b);
+        }
+    }
+}