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
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
|
// Copyright 2012 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "google_apis/common/test_util.h"
#include <string_view>
#include "base/files/file_util.h"
#include "base/json/json_file_value_serializer.h"
#include "base/json/json_reader.h"
#include "base/path_service.h"
#include "base/rand_util.h"
#include "base/run_loop.h"
#include "base/strings/string_number_conversions.h"
#include "base/strings/string_split.h"
#include "base/strings/string_util.h"
#include "base/strings/stringprintf.h"
#include "net/test/embedded_test_server/http_request.h"
#include "net/test/embedded_test_server/http_response.h"
#include "url/gurl.h"
namespace google_apis {
namespace test_util {
bool RemovePrefix(const std::string& input,
const std::string& prefix,
std::string* output) {
if (!base::StartsWith(input, prefix, base::CompareCase::SENSITIVE))
return false;
*output = input.substr(prefix.size());
return true;
}
base::FilePath GetTestFilePath(const std::string& relative_path) {
base::FilePath path;
if (!base::PathService::Get(base::DIR_SRC_TEST_DATA_ROOT, &path)) {
return base::FilePath();
}
path = path.AppendASCII("google_apis")
.AppendASCII("test")
.AppendASCII("data")
.Append(base::FilePath::FromUTF8Unsafe(relative_path));
return path;
}
GURL GetBaseUrlForTesting(int port) {
return GURL(base::StringPrintf("http://127.0.0.1:%d/", port));
}
void RunAndQuit(base::RunLoop* run_loop, base::OnceClosure closure) {
std::move(closure).Run();
run_loop->Quit();
}
bool WriteStringToFile(const base::FilePath& file_path,
std::string_view content) {
return base::WriteFile(file_path, content);
}
bool CreateFileOfSpecifiedSize(const base::FilePath& temp_dir,
size_t size,
base::FilePath* path,
std::string* data) {
if (!base::CreateTemporaryFileInDir(temp_dir, path))
return false;
if (size == 0) {
// Note: RandBytesAsString doesn't support generating an empty string.
data->clear();
return true;
}
*data = base::RandBytesAsString(size);
return WriteStringToFile(*path, *data);
}
std::unique_ptr<base::Value> LoadJSONFile(const std::string& relative_path) {
base::FilePath path = GetTestFilePath(relative_path);
std::string error;
JSONFileValueDeserializer deserializer(path);
std::unique_ptr<base::Value> value =
deserializer.Deserialize(nullptr, &error);
LOG_IF(WARNING, !value.get())
<< "Failed to parse " << path.value() << ": " << error;
return value;
}
// Returns a HttpResponse created from the given file path.
std::unique_ptr<net::test_server::BasicHttpResponse> CreateHttpResponseFromFile(
const base::FilePath& file_path) {
std::string content;
if (!base::ReadFileToString(file_path, &content))
return nullptr;
std::string content_type = "text/plain";
if (base::EndsWith(file_path.AsUTF8Unsafe(), ".json",
base::CompareCase::SENSITIVE))
content_type = "application/json";
std::unique_ptr<net::test_server::BasicHttpResponse> http_response(
new net::test_server::BasicHttpResponse);
http_response->set_code(net::HTTP_OK);
http_response->set_content(content);
http_response->set_content_type(content_type);
return http_response;
}
std::unique_ptr<net::test_server::HttpResponse> HandleDownloadFileRequest(
const GURL& base_url,
net::test_server::HttpRequest* out_request,
const net::test_server::HttpRequest& request) {
*out_request = request;
GURL absolute_url = base_url.Resolve(request.relative_url);
std::string remaining_path;
if (!RemovePrefix(absolute_url.path(), "/files/", &remaining_path))
return nullptr;
return CreateHttpResponseFromFile(GetTestFilePath(remaining_path));
}
bool ParseContentRangeHeader(const std::string& value,
int64_t* start_position,
int64_t* end_position,
int64_t* length) {
DCHECK(start_position);
DCHECK(end_position);
DCHECK(length);
std::string remaining;
if (!RemovePrefix(value, "bytes ", &remaining))
return false;
std::vector<std::string_view> parts = base::SplitStringPiece(
remaining, "/", base::TRIM_WHITESPACE, base::SPLIT_WANT_ALL);
if (parts.size() != 2U)
return false;
if (!base::StringToInt64(parts[1], length))
return false;
parts = base::SplitStringPiece(parts[0], "-", base::TRIM_WHITESPACE,
base::SPLIT_WANT_ALL);
if (parts.size() != 2U)
return false;
return (base::StringToInt64(parts[0], start_position) &&
base::StringToInt64(parts[1], end_position));
}
void AppendProgressCallbackResult(std::vector<ProgressInfo>* progress_values,
int64_t progress,
int64_t total) {
progress_values->push_back(ProgressInfo(progress, total));
}
TestGetContentCallback::TestGetContentCallback()
: callback_(base::BindRepeating(&TestGetContentCallback::OnGetContent,
base::Unretained(this))) {}
TestGetContentCallback::~TestGetContentCallback() = default;
std::string TestGetContentCallback::GetConcatenatedData() const {
std::string result;
for (size_t i = 0; i < data_.size(); ++i) {
result += *data_[i];
}
return result;
}
void TestGetContentCallback::OnGetContent(google_apis::ApiErrorCode error,
std::unique_ptr<std::string> data,
bool first_chunk) {
data_.push_back(std::move(data));
}
} // namespace test_util
} // namespace google_apis
|
