6aa8a836c0
Simply returning will report the test as PASSED when it didn't really do anything. SKIPPED is the correct result for these. Found by the Rotten Green Tests project.
113 lines
3.0 KiB
C++
113 lines
3.0 KiB
C++
//===- unittest/Support/ProcessTest.cpp -----------------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/ADT/Triple.h"
|
|
#include "llvm/Support/Error.h"
|
|
#include "llvm/Support/Host.h"
|
|
#include "llvm/Support/Process.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
#ifdef _WIN32
|
|
#include <windows.h>
|
|
#endif
|
|
|
|
namespace {
|
|
|
|
using namespace llvm;
|
|
using namespace sys;
|
|
|
|
TEST(ProcessTest, GetProcessIdTest) {
|
|
const Process::Pid pid = Process::getProcessId();
|
|
|
|
#ifdef _WIN32
|
|
EXPECT_EQ((DWORD)pid, ::GetCurrentProcessId());
|
|
#else
|
|
EXPECT_EQ(pid, ::getpid());
|
|
#endif
|
|
}
|
|
|
|
TEST(ProcessTest, GetRandomNumberTest) {
|
|
const unsigned r1 = Process::GetRandomNumber();
|
|
const unsigned r2 = Process::GetRandomNumber();
|
|
// It should be extremely unlikely that both r1 and r2 are 0.
|
|
EXPECT_NE((r1 | r2), 0u);
|
|
}
|
|
|
|
#ifdef _MSC_VER
|
|
#define setenv(name, var, ignore) _putenv_s(name, var)
|
|
#endif
|
|
|
|
#if HAVE_SETENV || _MSC_VER
|
|
TEST(ProcessTest, Basic) {
|
|
setenv("__LLVM_TEST_ENVIRON_VAR__", "abc", true);
|
|
Optional<std::string> val(Process::GetEnv("__LLVM_TEST_ENVIRON_VAR__"));
|
|
EXPECT_TRUE(val.hasValue());
|
|
EXPECT_STREQ("abc", val->c_str());
|
|
}
|
|
|
|
TEST(ProcessTest, None) {
|
|
Optional<std::string> val(
|
|
Process::GetEnv("__LLVM_TEST_ENVIRON_NO_SUCH_VAR__"));
|
|
EXPECT_FALSE(val.hasValue());
|
|
}
|
|
#endif
|
|
|
|
#ifdef _WIN32
|
|
|
|
TEST(ProcessTest, EmptyVal) {
|
|
SetEnvironmentVariableA("__LLVM_TEST_ENVIRON_VAR__", "");
|
|
Optional<std::string> val(Process::GetEnv("__LLVM_TEST_ENVIRON_VAR__"));
|
|
EXPECT_TRUE(val.hasValue());
|
|
EXPECT_STREQ("", val->c_str());
|
|
}
|
|
|
|
TEST(ProcessTest, Wchar) {
|
|
SetEnvironmentVariableW(L"__LLVM_TEST_ENVIRON_VAR__", L"abcdefghijklmnopqrs");
|
|
Optional<std::string> val(Process::GetEnv("__LLVM_TEST_ENVIRON_VAR__"));
|
|
EXPECT_TRUE(val.hasValue());
|
|
EXPECT_STREQ("abcdefghijklmnopqrs", val->c_str());
|
|
}
|
|
#endif
|
|
|
|
class PageSizeTest : public testing::Test {
|
|
Triple Host;
|
|
|
|
protected:
|
|
PageSizeTest() : Host(Triple::normalize(sys::getProcessTriple())) {}
|
|
|
|
bool isSupported() const {
|
|
// For now just on X86-64 and Aarch64. This can be expanded in the future.
|
|
return (Host.getArch() == Triple::x86_64 ||
|
|
Host.getArch() == Triple::aarch64) &&
|
|
Host.getOS() == Triple::Linux;
|
|
}
|
|
|
|
bool pageSizeAsExpected(unsigned PageSize) const {
|
|
switch (Host.getArch()) {
|
|
case Triple::x86_64:
|
|
return PageSize == 4096;
|
|
case Triple::aarch64:
|
|
// supported granule sizes are 4k, 16k and 64k
|
|
return PageSize == 4096 || PageSize == 16384 || PageSize == 65536;
|
|
default:
|
|
llvm_unreachable("unexpected arch!");
|
|
}
|
|
}
|
|
};
|
|
|
|
TEST_F(PageSizeTest, PageSize) {
|
|
if (!isSupported())
|
|
GTEST_SKIP();
|
|
|
|
llvm::Expected<unsigned> Result = llvm::sys::Process::getPageSize();
|
|
ASSERT_FALSE(!Result);
|
|
ASSERT_TRUE(pageSizeAsExpected(*Result));
|
|
}
|
|
|
|
} // end anonymous namespace
|