From 5fa5ecf852afc93e3d657bde1c5ec63b98efb209 Mon Sep 17 00:00:00 2001 From: Daniel Dunbar Date: Wed, 29 Feb 2012 20:30:56 +0000 Subject: Bitcode: Don't expose WriteBitcodeToStream to clients. llvm-svn: 151747 --- llvm/unittests/Bitcode/BitReaderTest.cpp | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) (limited to 'llvm/unittests/Bitcode/BitReaderTest.cpp') diff --git a/llvm/unittests/Bitcode/BitReaderTest.cpp b/llvm/unittests/Bitcode/BitReaderTest.cpp index 91e6c151558..68cfe2836a2 100644 --- a/llvm/unittests/Bitcode/BitReaderTest.cpp +++ b/llvm/unittests/Bitcode/BitReaderTest.cpp @@ -7,6 +7,7 @@ // //===----------------------------------------------------------------------===// +#include "llvm/ADT/SmallString.h" #include "llvm/Analysis/Verifier.h" #include "llvm/Bitcode/BitstreamWriter.h" #include "llvm/Bitcode/ReaderWriter.h" @@ -43,17 +44,16 @@ static Module *makeLLVMModule() { return Mod; } -static void writeModuleToBuffer(std::vector &Buffer) { +static void writeModuleToBuffer(SmallVectorImpl &Buffer) { Module *Mod = makeLLVMModule(); - BitstreamWriter Stream(Buffer); - WriteBitcodeToStream(Mod, Stream); + raw_svector_ostream OS(Buffer); + WriteBitcodeToFile(Mod, OS); } TEST(BitReaderTest, MaterializeFunctionsForBlockAddr) { // PR11677 - std::vector Mem; + SmallString<1024> Mem; writeModuleToBuffer(Mem); - StringRef Data((const char*)&Mem[0], Mem.size()); - MemoryBuffer *Buffer = MemoryBuffer::getMemBuffer(Data, "test", false); + MemoryBuffer *Buffer = MemoryBuffer::getMemBuffer(Mem.str(), "test", false); std::string errMsg; Module *m = getLazyBitcodeModule(Buffer, getGlobalContext(), &errMsg); PassManager passes; -- cgit v1.2.3