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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
|
#include "str.h"
#include <library/cpp/testing/unittest/registar.h>
#include <util/generic/typetraits.h>
template <typename T>
const T ReturnConstTemp();
Y_UNIT_TEST_SUITE(TStringInputOutputTest) {
Y_UNIT_TEST(Lvalue) {
TString str = "Hello, World!";
TStringInput input(str);
TString result = input.ReadAll();
UNIT_ASSERT_VALUES_EQUAL(result, str);
}
Y_UNIT_TEST(ConstRef) {
TString str = "Hello, World!";
const TString& r = str;
TStringInput input(r);
TString result = input.ReadAll();
UNIT_ASSERT_VALUES_EQUAL(result, str);
}
Y_UNIT_TEST(NonConstRef) {
TString str = "Hello, World!";
TString& r = str;
TStringInput input(r);
TString result = input.ReadAll();
UNIT_ASSERT_VALUES_EQUAL(result, str);
}
Y_UNIT_TEST(Transfer) {
TString inputString = "some_string";
TStringInput input(inputString);
TString outputString;
TStringOutput output(outputString);
TransferData(&input, &output);
UNIT_ASSERT_VALUES_EQUAL(inputString, outputString);
}
Y_UNIT_TEST(SkipReadAll) {
TString string0 = "All animals are equal, but some animals are more equal than others.";
TString string1;
for (size_t i = 1; i <= string0.size(); i++) {
string1 += string0.substr(0, i);
}
TStringInput input0(string1);
size_t left = 5;
while (left > 0) {
left -= input0.Skip(left);
}
TString string2 = input0.ReadAll();
UNIT_ASSERT_VALUES_EQUAL(string2, string1.substr(5));
}
Y_UNIT_TEST(OperatorBool) {
TStringStream str;
UNIT_ASSERT(!str);
str << "data";
UNIT_ASSERT(str);
str.Clear();
UNIT_ASSERT(!str);
}
Y_UNIT_TEST(TestReadTo) {
TString s("0123456789abc");
TString t;
TStringInput in0(s);
UNIT_ASSERT_VALUES_EQUAL(in0.ReadTo(t, '7'), 8);
UNIT_ASSERT_VALUES_EQUAL(t, "0123456");
UNIT_ASSERT_VALUES_EQUAL(in0.ReadTo(t, 'z'), 5);
UNIT_ASSERT_VALUES_EQUAL(t, "89abc");
}
Y_UNIT_TEST(WriteViaNextAndUndo) {
TString str1;
TStringOutput output(str1);
TString str2;
for (size_t i = 0; i < 10000; ++i) {
str2.push_back('a' + (i % 20));
}
size_t written = 0;
void* ptr = nullptr;
while (written < str2.size()) {
size_t bufferSize = output.Next(&ptr);
UNIT_ASSERT(ptr && bufferSize > 0);
size_t toWrite = Min(bufferSize, str2.size() - written);
memcpy(ptr, str2.begin() + written, toWrite);
written += toWrite;
if (toWrite < bufferSize) {
output.Undo(bufferSize - toWrite);
}
}
UNIT_ASSERT_STRINGS_EQUAL(str1, str2);
}
Y_UNIT_TEST(Write) {
TString str;
TStringOutput output(str);
output << "1"
<< "22"
<< "333"
<< "4444"
<< "55555";
UNIT_ASSERT_STRINGS_EQUAL(str, "1"
"22"
"333"
"4444"
"55555");
}
Y_UNIT_TEST(WriteChars) {
TString str;
TStringOutput output(str);
output << '1' << '2' << '3' << '4' << '5' << '6' << '7' << '8' << '9' << '0';
UNIT_ASSERT_STRINGS_EQUAL(str, "1234567890");
}
Y_UNIT_TEST(MoveConstructor) {
TString str;
TStringOutput output1(str);
output1 << "foo";
TStringOutput output2 = std::move(output1);
output2 << "bar";
UNIT_ASSERT_STRINGS_EQUAL(str, "foobar");
// Check old stream is in a valid state
output1 << "baz";
}
}
|
