llvm/unittests/Support/WithColorTest.cpp

   1 //===- WithColorTest.cpp --------------------------------------------------===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #include "llvm/Support/WithColor.h"
  10 #include "gtest/gtest.h"
  11
  12 using namespace llvm;
  13
  14 TEST(WithColorTest, ColorMode) {
  15   {
  16     std::string S;
  17     llvm::raw_string_ostream OS(S);
  18     OS.enable_colors(true);
  19
  20     WithColor(OS, HighlightColor::Error, ColorMode::Disable) << "test";
  21     EXPECT_EQ("test", OS.str());
  22   }
  23
  24   {
  25     std::string S;
  26     llvm::raw_string_ostream OS(S);
  27     OS.enable_colors(true);
  28
  29     WithColor(OS, HighlightColor::Error, ColorMode::Auto) << "test";
  30     EXPECT_EQ("test", OS.str());
  31   }
  32
  33 #ifdef LLVM_ON_UNIX
  34   {
  35     std::string S;
  36     llvm::raw_string_ostream OS(S);
  37     OS.enable_colors(true);
  38
  39     WithColor(OS, HighlightColor::Error, ColorMode::Enable) << "test";
  40     EXPECT_EQ("\x1B[0;1;31mtest\x1B[0m", OS.str());
  41   }
  42 #endif
  43 }