Limit total number of sockets in the system....

net/socket/tcp_client_socket_pool_unittest.cc

 // 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 "net/socket/tcp_client_socket_pool.h"
 
 #include "base/compiler_specific.h"
 #include "base/message_loop.h"
 #include "net/base/mock_host_resolver.h"
 #include "net/base/net_errors.h"
 #include "net/base/test_completion_callback.h"
 #include "net/socket/client_socket.h"
 #include "net/socket/client_socket_factory.h"
 #include "net/socket/client_socket_handle.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace net {
 
 namespace {
 
+const int kMaxSockets = 32;
 const int kMaxSocketsPerGroup = 6;
 
 // Note that the first and the last are the same, the first should be handled
 // before the last, since it was inserted first.
 const int kPriorities[10] = { 1, 7, 9, 5, 6, 2, 8, 3, 4, 1 };
 
 // This is the number of extra requests beyond the first few that use up all
 // available sockets in the socket group.
 const int kNumPendingRequests = arraysize(kPriorities);
 
@@ skipping 187 matching lines @@
   std::vector<TestSocketRequest*>* request_order_;
   TestCompletionCallback callback_;
 };
 
 int TestSocketRequest::completion_count = 0;
 
 class TCPClientSocketPoolTest : public testing::Test {
  protected:
   TCPClientSocketPoolTest()
       : host_resolver_(new MockHostResolver),
-        pool_(new TCPClientSocketPool(kMaxSocketsPerGroup,
+        pool_(new TCPClientSocketPool(kMaxSockets,
+                                      kMaxSocketsPerGroup,
                                       host_resolver_,
                                       &client_socket_factory_)) {
   }
 
   virtual void SetUp() {
     TestSocketRequest::completion_count = 0;
   }
 
   virtual void TearDown() {
     // The tests often call Reset() on handles at the end which may post
@@ skipping 372 matching lines @@
 
   EXPECT_EQ(kNumPendingRequests, TestSocketRequest::completion_count);
 }
 
 // Make sure that pending requests get serviced after active requests fail.
 TEST_F(TCPClientSocketPoolTest, FailingActiveRequestWithPendingRequests) {
   client_socket_factory_.set_client_socket_type(
       MockClientSocketFactory::MOCK_PENDING_FAILING_CLIENT_SOCKET);
 
   scoped_ptr<TestSocketRequest> reqs[kMaxSocketsPerGroup * 2 + 1];
+  ASSERT_LE(static_cast<int>(arraysize(reqs)), kMaxSockets);
 
   // Queue up all the requests
 
   HostResolver::RequestInfo info("www.google.com", 80);
   for (size_t i = 0; i < arraysize(reqs); ++i) {
     reqs[i].reset(new TestSocketRequest(pool_.get(), &request_order_));
     int rv = reqs[i]->handle.Init("a", info, 5, reqs[i].get());
     EXPECT_EQ(ERR_IO_PENDING, rv);
   }
 
   for (size_t i = 0; i < arraysize(reqs); ++i)
     EXPECT_EQ(ERR_CONNECTION_FAILED, reqs[i]->WaitForResult());
 }
 
 }  // namespace
 
 }  // namespace net