forked from OSchip/llvm-project
56 lines
1.5 KiB
C++
56 lines
1.5 KiB
C++
//===- YAMLTest.cpp - Tests for Object YAML -------------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/ObjectYAML/YAML.h"
|
|
#include "llvm/Support/YAMLTraits.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
using namespace llvm;
|
|
|
|
struct BinaryHolder {
|
|
yaml::BinaryRef Binary;
|
|
};
|
|
|
|
namespace llvm {
|
|
namespace yaml {
|
|
template <>
|
|
struct MappingTraits<BinaryHolder> {
|
|
static void mapping(IO &IO, BinaryHolder &BH) {
|
|
IO.mapRequired("Binary", BH.Binary);
|
|
}
|
|
};
|
|
} // end namespace yaml
|
|
} // end namespace llvm
|
|
|
|
TEST(ObjectYAML, BinaryRef) {
|
|
BinaryHolder BH;
|
|
SmallVector<char, 32> Buf;
|
|
llvm::raw_svector_ostream OS(Buf);
|
|
yaml::Output YOut(OS);
|
|
YOut << BH;
|
|
EXPECT_NE(OS.str().find("''"), StringRef::npos);
|
|
}
|
|
|
|
TEST(ObjectYAML, UnknownOption) {
|
|
StringRef InputYAML = "InvalidKey: InvalidValue\n"
|
|
"Binary: AAAA\n";
|
|
BinaryHolder BH;
|
|
yaml::Input Input(InputYAML);
|
|
// test 1: default in trying to parse invalid key is an error case.
|
|
Input >> BH;
|
|
EXPECT_EQ(Input.error().value(), 22);
|
|
|
|
// test 2: only warn about invalid key if actively set.
|
|
yaml::Input Input2(InputYAML);
|
|
BinaryHolder BH2;
|
|
Input2.setAllowUnknownKeys(true);
|
|
Input2 >> BH2;
|
|
EXPECT_EQ(BH2.Binary, yaml::BinaryRef("AAAA"));
|
|
EXPECT_EQ(Input2.error().value(), 0);
|
|
}
|