SupportTests::HomeDirectory: Don't try tests when $HOME is undefined.

[oota-llvm.git] / unittests / Support / Path.cpp

diff --git a/unittests/Support/Path.cpp b/unittests/Support/Path.cpp
index e8cd98b8cfa2b15408bf800f9f173ae70e9a68e3..044e80fe363dec1747a6fc0dd3de41b6f7fde9ae 100644 (file)
--- a/unittests/Support/Path.cpp
+++ b/unittests/Support/Path.cpp
@@ -300,19 +300,22 @@ TEST(Support, AbsolutePathIteratorEnd) {
 }
 
 TEST(Support, HomeDirectory) {
+  std::string expected;
 #ifdef LLVM_ON_WIN32
   wchar_t *path = ::_wgetenv(L"USERPROFILE");
   auto pathLen = ::wcslen(path);
   ArrayRef<char> ref{reinterpret_cast<char *>(path), pathLen * sizeof(wchar_t)};
-  std::string expected;
   convertUTF16ToUTF8String(ref, expected);
 #else
-  std::string expected{::getenv("HOME")};
+  if (char const *home = ::getenv("HOME"))
+    expected = home;
 #endif
-  SmallString<128> HomeDir;
-  auto status = path::home_directory(HomeDir);
-  EXPECT_TRUE(status ^ HomeDir.empty());
-  EXPECT_EQ(expected, HomeDir);
+  if (expected.length() > 0) {
+    SmallString<128> HomeDir;
+    auto status = path::home_directory(HomeDir);
+    EXPECT_TRUE(status ^ HomeDir.empty());
+    EXPECT_EQ(expected, HomeDir);
+  }
 }
 
 class FileSystemTest : public testing::Test {