intermediate changes

ref:2d4f292087954c9344efdabb7b2a67f466263c65

3 files changed, 93 insertions, 0 deletions

diff --git a/tools/enum_parser/parse_enum/benchmark_build/README.md b/tools/enum_parser/parse_enum/benchmark_build/README.md
new file mode 100644
index 0000000000..255592bc99
--- /dev/null
+++ b/tools/enum_parser/parse_enum/benchmark_build/README.md
@@ -0,0 +1,2 @@
+Модуль для проверки компилируемости больших перечислений и замера времени.
+Размечен как unittest, но проверки делаются во время компиляции.
diff --git a/tools/enum_parser/parse_enum/benchmark_build/lib/gen.py b/tools/enum_parser/parse_enum/benchmark_build/lib/gen.py
new file mode 100644
index 0000000000..783ab9f181
--- /dev/null
+++ b/tools/enum_parser/parse_enum/benchmark_build/lib/gen.py
@@ -0,0 +1,60 @@
+#!/usr/bin/env python3
+# - * - encoding: UTF-8 - * -
+
+from argparse import ArgumentParser
+import random
+import sys
+import math
+
+
+def parse_args():
+    parser = ArgumentParser(description="")
+    parser.add_argument('--range', type=int)
+    parser.add_argument('--enum', nargs=2, action="append", metavar=("NAME", "SIZE"))
+    parser.add_argument('--namespace', type=str)
+    args = parser.parse_args()
+    return args
+
+
+def gen_enum(name, n):
+    rg = random.Random(n)
+    h1 = list(range(n))
+    h2 = list(range(n))
+    rg.shuffle(h1)
+    rg.shuffle(h2)
+
+    print("enum class %s {" % name)
+    for  k, v in zip(h1, h2):
+        print("    V%x = 0x%04x," % (k, v))
+    print("};")
+    print()
+
+
+def main():
+    args = parse_args()
+
+    print("#pragma once\n\n")
+
+    gr = {}
+    for name, size in args.enum or []:
+        assert name not in gr
+        gr[name] = int(size)
+    if args.range:
+        step = max(int(math.sqrt(args.range)), 1)
+        for s in range(args.range, -1, -step):
+            gr["EDenseEnum%04d" % s] = s
+
+    if args.namespace:
+        print(f"namespace {args.namespace} {{")
+
+    for name, size in sorted(gr.items(), key=lambda kv: -kv[1]):
+        gen_enum(name, size)
+
+    if args.namespace:
+        print(f"}} // namespace {args.namespace}")
+
+    return 0
+
+
+if __name__ == '__main__':
+    sys.exit(main())
diff --git a/tools/enum_parser/parse_enum/benchmark_build/ut/huge_enums_fallback_ut.cpp b/tools/enum_parser/parse_enum/benchmark_build/ut/huge_enums_fallback_ut.cpp
new file mode 100644
index 0000000000..8d6800946f
--- /dev/null
+++ b/tools/enum_parser/parse_enum/benchmark_build/ut/huge_enums_fallback_ut.cpp
@@ -0,0 +1,31 @@
+#include <tools/enum_parser/parse_enum/benchmark_build/lib/enum_huge.h_serialized.h>
+#include <tools/enum_parser/parse_enum/benchmark_build/lib/enum_enormous.h_serialized.h>
+#include <library/cpp/testing/unittest/registar.h>
+#include <util/generic/serialized_enum.h>
+#include <util/string/cast.h>
+
+
+namespace {
+    template <class EEnum>
+    void EnumerateEnum() {
+        const auto& allValues = GetEnumAllValues<EEnum>();
+
+        TString s;
+        for (EEnum e : allValues) {
+            UNIT_ASSERT_NO_EXCEPTION(s = ToString(e));
+            UNIT_ASSERT_NO_EXCEPTION(e = FromString<EEnum>(s));
+        }
+
+        EEnum tmp;
+        UNIT_ASSERT_VALUES_EQUAL(TryFromString<EEnum>("", tmp), false);
+    }
+}
+
+Y_UNIT_TEST_SUITE(TTestHugeEnums) {
+    Y_UNIT_TEST(Huge) {
+        EnumerateEnum<NHuge::EHuge>();
+    }
+    Y_UNIT_TEST(Enormous) {
+        EnumerateEnum<EEnormous>();
+    }
+};