392 lines
15 KiB
C++
392 lines
15 KiB
C++
//===- lld/unittest/InputGraphTest.cpp -----------------------------------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
///
|
|
/// \file
|
|
/// \brief InputGraph Tests
|
|
///
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
#include "lld/Core/InputGraph.h"
|
|
#include "lld/Core/Resolver.h"
|
|
#include "lld/ReaderWriter/Simple.h"
|
|
|
|
using namespace lld;
|
|
|
|
namespace {
|
|
|
|
class MyLinkingContext : public LinkingContext {
|
|
public:
|
|
Writer &writer() const override { llvm_unreachable("no writer!"); }
|
|
|
|
bool validateImpl(raw_ostream &) override { return true; }
|
|
};
|
|
|
|
class MyFileNode : public SimpleFileNode {
|
|
public:
|
|
MyFileNode(StringRef path) : SimpleFileNode(path) {}
|
|
|
|
void resetNextIndex() override { FileNode::resetNextIndex(); }
|
|
};
|
|
|
|
class MyExpandFileNode : public SimpleFileNode {
|
|
public:
|
|
MyExpandFileNode(StringRef path) : SimpleFileNode(path) {}
|
|
|
|
/// \brief How do we want to expand the current node?
|
|
bool shouldExpand() const override { return true; }
|
|
|
|
/// \brief Get the elements that we want to expand with.
|
|
range<InputGraph::InputElementIterT> expandElements() override {
|
|
return make_range(_expandElements.begin(), _expandElements.end());
|
|
}
|
|
|
|
/// Process the input Elemenet
|
|
virtual bool addElement(std::unique_ptr<InputElement> element) {
|
|
_expandElements.push_back(std::move(element));
|
|
return true;
|
|
}
|
|
|
|
private:
|
|
InputGraph::InputElementVectorT _expandElements;
|
|
};
|
|
|
|
class InputGraphTest : public testing::Test {
|
|
public:
|
|
InputGraphTest() {
|
|
_ctx.setInputGraph(std::unique_ptr<InputGraph>(new InputGraph()));
|
|
}
|
|
|
|
InputGraph &getInputGraph() { return _ctx.getInputGraph(); }
|
|
int inputFileCount() { return _ctx.getInputGraph().size(); }
|
|
|
|
protected:
|
|
MyLinkingContext _ctx;
|
|
};
|
|
|
|
} // end anonymous namespace
|
|
|
|
TEST_F(InputGraphTest, Basic) {
|
|
EXPECT_EQ(0, inputFileCount());
|
|
ErrorOr<InputElement *> nextElement = getInputGraph().getNextInputElement();
|
|
EXPECT_EQ(InputGraphError::no_more_elements, nextElement.getError());
|
|
}
|
|
|
|
TEST_F(InputGraphTest, AddAFile) {
|
|
std::unique_ptr<MyFileNode> myfile(new MyFileNode("file1"));
|
|
EXPECT_EQ(true, getInputGraph().addInputElement(std::move(myfile)));
|
|
EXPECT_EQ(1, inputFileCount());
|
|
ErrorOr<InputElement *> nextElement = getInputGraph().getNextInputElement();
|
|
EXPECT_NE(InputGraphError::no_more_elements, nextElement.getError());
|
|
EXPECT_EQ(InputElement::Kind::File, (*nextElement)->kind());
|
|
FileNode *fileNode = dyn_cast<FileNode>(*nextElement);
|
|
EXPECT_EQ("file1", fileNode->getUserPath());
|
|
nextElement = getInputGraph().getNextInputElement();
|
|
EXPECT_EQ(InputGraphError::no_more_elements, nextElement.getError());
|
|
}
|
|
|
|
TEST_F(InputGraphTest, AddAFileWithLLDFiles) {
|
|
std::unique_ptr<MyFileNode> myfile(new MyFileNode("multi_files"));
|
|
std::vector<std::unique_ptr<File>> objfiles;
|
|
std::unique_ptr<SimpleFile> obj1(new SimpleFile("objfile1"));
|
|
std::unique_ptr<SimpleFile> obj2(new SimpleFile("objfile2"));
|
|
objfiles.push_back(std::move(obj1));
|
|
objfiles.push_back(std::move(obj2));
|
|
myfile->addFiles(std::move(objfiles));
|
|
EXPECT_EQ(true, getInputGraph().addInputElement(std::move(myfile)));
|
|
EXPECT_EQ(1, inputFileCount());
|
|
ErrorOr<InputElement *> nextElement = getInputGraph().getNextInputElement();
|
|
EXPECT_NE(InputGraphError::no_more_elements, nextElement.getError());
|
|
EXPECT_EQ(InputElement::Kind::File, (*nextElement)->kind());
|
|
FileNode *fileNode = dyn_cast<FileNode>(*nextElement);
|
|
|
|
EXPECT_EQ("multi_files", fileNode->getUserPath());
|
|
|
|
ErrorOr<File &> objfile = fileNode->getNextFile();
|
|
EXPECT_NE(InputGraphError::no_more_files, objfile.getError());
|
|
EXPECT_EQ("objfile1", (*objfile).path());
|
|
|
|
objfile = fileNode->getNextFile();
|
|
EXPECT_NE(InputGraphError::no_more_files, objfile.getError());
|
|
EXPECT_EQ("objfile2", (*objfile).path());
|
|
|
|
objfile = fileNode->getNextFile();
|
|
EXPECT_EQ(InputGraphError::no_more_files, objfile.getError());
|
|
|
|
fileNode->resetNextIndex();
|
|
|
|
objfile = fileNode->getNextFile();
|
|
EXPECT_NE(InputGraphError::no_more_files, objfile.getError());
|
|
EXPECT_EQ("objfile1", (*objfile).path());
|
|
|
|
nextElement = getInputGraph().getNextInputElement();
|
|
EXPECT_EQ(InputGraphError::no_more_elements, nextElement.getError());
|
|
}
|
|
|
|
TEST_F(InputGraphTest, AddNodeWithFilesAndGroup) {
|
|
std::unique_ptr<MyFileNode> myfile(new MyFileNode("multi_files1"));
|
|
std::vector<std::unique_ptr<File>> objfiles;
|
|
std::unique_ptr<SimpleFile> obj1(new SimpleFile("objfile1"));
|
|
std::unique_ptr<SimpleFile> obj2(new SimpleFile("objfile2"));
|
|
objfiles.push_back(std::move(obj1));
|
|
objfiles.push_back(std::move(obj2));
|
|
myfile->addFiles(std::move(objfiles));
|
|
EXPECT_EQ(true, getInputGraph().addInputElement(std::move(myfile)));
|
|
|
|
// Create a group node with two elements
|
|
// an file node which looks like an archive and
|
|
// two file nodes
|
|
std::unique_ptr<Group> mygroup(new Group());
|
|
std::unique_ptr<MyFileNode> myarchive(new MyFileNode("archive_file"));
|
|
std::vector<std::unique_ptr<File>> objfiles_group;
|
|
std::unique_ptr<SimpleFile> obj_1(new SimpleFile("objfile_1"));
|
|
std::unique_ptr<SimpleFile> obj_2(new SimpleFile("objfile_2"));
|
|
objfiles_group.push_back(std::move(obj_1));
|
|
objfiles_group.push_back(std::move(obj_2));
|
|
myarchive->addFiles(std::move(objfiles_group));
|
|
EXPECT_EQ(true, mygroup->addFile(std::move(myarchive)));
|
|
|
|
std::unique_ptr<MyFileNode> mygroupobjfile_1(
|
|
new MyFileNode("group_objfile1"));
|
|
std::vector<std::unique_ptr<File>> objfiles_group1;
|
|
std::unique_ptr<SimpleFile> mygroupobj1(new SimpleFile("group_objfile1"));
|
|
objfiles_group1.push_back(std::move(mygroupobj1));
|
|
mygroupobjfile_1->addFiles(std::move(objfiles_group1));
|
|
EXPECT_EQ(true, mygroup->addFile(std::move(mygroupobjfile_1)));
|
|
|
|
std::unique_ptr<MyFileNode> mygroupobjfile_2(
|
|
new MyFileNode("group_objfile2"));
|
|
std::vector<std::unique_ptr<File>> objfiles_group2;
|
|
std::unique_ptr<SimpleFile> mygroupobj2(new SimpleFile("group_objfile2"));
|
|
objfiles_group2.push_back(std::move(mygroupobj2));
|
|
mygroupobjfile_2->addFiles(std::move(objfiles_group2));
|
|
EXPECT_EQ(true, mygroup->addFile(std::move(mygroupobjfile_2)));
|
|
|
|
// Add the group to the InputGraph.
|
|
EXPECT_EQ(true, getInputGraph().addInputElement(std::move(mygroup)));
|
|
|
|
EXPECT_EQ(2, inputFileCount());
|
|
|
|
ErrorOr<InputElement *> nextElement = getInputGraph().getNextInputElement();
|
|
EXPECT_NE(InputGraphError::no_more_elements, nextElement.getError());
|
|
EXPECT_EQ(InputElement::Kind::File, (*nextElement)->kind());
|
|
FileNode *fileNode = dyn_cast<FileNode>(*nextElement);
|
|
|
|
EXPECT_EQ("multi_files1", fileNode->getUserPath());
|
|
|
|
ErrorOr<File &> objfile = fileNode->getNextFile();
|
|
EXPECT_NE(InputGraphError::no_more_files, objfile.getError());
|
|
EXPECT_EQ("objfile1", (*objfile).path());
|
|
|
|
objfile = fileNode->getNextFile();
|
|
EXPECT_NE(InputGraphError::no_more_files, objfile.getError());
|
|
EXPECT_EQ("objfile2", (*objfile).path());
|
|
|
|
objfile = fileNode->getNextFile();
|
|
EXPECT_EQ(InputGraphError::no_more_files, objfile.getError());
|
|
|
|
nextElement = getInputGraph().getNextInputElement();
|
|
Group *group = dyn_cast<Group>(*nextElement);
|
|
assert(group);
|
|
|
|
objfile = group->getNextFile();
|
|
EXPECT_NE(InputGraphError::no_more_files, objfile.getError());
|
|
EXPECT_EQ("objfile_1", (*objfile).path());
|
|
|
|
objfile = group->getNextFile();
|
|
EXPECT_NE(InputGraphError::no_more_files, objfile.getError());
|
|
EXPECT_EQ("objfile_2", (*objfile).path());
|
|
|
|
objfile = group->getNextFile();
|
|
EXPECT_NE(InputGraphError::no_more_files, objfile.getError());
|
|
EXPECT_EQ("group_objfile1", (*objfile).path());
|
|
|
|
objfile = group->getNextFile();
|
|
EXPECT_NE(InputGraphError::no_more_files, objfile.getError());
|
|
EXPECT_EQ("group_objfile2", (*objfile).path());
|
|
|
|
nextElement = getInputGraph().getNextInputElement();
|
|
EXPECT_EQ(InputGraphError::no_more_elements, nextElement.getError());
|
|
}
|
|
|
|
// Iterate through the group
|
|
TEST_F(InputGraphTest, AddNodeWithGroupIteration) {
|
|
std::unique_ptr<MyFileNode> myfile(new MyFileNode("multi_files1"));
|
|
std::vector<std::unique_ptr<File>> objfiles;
|
|
std::unique_ptr<SimpleFile> obj1(new SimpleFile("objfile1"));
|
|
std::unique_ptr<SimpleFile> obj2(new SimpleFile("objfile2"));
|
|
objfiles.push_back(std::move(obj1));
|
|
objfiles.push_back(std::move(obj2));
|
|
myfile->addFiles(std::move(objfiles));
|
|
EXPECT_EQ(true, getInputGraph().addInputElement(std::move(myfile)));
|
|
|
|
// Create a group node with two elements
|
|
// an file node which looks like an archive and
|
|
// two file nodes
|
|
std::unique_ptr<Group> mygroup(new Group());
|
|
std::unique_ptr<MyFileNode> myarchive(new MyFileNode("archive_file"));
|
|
std::vector<std::unique_ptr<File>> objfiles_group;
|
|
std::unique_ptr<SimpleFile> obj_1(new SimpleFile("objfile_1"));
|
|
std::unique_ptr<SimpleFile> obj_2(new SimpleFile("objfile_2"));
|
|
objfiles_group.push_back(std::move(obj_1));
|
|
objfiles_group.push_back(std::move(obj_2));
|
|
myarchive->addFiles(std::move(objfiles_group));
|
|
EXPECT_EQ(true, mygroup->addFile(std::move(myarchive)));
|
|
|
|
std::unique_ptr<MyFileNode> mygroupobjfile_1(
|
|
new MyFileNode("group_objfile1"));
|
|
std::vector<std::unique_ptr<File>> objfiles_group1;
|
|
std::unique_ptr<SimpleFile> mygroupobj1(new SimpleFile("group_objfile1"));
|
|
objfiles_group1.push_back(std::move(mygroupobj1));
|
|
mygroupobjfile_1->addFiles(std::move(objfiles_group1));
|
|
EXPECT_EQ(true, mygroup->addFile(std::move(mygroupobjfile_1)));
|
|
|
|
std::unique_ptr<MyFileNode> mygroupobjfile_2(
|
|
new MyFileNode("group_objfile2"));
|
|
std::vector<std::unique_ptr<File>> objfiles_group2;
|
|
std::unique_ptr<SimpleFile> mygroupobj2(new SimpleFile("group_objfile2"));
|
|
objfiles_group2.push_back(std::move(mygroupobj2));
|
|
mygroupobjfile_2->addFiles(std::move(objfiles_group2));
|
|
EXPECT_EQ(true, mygroup->addFile(std::move(mygroupobjfile_2)));
|
|
|
|
// Add the group to the InputGraph.
|
|
EXPECT_EQ(true, getInputGraph().addInputElement(std::move(mygroup)));
|
|
|
|
EXPECT_EQ(2, inputFileCount());
|
|
|
|
ErrorOr<InputElement *> nextElement = getInputGraph().getNextInputElement();
|
|
EXPECT_NE(InputGraphError::no_more_elements, nextElement.getError());
|
|
EXPECT_EQ(InputElement::Kind::File, (*nextElement)->kind());
|
|
FileNode *fileNode = dyn_cast<FileNode>(*nextElement);
|
|
|
|
EXPECT_EQ("multi_files1", fileNode->getUserPath());
|
|
|
|
ErrorOr<File &> objfile = fileNode->getNextFile();
|
|
EXPECT_NE(InputGraphError::no_more_files, objfile.getError());
|
|
EXPECT_EQ("objfile1", (*objfile).path());
|
|
|
|
objfile = fileNode->getNextFile();
|
|
EXPECT_NE(InputGraphError::no_more_files, objfile.getError());
|
|
EXPECT_EQ("objfile2", (*objfile).path());
|
|
|
|
objfile = fileNode->getNextFile();
|
|
EXPECT_EQ(InputGraphError::no_more_files, objfile.getError());
|
|
|
|
nextElement = getInputGraph().getNextInputElement();
|
|
Group *group = dyn_cast<Group>(*nextElement);
|
|
assert(group);
|
|
|
|
objfile = group->getNextFile();
|
|
EXPECT_NE(InputGraphError::no_more_files, objfile.getError());
|
|
EXPECT_EQ("objfile_1", (*objfile).path());
|
|
|
|
objfile = group->getNextFile();
|
|
EXPECT_NE(InputGraphError::no_more_files, objfile.getError());
|
|
EXPECT_EQ("objfile_2", (*objfile).path());
|
|
|
|
objfile = group->getNextFile();
|
|
EXPECT_NE(InputGraphError::no_more_files, objfile.getError());
|
|
EXPECT_EQ("group_objfile1", (*objfile).path());
|
|
|
|
objfile = group->getNextFile();
|
|
EXPECT_NE(InputGraphError::no_more_files, objfile.getError());
|
|
EXPECT_EQ("group_objfile2", (*objfile).path());
|
|
|
|
group->notifyProgress();
|
|
|
|
objfile = group->getNextFile();
|
|
EXPECT_NE(InputGraphError::no_more_files, objfile.getError());
|
|
EXPECT_EQ("objfile_1", (*objfile).path());
|
|
|
|
objfile = group->getNextFile();
|
|
EXPECT_NE(InputGraphError::no_more_files, objfile.getError());
|
|
EXPECT_EQ("objfile_2", (*objfile).path());
|
|
|
|
objfile = group->getNextFile();
|
|
EXPECT_NE(InputGraphError::no_more_files, objfile.getError());
|
|
EXPECT_EQ("group_objfile1", (*objfile).path());
|
|
|
|
objfile = group->getNextFile();
|
|
EXPECT_NE(InputGraphError::no_more_files, objfile.getError());
|
|
EXPECT_EQ("group_objfile2", (*objfile).path());
|
|
}
|
|
|
|
// Node expansion tests.
|
|
TEST_F(InputGraphTest, ExpandAndReplaceInputGraphNode) {
|
|
std::unique_ptr<MyFileNode> myfile(new MyFileNode("multi_files1"));
|
|
std::vector<std::unique_ptr<File>> objfiles;
|
|
std::unique_ptr<SimpleFile> obj1(new SimpleFile("objfile1"));
|
|
std::unique_ptr<SimpleFile> obj2(new SimpleFile("objfile2"));
|
|
objfiles.push_back(std::move(obj1));
|
|
objfiles.push_back(std::move(obj2));
|
|
myfile->addFiles(std::move(objfiles));
|
|
EXPECT_EQ(true, getInputGraph().addInputElement(std::move(myfile)));
|
|
objfiles.clear();
|
|
|
|
std::unique_ptr<MyExpandFileNode> expandFile(
|
|
new MyExpandFileNode("expand_node"));
|
|
|
|
std::unique_ptr<MyFileNode> filenode1(new MyFileNode("expand_file1"));
|
|
std::unique_ptr<SimpleFile> obj3(new SimpleFile("objfile3"));
|
|
objfiles.push_back(std::move(obj3));
|
|
filenode1->addFiles(std::move(objfiles));
|
|
expandFile->addElement(std::move(filenode1));
|
|
objfiles.clear();
|
|
|
|
std::unique_ptr<MyFileNode> filenode2(new MyFileNode("expand_file2"));
|
|
std::unique_ptr<SimpleFile> obj4(new SimpleFile("objfile4"));
|
|
objfiles.push_back(std::move(obj4));
|
|
filenode2->addFiles(std::move(objfiles));
|
|
expandFile->addElement(std::move(filenode2));
|
|
objfiles.clear();
|
|
|
|
// Add expand file to InputGraph
|
|
EXPECT_EQ(true, getInputGraph().addInputElement(std::move(expandFile)));
|
|
|
|
std::unique_ptr<MyFileNode> filenode3(new MyFileNode("obj_after_expand"));
|
|
std::unique_ptr<SimpleFile> obj5(new SimpleFile("objfile5"));
|
|
std::unique_ptr<SimpleFile> obj6(new SimpleFile("objfile6"));
|
|
objfiles.push_back(std::move(obj5));
|
|
objfiles.push_back(std::move(obj6));
|
|
filenode3->addFiles(std::move(objfiles));
|
|
|
|
// Add an extra obj after the expand node
|
|
EXPECT_EQ(true, getInputGraph().addInputElement(std::move(filenode3)));
|
|
|
|
getInputGraph().normalize();
|
|
|
|
ErrorOr<InputElement *> nextElement = getInputGraph().getNextInputElement();
|
|
EXPECT_NE(InputGraphError::no_more_elements, nextElement.getError());
|
|
EXPECT_EQ(InputElement::Kind::File, (*nextElement)->kind());
|
|
FileNode *fileNode = dyn_cast<FileNode>(*nextElement);
|
|
EXPECT_EQ("multi_files1", (*fileNode).getUserPath());
|
|
|
|
nextElement = getInputGraph().getNextInputElement();
|
|
EXPECT_NE(InputGraphError::no_more_elements, nextElement.getError());
|
|
EXPECT_EQ(InputElement::Kind::File, (*nextElement)->kind());
|
|
fileNode = dyn_cast<FileNode>(*nextElement);
|
|
EXPECT_EQ("expand_file1", (*fileNode).getUserPath());
|
|
|
|
nextElement = getInputGraph().getNextInputElement();
|
|
EXPECT_NE(InputGraphError::no_more_elements, nextElement.getError());
|
|
EXPECT_EQ(InputElement::Kind::File, (*nextElement)->kind());
|
|
fileNode = dyn_cast<FileNode>(*nextElement);
|
|
EXPECT_EQ("expand_file2", (*fileNode).getUserPath());
|
|
|
|
nextElement = getInputGraph().getNextInputElement();
|
|
EXPECT_NE(InputGraphError::no_more_elements, nextElement.getError());
|
|
EXPECT_EQ(InputElement::Kind::File, (*nextElement)->kind());
|
|
fileNode = dyn_cast<FileNode>(*nextElement);
|
|
EXPECT_EQ("obj_after_expand", (*fileNode).getUserPath());
|
|
|
|
nextElement = getInputGraph().getNextInputElement();
|
|
EXPECT_EQ(InputGraphError::no_more_elements, nextElement.getError());
|
|
}
|