[EditLineTest] Not always TERM is available, e.g. on some bots.

llvm-svn: 358918

1 files changed, 13 insertions, 3 deletions

diff --git a/lldb/unittests/Editline/EditlineTest.cpp b/lldb/unittests/Editline/EditlineTest.cpp
index 1a4bd37e767..2bfd1e7848e 100644
--- a/lldb/unittests/Editline/EditlineTest.cpp
+++ b/lldb/unittests/Editline/EditlineTest.cpp
@@ -244,17 +244,19 @@ private:
   EditlineAdapter _el_adapter;
   std::shared_ptr<std::thread> _sp_output_thread;
 
+protected:
+  bool _has_term = true;
+
 public:
   void SetUp() {
     FileSystem::Initialize();
 
     // We need a TERM set properly for editline to work as expected.
-    setenv("TERM", "vt100", 1);
+    if (setenv("TERM", "vt100", 1) != 0)
+      _has_term = false;
 
     // Validate the editline adapter.
     EXPECT_TRUE(_el_adapter.IsValid());
-    if (!_el_adapter.IsValid())
-      return;
 
     // Dump output.
     _sp_output_thread =
@@ -273,6 +275,10 @@ public:
 };
 
 TEST_F(EditlineTestFixture, EditlineReceivesSingleLineText) {
+  // Skip if we don't have a TERM.
+  if (!_has_term)
+    return;
+
   // Send it some text via our virtual keyboard.
   const std::string input_text("Hello, world");
   EXPECT_TRUE(GetEditlineAdapter().SendLine(input_text));
@@ -289,6 +295,10 @@ TEST_F(EditlineTestFixture, EditlineReceivesSingleLineText) {
 }
 
 TEST_F(EditlineTestFixture, EditlineReceivesMultiLineText) {
+  // Skip if we don't have a TERM.
+  if (!_has_term)
+    return;
+
   // Send it some text via our virtual keyboard.
   std::vector<std::string> input_lines;
   input_lines.push_back("int foo()");