commit ad596bf4fc406c0940d0f76f6b7c4ff22ecee7ac
author Treehugger Robot <treehugger-gerrit@google.com> Thu Jul 19 00:42:54 2018 +0000
committer Gerrit Code Review <noreply-gerritcodereview@google.com> Thu Jul 19 00:42:54 2018 +0000
tree f3cf2b2b64e9f5fbcec67d7f9d6cca12a8c1dcf7
parent 4502c4060a8beef5c8dfdddb74a577712a470368
parent bf74815b5281406a2cb8a87f652ac81c9e4a0899

Merge "setjmp is a macro."

tests/gtest_main.cpp

diff --git a/tests/gtest_main.cpp b/tests/gtest_main.cpp
index 1ee8b53..8aa94bb 100644
--- a/tests/gtest_main.cpp
+++ b/tests/gtest_main.cpp
@@ -78,15 +78,39 @@
 static constexpr const char* COLOR_GREEN  = "\033[0;32m";
 static constexpr const char* COLOR_YELLOW = "\033[0;33m";
 
-static void ColoredPrintf(const char* const color, const char* fmt, ...) {
-    va_list args;
-    va_start(args, fmt);
+static bool ShouldUseColor() {
+  const auto& gtest_color = ::testing::GTEST_FLAG(color);
+  if (gtest_color == "yes" || gtest_color == "true" || gtest_color == "t") {
+    return true;
+  }
+  if (gtest_color != "auto") {
+    return false;
+  }
 
+  bool stdout_is_tty = isatty(STDOUT_FILENO) != 0;
+  if (!stdout_is_tty) {
+    return false;
+  }
+
+  const char* const term = getenv("COLORTERM");
+  return term != nullptr && term[0] != 0;
+}
+
+static void ColoredPrintf(const char* const color, const char* fmt, ...) {
+  static const bool use_color = ShouldUseColor();
+
+  va_list args;
+  va_start(args, fmt);
+
+  if (!use_color) {
+    vprintf(fmt, args);
+  } else {
     printf("%s", color);
     vprintf(fmt, args);
     printf("%s", COLOR_RESET);
+  }
 
-    va_end(args);
+  va_end(args);
 }
 
 constexpr int DEFAULT_GLOBAL_TEST_RUN_DEADLINE_MS = 90000;