YT: Allow to define enum with custom string represenation

1 files changed, 14 insertions, 0 deletions

diff --git a/library/cpp/yt/misc/unittests/enum_ut.cpp b/library/cpp/yt/misc/unittests/enum_ut.cpp
index 6af11b04dc0..2a024be228c 100644
--- a/library/cpp/yt/misc/unittests/enum_ut.cpp
+++ b/library/cpp/yt/misc/unittests/enum_ut.cpp
@@ -37,6 +37,11 @@ DEFINE_AMBIGUOUS_ENUM_WITH_UNDERLYING_TYPE(EMultipleNames, int,
     ((D2)(100))
 );
 
+DEFINE_ENUM(ECustomString,
+    ((A) (1) ("1_a"))
+    ((B) (2) ("1_b"))
+);
+
 ////////////////////////////////////////////////////////////////////////////////
 
 template <class T, size_t N>
@@ -243,6 +248,15 @@ TEST(TEnumTest, MultipleNames)
     EXPECT_EQ("D1", ToString(EMultipleNames::D2));
 }
 
+TEST(TEnumTest, CustomString)
+{
+    EXPECT_EQ(ECustomString::A, TEnumTraits<ECustomString>::FromString("1_a"));
+    EXPECT_EQ(ECustomString::B, TEnumTraits<ECustomString>::FromString("1_b"));
+
+    EXPECT_EQ("1_a", ToString(ECustomString::A));
+    EXPECT_EQ("1_b", ToString(ECustomString::B));
+}
+
 ////////////////////////////////////////////////////////////////////////////////
 
 } // namespace