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
|
// Copyright 2015 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "content/browser/loader/async_resource_handler.h"
#include <stddef.h>
#include <string>
#include <utility>
#include "base/format_macros.h"
#include "base/strings/string_util.h"
#include "base/strings/stringprintf.h"
#include "build/build_config.h"
#include "content/public/common/content_switches.h"
#include "content/public/test/browser_test_utils.h"
#include "content/public/test/content_browser_test.h"
#include "content/public/test/content_browser_test_utils.h"
#include "content/shell/browser/shell.h"
#include "net/test/embedded_test_server/embedded_test_server.h"
#include "net/test/embedded_test_server/http_request.h"
#include "net/test/embedded_test_server/http_response.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace content {
namespace {
const char kPostPath[] = "/post";
const char kRedirectPostPath[] = "/redirect";
// ThreadSanitizer is too slow to perform the full upload, so tests
// using that build get an easier test which might not show two distinct
// progress events. See crbug.com/526985. In addition, OSX buildbots have
// experienced slowdowns on this test (crbug.com/548819), give them the easier
// test too.
#if defined(THREAD_SANITIZER) || defined(OS_MACOSX)
const size_t kPayloadSize = 1062882; // 2*3^12
#else
const size_t kPayloadSize = 28697814; // 2*3^15
#endif
std::unique_ptr<net::test_server::HttpResponse> HandlePostAndRedirectURLs(
const std::string& request_path,
const net::test_server::HttpRequest& request) {
std::unique_ptr<net::test_server::BasicHttpResponse> http_response(
new net::test_server::BasicHttpResponse());
if (base::StartsWith(request.relative_url, kRedirectPostPath,
base::CompareCase::SENSITIVE)) {
http_response->set_code(net::HTTP_TEMPORARY_REDIRECT);
http_response->AddCustomHeader("Location", kPostPath);
EXPECT_EQ(request.content.length(), kPayloadSize);
return std::move(http_response);
} else if (base::StartsWith(request.relative_url, kPostPath,
base::CompareCase::SENSITIVE)) {
http_response->set_content("hello");
http_response->set_content_type("text/plain");
EXPECT_EQ(request.content.length(), kPayloadSize);
return std::move(http_response);
} else {
return std::unique_ptr<net::test_server::HttpResponse>();
}
}
} // namespace
class AsyncResourceHandlerBrowserTest : public ContentBrowserTest {
};
IN_PROC_BROWSER_TEST_F(AsyncResourceHandlerBrowserTest, UploadProgress) {
net::EmbeddedTestServer* test_server = embedded_test_server();
ASSERT_TRUE(test_server->Start());
test_server->RegisterRequestHandler(
base::Bind(&HandlePostAndRedirectURLs, kPostPath));
NavigateToURL(shell(),
test_server->GetURL("/loader/async_resource_handler.html"));
std::string js_result;
EXPECT_TRUE(ExecuteScriptAndExtractString(
shell(), base::StringPrintf("WaitForAsyncXHR('%s', %" PRIuS ")",
kPostPath, kPayloadSize),
&js_result));
EXPECT_EQ(js_result, "success");
}
IN_PROC_BROWSER_TEST_F(AsyncResourceHandlerBrowserTest,
UploadProgressRedirect) {
net::EmbeddedTestServer* test_server = embedded_test_server();
ASSERT_TRUE(test_server->Start());
test_server->RegisterRequestHandler(
base::Bind(&HandlePostAndRedirectURLs, kRedirectPostPath));
NavigateToURL(shell(),
test_server->GetURL("/loader/async_resource_handler.html"));
std::string js_result;
EXPECT_TRUE(ExecuteScriptAndExtractString(
shell(), base::StringPrintf("WaitForAsyncXHR('%s', %" PRIuS ")",
kRedirectPostPath, kPayloadSize),
&js_result));
EXPECT_EQ(js_result, "success");
}
} // namespace content
|