intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 228 insertions, 0 deletions

diff --git a/library/cpp/json/ut/json_writer_ut.cpp b/library/cpp/json/ut/json_writer_ut.cpp
new file mode 100644
index 0000000000..ca11d34dad
--- /dev/null
+++ b/library/cpp/json/ut/json_writer_ut.cpp
@@ -0,0 +1,228 @@
+#include <library/cpp/json/json_writer.h>
+#include <library/cpp/testing/unittest/registar.h>
+
+#include <util/stream/str.h>
+
+using namespace NJson;
+
+Y_UNIT_TEST_SUITE(TJsonWriterTest) {
+    Y_UNIT_TEST(SimpleWriteTest) {
+        TString expected1 = "{\"key1\":1,\"key2\":2,\"key3\":3";
+        TString expected2 = expected1 + ",\"array\":[\"stroka\",false]";
+        TString expected3 = expected2 + "}";
+
+        TStringStream out;
+
+        TJsonWriter json(&out, false);
+        json.OpenMap();
+        json.Write("key1", (ui16)1);
+        json.WriteKey("key2");
+        json.Write((i32)2);
+        json.Write("key3", (ui64)3);
+
+        UNIT_ASSERT(out.Empty());
+        json.Flush();
+        UNIT_ASSERT_VALUES_EQUAL(out.Str(), expected1);
+
+        json.Write("array");
+        json.OpenArray();
+        json.Write("stroka");
+        json.Write(false);
+        json.CloseArray();
+
+        UNIT_ASSERT_VALUES_EQUAL(out.Str(), expected1);
+        json.Flush();
+        UNIT_ASSERT_VALUES_EQUAL(out.Str(), expected2);
+
+        json.CloseMap();
+
+        UNIT_ASSERT_VALUES_EQUAL(out.Str(), expected2);
+        json.Flush();
+        UNIT_ASSERT_VALUES_EQUAL(out.Str(), expected3);
+    }
+
+    Y_UNIT_TEST(SimpleWriteValueTest) {
+        TString expected = "{\"key1\":null,\"key2\":{\"subkey1\":[1,{\"subsubkey\":\"test2\"},null,true],\"subkey2\":\"test\"}}";
+        TJsonValue v;
+        v["key1"] = JSON_NULL;
+        v["key2"]["subkey1"].AppendValue(1);
+        v["key2"]["subkey1"].AppendValue(JSON_MAP)["subsubkey"] = "test2";
+        v["key2"]["subkey1"].AppendValue(JSON_NULL);
+        v["key2"]["subkey1"].AppendValue(true);
+        v["key2"]["subkey2"] = "test";
+        TStringStream out;
+        WriteJson(&out, &v);
+        UNIT_ASSERT_VALUES_EQUAL(out.Str(), expected);
+    }
+
+    Y_UNIT_TEST(FormatOutput) {
+        TString expected = "{\n  \"key1\":null,\n  \"key2\":\n    {\n      \"subkey1\":\n        [\n          1,\n          {\n            \"subsubkey\":\"test2\"\n          },\n          null,\n          true\n        ],\n      \"subkey2\":\"test\"\n    }\n}";
+        TJsonValue v;
+        v["key1"] = JSON_NULL;
+        v["key2"]["subkey1"].AppendValue(1);
+        v["key2"]["subkey1"].AppendValue(JSON_MAP)["subsubkey"] = "test2";
+        v["key2"]["subkey1"].AppendValue(JSON_NULL);
+        v["key2"]["subkey1"].AppendValue(true);
+        v["key2"]["subkey2"] = "test";
+        TStringStream out;
+        WriteJson(&out, &v, true);
+        UNIT_ASSERT_STRINGS_EQUAL(out.Str(), expected);
+    }
+
+    Y_UNIT_TEST(SortKeys) {
+        TString expected = "{\"a\":null,\"j\":null,\"n\":null,\"y\":null,\"z\":null}";
+        TJsonValue v;
+        v["z"] = JSON_NULL;
+        v["n"] = JSON_NULL;
+        v["a"] = JSON_NULL;
+        v["y"] = JSON_NULL;
+        v["j"] = JSON_NULL;
+        TStringStream out;
+        WriteJson(&out, &v, false, true);
+        UNIT_ASSERT_STRINGS_EQUAL(out.Str(), expected);
+    }
+
+    Y_UNIT_TEST(SimpleUnsignedIntegerWriteTest) {
+        {
+            TString expected = "{\"test\":1}";
+            TJsonValue v;
+            v.InsertValue("test", 1ull);
+            TStringStream out;
+            WriteJson(&out, &v);
+            UNIT_ASSERT_VALUES_EQUAL(out.Str(), expected);
+        } // 1
+
+        {
+            TString expected = "{\"test\":-1}";
+            TJsonValue v;
+            v.InsertValue("test", -1);
+            TStringStream out;
+            WriteJson(&out, &v);
+            UNIT_ASSERT_VALUES_EQUAL(out.Str(), expected);
+        } // -1
+
+        {
+            TString expected = "{\"test\":18446744073709551615}";
+            TJsonValue v;
+            v.InsertValue("test", 18446744073709551615ull);
+            TStringStream out;
+            WriteJson(&out, &v);
+            UNIT_ASSERT_VALUES_EQUAL(out.Str(), expected);
+        } // 18446744073709551615
+
+        {
+            TString expected = "{\"test\":[1,18446744073709551615]}";
+            TJsonValue v;
+            v.InsertValue("test", TJsonValue());
+            v["test"].AppendValue(1);
+            v["test"].AppendValue(18446744073709551615ull);
+            TStringStream out;
+            WriteJson(&out, &v);
+            UNIT_ASSERT_VALUES_EQUAL(out.Str(), expected);
+        } // 18446744073709551615
+    }     // SimpleUnsignedIntegerWriteTest
+
+    Y_UNIT_TEST(WriteOptionalTest) {
+        {
+            TString expected = "{\"test\":1}";
+
+            TStringStream out;
+
+            TJsonWriter json(&out, false);
+            json.OpenMap();
+            json.WriteOptional("test", MakeMaybe<int>(1));
+            json.CloseMap();
+            json.Flush();
+            UNIT_ASSERT_VALUES_EQUAL(out.Str(), expected);
+        }
+
+        {
+            TString expected = "{}";
+
+            TStringStream out;
+
+            TMaybe<int> nothing = Nothing();
+
+            TJsonWriter json(&out, false);
+            json.OpenMap();
+            json.WriteOptional("test", nothing);
+            json.CloseMap();
+            json.Flush();
+            UNIT_ASSERT_VALUES_EQUAL(out.Str(), expected);
+        }
+
+        {
+            TString expected = "{}";
+
+            TStringStream out;
+
+            TMaybe<int> empty;
+
+            TJsonWriter json(&out, false);
+            json.OpenMap();
+            json.WriteOptional("test", empty);
+            json.CloseMap();
+            json.Flush();
+            UNIT_ASSERT_VALUES_EQUAL(out.Str(), expected);
+        }
+
+        {
+            TString expected = "{}";
+
+            TStringStream out;
+
+            TJsonWriter json(&out, false);
+            json.OpenMap();
+            json.WriteOptional("test", Nothing());
+            json.CloseMap();
+            json.Flush();
+            UNIT_ASSERT_VALUES_EQUAL(out.Str(), expected);
+        }
+    }
+
+    Y_UNIT_TEST(Callback) {
+        NJsonWriter::TBuf json;
+        json.WriteString("A");
+        UNIT_ASSERT_VALUES_EQUAL(json.Str(), "\"A\"");
+        UNIT_ASSERT_VALUES_EQUAL(WrapJsonToCallback(json, ""), "\"A\"");
+        UNIT_ASSERT_VALUES_EQUAL(WrapJsonToCallback(json, "Foo"), "Foo(\"A\")");
+    }
+
+    Y_UNIT_TEST(FloatPrecision) {
+        const double value = 1517933989.4242;
+        const NJson::TJsonValue json(value);
+        NJson::TJsonWriterConfig config;
+        {
+            TString expected = "1517933989";
+            TString actual = NJson::WriteJson(json);
+            UNIT_ASSERT_VALUES_EQUAL(actual, expected);
+        }
+        {
+            TString expected = "1517933989";
+
+            TStringStream ss;
+            NJson::WriteJson(&ss, &json, config);
+            TString actual = ss.Str();
+            UNIT_ASSERT_VALUES_EQUAL(actual, expected);
+        }
+        {
+            config.DoubleNDigits = 13;
+            TString expected = "1517933989.424";
+
+            TStringStream ss;
+            NJson::WriteJson(&ss, &json, config);
+            TString actual = ss.Str();
+            UNIT_ASSERT_VALUES_EQUAL(actual, expected);
+        }
+        {
+            config.DoubleNDigits = 6;
+            config.FloatToStringMode = PREC_POINT_DIGITS;
+            TString expected = "1517933989.424200";
+
+            TStringStream ss;
+            NJson::WriteJson(&ss, &json, config);
+            TString actual = ss.Str();
+            UNIT_ASSERT_VALUES_EQUAL(actual, expected);
+        }
+    }
+}