mirror of
https://github.com/RPCS3/llvm-mirror.git
synced 2024-10-26 22:42:46 +02:00
8eaa591b52
This commit creates a new structure called 'SlotMapping' in the AsmParser library. This structure can be passed into the public parsing APIs from the AsmParser library in order to extract the data structures that map from slot numbers to unnamed global values and metadata nodes. This change is useful for MIR Serialization, as the MIR Parser has to lookup the unnamed global values and metadata nodes by their slot numbers. Reviewers: Duncan P. N. Exon Smith Differential Revision: http://reviews.llvm.org/D10551 llvm-svn: 240427
68 lines
1.9 KiB
C++
68 lines
1.9 KiB
C++
//===- llvm/unittest/AsmParser/AsmParserTest.cpp - asm parser unittests ---===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/ADT/StringRef.h"
|
|
#include "llvm/AsmParser/Parser.h"
|
|
#include "llvm/AsmParser/SlotMapping.h"
|
|
#include "llvm/IR/LLVMContext.h"
|
|
#include "llvm/IR/Module.h"
|
|
#include "llvm/Support/SourceMgr.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
using namespace llvm;
|
|
|
|
namespace {
|
|
|
|
TEST(AsmParserTest, NullTerminatedInput) {
|
|
LLVMContext &Ctx = getGlobalContext();
|
|
StringRef Source = "; Empty module \n";
|
|
SMDiagnostic Error;
|
|
auto Mod = parseAssemblyString(Source, Error, Ctx);
|
|
|
|
EXPECT_TRUE(Mod != nullptr);
|
|
EXPECT_TRUE(Error.getMessage().empty());
|
|
}
|
|
|
|
#ifdef GTEST_HAS_DEATH_TEST
|
|
#ifndef NDEBUG
|
|
|
|
TEST(AsmParserTest, NonNullTerminatedInput) {
|
|
LLVMContext &Ctx = getGlobalContext();
|
|
StringRef Source = "; Empty module \n\1\2";
|
|
SMDiagnostic Error;
|
|
std::unique_ptr<Module> Mod;
|
|
EXPECT_DEATH(Mod = parseAssemblyString(Source.substr(0, Source.size() - 2),
|
|
Error, Ctx),
|
|
"Buffer is not null terminated!");
|
|
}
|
|
|
|
#endif
|
|
#endif
|
|
|
|
TEST(AsmParserTest, SlotMappingTest) {
|
|
LLVMContext &Ctx = getGlobalContext();
|
|
StringRef Source = "@0 = global i32 0\n !0 = !{}\n !42 = !{i32 42}";
|
|
SMDiagnostic Error;
|
|
SlotMapping Mapping;
|
|
auto Mod = parseAssemblyString(Source, Error, Ctx, &Mapping);
|
|
|
|
EXPECT_TRUE(Mod != nullptr);
|
|
EXPECT_TRUE(Error.getMessage().empty());
|
|
|
|
ASSERT_EQ(Mapping.GlobalValues.size(), 1u);
|
|
EXPECT_TRUE(isa<GlobalVariable>(Mapping.GlobalValues[0]));
|
|
|
|
EXPECT_EQ(Mapping.MetadataNodes.size(), 2u);
|
|
EXPECT_EQ(Mapping.MetadataNodes.count(0), 1u);
|
|
EXPECT_EQ(Mapping.MetadataNodes.count(42), 1u);
|
|
EXPECT_EQ(Mapping.MetadataNodes.count(1), 0u);
|
|
}
|
|
|
|
} // end anonymous namespace
|