mirror of
https://github.com/Gericom/teak-llvm.git
synced 2025-06-22 13:05:52 -04:00
a0a738b184
We can't put the unittest source dir map in the configuration specific directory because VS doesn't have a configure-specific directory, instead it only knows this at runtime. So we have to remove this from the path. This in turn means that the path will be slightly different in VS configurations vs non vs configurations. In the former, the source map will be in the parent directory of the executable, and in the latter it will be in the same directory as the executable. So check both. llvm-svn: 341590
56 lines
1.7 KiB
C++
56 lines
1.7 KiB
C++
|
|
#include "llvm/Testing/Support/SupportHelpers.h"
|
|
|
|
#include "llvm/ADT/SmallString.h"
|
|
#include "llvm/ADT/Twine.h"
|
|
#include "llvm/Support/Error.h"
|
|
#include "llvm/Support/FileSystem.h"
|
|
#include "llvm/Support/MemoryBuffer.h"
|
|
#include "llvm/Support/Path.h"
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
using namespace llvm;
|
|
using namespace llvm::unittest;
|
|
|
|
static std::pair<bool, SmallString<128>> findSrcDirMap(StringRef Argv0) {
|
|
SmallString<128> BaseDir = llvm::sys::path::parent_path(Argv0);
|
|
|
|
llvm::sys::fs::make_absolute(BaseDir);
|
|
|
|
SmallString<128> PathInSameDir = BaseDir;
|
|
llvm::sys::path::append(PathInSameDir, "llvm.srcdir.txt");
|
|
|
|
if (llvm::sys::fs::is_regular_file(PathInSameDir))
|
|
return std::make_pair(true, std::move(PathInSameDir));
|
|
|
|
SmallString<128> PathInParentDir = llvm::sys::path::parent_path(BaseDir);
|
|
|
|
llvm::sys::path::append(PathInParentDir, "llvm.srcdir.txt");
|
|
if (llvm::sys::fs::is_regular_file(PathInParentDir))
|
|
return std::make_pair(true, std::move(PathInParentDir));
|
|
|
|
return std::pair<bool, SmallString<128>>(false, "");
|
|
}
|
|
|
|
static bool isInParentDir(StringRef Argv0) {}
|
|
|
|
SmallString<128> llvm::unittest::getInputFileDirectory(const char *Argv0) {
|
|
bool Found = false;
|
|
SmallString<128> InputFilePath;
|
|
std::tie(Found, InputFilePath) = findSrcDirMap(Argv0);
|
|
|
|
EXPECT_TRUE(Found) << "Unit test source directory file does not exist.";
|
|
|
|
auto File = MemoryBuffer::getFile(InputFilePath);
|
|
|
|
EXPECT_TRUE(static_cast<bool>(File))
|
|
<< "Could not open unit test source directory file.";
|
|
|
|
InputFilePath.clear();
|
|
InputFilePath.append((*File)->getBuffer().trim());
|
|
llvm::sys::path::append(InputFilePath, "Inputs");
|
|
llvm::sys::path::native(InputFilePath);
|
|
return InputFilePath;
|
|
}
|