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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/elevation_service/elevator.h"
#include <algorithm>
#include <string>
#include <vector>
#include "testing/gtest/include/gtest/gtest.h"
namespace elevation_service {
class ElevatorTest : public ::testing::Test {};
TEST_F(ElevatorTest, StringHandlingTest) {
{
std::string str;
const std::string kFirstString("hello again");
Elevator::AppendStringWithLength(kFirstString, str);
EXPECT_EQ(sizeof(uint32_t) + kFirstString.length(), str.length());
const std::string kSecondString("another string");
Elevator::AppendStringWithLength(kSecondString, str);
EXPECT_EQ(sizeof(uint32_t) + kFirstString.length() + sizeof(uint32_t) +
kSecondString.length(),
str.length());
const std::string first_str = Elevator::PopFromStringFront(str);
EXPECT_EQ(kFirstString, first_str);
EXPECT_EQ(sizeof(uint32_t) + kSecondString.length(), str.length());
const std::string second_str = Elevator::PopFromStringFront(str);
EXPECT_EQ(kSecondString, second_str);
const std::string error_str = Elevator::PopFromStringFront(str);
EXPECT_TRUE(error_str.empty());
std::string str2;
Elevator::AppendStringWithLength("", str2);
EXPECT_EQ(sizeof(uint32_t), str2.length());
std::string empty_str = Elevator::PopFromStringFront(str2);
EXPECT_TRUE(empty_str.empty());
}
{
// Test data with embedded nulls to make sure it's handled correctly.
const std::vector<uint8_t> test_data1{6, 5, 4, 3, 2, 1, 0, 1, 2,
3, 4, 5, 6, 0, 1, 2, 3, 4};
const std::vector<uint8_t> test_data2{0, 1, 4, 0, 1, 4, 5, 0};
std::string str;
Elevator::AppendStringWithLength(
std::string(test_data1.cbegin(), test_data1.cend()), str);
Elevator::AppendStringWithLength(
std::string(test_data2.cbegin(), test_data2.cend()), str);
EXPECT_EQ(sizeof(uint32_t) + test_data1.size() + sizeof(uint32_t) +
test_data2.size(),
str.length());
const std::string first_str = Elevator::PopFromStringFront(str);
EXPECT_TRUE(
std::equal(test_data1.cbegin(), test_data1.cend(), first_str.cbegin()));
const std::string second_str = Elevator::PopFromStringFront(str);
EXPECT_TRUE(std::equal(test_data2.cbegin(), test_data2.cend(),
second_str.cbegin()));
}
}
} // namespace elevation_service
|
