10184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler Carruth//===- unittest/Support/ProcessTest.cpp -----------------------------------===// 20184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler Carruth// 30184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler Carruth// The LLVM Compiler Infrastructure 40184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler Carruth// 50184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler Carruth// This file is distributed under the University of Illinois Open Source 60184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler Carruth// License. See LICENSE.TXT for details. 70184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler Carruth// 80184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler Carruth//===----------------------------------------------------------------------===// 90184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler Carruth 100184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler Carruth#include "llvm/Support/Process.h" 110184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler Carruth#include "gtest/gtest.h" 120184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler Carruth 130184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler Carruth#ifdef LLVM_ON_WIN32 1436b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines#include <windows.h> 150184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler Carruth#endif 160184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler Carruth 170184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler Carruthnamespace { 180184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler Carruth 190184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler Carruthusing namespace llvm; 200184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler Carruthusing namespace sys; 210184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler Carruth 220184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler CarruthTEST(ProcessTest, SelfProcess) { 230184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler Carruth EXPECT_TRUE(process::get_self()); 240184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler Carruth EXPECT_EQ(process::get_self(), process::get_self()); 250184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler Carruth 260184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler Carruth#if defined(LLVM_ON_UNIX) 270184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler Carruth EXPECT_EQ(getpid(), process::get_self()->get_id()); 280184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler Carruth#elif defined(LLVM_ON_WIN32) 290c793018079803628b5aeee6bbc1e6464fbefd65Aaron Ballman EXPECT_EQ(GetCurrentProcessId(), process::get_self()->get_id()); 300184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler Carruth#endif 31814afe91ccad0e5e1f767303d780fa0318fa5212Chandler Carruth 32814afe91ccad0e5e1f767303d780fa0318fa5212Chandler Carruth EXPECT_LT(1u, process::get_self()->page_size()); 3373c35d86b9d5236be5b3f49bc8df11008b96636eChandler Carruth 3473c35d86b9d5236be5b3f49bc8df11008b96636eChandler Carruth EXPECT_LT(TimeValue::MinTime, process::get_self()->get_user_time()); 3573c35d86b9d5236be5b3f49bc8df11008b96636eChandler Carruth EXPECT_GT(TimeValue::MaxTime, process::get_self()->get_user_time()); 3673c35d86b9d5236be5b3f49bc8df11008b96636eChandler Carruth EXPECT_LT(TimeValue::MinTime, process::get_self()->get_system_time()); 3773c35d86b9d5236be5b3f49bc8df11008b96636eChandler Carruth EXPECT_GT(TimeValue::MaxTime, process::get_self()->get_system_time()); 3873c35d86b9d5236be5b3f49bc8df11008b96636eChandler Carruth EXPECT_LT(TimeValue::MinTime, process::get_self()->get_wall_time()); 3973c35d86b9d5236be5b3f49bc8df11008b96636eChandler Carruth EXPECT_GT(TimeValue::MaxTime, process::get_self()->get_wall_time()); 400184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler Carruth} 410184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler Carruth 4236b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen HinesTEST(ProcessTest, GetRandomNumberTest) { 4336b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines const unsigned r1 = Process::GetRandomNumber(); 4436b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines const unsigned r2 = Process::GetRandomNumber(); 4536b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines // It should be extremely unlikely that both r1 and r2 are 0. 4636b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines EXPECT_NE((r1 | r2), 0u); 4736b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines} 4836b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines 493014066976dd24d8ca565dd5fabe4d5a6144574fRui Ueyama#ifdef _MSC_VER 50f42d4247ae1138c6deed50f92dcd1a4f34e07decRui Ueyama#define setenv(name, var, ignore) _putenv_s(name, var) 51f42d4247ae1138c6deed50f92dcd1a4f34e07decRui Ueyama#endif 52f42d4247ae1138c6deed50f92dcd1a4f34e07decRui Ueyama 533014066976dd24d8ca565dd5fabe4d5a6144574fRui Ueyama#if HAVE_SETENV || _MSC_VER 54f42d4247ae1138c6deed50f92dcd1a4f34e07decRui UeyamaTEST(ProcessTest, Basic) { 55f42d4247ae1138c6deed50f92dcd1a4f34e07decRui Ueyama setenv("__LLVM_TEST_ENVIRON_VAR__", "abc", true); 56f42d4247ae1138c6deed50f92dcd1a4f34e07decRui Ueyama Optional<std::string> val(Process::GetEnv("__LLVM_TEST_ENVIRON_VAR__")); 57f42d4247ae1138c6deed50f92dcd1a4f34e07decRui Ueyama EXPECT_TRUE(val.hasValue()); 58f42d4247ae1138c6deed50f92dcd1a4f34e07decRui Ueyama EXPECT_STREQ("abc", val->c_str()); 59f42d4247ae1138c6deed50f92dcd1a4f34e07decRui Ueyama} 60f42d4247ae1138c6deed50f92dcd1a4f34e07decRui Ueyama 61f42d4247ae1138c6deed50f92dcd1a4f34e07decRui UeyamaTEST(ProcessTest, None) { 62f42d4247ae1138c6deed50f92dcd1a4f34e07decRui Ueyama Optional<std::string> val( 63f42d4247ae1138c6deed50f92dcd1a4f34e07decRui Ueyama Process::GetEnv("__LLVM_TEST_ENVIRON_NO_SUCH_VAR__")); 64f42d4247ae1138c6deed50f92dcd1a4f34e07decRui Ueyama EXPECT_FALSE(val.hasValue()); 65f42d4247ae1138c6deed50f92dcd1a4f34e07decRui Ueyama} 66f42d4247ae1138c6deed50f92dcd1a4f34e07decRui Ueyama#endif 67f42d4247ae1138c6deed50f92dcd1a4f34e07decRui Ueyama 68f42d4247ae1138c6deed50f92dcd1a4f34e07decRui Ueyama#ifdef LLVM_ON_WIN32 69f42d4247ae1138c6deed50f92dcd1a4f34e07decRui UeyamaTEST(ProcessTest, Wchar) { 70f42d4247ae1138c6deed50f92dcd1a4f34e07decRui Ueyama SetEnvironmentVariableW(L"__LLVM_TEST_ENVIRON_VAR__", L"abcdefghijklmnopqrs"); 71f42d4247ae1138c6deed50f92dcd1a4f34e07decRui Ueyama Optional<std::string> val(Process::GetEnv("__LLVM_TEST_ENVIRON_VAR__")); 72f42d4247ae1138c6deed50f92dcd1a4f34e07decRui Ueyama EXPECT_TRUE(val.hasValue()); 73f42d4247ae1138c6deed50f92dcd1a4f34e07decRui Ueyama EXPECT_STREQ("abcdefghijklmnopqrs", val->c_str()); 74f42d4247ae1138c6deed50f92dcd1a4f34e07decRui Ueyama} 75f42d4247ae1138c6deed50f92dcd1a4f34e07decRui Ueyama#endif 76f42d4247ae1138c6deed50f92dcd1a4f34e07decRui Ueyama 770184a841d3914bb78c7c6fa87ce9da86a2d5992aChandler Carruth} // end anonymous namespace 78