Support Save/Load TYsonString

1 files changed, 46 insertions, 0 deletions

diff --git a/library/cpp/yt/yson_string/unittests/saveload_ut.cpp b/library/cpp/yt/yson_string/unittests/saveload_ut.cpp
new file mode 100644
index 0000000000..33a98a30ea
--- /dev/null
+++ b/library/cpp/yt/yson_string/unittests/saveload_ut.cpp
@@ -0,0 +1,46 @@
+#include <library/cpp/testing/gtest/gtest.h>
+
+#include <library/cpp/testing/gtest_extensions/assertions.h>
+
+#include <library/cpp/yt/yson_string/string.h>
+
+namespace NYT::NYson {
+namespace {
+
+////////////////////////////////////////////////////////////////////////////////
+
+TEST(TYsonStringTest, SaveLoadNull)
+{
+    const TYsonString expected;
+    TStringStream s;
+    ::Save(&s, expected);
+    TYsonString result;
+    ::Load(&s, result);
+    EXPECT_EQ(expected, result);
+}
+
+TEST(TYsonStringTest, SaveLoadString)
+{
+    const TYsonString expected(TString("My tests data"));
+    TStringStream s;
+    ::Save(&s, expected);
+    TYsonString result;
+    ::Load(&s, result);
+    EXPECT_EQ(expected, result);
+}
+
+TEST(TYsonStringTest, SaveLoadSharedRef)
+{
+    TSharedRef ref = TSharedRef::FromString("My tests data");
+    const TYsonString expected(ref);
+    TStringStream s;
+    ::Save(&s, expected);
+    TYsonString result;
+    ::Load(&s, result);
+    EXPECT_EQ(expected, result);
+}
+
+////////////////////////////////////////////////////////////////////////////////
+
+} // namespace
+} // namespace NYT::NYson