Use shared shared URLRequestContextGetter for all gcd_private sessions

chrome/browser/local_discovery/privet_http_unittest.cc

 // Copyright 2013 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/bind.h"
 #include "base/command_line.h"
 #include "base/json/json_reader.h"
 #include "base/json/json_writer.h"
 #include "base/location.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "base/single_thread_task_runner.h"
 #include "base/thread_task_runner_handle.h"
+#include "chrome/browser/extensions/api/gcd_private/privet_v3_context_getter.h"
 #include "chrome/browser/local_discovery/privet_http_impl.h"
 #include "chrome/common/chrome_switches.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/test/test_browser_thread_bundle.h"
 #include "net/base/host_port_pair.h"
 #include "net/base/net_errors.h"
 #include "net/test/embedded_test_server/embedded_test_server.h"
 #include "net/url_request/test_url_fetcher_factory.h"
 #include "net/url_request/url_fetcher.h"
 #include "net/url_request/url_request_test_util.h"
@@ skipping 1011 matching lines @@
 
 class PrivetHttpWithServerTest : public ::testing::Test,
                                  public PrivetURLFetcher::Delegate {
  protected:
   PrivetHttpWithServerTest()
       : thread_bundle_(content::TestBrowserThreadBundle::REAL_IO_THREAD) {}
 
   void SetUp() override {
     base::CommandLine::ForCurrentProcess()->AppendSwitch(
         switches::kEnablePrivetV3);
-    context_getter_ = new net::TestURLRequestContextGetter(
-        BrowserThread::GetMessageLoopProxyForThread(BrowserThread::IO));
   }
 
   void OnError(PrivetURLFetcher* fetcher,
                PrivetURLFetcher::ErrorType error) override {
     done_ = true;
     success_ = false;
     error_ = error;
 
     base::MessageLoop::current()->PostTask(FROM_HERE, quit_);
   }
@@ skipping 20 matching lines @@
     server_.reset(new EmbeddedTestServer(type));
     ASSERT_TRUE(server_->InitializeAndWaitUntilReady());
 
     base::FilePath test_data_dir;
     ASSERT_TRUE(PathService::Get(base::DIR_SOURCE_ROOT, &test_data_dir));
     server_->ServeFilesFromDirectory(
         test_data_dir.Append(FILE_PATH_LITERAL("chrome/test/data")));
   }
 
   void CreateClient() {
+    context_getter_ = new extensions::PrivetV3ContextGetter(
+        BrowserThread::GetMessageLoopProxyForThread(BrowserThread::IO));
     client_.reset(new PrivetHTTPClientImpl("test", server_->host_port_pair(),
                                            context_getter_));
   }
 
   void Run() {
     success_ = false;
     done_ = false;
 
     base::RunLoop run_loop;
     quit_ = run_loop.QuitClosure();
 
     scoped_ptr<PrivetURLFetcher> fetcher = client_->CreateURLFetcher(
         server_->GetURL("/simple.html"), net::URLFetcher::GET, this);
 
     fetcher->SetMaxRetries(1);
     fetcher->V3Mode();
     fetcher->Start();
 
     run_loop.Run();
 
     EXPECT_TRUE(done_);
   }
 
   bool success_ = false;
   bool done_ = false;
   PrivetURLFetcher::ErrorType error_ = PrivetURLFetcher::ErrorType();
   content::TestBrowserThreadBundle thread_bundle_;
-  scoped_refptr<net::TestURLRequestContextGetter> context_getter_;
+  scoped_refptr<extensions::PrivetV3ContextGetter> context_getter_;
   scoped_ptr<EmbeddedTestServer> server_;
   scoped_ptr<PrivetHTTPClientImpl> client_;
 
   base::Closure quit_;
 };
 
 TEST_F(PrivetHttpWithServerTest, HttpServer) {
   CreateServer(EmbeddedTestServer::TYPE_HTTP);
 
   CreateClient();
   Run();
   EXPECT_TRUE(success_);
 
   CreateClient();
-  net::SHA256HashValue fingerprint = {};
-  client_->SwitchToHttps(server_->host_port_pair().port(), fingerprint);
+  client_->SwitchToHttps(server_->host_port_pair().port());
   Run();
   EXPECT_FALSE(success_);
   EXPECT_EQ(PrivetURLFetcher::UNKNOWN_ERROR, error_);
 }
 
 TEST_F(PrivetHttpWithServerTest, HttpsServer) {
   CreateServer(EmbeddedTestServer::TYPE_HTTPS);
 
   CreateClient();
   Run();
   EXPECT_FALSE(success_);
   EXPECT_EQ(PrivetURLFetcher::UNKNOWN_ERROR, error_);
 
   CreateClient();
   net::SHA256HashValue fingerprint =
       net::X509Certificate::CalculateFingerprint256(
           server_->GetCertificate()->os_cert_handle());
-  client_->SwitchToHttps(server_->host_port_pair().port(), fingerprint);
+  {
+    base::RunLoop run_loop;
+    context_getter_->AddPairedHost(client_->GetHost(), fingerprint,
+                                   run_loop.QuitClosure());
+    run_loop.Run();
+  }
+  client_->SwitchToHttps(server_->host_port_pair().port());
   Run();
   EXPECT_TRUE(success_);
 
   CreateClient();
   fingerprint = {};
-  client_->SwitchToHttps(server_->host_port_pair().port(), fingerprint);
+  client_->SwitchToHttps(server_->host_port_pair().port());
   Run();
   EXPECT_FALSE(success_);
   EXPECT_EQ(PrivetURLFetcher::REQUEST_CANCELED, error_);
 }
 
 }  // namespace
 
 }  // namespace local_discovery