1 | //===- IndentedOstreamTest.cpp - Indented raw ostream Tests ---------------===// |
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 "mlir/Support/IndentedOstream.h" |
10 | #include "gmock/gmock.h" |
11 | |
12 | using namespace mlir; |
13 | using ::testing::StrEq; |
14 | |
15 | TEST(FormatTest, SingleLine) { |
16 | std::string str; |
17 | llvm::raw_string_ostream os(str); |
18 | raw_indented_ostream ros(os); |
19 | ros << 10; |
20 | EXPECT_THAT(str, StrEq("10" )); |
21 | } |
22 | |
23 | TEST(FormatTest, SimpleMultiLine) { |
24 | std::string str; |
25 | llvm::raw_string_ostream os(str); |
26 | raw_indented_ostream ros(os); |
27 | ros << "a" ; |
28 | ros << "b" ; |
29 | ros << "\n" ; |
30 | ros << "c" ; |
31 | ros << "\n" ; |
32 | EXPECT_THAT(str, StrEq("ab\nc\n" )); |
33 | } |
34 | |
35 | TEST(FormatTest, SimpleMultiLineIndent) { |
36 | std::string str; |
37 | llvm::raw_string_ostream os(str); |
38 | raw_indented_ostream ros(os); |
39 | ros.indent(with: 2) << "a" ; |
40 | ros.indent(with: 4) << "b" ; |
41 | ros << "\n" ; |
42 | ros << "c" ; |
43 | ros << "\n" ; |
44 | EXPECT_THAT(str, StrEq(" a b\n c\n" )); |
45 | } |
46 | |
47 | TEST(FormatTest, SingleRegion) { |
48 | std::string str; |
49 | llvm::raw_string_ostream os(str); |
50 | raw_indented_ostream ros(os); |
51 | ros << "before\n" ; |
52 | { |
53 | raw_indented_ostream::DelimitedScope scope(ros); |
54 | ros << "inside " << 10; |
55 | ros << "\n two\n" ; |
56 | { |
57 | raw_indented_ostream::DelimitedScope scope(ros, "{\n" , "\n}\n" ); |
58 | ros << "inner inner" ; |
59 | } |
60 | } |
61 | ros << "after" ; |
62 | const auto *expected = |
63 | R"(before |
64 | inside 10 |
65 | two |
66 | { |
67 | inner inner |
68 | } |
69 | after)" ; |
70 | EXPECT_THAT(str, StrEq(expected)); |
71 | |
72 | // Repeat the above with inline form. |
73 | str.clear(); |
74 | ros << "before\n" ; |
75 | ros.scope().os << "inside " << 10 << "\n two\n" ; |
76 | ros.scope().os.scope(open: "{\n" , close: "\n}\n" ).os << "inner inner" ; |
77 | ros << "after" ; |
78 | EXPECT_THAT(os.str(), StrEq(expected)); |
79 | } |
80 | |
81 | TEST(FormatTest, Reindent) { |
82 | std::string str; |
83 | llvm::raw_string_ostream os(str); |
84 | raw_indented_ostream ros(os); |
85 | |
86 | // String to print with some additional empty lines at the start and lines |
87 | // with just spaces. |
88 | const auto *desc = R"( |
89 | |
90 | |
91 | First line |
92 | second line |
93 | |
94 | |
95 | )" ; |
96 | ros.printReindented(str: desc); |
97 | const auto *expected = |
98 | R"(First line |
99 | second line |
100 | |
101 | |
102 | )" ; |
103 | EXPECT_THAT(str, StrEq(expected)); |
104 | } |
105 | |
106 | TEST(FormatTest, ReindentLineEndings) { |
107 | std::string str; |
108 | llvm::raw_string_ostream os(str); |
109 | raw_indented_ostream ros(os); |
110 | |
111 | // Similar string as the previous test, but with \r\n (Windows style) line |
112 | // breaks. Note that C++'s internal string representation uses \n, so just |
113 | // running the previous test as-is on Windows is not sufficient. |
114 | const auto *desc = |
115 | "\r\n\r\n\r\n First line\r\n second line" ; |
116 | ros.printReindented(str: desc); |
117 | const auto *expected = "First line\r\n second line" ; |
118 | EXPECT_THAT(str, StrEq(expected)); |
119 | } |
120 | |