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
|
// Copyright (c) 2006-2008 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 "base/string_util.h"
#include "net/http/winhttp_request_throttle.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace {
// Converts an int i to an HINTERNET (void *) request handle.
HINTERNET RequestHandle(int i) {
return reinterpret_cast<HINTERNET>(static_cast<intptr_t>(i));
}
class MockRequestThrottle : public net::WinHttpRequestThrottle {
public:
MockRequestThrottle() : last_sent_request_(NULL) { }
// The request handle of the last sent request. This allows us to determine
// whether a submitted request was sent or queued.
HINTERNET last_sent_request() const { return last_sent_request_; }
protected:
virtual BOOL SendRequest(HINTERNET request_handle,
DWORD total_size,
DWORD_PTR context,
bool report_async_error) {
last_sent_request_ = request_handle;
return TRUE;
}
private:
HINTERNET last_sent_request_;
DISALLOW_EVIL_CONSTRUCTORS(MockRequestThrottle);
};
} // namespace
namespace net {
TEST(WinHttpRequestThrottleTest, OneServer) {
MockRequestThrottle throttle;
std::string server("http://www.foo.com");
HINTERNET request_handle;
// Submit 20 requests to the request throttle.
// Expected outcome: 6 requests should be in progress, and requests 7-20
// should be queued.
for (int i = 1; i <= 20; i++) {
request_handle = RequestHandle(i);
EXPECT_TRUE(throttle.SubmitRequest(server, request_handle, 0, 0));
if (i <= 6)
EXPECT_EQ(request_handle, throttle.last_sent_request());
else
EXPECT_EQ(RequestHandle(6), throttle.last_sent_request());
}
// Notify the request throttle of the completion of 10 requests.
// Expected outcome: 6 requests should be in progress, and requests 17-20
// should be queued.
for (int j = 0; j < 10; j++) {
throttle.NotifyRequestDone(server);
EXPECT_EQ(RequestHandle(7 + j), throttle.last_sent_request());
}
// Remove request 17, which is queued.
// Expected outcome: Requests 18-20 should remain queued.
request_handle = RequestHandle(17);
throttle.RemoveRequest(server, request_handle);
EXPECT_EQ(RequestHandle(16), throttle.last_sent_request());
// Remove request 16, which is in progress.
// Expected outcome: The request throttle should send request 18.
// Requests 19-20 should remained queued.
request_handle = RequestHandle(16);
throttle.RemoveRequest(server, request_handle);
EXPECT_EQ(RequestHandle(18), throttle.last_sent_request());
// Notify the request throttle of the completion of the remaining
// 8 requests.
for (int j = 0; j < 8; j++) {
throttle.NotifyRequestDone(server);
if (j < 2)
EXPECT_EQ(RequestHandle(19 + j), throttle.last_sent_request());
else
EXPECT_EQ(RequestHandle(20), throttle.last_sent_request());
}
}
// Submit requests to a large number (> 64) of servers to force the garbage
// collection of idle PerServerThrottles.
TEST(WinHttpRequestThrottleTest, GarbageCollect) {
MockRequestThrottle throttle;
for (int i = 0; i < 150; i++) {
std::string server("http://www.foo");
server.append(IntToString(i));
server.append(".com");
throttle.SubmitRequest(server, RequestHandle(1), 0, 0);
throttle.NotifyRequestDone(server);
if (i < 64)
EXPECT_EQ(i + 1, throttle.throttles_.size());
else if (i < 129)
EXPECT_EQ(i - 64, throttle.throttles_.size());
else
EXPECT_EQ(i - 129, throttle.throttles_.size());
}
}
} // namespace net
|
