summaryrefslogtreecommitdiffstats
path: root/gnu/llvm/unittests/Support/YAMLIOTest.cpp
diff options
context:
space:
mode:
authorpatrick <patrick@openbsd.org>2020-08-03 15:06:44 +0000
committerpatrick <patrick@openbsd.org>2020-08-03 15:06:44 +0000
commitb64793999546ed8adebaeebd9d8345d18db8927d (patch)
tree4357c27b561d73b0e089727c6ed659f2ceff5f47 /gnu/llvm/unittests/Support/YAMLIOTest.cpp
parentAdd support for UTF-8 DISPLAY-HINTs with octet length. For now only (diff)
downloadwireguard-openbsd-b64793999546ed8adebaeebd9d8345d18db8927d.tar.xz
wireguard-openbsd-b64793999546ed8adebaeebd9d8345d18db8927d.zip
Remove LLVM 8.0.1 files.
Diffstat (limited to 'gnu/llvm/unittests/Support/YAMLIOTest.cpp')
-rw-r--r--gnu/llvm/unittests/Support/YAMLIOTest.cpp2877
1 files changed, 0 insertions, 2877 deletions
diff --git a/gnu/llvm/unittests/Support/YAMLIOTest.cpp b/gnu/llvm/unittests/Support/YAMLIOTest.cpp
deleted file mode 100644
index 17f38f9b86c..00000000000
--- a/gnu/llvm/unittests/Support/YAMLIOTest.cpp
+++ /dev/null
@@ -1,2877 +0,0 @@
-//===- unittest/Support/YAMLIOTest.cpp ------------------------------------===//
-//
-// The LLVM Compiler Infrastructure
-//
-// This file is distributed under the University of Illinois Open Source
-// License. See LICENSE.TXT for details.
-//
-//===----------------------------------------------------------------------===//
-
-#include "llvm/ADT/StringMap.h"
-#include "llvm/ADT/StringRef.h"
-#include "llvm/ADT/Twine.h"
-#include "llvm/Support/Casting.h"
-#include "llvm/Support/Endian.h"
-#include "llvm/Support/Format.h"
-#include "llvm/Support/YAMLTraits.h"
-#include "gmock/gmock.h"
-#include "gtest/gtest.h"
-
-using llvm::yaml::Hex16;
-using llvm::yaml::Hex32;
-using llvm::yaml::Hex64;
-using llvm::yaml::Hex8;
-using llvm::yaml::Input;
-using llvm::yaml::IO;
-using llvm::yaml::isNumeric;
-using llvm::yaml::MappingNormalization;
-using llvm::yaml::MappingTraits;
-using llvm::yaml::Output;
-using llvm::yaml::ScalarTraits;
-using ::testing::StartsWith;
-
-
-
-
-static void suppressErrorMessages(const llvm::SMDiagnostic &, void *) {
-}
-
-
-
-//===----------------------------------------------------------------------===//
-// Test MappingTraits
-//===----------------------------------------------------------------------===//
-
-struct FooBar {
- int foo;
- int bar;
-};
-typedef std::vector<FooBar> FooBarSequence;
-
-LLVM_YAML_IS_SEQUENCE_VECTOR(FooBar)
-
-struct FooBarContainer {
- FooBarSequence fbs;
-};
-
-namespace llvm {
-namespace yaml {
- template <>
- struct MappingTraits<FooBar> {
- static void mapping(IO &io, FooBar& fb) {
- io.mapRequired("foo", fb.foo);
- io.mapRequired("bar", fb.bar);
- }
- };
-
- template <> struct MappingTraits<FooBarContainer> {
- static void mapping(IO &io, FooBarContainer &fb) {
- io.mapRequired("fbs", fb.fbs);
- }
- };
-}
-}
-
-
-//
-// Test the reading of a yaml mapping
-//
-TEST(YAMLIO, TestMapRead) {
- FooBar doc;
- {
- Input yin("---\nfoo: 3\nbar: 5\n...\n");
- yin >> doc;
-
- EXPECT_FALSE(yin.error());
- EXPECT_EQ(doc.foo, 3);
- EXPECT_EQ(doc.bar, 5);
- }
-
- {
- Input yin("{foo: 3, bar: 5}");
- yin >> doc;
-
- EXPECT_FALSE(yin.error());
- EXPECT_EQ(doc.foo, 3);
- EXPECT_EQ(doc.bar, 5);
- }
-}
-
-TEST(YAMLIO, TestMalformedMapRead) {
- FooBar doc;
- Input yin("{foo: 3; bar: 5}", nullptr, suppressErrorMessages);
- yin >> doc;
- EXPECT_TRUE(!!yin.error());
-}
-
-//
-// Test the reading of a yaml sequence of mappings
-//
-TEST(YAMLIO, TestSequenceMapRead) {
- FooBarSequence seq;
- Input yin("---\n - foo: 3\n bar: 5\n - foo: 7\n bar: 9\n...\n");
- yin >> seq;
-
- EXPECT_FALSE(yin.error());
- EXPECT_EQ(seq.size(), 2UL);
- FooBar& map1 = seq[0];
- FooBar& map2 = seq[1];
- EXPECT_EQ(map1.foo, 3);
- EXPECT_EQ(map1.bar, 5);
- EXPECT_EQ(map2.foo, 7);
- EXPECT_EQ(map2.bar, 9);
-}
-
-//
-// Test the reading of a map containing a yaml sequence of mappings
-//
-TEST(YAMLIO, TestContainerSequenceMapRead) {
- {
- FooBarContainer cont;
- Input yin2("---\nfbs:\n - foo: 3\n bar: 5\n - foo: 7\n bar: 9\n...\n");
- yin2 >> cont;
-
- EXPECT_FALSE(yin2.error());
- EXPECT_EQ(cont.fbs.size(), 2UL);
- EXPECT_EQ(cont.fbs[0].foo, 3);
- EXPECT_EQ(cont.fbs[0].bar, 5);
- EXPECT_EQ(cont.fbs[1].foo, 7);
- EXPECT_EQ(cont.fbs[1].bar, 9);
- }
-
- {
- FooBarContainer cont;
- Input yin("---\nfbs:\n...\n");
- yin >> cont;
- // Okay: Empty node represents an empty array.
- EXPECT_FALSE(yin.error());
- EXPECT_EQ(cont.fbs.size(), 0UL);
- }
-
- {
- FooBarContainer cont;
- Input yin("---\nfbs: !!null null\n...\n");
- yin >> cont;
- // Okay: null represents an empty array.
- EXPECT_FALSE(yin.error());
- EXPECT_EQ(cont.fbs.size(), 0UL);
- }
-
- {
- FooBarContainer cont;
- Input yin("---\nfbs: ~\n...\n");
- yin >> cont;
- // Okay: null represents an empty array.
- EXPECT_FALSE(yin.error());
- EXPECT_EQ(cont.fbs.size(), 0UL);
- }
-
- {
- FooBarContainer cont;
- Input yin("---\nfbs: null\n...\n");
- yin >> cont;
- // Okay: null represents an empty array.
- EXPECT_FALSE(yin.error());
- EXPECT_EQ(cont.fbs.size(), 0UL);
- }
-}
-
-//
-// Test the reading of a map containing a malformed yaml sequence
-//
-TEST(YAMLIO, TestMalformedContainerSequenceMapRead) {
- {
- FooBarContainer cont;
- Input yin("---\nfbs:\n foo: 3\n bar: 5\n...\n", nullptr,
- suppressErrorMessages);
- yin >> cont;
- // Error: fbs is not a sequence.
- EXPECT_TRUE(!!yin.error());
- EXPECT_EQ(cont.fbs.size(), 0UL);
- }
-
- {
- FooBarContainer cont;
- Input yin("---\nfbs: 'scalar'\n...\n", nullptr, suppressErrorMessages);
- yin >> cont;
- // This should be an error.
- EXPECT_TRUE(!!yin.error());
- EXPECT_EQ(cont.fbs.size(), 0UL);
- }
-}
-
-//
-// Test writing then reading back a sequence of mappings
-//
-TEST(YAMLIO, TestSequenceMapWriteAndRead) {
- std::string intermediate;
- {
- FooBar entry1;
- entry1.foo = 10;
- entry1.bar = -3;
- FooBar entry2;
- entry2.foo = 257;
- entry2.bar = 0;
- FooBarSequence seq;
- seq.push_back(entry1);
- seq.push_back(entry2);
-
- llvm::raw_string_ostream ostr(intermediate);
- Output yout(ostr);
- yout << seq;
- }
-
- {
- Input yin(intermediate);
- FooBarSequence seq2;
- yin >> seq2;
-
- EXPECT_FALSE(yin.error());
- EXPECT_EQ(seq2.size(), 2UL);
- FooBar& map1 = seq2[0];
- FooBar& map2 = seq2[1];
- EXPECT_EQ(map1.foo, 10);
- EXPECT_EQ(map1.bar, -3);
- EXPECT_EQ(map2.foo, 257);
- EXPECT_EQ(map2.bar, 0);
- }
-}
-
-//
-// Test YAML filename handling.
-//
-static void testErrorFilename(const llvm::SMDiagnostic &Error, void *) {
- EXPECT_EQ(Error.getFilename(), "foo.yaml");
-}
-
-TEST(YAMLIO, TestGivenFilename) {
- auto Buffer = llvm::MemoryBuffer::getMemBuffer("{ x: 42 }", "foo.yaml");
- Input yin(*Buffer, nullptr, testErrorFilename);
- FooBar Value;
- yin >> Value;
-
- EXPECT_TRUE(!!yin.error());
-}
-
-struct WithStringField {
- std::string str1;
- std::string str2;
- std::string str3;
-};
-
-namespace llvm {
-namespace yaml {
-template <> struct MappingTraits<WithStringField> {
- static void mapping(IO &io, WithStringField &fb) {
- io.mapRequired("str1", fb.str1);
- io.mapRequired("str2", fb.str2);
- io.mapRequired("str3", fb.str3);
- }
-};
-} // namespace yaml
-} // namespace llvm
-
-TEST(YAMLIO, MultilineStrings) {
- WithStringField Original;
- Original.str1 = "a multiline string\nfoobarbaz";
- Original.str2 = "another one\rfoobarbaz";
- Original.str3 = "a one-line string";
-
- std::string Serialized;
- {
- llvm::raw_string_ostream OS(Serialized);
- Output YOut(OS);
- YOut << Original;
- }
- auto Expected = "---\n"
- "str1: 'a multiline string\n"
- "foobarbaz'\n"
- "str2: 'another one\r"
- "foobarbaz'\n"
- "str3: a one-line string\n"
- "...\n";
- ASSERT_EQ(Serialized, Expected);
-
- // Also check it parses back without the errors.
- WithStringField Deserialized;
- {
- Input YIn(Serialized);
- YIn >> Deserialized;
- ASSERT_FALSE(YIn.error())
- << "Parsing error occurred during deserialization. Serialized string:\n"
- << Serialized;
- }
- EXPECT_EQ(Original.str1, Deserialized.str1);
- EXPECT_EQ(Original.str2, Deserialized.str2);
- EXPECT_EQ(Original.str3, Deserialized.str3);
-}
-
-TEST(YAMLIO, NoQuotesForTab) {
- WithStringField WithTab;
- WithTab.str1 = "aba\tcaba";
- std::string Serialized;
- {
- llvm::raw_string_ostream OS(Serialized);
- Output YOut(OS);
- YOut << WithTab;
- }
- auto ExpectedPrefix = "---\n"
- "str1: aba\tcaba\n";
- EXPECT_THAT(Serialized, StartsWith(ExpectedPrefix));
-}
-
-//===----------------------------------------------------------------------===//
-// Test built-in types
-//===----------------------------------------------------------------------===//
-
-struct BuiltInTypes {
- llvm::StringRef str;
- std::string stdstr;
- uint64_t u64;
- uint32_t u32;
- uint16_t u16;
- uint8_t u8;
- bool b;
- int64_t s64;
- int32_t s32;
- int16_t s16;
- int8_t s8;
- float f;
- double d;
- Hex8 h8;
- Hex16 h16;
- Hex32 h32;
- Hex64 h64;
-};
-
-namespace llvm {
-namespace yaml {
- template <>
- struct MappingTraits<BuiltInTypes> {
- static void mapping(IO &io, BuiltInTypes& bt) {
- io.mapRequired("str", bt.str);
- io.mapRequired("stdstr", bt.stdstr);
- io.mapRequired("u64", bt.u64);
- io.mapRequired("u32", bt.u32);
- io.mapRequired("u16", bt.u16);
- io.mapRequired("u8", bt.u8);
- io.mapRequired("b", bt.b);
- io.mapRequired("s64", bt.s64);
- io.mapRequired("s32", bt.s32);
- io.mapRequired("s16", bt.s16);
- io.mapRequired("s8", bt.s8);
- io.mapRequired("f", bt.f);
- io.mapRequired("d", bt.d);
- io.mapRequired("h8", bt.h8);
- io.mapRequired("h16", bt.h16);
- io.mapRequired("h32", bt.h32);
- io.mapRequired("h64", bt.h64);
- }
- };
-}
-}
-
-
-//
-// Test the reading of all built-in scalar conversions
-//
-TEST(YAMLIO, TestReadBuiltInTypes) {
- BuiltInTypes map;
- Input yin("---\n"
- "str: hello there\n"
- "stdstr: hello where?\n"
- "u64: 5000000000\n"
- "u32: 4000000000\n"
- "u16: 65000\n"
- "u8: 255\n"
- "b: false\n"
- "s64: -5000000000\n"
- "s32: -2000000000\n"
- "s16: -32000\n"
- "s8: -127\n"
- "f: 137.125\n"
- "d: -2.8625\n"
- "h8: 0xFF\n"
- "h16: 0x8765\n"
- "h32: 0xFEDCBA98\n"
- "h64: 0xFEDCBA9876543210\n"
- "...\n");
- yin >> map;
-
- EXPECT_FALSE(yin.error());
- EXPECT_TRUE(map.str.equals("hello there"));
- EXPECT_TRUE(map.stdstr == "hello where?");
- EXPECT_EQ(map.u64, 5000000000ULL);
- EXPECT_EQ(map.u32, 4000000000U);
- EXPECT_EQ(map.u16, 65000);
- EXPECT_EQ(map.u8, 255);
- EXPECT_EQ(map.b, false);
- EXPECT_EQ(map.s64, -5000000000LL);
- EXPECT_EQ(map.s32, -2000000000L);
- EXPECT_EQ(map.s16, -32000);
- EXPECT_EQ(map.s8, -127);
- EXPECT_EQ(map.f, 137.125);
- EXPECT_EQ(map.d, -2.8625);
- EXPECT_EQ(map.h8, Hex8(255));
- EXPECT_EQ(map.h16, Hex16(0x8765));
- EXPECT_EQ(map.h32, Hex32(0xFEDCBA98));
- EXPECT_EQ(map.h64, Hex64(0xFEDCBA9876543210LL));
-}
-
-
-//
-// Test writing then reading back all built-in scalar types
-//
-TEST(YAMLIO, TestReadWriteBuiltInTypes) {
- std::string intermediate;
- {
- BuiltInTypes map;
- map.str = "one two";
- map.stdstr = "three four";
- map.u64 = 6000000000ULL;
- map.u32 = 3000000000U;
- map.u16 = 50000;
- map.u8 = 254;
- map.b = true;
- map.s64 = -6000000000LL;
- map.s32 = -2000000000;
- map.s16 = -32000;
- map.s8 = -128;
- map.f = 3.25;
- map.d = -2.8625;
- map.h8 = 254;
- map.h16 = 50000;
- map.h32 = 3000000000U;
- map.h64 = 6000000000LL;
-
- llvm::raw_string_ostream ostr(intermediate);
- Output yout(ostr);
- yout << map;
- }
-
- {
- Input yin(intermediate);
- BuiltInTypes map;
- yin >> map;
-
- EXPECT_FALSE(yin.error());
- EXPECT_TRUE(map.str.equals("one two"));
- EXPECT_TRUE(map.stdstr == "three four");
- EXPECT_EQ(map.u64, 6000000000ULL);
- EXPECT_EQ(map.u32, 3000000000U);
- EXPECT_EQ(map.u16, 50000);
- EXPECT_EQ(map.u8, 254);
- EXPECT_EQ(map.b, true);
- EXPECT_EQ(map.s64, -6000000000LL);
- EXPECT_EQ(map.s32, -2000000000L);
- EXPECT_EQ(map.s16, -32000);
- EXPECT_EQ(map.s8, -128);
- EXPECT_EQ(map.f, 3.25);
- EXPECT_EQ(map.d, -2.8625);
- EXPECT_EQ(map.h8, Hex8(254));
- EXPECT_EQ(map.h16, Hex16(50000));
- EXPECT_EQ(map.h32, Hex32(3000000000U));
- EXPECT_EQ(map.h64, Hex64(6000000000LL));
- }
-}
-
-//===----------------------------------------------------------------------===//
-// Test endian-aware types
-//===----------------------------------------------------------------------===//
-
-struct EndianTypes {
- typedef llvm::support::detail::packed_endian_specific_integral<
- float, llvm::support::little, llvm::support::unaligned>
- ulittle_float;
- typedef llvm::support::detail::packed_endian_specific_integral<
- double, llvm::support::little, llvm::support::unaligned>
- ulittle_double;
-
- llvm::support::ulittle64_t u64;
- llvm::support::ulittle32_t u32;
- llvm::support::ulittle16_t u16;
- llvm::support::little64_t s64;
- llvm::support::little32_t s32;
- llvm::support::little16_t s16;
- ulittle_float f;
- ulittle_double d;
-};
-
-namespace llvm {
-namespace yaml {
-template <> struct MappingTraits<EndianTypes> {
- static void mapping(IO &io, EndianTypes &et) {
- io.mapRequired("u64", et.u64);
- io.mapRequired("u32", et.u32);
- io.mapRequired("u16", et.u16);
- io.mapRequired("s64", et.s64);
- io.mapRequired("s32", et.s32);
- io.mapRequired("s16", et.s16);
- io.mapRequired("f", et.f);
- io.mapRequired("d", et.d);
- }
-};
-}
-}
-
-//
-// Test the reading of all endian scalar conversions
-//
-TEST(YAMLIO, TestReadEndianTypes) {
- EndianTypes map;
- Input yin("---\n"
- "u64: 5000000000\n"
- "u32: 4000000000\n"
- "u16: 65000\n"
- "s64: -5000000000\n"
- "s32: -2000000000\n"
- "s16: -32000\n"
- "f: 3.25\n"
- "d: -2.8625\n"
- "...\n");
- yin >> map;
-
- EXPECT_FALSE(yin.error());
- EXPECT_EQ(map.u64, 5000000000ULL);
- EXPECT_EQ(map.u32, 4000000000U);
- EXPECT_EQ(map.u16, 65000);
- EXPECT_EQ(map.s64, -5000000000LL);
- EXPECT_EQ(map.s32, -2000000000L);
- EXPECT_EQ(map.s16, -32000);
- EXPECT_EQ(map.f, 3.25f);
- EXPECT_EQ(map.d, -2.8625);
-}
-
-//
-// Test writing then reading back all endian-aware scalar types
-//
-TEST(YAMLIO, TestReadWriteEndianTypes) {
- std::string intermediate;
- {
- EndianTypes map;
- map.u64 = 6000000000ULL;
- map.u32 = 3000000000U;
- map.u16 = 50000;
- map.s64 = -6000000000LL;
- map.s32 = -2000000000;
- map.s16 = -32000;
- map.f = 3.25f;
- map.d = -2.8625;
-
- llvm::raw_string_ostream ostr(intermediate);
- Output yout(ostr);
- yout << map;
- }
-
- {
- Input yin(intermediate);
- EndianTypes map;
- yin >> map;
-
- EXPECT_FALSE(yin.error());
- EXPECT_EQ(map.u64, 6000000000ULL);
- EXPECT_EQ(map.u32, 3000000000U);
- EXPECT_EQ(map.u16, 50000);
- EXPECT_EQ(map.s64, -6000000000LL);
- EXPECT_EQ(map.s32, -2000000000L);
- EXPECT_EQ(map.s16, -32000);
- EXPECT_EQ(map.f, 3.25f);
- EXPECT_EQ(map.d, -2.8625);
- }
-}
-
-struct StringTypes {
- llvm::StringRef str1;
- llvm::StringRef str2;
- llvm::StringRef str3;
- llvm::StringRef str4;
- llvm::StringRef str5;
- llvm::StringRef str6;
- llvm::StringRef str7;
- llvm::StringRef str8;
- llvm::StringRef str9;
- llvm::StringRef str10;
- llvm::StringRef str11;
- std::string stdstr1;
- std::string stdstr2;
- std::string stdstr3;
- std::string stdstr4;
- std::string stdstr5;
- std::string stdstr6;
- std::string stdstr7;
- std::string stdstr8;
- std::string stdstr9;
- std::string stdstr10;
- std::string stdstr11;
- std::string stdstr12;
-};
-
-namespace llvm {
-namespace yaml {
- template <>
- struct MappingTraits<StringTypes> {
- static void mapping(IO &io, StringTypes& st) {
- io.mapRequired("str1", st.str1);
- io.mapRequired("str2", st.str2);
- io.mapRequired("str3", st.str3);
- io.mapRequired("str4", st.str4);
- io.mapRequired("str5", st.str5);
- io.mapRequired("str6", st.str6);
- io.mapRequired("str7", st.str7);
- io.mapRequired("str8", st.str8);
- io.mapRequired("str9", st.str9);
- io.mapRequired("str10", st.str10);
- io.mapRequired("str11", st.str11);
- io.mapRequired("stdstr1", st.stdstr1);
- io.mapRequired("stdstr2", st.stdstr2);
- io.mapRequired("stdstr3", st.stdstr3);
- io.mapRequired("stdstr4", st.stdstr4);
- io.mapRequired("stdstr5", st.stdstr5);
- io.mapRequired("stdstr6", st.stdstr6);
- io.mapRequired("stdstr7", st.stdstr7);
- io.mapRequired("stdstr8", st.stdstr8);
- io.mapRequired("stdstr9", st.stdstr9);
- io.mapRequired("stdstr10", st.stdstr10);
- io.mapRequired("stdstr11", st.stdstr11);
- io.mapRequired("stdstr12", st.stdstr12);
- }
- };
-}
-}
-
-TEST(YAMLIO, TestReadWriteStringTypes) {
- std::string intermediate;
- {
- StringTypes map;
- map.str1 = "'aaa";
- map.str2 = "\"bbb";
- map.str3 = "`ccc";
- map.str4 = "@ddd";
- map.str5 = "";
- map.str6 = "0000000004000000";
- map.str7 = "true";
- map.str8 = "FALSE";
- map.str9 = "~";
- map.str10 = "0.2e20";
- map.str11 = "0x30";
- map.stdstr1 = "'eee";
- map.stdstr2 = "\"fff";
- map.stdstr3 = "`ggg";
- map.stdstr4 = "@hhh";
- map.stdstr5 = "";
- map.stdstr6 = "0000000004000000";
- map.stdstr7 = "true";
- map.stdstr8 = "FALSE";
- map.stdstr9 = "~";
- map.stdstr10 = "0.2e20";
- map.stdstr11 = "0x30";
- map.stdstr12 = "- match";
-
- llvm::raw_string_ostream ostr(intermediate);
- Output yout(ostr);
- yout << map;
- }
-
- llvm::StringRef flowOut(intermediate);
- EXPECT_NE(llvm::StringRef::npos, flowOut.find("'''aaa"));
- EXPECT_NE(llvm::StringRef::npos, flowOut.find("'\"bbb'"));
- EXPECT_NE(llvm::StringRef::npos, flowOut.find("'`ccc'"));
- EXPECT_NE(llvm::StringRef::npos, flowOut.find("'@ddd'"));
- EXPECT_NE(llvm::StringRef::npos, flowOut.find("''\n"));
- EXPECT_NE(llvm::StringRef::npos, flowOut.find("'0000000004000000'\n"));
- EXPECT_NE(llvm::StringRef::npos, flowOut.find("'true'\n"));
- EXPECT_NE(llvm::StringRef::npos, flowOut.find("'FALSE'\n"));
- EXPECT_NE(llvm::StringRef::npos, flowOut.find("'~'\n"));
- EXPECT_NE(llvm::StringRef::npos, flowOut.find("'0.2e20'\n"));
- EXPECT_NE(llvm::StringRef::npos, flowOut.find("'0x30'\n"));
- EXPECT_NE(llvm::StringRef::npos, flowOut.find("'- match'\n"));
- EXPECT_NE(std::string::npos, flowOut.find("'''eee"));
- EXPECT_NE(std::string::npos, flowOut.find("'\"fff'"));
- EXPECT_NE(std::string::npos, flowOut.find("'`ggg'"));
- EXPECT_NE(std::string::npos, flowOut.find("'@hhh'"));
- EXPECT_NE(std::string::npos, flowOut.find("''\n"));
- EXPECT_NE(std::string::npos, flowOut.find("'0000000004000000'\n"));
-
- {
- Input yin(intermediate);
- StringTypes map;
- yin >> map;
-
- EXPECT_FALSE(yin.error());
- EXPECT_TRUE(map.str1.equals("'aaa"));
- EXPECT_TRUE(map.str2.equals("\"bbb"));
- EXPECT_TRUE(map.str3.equals("`ccc"));
- EXPECT_TRUE(map.str4.equals("@ddd"));
- EXPECT_TRUE(map.str5.equals(""));
- EXPECT_TRUE(map.str6.equals("0000000004000000"));
- EXPECT_TRUE(map.stdstr1 == "'eee");
- EXPECT_TRUE(map.stdstr2 == "\"fff");
- EXPECT_TRUE(map.stdstr3 == "`ggg");
- EXPECT_TRUE(map.stdstr4 == "@hhh");
- EXPECT_TRUE(map.stdstr5 == "");
- EXPECT_TRUE(map.stdstr6 == "0000000004000000");
- }
-}
-
-//===----------------------------------------------------------------------===//
-// Test ScalarEnumerationTraits
-//===----------------------------------------------------------------------===//
-
-enum Colors {
- cRed,
- cBlue,
- cGreen,
- cYellow
-};
-
-struct ColorMap {
- Colors c1;
- Colors c2;
- Colors c3;
- Colors c4;
- Colors c5;
- Colors c6;
-};
-
-namespace llvm {
-namespace yaml {
- template <>
- struct ScalarEnumerationTraits<Colors> {
- static void enumeration(IO &io, Colors &value) {
- io.enumCase(value, "red", cRed);
- io.enumCase(value, "blue", cBlue);
- io.enumCase(value, "green", cGreen);
- io.enumCase(value, "yellow",cYellow);
- }
- };
- template <>
- struct MappingTraits<ColorMap> {
- static void mapping(IO &io, ColorMap& c) {
- io.mapRequired("c1", c.c1);
- io.mapRequired("c2", c.c2);
- io.mapRequired("c3", c.c3);
- io.mapOptional("c4", c.c4, cBlue); // supplies default
- io.mapOptional("c5", c.c5, cYellow); // supplies default
- io.mapOptional("c6", c.c6, cRed); // supplies default
- }
- };
-}
-}
-
-
-//
-// Test reading enumerated scalars
-//
-TEST(YAMLIO, TestEnumRead) {
- ColorMap map;
- Input yin("---\n"
- "c1: blue\n"
- "c2: red\n"
- "c3: green\n"
- "c5: yellow\n"
- "...\n");
- yin >> map;
-
- EXPECT_FALSE(yin.error());
- EXPECT_EQ(cBlue, map.c1);
- EXPECT_EQ(cRed, map.c2);
- EXPECT_EQ(cGreen, map.c3);
- EXPECT_EQ(cBlue, map.c4); // tests default
- EXPECT_EQ(cYellow,map.c5); // tests overridden
- EXPECT_EQ(cRed, map.c6); // tests default
-}
-
-
-
-//===----------------------------------------------------------------------===//
-// Test ScalarBitSetTraits
-//===----------------------------------------------------------------------===//
-
-enum MyFlags {
- flagNone = 0,
- flagBig = 1 << 0,
- flagFlat = 1 << 1,
- flagRound = 1 << 2,
- flagPointy = 1 << 3
-};
-inline MyFlags operator|(MyFlags a, MyFlags b) {
- return static_cast<MyFlags>(
- static_cast<uint32_t>(a) | static_cast<uint32_t>(b));
-}
-
-struct FlagsMap {
- MyFlags f1;
- MyFlags f2;
- MyFlags f3;
- MyFlags f4;
-};
-
-
-namespace llvm {
-namespace yaml {
- template <>
- struct ScalarBitSetTraits<MyFlags> {
- static void bitset(IO &io, MyFlags &value) {
- io.bitSetCase(value, "big", flagBig);
- io.bitSetCase(value, "flat", flagFlat);
- io.bitSetCase(value, "round", flagRound);
- io.bitSetCase(value, "pointy",flagPointy);
- }
- };
- template <>
- struct MappingTraits<FlagsMap> {
- static void mapping(IO &io, FlagsMap& c) {
- io.mapRequired("f1", c.f1);
- io.mapRequired("f2", c.f2);
- io.mapRequired("f3", c.f3);
- io.mapOptional("f4", c.f4, MyFlags(flagRound));
- }
- };
-}
-}
-
-
-//
-// Test reading flow sequence representing bit-mask values
-//
-TEST(YAMLIO, TestFlagsRead) {
- FlagsMap map;
- Input yin("---\n"
- "f1: [ big ]\n"
- "f2: [ round, flat ]\n"
- "f3: []\n"
- "...\n");
- yin >> map;
-
- EXPECT_FALSE(yin.error());
- EXPECT_EQ(flagBig, map.f1);
- EXPECT_EQ(flagRound|flagFlat, map.f2);
- EXPECT_EQ(flagNone, map.f3); // check empty set
- EXPECT_EQ(flagRound, map.f4); // check optional key
-}
-
-
-//
-// Test writing then reading back bit-mask values
-//
-TEST(YAMLIO, TestReadWriteFlags) {
- std::string intermediate;
- {
- FlagsMap map;
- map.f1 = flagBig;
- map.f2 = flagRound | flagFlat;
- map.f3 = flagNone;
- map.f4 = flagNone;
-
- llvm::raw_string_ostream ostr(intermediate);
- Output yout(ostr);
- yout << map;
- }
-
- {
- Input yin(intermediate);
- FlagsMap map2;
- yin >> map2;
-
- EXPECT_FALSE(yin.error());
- EXPECT_EQ(flagBig, map2.f1);
- EXPECT_EQ(flagRound|flagFlat, map2.f2);
- EXPECT_EQ(flagNone, map2.f3);
- //EXPECT_EQ(flagRound, map2.f4); // check optional key
- }
-}
-
-
-
-//===----------------------------------------------------------------------===//
-// Test ScalarTraits
-//===----------------------------------------------------------------------===//
-
-struct MyCustomType {
- int length;
- int width;
-};
-
-struct MyCustomTypeMap {
- MyCustomType f1;
- MyCustomType f2;
- int f3;
-};
-
-
-namespace llvm {
-namespace yaml {
- template <>
- struct MappingTraits<MyCustomTypeMap> {
- static void mapping(IO &io, MyCustomTypeMap& s) {
- io.mapRequired("f1", s.f1);
- io.mapRequired("f2", s.f2);
- io.mapRequired("f3", s.f3);
- }
- };
- // MyCustomType is formatted as a yaml scalar. A value of
- // {length=3, width=4} would be represented in yaml as "3 by 4".
- template<>
- struct ScalarTraits<MyCustomType> {
- static void output(const MyCustomType &value, void* ctxt, llvm::raw_ostream &out) {
- out << llvm::format("%d by %d", value.length, value.width);
- }
- static StringRef input(StringRef scalar, void* ctxt, MyCustomType &value) {
- size_t byStart = scalar.find("by");
- if ( byStart != StringRef::npos ) {
- StringRef lenStr = scalar.slice(0, byStart);
- lenStr = lenStr.rtrim();
- if ( lenStr.getAsInteger(0, value.length) ) {
- return "malformed length";
- }
- StringRef widthStr = scalar.drop_front(byStart+2);
- widthStr = widthStr.ltrim();
- if ( widthStr.getAsInteger(0, value.width) ) {
- return "malformed width";
- }
- return StringRef();
- }
- else {
- return "malformed by";
- }
- }
- static QuotingType mustQuote(StringRef) { return QuotingType::Single; }
- };
-}
-}
-
-
-//
-// Test writing then reading back custom values
-//
-TEST(YAMLIO, TestReadWriteMyCustomType) {
- std::string intermediate;
- {
- MyCustomTypeMap map;
- map.f1.length = 1;
- map.f1.width = 4;
- map.f2.length = 100;
- map.f2.width = 400;
- map.f3 = 10;
-
- llvm::raw_string_ostream ostr(intermediate);
- Output yout(ostr);
- yout << map;
- }
-
- {
- Input yin(intermediate);
- MyCustomTypeMap map2;
- yin >> map2;
-
- EXPECT_FALSE(yin.error());
- EXPECT_EQ(1, map2.f1.length);
- EXPECT_EQ(4, map2.f1.width);
- EXPECT_EQ(100, map2.f2.length);
- EXPECT_EQ(400, map2.f2.width);
- EXPECT_EQ(10, map2.f3);
- }
-}
-
-
-//===----------------------------------------------------------------------===//
-// Test BlockScalarTraits
-//===----------------------------------------------------------------------===//
-
-struct MultilineStringType {
- std::string str;
-};
-
-struct MultilineStringTypeMap {
- MultilineStringType name;
- MultilineStringType description;
- MultilineStringType ingredients;
- MultilineStringType recipes;
- MultilineStringType warningLabels;
- MultilineStringType documentation;
- int price;
-};
-
-namespace llvm {
-namespace yaml {
- template <>
- struct MappingTraits<MultilineStringTypeMap> {
- static void mapping(IO &io, MultilineStringTypeMap& s) {
- io.mapRequired("name", s.name);
- io.mapRequired("description", s.description);
- io.mapRequired("ingredients", s.ingredients);
- io.mapRequired("recipes", s.recipes);
- io.mapRequired("warningLabels", s.warningLabels);
- io.mapRequired("documentation", s.documentation);
- io.mapRequired("price", s.price);
- }
- };
-
- // MultilineStringType is formatted as a yaml block literal scalar. A value of
- // "Hello\nWorld" would be represented in yaml as
- // |
- // Hello
- // World
- template <>
- struct BlockScalarTraits<MultilineStringType> {
- static void output(const MultilineStringType &value, void *ctxt,
- llvm::raw_ostream &out) {
- out << value.str;
- }
- static StringRef input(StringRef scalar, void *ctxt,
- MultilineStringType &value) {
- value.str = scalar.str();
- return StringRef();
- }
- };
-}
-}
-
-LLVM_YAML_IS_DOCUMENT_LIST_VECTOR(MultilineStringType)
-
-//
-// Test writing then reading back custom values
-//
-TEST(YAMLIO, TestReadWriteMultilineStringType) {
- std::string intermediate;
- {
- MultilineStringTypeMap map;
- map.name.str = "An Item";
- map.description.str = "Hello\nWorld";
- map.ingredients.str = "SubItem 1\nSub Item 2\n\nSub Item 3\n";
- map.recipes.str = "\n\nTest 1\n\n\n";
- map.warningLabels.str = "";
- map.documentation.str = "\n\n";
- map.price = 350;
-
- llvm::raw_string_ostream ostr(intermediate);
- Output yout(ostr);
- yout << map;
- }
- {
- Input yin(intermediate);
- MultilineStringTypeMap map2;
- yin >> map2;
-
- EXPECT_FALSE(yin.error());
- EXPECT_EQ(map2.name.str, "An Item\n");
- EXPECT_EQ(map2.description.str, "Hello\nWorld\n");
- EXPECT_EQ(map2.ingredients.str, "SubItem 1\nSub Item 2\n\nSub Item 3\n");
- EXPECT_EQ(map2.recipes.str, "\n\nTest 1\n");
- EXPECT_TRUE(map2.warningLabels.str.empty());
- EXPECT_TRUE(map2.documentation.str.empty());
- EXPECT_EQ(map2.price, 350);
- }
-}
-
-//
-// Test writing then reading back custom values
-//
-TEST(YAMLIO, TestReadWriteBlockScalarDocuments) {
- std::string intermediate;
- {
- std::vector<MultilineStringType> documents;
- MultilineStringType doc;
- doc.str = "Hello\nWorld";
- documents.push_back(doc);
-
- llvm::raw_string_ostream ostr(intermediate);
- Output yout(ostr);
- yout << documents;
-
- // Verify that the block scalar header was written out on the same line
- // as the document marker.
- EXPECT_NE(llvm::StringRef::npos, llvm::StringRef(ostr.str()).find("--- |"));
- }
- {
- Input yin(intermediate);
- std::vector<MultilineStringType> documents2;
- yin >> documents2;
-
- EXPECT_FALSE(yin.error());
- EXPECT_EQ(documents2.size(), size_t(1));
- EXPECT_EQ(documents2[0].str, "Hello\nWorld\n");
- }
-}
-
-TEST(YAMLIO, TestReadWriteBlockScalarValue) {
- std::string intermediate;
- {
- MultilineStringType doc;
- doc.str = "Just a block\nscalar doc";
-
- llvm::raw_string_ostream ostr(intermediate);
- Output yout(ostr);
- yout << doc;
- }
- {
- Input yin(intermediate);
- MultilineStringType doc;
- yin >> doc;
-
- EXPECT_FALSE(yin.error());
- EXPECT_EQ(doc.str, "Just a block\nscalar doc\n");
- }
-}
-
-//===----------------------------------------------------------------------===//
-// Test flow sequences
-//===----------------------------------------------------------------------===//
-
-LLVM_YAML_STRONG_TYPEDEF(int, MyNumber)
-LLVM_YAML_IS_FLOW_SEQUENCE_VECTOR(MyNumber)
-LLVM_YAML_STRONG_TYPEDEF(llvm::StringRef, MyString)
-LLVM_YAML_IS_FLOW_SEQUENCE_VECTOR(MyString)
-
-namespace llvm {
-namespace yaml {
- template<>
- struct ScalarTraits<MyNumber> {
- static void output(const MyNumber &value, void *, llvm::raw_ostream &out) {
- out << value;
- }
-
- static StringRef input(StringRef scalar, void *, MyNumber &value) {
- long long n;
- if ( getAsSignedInteger(scalar, 0, n) )
- return "invalid number";
- value = n;
- return StringRef();
- }
-
- static QuotingType mustQuote(StringRef) { return QuotingType::None; }
- };
-
- template <> struct ScalarTraits<MyString> {
- using Impl = ScalarTraits<StringRef>;
- static void output(const MyString &V, void *Ctx, raw_ostream &OS) {
- Impl::output(V, Ctx, OS);
- }
- static StringRef input(StringRef S, void *Ctx, MyString &V) {
- return Impl::input(S, Ctx, V.value);
- }
- static QuotingType mustQuote(StringRef S) {
- return Impl::mustQuote(S);
- }
- };
-}
-}
-
-struct NameAndNumbers {
- llvm::StringRef name;
- std::vector<MyString> strings;
- std::vector<MyNumber> single;
- std::vector<MyNumber> numbers;
-};
-
-namespace llvm {
-namespace yaml {
- template <>
- struct MappingTraits<NameAndNumbers> {
- static void mapping(IO &io, NameAndNumbers& nn) {
- io.mapRequired("name", nn.name);
- io.mapRequired("strings", nn.strings);
- io.mapRequired("single", nn.single);
- io.mapRequired("numbers", nn.numbers);
- }
- };
-}
-}
-
-typedef std::vector<MyNumber> MyNumberFlowSequence;
-
-LLVM_YAML_IS_SEQUENCE_VECTOR(MyNumberFlowSequence)
-
-struct NameAndNumbersFlow {
- llvm::StringRef name;
- std::vector<MyNumberFlowSequence> sequenceOfNumbers;
-};
-
-namespace llvm {
-namespace yaml {
- template <>
- struct MappingTraits<NameAndNumbersFlow> {
- static void mapping(IO &io, NameAndNumbersFlow& nn) {
- io.mapRequired("name", nn.name);
- io.mapRequired("sequenceOfNumbers", nn.sequenceOfNumbers);
- }
- };
-}
-}
-
-//
-// Test writing then reading back custom values
-//
-TEST(YAMLIO, TestReadWriteMyFlowSequence) {
- std::string intermediate;
- {
- NameAndNumbers map;
- map.name = "hello";
- map.strings.push_back(llvm::StringRef("one"));
- map.strings.push_back(llvm::StringRef("two"));
- map.single.push_back(1);
- map.numbers.push_back(10);
- map.numbers.push_back(-30);
- map.numbers.push_back(1024);
-
- llvm::raw_string_ostream ostr(intermediate);
- Output yout(ostr);
- yout << map;
-
- // Verify sequences were written in flow style
- ostr.flush();
- llvm::StringRef flowOut(intermediate);
- EXPECT_NE(llvm::StringRef::npos, flowOut.find("one, two"));
- EXPECT_NE(llvm::StringRef::npos, flowOut.find("10, -30, 1024"));
- }
-
- {
- Input yin(intermediate);
- NameAndNumbers map2;
- yin >> map2;
-
- EXPECT_FALSE(yin.error());
- EXPECT_TRUE(map2.name.equals("hello"));
- EXPECT_EQ(map2.strings.size(), 2UL);
- EXPECT_TRUE(map2.strings[0].value.equals("one"));
- EXPECT_TRUE(map2.strings[1].value.equals("two"));
- EXPECT_EQ(map2.single.size(), 1UL);
- EXPECT_EQ(1, map2.single[0]);
- EXPECT_EQ(map2.numbers.size(), 3UL);
- EXPECT_EQ(10, map2.numbers[0]);
- EXPECT_EQ(-30, map2.numbers[1]);
- EXPECT_EQ(1024, map2.numbers[2]);
- }
-}
-
-
-//
-// Test writing then reading back a sequence of flow sequences.
-//
-TEST(YAMLIO, TestReadWriteSequenceOfMyFlowSequence) {
- std::string intermediate;
- {
- NameAndNumbersFlow map;
- map.name = "hello";
- MyNumberFlowSequence single = { 0 };
- MyNumberFlowSequence numbers = { 12, 1, -512 };
- map.sequenceOfNumbers.push_back(single);
- map.sequenceOfNumbers.push_back(numbers);
- map.sequenceOfNumbers.push_back(MyNumberFlowSequence());
-
- llvm::raw_string_ostream ostr(intermediate);
- Output yout(ostr);
- yout << map;
-
- // Verify sequences were written in flow style
- // and that the parent sequence used '-'.
- ostr.flush();
- llvm::StringRef flowOut(intermediate);
- EXPECT_NE(llvm::StringRef::npos, flowOut.find("- [ 0 ]"));
- EXPECT_NE(llvm::StringRef::npos, flowOut.find("- [ 12, 1, -512 ]"));
- EXPECT_NE(llvm::StringRef::npos, flowOut.find("- [ ]"));
- }
-
- {
- Input yin(intermediate);
- NameAndNumbersFlow map2;
- yin >> map2;
-
- EXPECT_FALSE(yin.error());
- EXPECT_TRUE(map2.name.equals("hello"));
- EXPECT_EQ(map2.sequenceOfNumbers.size(), 3UL);
- EXPECT_EQ(map2.sequenceOfNumbers[0].size(), 1UL);
- EXPECT_EQ(0, map2.sequenceOfNumbers[0][0]);
- EXPECT_EQ(map2.sequenceOfNumbers[1].size(), 3UL);
- EXPECT_EQ(12, map2.sequenceOfNumbers[1][0]);
- EXPECT_EQ(1, map2.sequenceOfNumbers[1][1]);
- EXPECT_EQ(-512, map2.sequenceOfNumbers[1][2]);
- EXPECT_TRUE(map2.sequenceOfNumbers[2].empty());
- }
-}
-
-//===----------------------------------------------------------------------===//
-// Test normalizing/denormalizing
-//===----------------------------------------------------------------------===//
-
-LLVM_YAML_STRONG_TYPEDEF(uint32_t, TotalSeconds)
-
-typedef std::vector<TotalSeconds> SecondsSequence;
-
-LLVM_YAML_IS_SEQUENCE_VECTOR(TotalSeconds)
-
-
-namespace llvm {
-namespace yaml {
- template <>
- struct MappingTraits<TotalSeconds> {
-
- class NormalizedSeconds {
- public:
- NormalizedSeconds(IO &io)
- : hours(0), minutes(0), seconds(0) {
- }
- NormalizedSeconds(IO &, TotalSeconds &secs)
- : hours(secs/3600),
- minutes((secs - (hours*3600))/60),
- seconds(secs % 60) {
- }
- TotalSeconds denormalize(IO &) {
- return TotalSeconds(hours*3600 + minutes*60 + seconds);
- }
-
- uint32_t hours;
- uint8_t minutes;
- uint8_t seconds;
- };
-
- static void mapping(IO &io, TotalSeconds &secs) {
- MappingNormalization<NormalizedSeconds, TotalSeconds> keys(io, secs);
-
- io.mapOptional("hours", keys->hours, (uint32_t)0);
- io.mapOptional("minutes", keys->minutes, (uint8_t)0);
- io.mapRequired("seconds", keys->seconds);
- }
- };
-}
-}
-
-
-//
-// Test the reading of a yaml sequence of mappings
-//
-TEST(YAMLIO, TestReadMySecondsSequence) {
- SecondsSequence seq;
- Input yin("---\n - hours: 1\n seconds: 5\n - seconds: 59\n...\n");
- yin >> seq;
-
- EXPECT_FALSE(yin.error());
- EXPECT_EQ(seq.size(), 2UL);
- EXPECT_EQ(seq[0], 3605U);
- EXPECT_EQ(seq[1], 59U);
-}
-
-
-//
-// Test writing then reading back custom values
-//
-TEST(YAMLIO, TestReadWriteMySecondsSequence) {
- std::string intermediate;
- {
- SecondsSequence seq;
- seq.push_back(4000);
- seq.push_back(500);
- seq.push_back(59);
-
- llvm::raw_string_ostream ostr(intermediate);
- Output yout(ostr);
- yout << seq;
- }
- {
- Input yin(intermediate);
- SecondsSequence seq2;
- yin >> seq2;
-
- EXPECT_FALSE(yin.error());
- EXPECT_EQ(seq2.size(), 3UL);
- EXPECT_EQ(seq2[0], 4000U);
- EXPECT_EQ(seq2[1], 500U);
- EXPECT_EQ(seq2[2], 59U);
- }
-}
-
-
-//===----------------------------------------------------------------------===//
-// Test dynamic typing
-//===----------------------------------------------------------------------===//
-
-enum AFlags {
- a1,
- a2,
- a3
-};
-
-enum BFlags {
- b1,
- b2,
- b3
-};
-
-enum Kind {
- kindA,
- kindB
-};
-
-struct KindAndFlags {
- KindAndFlags() : kind(kindA), flags(0) { }
- KindAndFlags(Kind k, uint32_t f) : kind(k), flags(f) { }
- Kind kind;
- uint32_t flags;
-};
-
-typedef std::vector<KindAndFlags> KindAndFlagsSequence;
-
-LLVM_YAML_IS_SEQUENCE_VECTOR(KindAndFlags)
-
-namespace llvm {
-namespace yaml {
- template <>
- struct ScalarEnumerationTraits<AFlags> {
- static void enumeration(IO &io, AFlags &value) {
- io.enumCase(value, "a1", a1);
- io.enumCase(value, "a2", a2);
- io.enumCase(value, "a3", a3);
- }
- };
- template <>
- struct ScalarEnumerationTraits<BFlags> {
- static void enumeration(IO &io, BFlags &value) {
- io.enumCase(value, "b1", b1);
- io.enumCase(value, "b2", b2);
- io.enumCase(value, "b3", b3);
- }
- };
- template <>
- struct ScalarEnumerationTraits<Kind> {
- static void enumeration(IO &io, Kind &value) {
- io.enumCase(value, "A", kindA);
- io.enumCase(value, "B", kindB);
- }
- };
- template <>
- struct MappingTraits<KindAndFlags> {
- static void mapping(IO &io, KindAndFlags& kf) {
- io.mapRequired("kind", kf.kind);
- // Type of "flags" field varies depending on "kind" field.
- // Use memcpy here to avoid breaking strict aliasing rules.
- if (kf.kind == kindA) {
- AFlags aflags = static_cast<AFlags>(kf.flags);
- io.mapRequired("flags", aflags);
- kf.flags = aflags;
- } else {
- BFlags bflags = static_cast<BFlags>(kf.flags);
- io.mapRequired("flags", bflags);
- kf.flags = bflags;
- }
- }
- };
-}
-}
-
-
-//
-// Test the reading of a yaml sequence dynamic types
-//
-TEST(YAMLIO, TestReadKindAndFlagsSequence) {
- KindAndFlagsSequence seq;
- Input yin("---\n - kind: A\n flags: a2\n - kind: B\n flags: b1\n...\n");
- yin >> seq;
-
- EXPECT_FALSE(yin.error());
- EXPECT_EQ(seq.size(), 2UL);
- EXPECT_EQ(seq[0].kind, kindA);
- EXPECT_EQ(seq[0].flags, (uint32_t)a2);
- EXPECT_EQ(seq[1].kind, kindB);
- EXPECT_EQ(seq[1].flags, (uint32_t)b1);
-}
-
-//
-// Test writing then reading back dynamic types
-//
-TEST(YAMLIO, TestReadWriteKindAndFlagsSequence) {
- std::string intermediate;
- {
- KindAndFlagsSequence seq;
- seq.push_back(KindAndFlags(kindA,a1));
- seq.push_back(KindAndFlags(kindB,b1));
- seq.push_back(KindAndFlags(kindA,a2));
- seq.push_back(KindAndFlags(kindB,b2));
- seq.push_back(KindAndFlags(kindA,a3));
-
- llvm::raw_string_ostream ostr(intermediate);
- Output yout(ostr);
- yout << seq;
- }
- {
- Input yin(intermediate);
- KindAndFlagsSequence seq2;
- yin >> seq2;
-
- EXPECT_FALSE(yin.error());
- EXPECT_EQ(seq2.size(), 5UL);
- EXPECT_EQ(seq2[0].kind, kindA);
- EXPECT_EQ(seq2[0].flags, (uint32_t)a1);
- EXPECT_EQ(seq2[1].kind, kindB);
- EXPECT_EQ(seq2[1].flags, (uint32_t)b1);
- EXPECT_EQ(seq2[2].kind, kindA);
- EXPECT_EQ(seq2[2].flags, (uint32_t)a2);
- EXPECT_EQ(seq2[3].kind, kindB);
- EXPECT_EQ(seq2[3].flags, (uint32_t)b2);
- EXPECT_EQ(seq2[4].kind, kindA);
- EXPECT_EQ(seq2[4].flags, (uint32_t)a3);
- }
-}
-
-
-//===----------------------------------------------------------------------===//
-// Test document list
-//===----------------------------------------------------------------------===//
-
-struct FooBarMap {
- int foo;
- int bar;
-};
-typedef std::vector<FooBarMap> FooBarMapDocumentList;
-
-LLVM_YAML_IS_DOCUMENT_LIST_VECTOR(FooBarMap)
-
-
-namespace llvm {
-namespace yaml {
- template <>
- struct MappingTraits<FooBarMap> {
- static void mapping(IO &io, FooBarMap& fb) {
- io.mapRequired("foo", fb.foo);
- io.mapRequired("bar", fb.bar);
- }
- };
-}
-}
-
-
-//
-// Test the reading of a yaml mapping
-//
-TEST(YAMLIO, TestDocRead) {
- FooBarMap doc;
- Input yin("---\nfoo: 3\nbar: 5\n...\n");
- yin >> doc;
-
- EXPECT_FALSE(yin.error());
- EXPECT_EQ(doc.foo, 3);
- EXPECT_EQ(doc.bar,5);
-}
-
-
-
-//
-// Test writing then reading back a sequence of mappings
-//
-TEST(YAMLIO, TestSequenceDocListWriteAndRead) {
- std::string intermediate;
- {
- FooBarMap doc1;
- doc1.foo = 10;
- doc1.bar = -3;
- FooBarMap doc2;
- doc2.foo = 257;
- doc2.bar = 0;
- std::vector<FooBarMap> docList;
- docList.push_back(doc1);
- docList.push_back(doc2);
-
- llvm::raw_string_ostream ostr(intermediate);
- Output yout(ostr);
- yout << docList;
- }
-
-
- {
- Input yin(intermediate);
- std::vector<FooBarMap> docList2;
- yin >> docList2;
-
- EXPECT_FALSE(yin.error());
- EXPECT_EQ(docList2.size(), 2UL);
- FooBarMap& map1 = docList2[0];
- FooBarMap& map2 = docList2[1];
- EXPECT_EQ(map1.foo, 10);
- EXPECT_EQ(map1.bar, -3);
- EXPECT_EQ(map2.foo, 257);
- EXPECT_EQ(map2.bar, 0);
- }
-}
-
-//===----------------------------------------------------------------------===//
-// Test document tags
-//===----------------------------------------------------------------------===//
-
-struct MyDouble {
- MyDouble() : value(0.0) { }
- MyDouble(double x) : value(x) { }
- double value;
-};
-
-LLVM_YAML_IS_DOCUMENT_LIST_VECTOR(MyDouble)
-
-
-namespace llvm {
-namespace yaml {
- template <>
- struct MappingTraits<MyDouble> {
- static void mapping(IO &io, MyDouble &d) {
- if (io.mapTag("!decimal", true)) {
- mappingDecimal(io, d);
- } else if (io.mapTag("!fraction")) {
- mappingFraction(io, d);
- }
- }
- static void mappingDecimal(IO &io, MyDouble &d) {
- io.mapRequired("value", d.value);
- }
- static void mappingFraction(IO &io, MyDouble &d) {
- double num, denom;
- io.mapRequired("numerator", num);
- io.mapRequired("denominator", denom);
- // convert fraction to double
- d.value = num/denom;
- }
- };
- }
-}
-
-
-//
-// Test the reading of two different tagged yaml documents.
-//
-TEST(YAMLIO, TestTaggedDocuments) {
- std::vector<MyDouble> docList;
- Input yin("--- !decimal\nvalue: 3.0\n"
- "--- !fraction\nnumerator: 9.0\ndenominator: 2\n...\n");
- yin >> docList;
- EXPECT_FALSE(yin.error());
- EXPECT_EQ(docList.size(), 2UL);
- EXPECT_EQ(docList[0].value, 3.0);
- EXPECT_EQ(docList[1].value, 4.5);
-}
-
-
-
-//
-// Test writing then reading back tagged documents
-//
-TEST(YAMLIO, TestTaggedDocumentsWriteAndRead) {
- std::string intermediate;
- {
- MyDouble a(10.25);
- MyDouble b(-3.75);
- std::vector<MyDouble> docList;
- docList.push_back(a);
- docList.push_back(b);
-
- llvm::raw_string_ostream ostr(intermediate);
- Output yout(ostr);
- yout << docList;
- }
-
- {
- Input yin(intermediate);
- std::vector<MyDouble> docList2;
- yin >> docList2;
-
- EXPECT_FALSE(yin.error());
- EXPECT_EQ(docList2.size(), 2UL);
- EXPECT_EQ(docList2[0].value, 10.25);
- EXPECT_EQ(docList2[1].value, -3.75);
- }
-}
-
-
-//===----------------------------------------------------------------------===//
-// Test mapping validation
-//===----------------------------------------------------------------------===//
-
-struct MyValidation {
- double value;
-};
-
-LLVM_YAML_IS_DOCUMENT_LIST_VECTOR(MyValidation)
-
-namespace llvm {
-namespace yaml {
- template <>
- struct MappingTraits<MyValidation> {
- static void mapping(IO &io, MyValidation &d) {
- io.mapRequired("value", d.value);
- }
- static StringRef validate(IO &io, MyValidation &d) {
- if (d.value < 0)
- return "negative value";
- return StringRef();
- }
- };
- }
-}
-
-
-//
-// Test that validate() is called and complains about the negative value.
-//
-TEST(YAMLIO, TestValidatingInput) {
- std::vector<MyValidation> docList;
- Input yin("--- \nvalue: 3.0\n"
- "--- \nvalue: -1.0\n...\n",
- nullptr, suppressErrorMessages);
- yin >> docList;
- EXPECT_TRUE(!!yin.error());
-}
-
-//===----------------------------------------------------------------------===//
-// Test flow mapping
-//===----------------------------------------------------------------------===//
-
-struct FlowFooBar {
- int foo;
- int bar;
-
- FlowFooBar() : foo(0), bar(0) {}
- FlowFooBar(int foo, int bar) : foo(foo), bar(bar) {}
-};
-
-typedef std::vector<FlowFooBar> FlowFooBarSequence;
-
-LLVM_YAML_IS_SEQUENCE_VECTOR(FlowFooBar)
-
-struct FlowFooBarDoc {
- FlowFooBar attribute;
- FlowFooBarSequence seq;
-};
-
-namespace llvm {
-namespace yaml {
- template <>
- struct MappingTraits<FlowFooBar> {
- static void mapping(IO &io, FlowFooBar &fb) {
- io.mapRequired("foo", fb.foo);
- io.mapRequired("bar", fb.bar);
- }
-
- static const bool flow = true;
- };
-
- template <>
- struct MappingTraits<FlowFooBarDoc> {
- static void mapping(IO &io, FlowFooBarDoc &fb) {
- io.mapRequired("attribute", fb.attribute);
- io.mapRequired("seq", fb.seq);
- }
- };
-}
-}
-
-//
-// Test writing then reading back custom mappings
-//
-TEST(YAMLIO, TestReadWriteMyFlowMapping) {
- std::string intermediate;
- {
- FlowFooBarDoc doc;
- doc.attribute = FlowFooBar(42, 907);
- doc.seq.push_back(FlowFooBar(1, 2));
- doc.seq.push_back(FlowFooBar(0, 0));
- doc.seq.push_back(FlowFooBar(-1, 1024));
-
- llvm::raw_string_ostream ostr(intermediate);
- Output yout(ostr);
- yout << doc;
-
- // Verify that mappings were written in flow style
- ostr.flush();
- llvm::StringRef flowOut(intermediate);
- EXPECT_NE(llvm::StringRef::npos, flowOut.find("{ foo: 42, bar: 907 }"));
- EXPECT_NE(llvm::StringRef::npos, flowOut.find("- { foo: 1, bar: 2 }"));
- EXPECT_NE(llvm::StringRef::npos, flowOut.find("- { foo: 0, bar: 0 }"));
- EXPECT_NE(llvm::StringRef::npos, flowOut.find("- { foo: -1, bar: 1024 }"));
- }
-
- {
- Input yin(intermediate);
- FlowFooBarDoc doc2;
- yin >> doc2;
-
- EXPECT_FALSE(yin.error());
- EXPECT_EQ(doc2.attribute.foo, 42);
- EXPECT_EQ(doc2.attribute.bar, 907);
- EXPECT_EQ(doc2.seq.size(), 3UL);
- EXPECT_EQ(doc2.seq[0].foo, 1);
- EXPECT_EQ(doc2.seq[0].bar, 2);
- EXPECT_EQ(doc2.seq[1].foo, 0);
- EXPECT_EQ(doc2.seq[1].bar, 0);
- EXPECT_EQ(doc2.seq[2].foo, -1);
- EXPECT_EQ(doc2.seq[2].bar, 1024);
- }
-}
-
-//===----------------------------------------------------------------------===//
-// Test error handling
-//===----------------------------------------------------------------------===//
-
-//
-// Test error handling of unknown enumerated scalar
-//
-TEST(YAMLIO, TestColorsReadError) {
- ColorMap map;
- Input yin("---\n"
- "c1: blue\n"
- "c2: purple\n"
- "c3: green\n"
- "...\n",
- /*Ctxt=*/nullptr,
- suppressErrorMessages);
- yin >> map;
- EXPECT_TRUE(!!yin.error());
-}
-
-
-//
-// Test error handling of flow sequence with unknown value
-//
-TEST(YAMLIO, TestFlagsReadError) {
- FlagsMap map;
- Input yin("---\n"
- "f1: [ big ]\n"
- "f2: [ round, hollow ]\n"
- "f3: []\n"
- "...\n",
- /*Ctxt=*/nullptr,
- suppressErrorMessages);
- yin >> map;
-
- EXPECT_TRUE(!!yin.error());
-}
-
-
-//
-// Test error handling reading built-in uint8_t type
-//
-TEST(YAMLIO, TestReadBuiltInTypesUint8Error) {
- std::vector<uint8_t> seq;
- Input yin("---\n"
- "- 255\n"
- "- 0\n"
- "- 257\n"
- "...\n",
- /*Ctxt=*/nullptr,
- suppressErrorMessages);
- yin >> seq;
-
- EXPECT_TRUE(!!yin.error());
-}
-
-
-//
-// Test error handling reading built-in uint16_t type
-//
-TEST(YAMLIO, TestReadBuiltInTypesUint16Error) {
- std::vector<uint16_t> seq;
- Input yin("---\n"
- "- 65535\n"
- "- 0\n"
- "- 66000\n"
- "...\n",
- /*Ctxt=*/nullptr,
- suppressErrorMessages);
- yin >> seq;
-
- EXPECT_TRUE(!!yin.error());
-}
-
-
-//
-// Test error handling reading built-in uint32_t type
-//
-TEST(YAMLIO, TestReadBuiltInTypesUint32Error) {
- std::vector<uint32_t> seq;
- Input yin("---\n"
- "- 4000000000\n"
- "- 0\n"
- "- 5000000000\n"
- "...\n",
- /*Ctxt=*/nullptr,
- suppressErrorMessages);
- yin >> seq;
-
- EXPECT_TRUE(!!yin.error());
-}
-
-
-//
-// Test error handling reading built-in uint64_t type
-//
-TEST(YAMLIO, TestReadBuiltInTypesUint64Error) {
- std::vector<uint64_t> seq;
- Input yin("---\n"
- "- 18446744073709551615\n"
- "- 0\n"
- "- 19446744073709551615\n"
- "...\n",
- /*Ctxt=*/nullptr,
- suppressErrorMessages);
- yin >> seq;
-
- EXPECT_TRUE(!!yin.error());
-}
-
-
-//
-// Test error handling reading built-in int8_t type
-//
-TEST(YAMLIO, TestReadBuiltInTypesint8OverError) {
- std::vector<int8_t> seq;
- Input yin("---\n"
- "- -128\n"
- "- 0\n"
- "- 127\n"
- "- 128\n"
- "...\n",
- /*Ctxt=*/nullptr,
- suppressErrorMessages);
- yin >> seq;
-
- EXPECT_TRUE(!!yin.error());
-}
-
-//
-// Test error handling reading built-in int8_t type
-//
-TEST(YAMLIO, TestReadBuiltInTypesint8UnderError) {
- std::vector<int8_t> seq;
- Input yin("---\n"
- "- -128\n"
- "- 0\n"
- "- 127\n"
- "- -129\n"
- "...\n",
- /*Ctxt=*/nullptr,
- suppressErrorMessages);
- yin >> seq;
-
- EXPECT_TRUE(!!yin.error());
-}
-
-
-//
-// Test error handling reading built-in int16_t type
-//
-TEST(YAMLIO, TestReadBuiltInTypesint16UnderError) {
- std::vector<int16_t> seq;
- Input yin("---\n"
- "- 32767\n"
- "- 0\n"
- "- -32768\n"
- "- -32769\n"
- "...\n",
- /*Ctxt=*/nullptr,
- suppressErrorMessages);
- yin >> seq;
-
- EXPECT_TRUE(!!yin.error());
-}
-
-
-//
-// Test error handling reading built-in int16_t type
-//
-TEST(YAMLIO, TestReadBuiltInTypesint16OverError) {
- std::vector<int16_t> seq;
- Input yin("---\n"
- "- 32767\n"
- "- 0\n"
- "- -32768\n"
- "- 32768\n"
- "...\n",
- /*Ctxt=*/nullptr,
- suppressErrorMessages);
- yin >> seq;
-
- EXPECT_TRUE(!!yin.error());
-}
-
-
-//
-// Test error handling reading built-in int32_t type
-//
-TEST(YAMLIO, TestReadBuiltInTypesint32UnderError) {
- std::vector<int32_t> seq;
- Input yin("---\n"
- "- 2147483647\n"
- "- 0\n"
- "- -2147483648\n"
- "- -2147483649\n"
- "...\n",
- /*Ctxt=*/nullptr,
- suppressErrorMessages);
- yin >> seq;
-
- EXPECT_TRUE(!!yin.error());
-}
-
-//
-// Test error handling reading built-in int32_t type
-//
-TEST(YAMLIO, TestReadBuiltInTypesint32OverError) {
- std::vector<int32_t> seq;
- Input yin("---\n"
- "- 2147483647\n"
- "- 0\n"
- "- -2147483648\n"
- "- 2147483649\n"
- "...\n",
- /*Ctxt=*/nullptr,
- suppressErrorMessages);
- yin >> seq;
-
- EXPECT_TRUE(!!yin.error());
-}
-
-
-//
-// Test error handling reading built-in int64_t type
-//
-TEST(YAMLIO, TestReadBuiltInTypesint64UnderError) {
- std::vector<int64_t> seq;
- Input yin("---\n"
- "- -9223372036854775808\n"
- "- 0\n"
- "- 9223372036854775807\n"
- "- -9223372036854775809\n"
- "...\n",
- /*Ctxt=*/nullptr,
- suppressErrorMessages);
- yin >> seq;
-
- EXPECT_TRUE(!!yin.error());
-}
-
-//
-// Test error handling reading built-in int64_t type
-//
-TEST(YAMLIO, TestReadBuiltInTypesint64OverError) {
- std::vector<int64_t> seq;
- Input yin("---\n"
- "- -9223372036854775808\n"
- "- 0\n"
- "- 9223372036854775807\n"
- "- 9223372036854775809\n"
- "...\n",
- /*Ctxt=*/nullptr,
- suppressErrorMessages);
- yin >> seq;
-
- EXPECT_TRUE(!!yin.error());
-}
-
-//
-// Test error handling reading built-in float type
-//
-TEST(YAMLIO, TestReadBuiltInTypesFloatError) {
- std::vector<float> seq;
- Input yin("---\n"
- "- 0.0\n"
- "- 1000.1\n"
- "- -123.456\n"
- "- 1.2.3\n"
- "...\n",
- /*Ctxt=*/nullptr,
- suppressErrorMessages);
- yin >> seq;
-
- EXPECT_TRUE(!!yin.error());
-}
-
-//
-// Test error handling reading built-in float type
-//
-TEST(YAMLIO, TestReadBuiltInTypesDoubleError) {
- std::vector<double> seq;
- Input yin("---\n"
- "- 0.0\n"
- "- 1000.1\n"
- "- -123.456\n"
- "- 1.2.3\n"
- "...\n",
- /*Ctxt=*/nullptr,
- suppressErrorMessages);
- yin >> seq;
-
- EXPECT_TRUE(!!yin.error());
-}
-
-//
-// Test error handling reading built-in Hex8 type
-//
-LLVM_YAML_IS_SEQUENCE_VECTOR(Hex8)
-TEST(YAMLIO, TestReadBuiltInTypesHex8Error) {
- std::vector<Hex8> seq;
- Input yin("---\n"
- "- 0x12\n"
- "- 0xFE\n"
- "- 0x123\n"
- "...\n",
- /*Ctxt=*/nullptr,
- suppressErrorMessages);
- yin >> seq;
-
- EXPECT_TRUE(!!yin.error());
-}
-
-
-//
-// Test error handling reading built-in Hex16 type
-//
-LLVM_YAML_IS_SEQUENCE_VECTOR(Hex16)
-TEST(YAMLIO, TestReadBuiltInTypesHex16Error) {
- std::vector<Hex16> seq;
- Input yin("---\n"
- "- 0x0012\n"
- "- 0xFEFF\n"
- "- 0x12345\n"
- "...\n",
- /*Ctxt=*/nullptr,
- suppressErrorMessages);
- yin >> seq;
-
- EXPECT_TRUE(!!yin.error());
-}
-
-//
-// Test error handling reading built-in Hex32 type
-//
-LLVM_YAML_IS_SEQUENCE_VECTOR(Hex32)
-TEST(YAMLIO, TestReadBuiltInTypesHex32Error) {
- std::vector<Hex32> seq;
- Input yin("---\n"
- "- 0x0012\n"
- "- 0xFEFF0000\n"
- "- 0x1234556789\n"
- "...\n",
- /*Ctxt=*/nullptr,
- suppressErrorMessages);
- yin >> seq;
-
- EXPECT_TRUE(!!yin.error());
-}
-
-//
-// Test error handling reading built-in Hex64 type
-//
-LLVM_YAML_IS_SEQUENCE_VECTOR(Hex64)
-TEST(YAMLIO, TestReadBuiltInTypesHex64Error) {
- std::vector<Hex64> seq;
- Input yin("---\n"
- "- 0x0012\n"
- "- 0xFFEEDDCCBBAA9988\n"
- "- 0x12345567890ABCDEF0\n"
- "...\n",
- /*Ctxt=*/nullptr,
- suppressErrorMessages);
- yin >> seq;
-
- EXPECT_TRUE(!!yin.error());
-}
-
-TEST(YAMLIO, TestMalformedMapFailsGracefully) {
- FooBar doc;
- {
- // We pass the suppressErrorMessages handler to handle the error
- // message generated in the constructor of Input.
- Input yin("{foo:3, bar: 5}", /*Ctxt=*/nullptr, suppressErrorMessages);
- yin >> doc;
- EXPECT_TRUE(!!yin.error());
- }
-
- {
- Input yin("---\nfoo:3\nbar: 5\n...\n", /*Ctxt=*/nullptr, suppressErrorMessages);
- yin >> doc;
- EXPECT_TRUE(!!yin.error());
- }
-}
-
-struct OptionalTest {
- std::vector<int> Numbers;
-};
-
-struct OptionalTestSeq {
- std::vector<OptionalTest> Tests;
-};
-
-LLVM_YAML_IS_SEQUENCE_VECTOR(OptionalTest)
-namespace llvm {
-namespace yaml {
- template <>
- struct MappingTraits<OptionalTest> {
- static void mapping(IO& IO, OptionalTest &OT) {
- IO.mapOptional("Numbers", OT.Numbers);
- }
- };
-
- template <>
- struct MappingTraits<OptionalTestSeq> {
- static void mapping(IO &IO, OptionalTestSeq &OTS) {
- IO.mapOptional("Tests", OTS.Tests);
- }
- };
-}
-}
-
-TEST(YAMLIO, SequenceElideTest) {
- // Test that writing out a purely optional structure with its fields set to
- // default followed by other data is properly read back in.
- OptionalTestSeq Seq;
- OptionalTest One, Two, Three, Four;
- int N[] = {1, 2, 3};
- Three.Numbers.assign(N, N + 3);
- Seq.Tests.push_back(One);
- Seq.Tests.push_back(Two);
- Seq.Tests.push_back(Three);
- Seq.Tests.push_back(Four);
-
- std::string intermediate;
- {
- llvm::raw_string_ostream ostr(intermediate);
- Output yout(ostr);
- yout << Seq;
- }
-
- Input yin(intermediate);
- OptionalTestSeq Seq2;
- yin >> Seq2;
-
- EXPECT_FALSE(yin.error());
-
- EXPECT_EQ(4UL, Seq2.Tests.size());
-
- EXPECT_TRUE(Seq2.Tests[0].Numbers.empty());
- EXPECT_TRUE(Seq2.Tests[1].Numbers.empty());
-
- EXPECT_EQ(1, Seq2.Tests[2].Numbers[0]);
- EXPECT_EQ(2, Seq2.Tests[2].Numbers[1]);
- EXPECT_EQ(3, Seq2.Tests[2].Numbers[2]);
-
- EXPECT_TRUE(Seq2.Tests[3].Numbers.empty());
-}
-
-TEST(YAMLIO, TestEmptyStringFailsForMapWithRequiredFields) {
- FooBar doc;
- Input yin("");
- yin >> doc;
- EXPECT_TRUE(!!yin.error());
-}
-
-TEST(YAMLIO, TestEmptyStringSucceedsForMapWithOptionalFields) {
- OptionalTest doc;
- Input yin("");
- yin >> doc;
- EXPECT_FALSE(yin.error());
-}
-
-TEST(YAMLIO, TestEmptyStringSucceedsForSequence) {
- std::vector<uint8_t> seq;
- Input yin("", /*Ctxt=*/nullptr, suppressErrorMessages);
- yin >> seq;
-
- EXPECT_FALSE(yin.error());
- EXPECT_TRUE(seq.empty());
-}
-
-struct FlowMap {
- llvm::StringRef str1, str2, str3;
- FlowMap(llvm::StringRef str1, llvm::StringRef str2, llvm::StringRef str3)
- : str1(str1), str2(str2), str3(str3) {}
-};
-
-struct FlowSeq {
- llvm::StringRef str;
- FlowSeq(llvm::StringRef S) : str(S) {}
- FlowSeq() = default;
-};
-
-namespace llvm {
-namespace yaml {
- template <>
- struct MappingTraits<FlowMap> {
- static void mapping(IO &io, FlowMap &fm) {
- io.mapRequired("str1", fm.str1);
- io.mapRequired("str2", fm.str2);
- io.mapRequired("str3", fm.str3);
- }
-
- static const bool flow = true;
- };
-
-template <>
-struct ScalarTraits<FlowSeq> {
- static void output(const FlowSeq &value, void*, llvm::raw_ostream &out) {
- out << value.str;
- }
- static StringRef input(StringRef scalar, void*, FlowSeq &value) {
- value.str = scalar;
- return "";
- }
-
- static QuotingType mustQuote(StringRef S) { return QuotingType::None; }
-};
-}
-}
-
-LLVM_YAML_IS_FLOW_SEQUENCE_VECTOR(FlowSeq)
-
-TEST(YAMLIO, TestWrapFlow) {
- std::string out;
- llvm::raw_string_ostream ostr(out);
- FlowMap Map("This is str1", "This is str2", "This is str3");
- std::vector<FlowSeq> Seq;
- Seq.emplace_back("This is str1");
- Seq.emplace_back("This is str2");
- Seq.emplace_back("This is str3");
-
- {
- // 20 is just bellow the total length of the first mapping field.
- // We should wreap at every element.
- Output yout(ostr, nullptr, 15);
-
- yout << Map;
- ostr.flush();
- EXPECT_EQ(out,
- "---\n"
- "{ str1: This is str1, \n"
- " str2: This is str2, \n"
- " str3: This is str3 }\n"
- "...\n");
- out.clear();
-
- yout << Seq;
- ostr.flush();
- EXPECT_EQ(out,
- "---\n"
- "[ This is str1, \n"
- " This is str2, \n"
- " This is str3 ]\n"
- "...\n");
- out.clear();
- }
- {
- // 25 will allow the second field to be output on the first line.
- Output yout(ostr, nullptr, 25);
-
- yout << Map;
- ostr.flush();
- EXPECT_EQ(out,
- "---\n"
- "{ str1: This is str1, str2: This is str2, \n"
- " str3: This is str3 }\n"
- "...\n");
- out.clear();
-
- yout << Seq;
- ostr.flush();
- EXPECT_EQ(out,
- "---\n"
- "[ This is str1, This is str2, \n"
- " This is str3 ]\n"
- "...\n");
- out.clear();
- }
- {
- // 0 means no wrapping.
- Output yout(ostr, nullptr, 0);
-
- yout << Map;
- ostr.flush();
- EXPECT_EQ(out,
- "---\n"
- "{ str1: This is str1, str2: This is str2, str3: This is str3 }\n"
- "...\n");
- out.clear();
-
- yout << Seq;
- ostr.flush();
- EXPECT_EQ(out,
- "---\n"
- "[ This is str1, This is str2, This is str3 ]\n"
- "...\n");
- out.clear();
- }
-}
-
-struct MappingContext {
- int A = 0;
-};
-struct SimpleMap {
- int B = 0;
- int C = 0;
-};
-
-struct NestedMap {
- NestedMap(MappingContext &Context) : Context(Context) {}
- SimpleMap Simple;
- MappingContext &Context;
-};
-
-namespace llvm {
-namespace yaml {
-template <> struct MappingContextTraits<SimpleMap, MappingContext> {
- static void mapping(IO &io, SimpleMap &sm, MappingContext &Context) {
- io.mapRequired("B", sm.B);
- io.mapRequired("C", sm.C);
- ++Context.A;
- io.mapRequired("Context", Context.A);
- }
-};
-
-template <> struct MappingTraits<NestedMap> {
- static void mapping(IO &io, NestedMap &nm) {
- io.mapRequired("Simple", nm.Simple, nm.Context);
- }
-};
-}
-}
-
-TEST(YAMLIO, TestMapWithContext) {
- MappingContext Context;
- NestedMap Nested(Context);
- std::string out;
- llvm::raw_string_ostream ostr(out);
-
- Output yout(ostr, nullptr, 15);
-
- yout << Nested;
- ostr.flush();
- EXPECT_EQ(1, Context.A);
- EXPECT_EQ("---\n"
- "Simple: \n"
- " B: 0\n"
- " C: 0\n"
- " Context: 1\n"
- "...\n",
- out);
-
- out.clear();
-
- Nested.Simple.B = 2;
- Nested.Simple.C = 3;
- yout << Nested;
- ostr.flush();
- EXPECT_EQ(2, Context.A);
- EXPECT_EQ("---\n"
- "Simple: \n"
- " B: 2\n"
- " C: 3\n"
- " Context: 2\n"
- "...\n",
- out);
- out.clear();
-}
-
-LLVM_YAML_IS_STRING_MAP(int)
-
-TEST(YAMLIO, TestCustomMapping) {
- std::map<std::string, int> x;
- x["foo"] = 1;
- x["bar"] = 2;
-
- std::string out;
- llvm::raw_string_ostream ostr(out);
- Output xout(ostr, nullptr, 0);
-
- xout << x;
- ostr.flush();
- EXPECT_EQ("---\n"
- "bar: 2\n"
- "foo: 1\n"
- "...\n",
- out);
-
- Input yin(out);
- std::map<std::string, int> y;
- yin >> y;
- EXPECT_EQ(2ul, y.size());
- EXPECT_EQ(1, y["foo"]);
- EXPECT_EQ(2, y["bar"]);
-}
-
-LLVM_YAML_IS_STRING_MAP(FooBar)
-
-TEST(YAMLIO, TestCustomMappingStruct) {
- std::map<std::string, FooBar> x;
- x["foo"].foo = 1;
- x["foo"].bar = 2;
- x["bar"].foo = 3;
- x["bar"].bar = 4;
-
- std::string out;
- llvm::raw_string_ostream ostr(out);
- Output xout(ostr, nullptr, 0);
-
- xout << x;
- ostr.flush();
- EXPECT_EQ("---\n"
- "bar: \n"
- " foo: 3\n"
- " bar: 4\n"
- "foo: \n"
- " foo: 1\n"
- " bar: 2\n"
- "...\n",
- out);
-
- Input yin(out);
- std::map<std::string, FooBar> y;
- yin >> y;
- EXPECT_EQ(2ul, y.size());
- EXPECT_EQ(1, y["foo"].foo);
- EXPECT_EQ(2, y["foo"].bar);
- EXPECT_EQ(3, y["bar"].foo);
- EXPECT_EQ(4, y["bar"].bar);
-}
-
-static void TestEscaped(llvm::StringRef Input, llvm::StringRef Expected) {
- std::string out;
- llvm::raw_string_ostream ostr(out);
- Output xout(ostr, nullptr, 0);
-
- llvm::yaml::EmptyContext Ctx;
- yamlize(xout, Input, true, Ctx);
-
- ostr.flush();
-
- // Make a separate StringRef so we get nice byte-by-byte output.
- llvm::StringRef Got(out);
- EXPECT_EQ(Expected, Got);
-}
-
-TEST(YAMLIO, TestEscaped) {
- // Single quote
- TestEscaped("@abc@", "'@abc@'");
- // No quote
- TestEscaped("abc", "abc");
- // Forward slash quoted
- TestEscaped("abc/", "'abc/'");
- // Double quote non-printable
- TestEscaped("\01@abc@", "\"\\x01@abc@\"");
- // Double quote inside single quote
- TestEscaped("abc\"fdf", "'abc\"fdf'");
- // Double quote inside double quote
- TestEscaped("\01bc\"fdf", "\"\\x01bc\\\"fdf\"");
- // Single quote inside single quote
- TestEscaped("abc'fdf", "'abc''fdf'");
- // UTF8
- TestEscaped("/*параметр*/", "\"/*параметр*/\"");
- // UTF8 with single quote inside double quote
- TestEscaped("parameter 'параметр' is unused",
- "\"parameter 'параметр' is unused\"");
-
- // String with embedded non-printable multibyte UTF-8 sequence (U+200B
- // zero-width space). The thing to test here is that we emit a
- // unicode-scalar level escape like \uNNNN (at the YAML level), and don't
- // just pass the UTF-8 byte sequence through as with quoted printables.
- {
- const unsigned char foobar[10] = {'f', 'o', 'o',
- 0xE2, 0x80, 0x8B, // UTF-8 of U+200B
- 'b', 'a', 'r',
- 0x0};
- TestEscaped((char const *)foobar, "\"foo\\u200Bbar\"");
- }
-}
-
-TEST(YAMLIO, Numeric) {
- EXPECT_TRUE(isNumeric(".inf"));
- EXPECT_TRUE(isNumeric(".INF"));
- EXPECT_TRUE(isNumeric(".Inf"));
- EXPECT_TRUE(isNumeric("-.inf"));
- EXPECT_TRUE(isNumeric("+.inf"));
-
- EXPECT_TRUE(isNumeric(".nan"));
- EXPECT_TRUE(isNumeric(".NaN"));
- EXPECT_TRUE(isNumeric(".NAN"));
-
- EXPECT_TRUE(isNumeric("0"));
- EXPECT_TRUE(isNumeric("0."));
- EXPECT_TRUE(isNumeric("0.0"));
- EXPECT_TRUE(isNumeric("-0.0"));
- EXPECT_TRUE(isNumeric("+0.0"));
-
- EXPECT_TRUE(isNumeric("12345"));
- EXPECT_TRUE(isNumeric("012345"));
- EXPECT_TRUE(isNumeric("+12.0"));
- EXPECT_TRUE(isNumeric(".5"));
- EXPECT_TRUE(isNumeric("+.5"));
- EXPECT_TRUE(isNumeric("-1.0"));
-
- EXPECT_TRUE(isNumeric("2.3e4"));
- EXPECT_TRUE(isNumeric("-2E+05"));
- EXPECT_TRUE(isNumeric("+12e03"));
- EXPECT_TRUE(isNumeric("6.8523015e+5"));
-
- EXPECT_TRUE(isNumeric("1.e+1"));
- EXPECT_TRUE(isNumeric(".0e+1"));
-
- EXPECT_TRUE(isNumeric("0x2aF3"));
- EXPECT_TRUE(isNumeric("0o01234567"));
-
- EXPECT_FALSE(isNumeric("not a number"));
- EXPECT_FALSE(isNumeric("."));
- EXPECT_FALSE(isNumeric(".e+1"));
- EXPECT_FALSE(isNumeric(".1e"));
- EXPECT_FALSE(isNumeric(".1e+"));
- EXPECT_FALSE(isNumeric(".1e++1"));
-
- EXPECT_FALSE(isNumeric("ABCD"));
- EXPECT_FALSE(isNumeric("+0x2AF3"));
- EXPECT_FALSE(isNumeric("-0x2AF3"));
- EXPECT_FALSE(isNumeric("0x2AF3Z"));
- EXPECT_FALSE(isNumeric("0o012345678"));
- EXPECT_FALSE(isNumeric("0xZ"));
- EXPECT_FALSE(isNumeric("-0o012345678"));
- EXPECT_FALSE(isNumeric("000003A8229434B839616A25C16B0291F77A438B"));
-
- EXPECT_FALSE(isNumeric(""));
- EXPECT_FALSE(isNumeric("."));
- EXPECT_FALSE(isNumeric(".e+1"));
- EXPECT_FALSE(isNumeric(".e+"));
- EXPECT_FALSE(isNumeric(".e"));
- EXPECT_FALSE(isNumeric("e1"));
-
- // Deprecated formats: as for YAML 1.2 specification, the following are not
- // valid numbers anymore:
- //
- // * Sexagecimal numbers
- // * Decimal numbers with comma s the delimiter
- // * "inf", "nan" without '.' prefix
- EXPECT_FALSE(isNumeric("3:25:45"));
- EXPECT_FALSE(isNumeric("+12,345"));
- EXPECT_FALSE(isNumeric("-inf"));
- EXPECT_FALSE(isNumeric("1,230.15"));
-}
-
-//===----------------------------------------------------------------------===//
-// Test PolymorphicTraits and TaggedScalarTraits
-//===----------------------------------------------------------------------===//
-
-struct Poly {
- enum NodeKind {
- NK_Scalar,
- NK_Seq,
- NK_Map,
- } Kind;
-
- Poly(NodeKind Kind) : Kind(Kind) {}
-
- virtual ~Poly() = default;
-
- NodeKind getKind() const { return Kind; }
-};
-
-struct Scalar : Poly {
- enum ScalarKind {
- SK_Unknown,
- SK_Double,
- SK_Bool,
- } SKind;
-
- union {
- double DoubleValue;
- bool BoolValue;
- };
-
- Scalar() : Poly(NK_Scalar), SKind(SK_Unknown) {}
- Scalar(double DoubleValue)
- : Poly(NK_Scalar), SKind(SK_Double), DoubleValue(DoubleValue) {}
- Scalar(bool BoolValue)
- : Poly(NK_Scalar), SKind(SK_Bool), BoolValue(BoolValue) {}
-
- static bool classof(const Poly *N) { return N->getKind() == NK_Scalar; }
-};
-
-struct Seq : Poly, std::vector<std::unique_ptr<Poly>> {
- Seq() : Poly(NK_Seq) {}
-
- static bool classof(const Poly *N) { return N->getKind() == NK_Seq; }
-};
-
-struct Map : Poly, llvm::StringMap<std::unique_ptr<Poly>> {
- Map() : Poly(NK_Map) {}
-
- static bool classof(const Poly *N) { return N->getKind() == NK_Map; }
-};
-
-namespace llvm {
-namespace yaml {
-
-template <> struct PolymorphicTraits<std::unique_ptr<Poly>> {
- static NodeKind getKind(const std::unique_ptr<Poly> &N) {
- if (isa<Scalar>(*N))
- return NodeKind::Scalar;
- if (isa<Seq>(*N))
- return NodeKind::Sequence;
- if (isa<Map>(*N))
- return NodeKind::Map;
- llvm_unreachable("unsupported node type");
- }
-
- static Scalar &getAsScalar(std::unique_ptr<Poly> &N) {
- if (!N || !isa<Scalar>(*N))
- N = llvm::make_unique<Scalar>();
- return *cast<Scalar>(N.get());
- }
-
- static Seq &getAsSequence(std::unique_ptr<Poly> &N) {
- if (!N || !isa<Seq>(*N))
- N = llvm::make_unique<Seq>();
- return *cast<Seq>(N.get());
- }
-
- static Map &getAsMap(std::unique_ptr<Poly> &N) {
- if (!N || !isa<Map>(*N))
- N = llvm::make_unique<Map>();
- return *cast<Map>(N.get());
- }
-};
-
-template <> struct TaggedScalarTraits<Scalar> {
- static void output(const Scalar &S, void *Ctxt, raw_ostream &ScalarOS,
- raw_ostream &TagOS) {
- switch (S.SKind) {
- case Scalar::SK_Unknown:
- report_fatal_error("output unknown scalar");
- break;
- case Scalar::SK_Double:
- TagOS << "!double";
- ScalarTraits<double>::output(S.DoubleValue, Ctxt, ScalarOS);
- break;
- case Scalar::SK_Bool:
- TagOS << "!bool";
- ScalarTraits<bool>::output(S.BoolValue, Ctxt, ScalarOS);
- break;
- }
- }
-
- static StringRef input(StringRef ScalarStr, StringRef Tag, void *Ctxt,
- Scalar &S) {
- S.SKind = StringSwitch<Scalar::ScalarKind>(Tag)
- .Case("!double", Scalar::SK_Double)
- .Case("!bool", Scalar::SK_Bool)
- .Default(Scalar::SK_Unknown);
- switch (S.SKind) {
- case Scalar::SK_Unknown:
- return StringRef("unknown scalar tag");
- case Scalar::SK_Double:
- return ScalarTraits<double>::input(ScalarStr, Ctxt, S.DoubleValue);
- case Scalar::SK_Bool:
- return ScalarTraits<bool>::input(ScalarStr, Ctxt, S.BoolValue);
- }
- llvm_unreachable("unknown scalar kind");
- }
-
- static QuotingType mustQuote(const Scalar &S, StringRef Str) {
- switch (S.SKind) {
- case Scalar::SK_Unknown:
- report_fatal_error("quote unknown scalar");
- case Scalar::SK_Double:
- return ScalarTraits<double>::mustQuote(Str);
- case Scalar::SK_Bool:
- return ScalarTraits<bool>::mustQuote(Str);
- }
- llvm_unreachable("unknown scalar kind");
- }
-};
-
-template <> struct CustomMappingTraits<Map> {
- static void inputOne(IO &IO, StringRef Key, Map &M) {
- IO.mapRequired(Key.str().c_str(), M[Key]);
- }
-
- static void output(IO &IO, Map &M) {
- for (auto &N : M)
- IO.mapRequired(N.getKey().str().c_str(), N.getValue());
- }
-};
-
-template <> struct SequenceTraits<Seq> {
- static size_t size(IO &IO, Seq &A) { return A.size(); }
-
- static std::unique_ptr<Poly> &element(IO &IO, Seq &A, size_t Index) {
- if (Index >= A.size())
- A.resize(Index + 1);
- return A[Index];
- }
-};
-
-} // namespace yaml
-} // namespace llvm
-
-TEST(YAMLIO, TestReadWritePolymorphicScalar) {
- std::string intermediate;
- std::unique_ptr<Poly> node = llvm::make_unique<Scalar>(true);
-
- llvm::raw_string_ostream ostr(intermediate);
- Output yout(ostr);
-#ifdef GTEST_HAS_DEATH_TEST
-#ifndef NDEBUG
- EXPECT_DEATH(yout << node, "plain scalar documents are not supported");
-#endif
-#endif
-}
-
-TEST(YAMLIO, TestReadWritePolymorphicSeq) {
- std::string intermediate;
- {
- auto seq = llvm::make_unique<Seq>();
- seq->push_back(llvm::make_unique<Scalar>(true));
- seq->push_back(llvm::make_unique<Scalar>(1.0));
- auto node = llvm::unique_dyn_cast<Poly>(seq);
-
- llvm::raw_string_ostream ostr(intermediate);
- Output yout(ostr);
- yout << node;
- }
- {
- Input yin(intermediate);
- std::unique_ptr<Poly> node;
- yin >> node;
-
- EXPECT_FALSE(yin.error());
- auto seq = llvm::dyn_cast<Seq>(node.get());
- ASSERT_TRUE(seq);
- ASSERT_EQ(seq->size(), 2u);
- auto first = llvm::dyn_cast<Scalar>((*seq)[0].get());
- ASSERT_TRUE(first);
- EXPECT_EQ(first->SKind, Scalar::SK_Bool);
- EXPECT_TRUE(first->BoolValue);
- auto second = llvm::dyn_cast<Scalar>((*seq)[1].get());
- ASSERT_TRUE(second);
- EXPECT_EQ(second->SKind, Scalar::SK_Double);
- EXPECT_EQ(second->DoubleValue, 1.0);
- }
-}
-
-TEST(YAMLIO, TestReadWritePolymorphicMap) {
- std::string intermediate;
- {
- auto map = llvm::make_unique<Map>();
- (*map)["foo"] = llvm::make_unique<Scalar>(false);
- (*map)["bar"] = llvm::make_unique<Scalar>(2.0);
- std::unique_ptr<Poly> node = llvm::unique_dyn_cast<Poly>(map);
-
- llvm::raw_string_ostream ostr(intermediate);
- Output yout(ostr);
- yout << node;
- }
- {
- Input yin(intermediate);
- std::unique_ptr<Poly> node;
- yin >> node;
-
- EXPECT_FALSE(yin.error());
- auto map = llvm::dyn_cast<Map>(node.get());
- ASSERT_TRUE(map);
- auto foo = llvm::dyn_cast<Scalar>((*map)["foo"].get());
- ASSERT_TRUE(foo);
- EXPECT_EQ(foo->SKind, Scalar::SK_Bool);
- EXPECT_FALSE(foo->BoolValue);
- auto bar = llvm::dyn_cast<Scalar>((*map)["bar"].get());
- ASSERT_TRUE(bar);
- EXPECT_EQ(bar->SKind, Scalar::SK_Double);
- EXPECT_EQ(bar->DoubleValue, 2.0);
- }
-}