aboutsummaryrefslogtreecommitdiffstats
path: root/library/cpp/yaml/as/ut/tstring_test.cpp
blob: 4ea77b5a7ad3b61f2fa65edd02ca4d5a12731f92 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
#include <library/cpp/testing/unittest/registar.h> 
#include <library/cpp/yaml/as/tstring.h>

#include <contrib/libs/yaml-cpp/include/yaml-cpp/yaml.h>

#include <array>

Y_UNIT_TEST_SUITE(YamlTstringTest) {
    Y_UNIT_TEST(As) {
        const auto* key1 = "key1";
        const auto* key2 = "key2";
        const TString string = "string";
        const TString anotherString = "another string";

        YAML::Node map;
        map[key1] = std::string(string);

        UNIT_ASSERT_VALUES_EQUAL(map[key1].as<TString>(), string);
        UNIT_ASSERT_VALUES_EQUAL(map[key1].as<TString>(anotherString), string);

        UNIT_ASSERT_VALUES_EQUAL(map[key2].as<TString>(anotherString), anotherString);

        UNIT_ASSERT_EXCEPTION(map[key1].as<int>(), YAML::BadConversion);

        UNIT_ASSERT_EXCEPTION(map[key2].as<int>(), YAML::BadConversion);
        UNIT_ASSERT_EXCEPTION(map[key2].as<TString>(), YAML::BadConversion);
    }

    Y_UNIT_TEST(Convert) {
        const TString string("string");
        auto node = YAML::Node(std::string(string));

        UNIT_ASSERT_VALUES_EQUAL(node.Scalar(), YAML::convert<TString>::encode(string).Scalar());

        TString buffer;
        YAML::convert<TString>::decode(node, buffer);
        UNIT_ASSERT_VALUES_EQUAL(string, buffer);
    }

    Y_UNIT_TEST(NullCharactersInString) {
        const std::array<char, 3> characters = {{'a', '\0', 'b'}};
        const std::string testString(characters.data(), characters.size());
        const TString refString(characters.data(), characters.size());
        const std::string key = "key";

        YAML::Node node;
        node[key] = testString;

        UNIT_ASSERT_VALUES_EQUAL(node[key].as<TString>(), refString);
        UNIT_ASSERT_VALUES_EQUAL(node[key].as<TString>(TString("fallback")), refString);

        auto stringNode = YAML::Node(testString);
        UNIT_ASSERT_VALUES_EQUAL(stringNode.Scalar(), YAML::convert<TString>::encode(refString).Scalar());

        TString buffer;
        YAML::convert<TString>::decode(stringNode, buffer);
        UNIT_ASSERT_VALUES_EQUAL(refString, buffer);
    }
}