1 // Copyright 2017 Google Inc.
3 // Licensed under the Apache License, Version 2.0 (the "License");
4 // you may not use this file except in compliance with the License.
5 // You may obtain a copy of the License at
7 // http://www.apache.org/licenses/LICENSE-2.0
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS,
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 // See the License for the specific language governing permissions and
13 // limitations under the License.
15 #include "internal/stack_line_reader.h"
16 #include "filesystem_for_testing.h"
18 #include "gtest/gtest.h"
20 namespace cpu_features {
22 bool operator==(const StringView& a, const StringView& b) {
23 return CpuFeatures_StringView_IsEquals(a, b);
28 std::string ToString(StringView view) { return {view.ptr, view.size}; }
30 TEST(StackLineReaderTest, Empty) {
31 auto& fs = GetEmptyFilesystem();
32 auto* file = fs.CreateFile("/proc/cpuinfo", "");
33 StackLineReader reader;
34 StackLineReader_Initialize(&reader, file->GetFileDescriptor());
36 const auto result = StackLineReader_NextLine(&reader);
37 EXPECT_TRUE(result.eof);
38 EXPECT_TRUE(result.full_line);
39 EXPECT_EQ(result.line, str(""));
43 TEST(StackLineReaderTest, ManySmallLines) {
44 auto& fs = GetEmptyFilesystem();
45 auto* file = fs.CreateFile("/proc/cpuinfo", "a\nb\nc");
47 StackLineReader reader;
48 StackLineReader_Initialize(&reader, file->GetFileDescriptor());
50 const auto result = StackLineReader_NextLine(&reader);
51 EXPECT_FALSE(result.eof);
52 EXPECT_TRUE(result.full_line);
53 EXPECT_EQ(result.line, str("a"));
56 const auto result = StackLineReader_NextLine(&reader);
57 EXPECT_FALSE(result.eof);
58 EXPECT_TRUE(result.full_line);
59 EXPECT_EQ(result.line, str("b"));
62 const auto result = StackLineReader_NextLine(&reader);
63 EXPECT_TRUE(result.eof);
64 EXPECT_TRUE(result.full_line);
65 EXPECT_EQ(result.line, str("c"));
69 TEST(StackLineReaderTest, TruncatedLine) {
70 auto& fs = GetEmptyFilesystem();
71 auto* file = fs.CreateFile("/proc/cpuinfo", R"(First
73 More than 16 characters, this will be truncated.
76 StackLineReader reader;
77 StackLineReader_Initialize(&reader, file->GetFileDescriptor());
79 const auto result = StackLineReader_NextLine(&reader);
80 EXPECT_FALSE(result.eof);
81 EXPECT_TRUE(result.full_line);
82 EXPECT_EQ(result.line, str("First"));
85 const auto result = StackLineReader_NextLine(&reader);
86 EXPECT_FALSE(result.eof);
87 EXPECT_TRUE(result.full_line);
88 EXPECT_EQ(result.line, str("Second"));
91 const auto result = StackLineReader_NextLine(&reader);
92 EXPECT_FALSE(result.eof);
93 EXPECT_FALSE(result.full_line);
94 EXPECT_EQ(result.line, str("More than 16 cha"));
97 const auto result = StackLineReader_NextLine(&reader);
98 EXPECT_TRUE(result.eof);
99 EXPECT_TRUE(result.full_line);
100 EXPECT_EQ(result.line, str("last"));
104 TEST(StackLineReaderTest, TruncatedLines) {
105 auto& fs = GetEmptyFilesystem();
106 auto* file = fs.CreateFile("/proc/cpuinfo", R"(More than 16 characters
107 Another line that is too long)");
109 StackLineReader reader;
110 StackLineReader_Initialize(&reader, file->GetFileDescriptor());
112 const auto result = StackLineReader_NextLine(&reader);
113 EXPECT_FALSE(result.eof);
114 EXPECT_FALSE(result.full_line);
115 EXPECT_EQ(result.line, str("More than 16 cha"));
118 const auto result = StackLineReader_NextLine(&reader);
119 EXPECT_FALSE(result.eof);
120 EXPECT_FALSE(result.full_line);
121 EXPECT_EQ(result.line, str("Another line tha"));
124 const auto result = StackLineReader_NextLine(&reader);
125 EXPECT_TRUE(result.eof);
126 EXPECT_TRUE(result.full_line);
127 EXPECT_EQ(result.line, str(""));
132 } // namespace cpu_features