aboutsummaryrefslogtreecommitdiff
path: root/llvm/unittests/Support/Format.cpp
blob: c4e421fe9dc4c3f8d39ec97a9f8d9bdcadde3dac (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
//===----------------------------------------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

#include "llvm/Support/Format.h"
#include "gtest/gtest.h"

using namespace llvm;

namespace {

template <typename FormatTy>
std::string printToString(unsigned MaxN, FormatTy &&Fmt) {
  std::vector<char> Dst(MaxN + 2);
  int N = Fmt.snprint(Dst.data(), Dst.size());
  Dst.back() = 0;
  return N < 0 ? "" : Dst.data();
}

template <typename Expected, typename Arg>
constexpr bool checkDecayTypeEq(const Arg &arg) {
  return std::is_same_v<detail::decay_if_c_char_array_t<Arg>, Expected>;
}

TEST(Format, DecayIfCCharArray) {
  char Array[] = "Array";
  const char ConstArray[] = "ConstArray";
  char PtrBuf[] = "Ptr";
  char *Ptr = PtrBuf;
  const char *PtrToConst = "PtrToConst";

  EXPECT_EQ("        Literal", printToString(20, format("%15s", "Literal")));
  EXPECT_EQ("          Array", printToString(20, format("%15s", Array)));
  EXPECT_EQ("     ConstArray", printToString(20, format("%15s", ConstArray)));
  EXPECT_EQ("            Ptr", printToString(20, format("%15s", Ptr)));
  EXPECT_EQ("     PtrToConst", printToString(20, format("%15s", PtrToConst)));

  EXPECT_TRUE(checkDecayTypeEq<const char *>("Literal"));
  EXPECT_TRUE(checkDecayTypeEq<const char *>(Array));
  EXPECT_TRUE(checkDecayTypeEq<const char *>(ConstArray));
  EXPECT_TRUE(checkDecayTypeEq<char *>(Ptr));
  EXPECT_TRUE(checkDecayTypeEq<const char *>(PtrToConst));
  EXPECT_TRUE(checkDecayTypeEq<char>(PtrToConst[0]));
  EXPECT_TRUE(
      checkDecayTypeEq<const char *>(static_cast<const char *>("Literal")));

  wchar_t WCharArray[] = L"WCharArray";
  EXPECT_TRUE(checkDecayTypeEq<wchar_t[11]>(WCharArray));
  EXPECT_TRUE(checkDecayTypeEq<wchar_t>(WCharArray[0]));
}

} // namespace