From 2f672d610e97a671c742689e12560eaeadf94f15 Mon Sep 17 00:00:00 2001
From: Sean Silva <silvas@purdue.edu>
Date: Tue, 9 Jul 2013 00:54:46 +0000
Subject: Make BinaryRef output correctly in case of empty data.

Previously, it would simply output nothing, but it should output an
empty string `""`.

llvm-svn: 185894
---
 llvm/unittests/Object/YAMLTest.cpp | 40 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 40 insertions(+)
 create mode 100644 llvm/unittests/Object/YAMLTest.cpp

(limited to 'llvm/unittests/Object/YAMLTest.cpp')

diff --git a/llvm/unittests/Object/YAMLTest.cpp b/llvm/unittests/Object/YAMLTest.cpp
new file mode 100644
index 00000000000..3428e94d64d
--- /dev/null
+++ b/llvm/unittests/Object/YAMLTest.cpp
@@ -0,0 +1,40 @@
+//===- llvm/unittest/Object/YAMLTest.cpp - Tests for Object YAML ----------===//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+
+#include "llvm/Object/YAML.h"
+#include "llvm/Support/YAMLTraits.h"
+#include "gtest/gtest.h"
+
+using namespace llvm;
+
+namespace {
+struct BinaryHolder {
+  object::yaml::BinaryRef Binary;
+};
+} // end anonymous namespace
+
+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);
+}
-- 
cgit v1.2.3