// 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