Update marking proxies as bad in ProxyList/ProxyService to use a std::vector

net/proxy/proxy_service_unittest.cc

 // Copyright (c) 2012 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/proxy/proxy_service.h"
 
 #include <vector>
 
 #include "base/format_macros.h"
 #include "base/logging.h"
@@ skipping 1558 matching lines @@
   EXPECT_EQ(OK, rv);
   EXPECT_TRUE(info[0].is_direct());
 
   // Request for a .com domain hits the proxy.
   rv = service.ResolveProxy(url2, LOAD_NORMAL, &info[1], callback[1].callback(),
                             NULL, NULL, BoundNetLog());
   EXPECT_EQ(OK, rv);
   EXPECT_EQ("foopy1:8080", info[1].proxy_server().ToURI());
 }
 
+TEST_F(ProxyServiceTest, MarkProxiesAsBadTests) {
+  ProxyConfig config;
+  config.proxy_rules().ParseFromString(
+      "http=foopy1:8080;http=foopy2:8080;http=foopy3.8080;http=foopy4:8080");
+  config.set_auto_detect(false);
+
+  ProxyList proxy_list;
+  std::vector<ProxyServer> additional_bad_proxies;
+  for (const ProxyServer& proxy_server :
+       config.proxy_rules().proxies_for_http.GetAll()) {
+    proxy_list.AddProxyServer(proxy_server);
+    if (proxy_server == config.proxy_rules().proxies_for_http.Get())
+      continue;
+
+    additional_bad_proxies.push_back(proxy_server);
+  }
+
+  EXPECT_EQ(3u, additional_bad_proxies.size());
+
+  ProxyService service(new MockProxyConfigService(config), nullptr, NULL);
+  ProxyInfo proxy_info;
+  proxy_info.UseProxyList(proxy_list);
+  const ProxyRetryInfoMap& retry_info = service.proxy_retry_info();
+  service.MarkProxiesAsBadUntil(proxy_info, base::TimeDelta::FromSeconds(1),
+                                additional_bad_proxies, BoundNetLog());
+  ASSERT_EQ(4u, retry_info.size());
+  for (const ProxyServer& proxy_server :
+       config.proxy_rules().proxies_for_http.GetAll()) {
+    ProxyRetryInfoMap::const_iterator i =
+        retry_info.find(proxy_server.host_port_pair().ToString());
+    ASSERT_TRUE(i != retry_info.end());
+  }
+}
 
 TEST_F(ProxyServiceTest, PerProtocolProxyTests) {
   ProxyConfig config;
   config.proxy_rules().ParseFromString("http=foopy1:8080;https=foopy2:8080");
   config.set_auto_detect(false);
   {
     ProxyService service(new MockProxyConfigService(config), nullptr, NULL);
     GURL test_url("http://www.msn.com");
     ProxyInfo info;
     TestCompletionCallback callback;
@@ skipping 1651 matching lines @@
       url, LOAD_NORMAL, &info, NULL, log.bound());
   EXPECT_TRUE(synchronous_success);
   EXPECT_FALSE(info.is_direct());
   EXPECT_EQ("foopy1", info.proxy_server().host_port_pair().host());
 
   // No request should have been queued.
   EXPECT_EQ(0u, factory->pending_requests().size());
 }
 
 }  // namespace net