1 // Copyright 2017 Google LLC
2 //
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
6 //
7 // http://www.apache.org/licenses/LICENSE-2.0
8 //
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.
14
15 #include "internal/stack_line_reader.h"
16
17 #include "filesystem_for_testing.h"
18 #include "gtest/gtest.h"
19
20 namespace cpu_features {
21
operator ==(const StringView & a,const StringView & b)22 bool operator==(const StringView& a, const StringView& b) {
23 return CpuFeatures_StringView_IsEquals(a, b);
24 }
25
26 namespace {
27
ToString(StringView view)28 std::string ToString(StringView view) { return {view.ptr, view.size}; }
29
TEST(StackLineReaderTest,Empty)30 TEST(StackLineReaderTest, Empty) {
31 auto& fs = GetEmptyFilesystem();
32 auto* file = fs.CreateFile("/proc/cpuinfo", "");
33 StackLineReader reader;
34 StackLineReader_Initialize(&reader, file->GetFileDescriptor());
35 {
36 const auto result = StackLineReader_NextLine(&reader);
37 EXPECT_TRUE(result.eof);
38 EXPECT_TRUE(result.full_line);
39 EXPECT_EQ(result.line, str(""));
40 }
41 }
42
TEST(StackLineReaderTest,ManySmallLines)43 TEST(StackLineReaderTest, ManySmallLines) {
44 auto& fs = GetEmptyFilesystem();
45 auto* file = fs.CreateFile("/proc/cpuinfo", "a\nb\nc");
46
47 StackLineReader reader;
48 StackLineReader_Initialize(&reader, file->GetFileDescriptor());
49 {
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"));
54 }
55 {
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"));
60 }
61 {
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"));
66 }
67 }
68
TEST(StackLineReaderTest,TruncatedLine)69 TEST(StackLineReaderTest, TruncatedLine) {
70 auto& fs = GetEmptyFilesystem();
71 auto* file = fs.CreateFile("/proc/cpuinfo", R"(First
72 Second
73 More than 16 characters, this will be truncated.
74 last)");
75
76 StackLineReader reader;
77 StackLineReader_Initialize(&reader, file->GetFileDescriptor());
78 {
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"));
83 }
84 {
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"));
89 }
90 {
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"));
95 }
96 {
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"));
101 }
102 }
103
TEST(StackLineReaderTest,TruncatedLines)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)");
108
109 StackLineReader reader;
110 StackLineReader_Initialize(&reader, file->GetFileDescriptor());
111 {
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"));
116 }
117 {
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"));
122 }
123 {
124 const auto result = StackLineReader_NextLine(&reader);
125 EXPECT_TRUE(result.eof);
126 EXPECT_TRUE(result.full_line);
127 EXPECT_EQ(result.line, str(""));
128 }
129 }
130
131 } // namespace
132 } // namespace cpu_features
133