| Index: net/http/winhttp_request_throttle_unittest.cc
|
| ===================================================================
|
| --- net/http/winhttp_request_throttle_unittest.cc (revision 7967)
|
| +++ net/http/winhttp_request_throttle_unittest.cc (working copy)
|
| @@ -1,112 +0,0 @@
|
| -// 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
|
| -
|
|
|