JobController 1: Adding a new class HttpStreamFactoryImpl::JobController

net/http/http_stream_factory_test_util.cc

+// Copyright (c) 2016 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/http/http_stream_factory_test_util.h"
+
+#include "net/proxy/proxy_info.h"
+
+using ::testing::_;
+
+namespace net {
+MockHttpStreamRequestDelegate::MockHttpStreamRequestDelegate() {}
+
+MockHttpStreamRequestDelegate::~MockHttpStreamRequestDelegate() {}
+
+TestHttpStreamFactoryImplJob::TestHttpStreamFactoryImplJob(
+    HttpStreamFactoryImpl::Job::Delegate* delegate,
+    HttpStreamFactoryImpl::JobType job_type,
+    HttpNetworkSession* session,
+    const HttpRequestInfo& request_info,
+    RequestPriority priority,
+    const SSLConfig& server_ssl_config,
+    const SSLConfig& proxy_ssl_config,
+    HostPortPair destination,
+    GURL origin_url,
+    NetLog* net_log)
+    : HttpStreamFactoryImpl::Job(delegate,
+                                 job_type,
+                                 session,
+                                 request_info,
+                                 priority,
+                                 server_ssl_config,
+                                 proxy_ssl_config,
+                                 destination,
+                                 origin_url,
+                                 net_log) {}
+
+TestHttpStreamFactoryImplJob::TestHttpStreamFactoryImplJob(
+    HttpStreamFactoryImpl::Job::Delegate* delegate,
+    HttpStreamFactoryImpl::JobType job_type,
+    HttpNetworkSession* session,
+    const HttpRequestInfo& request_info,
+    RequestPriority priority,
+    const SSLConfig& server_ssl_config,
+    const SSLConfig& proxy_ssl_config,
+    HostPortPair destination,
+    GURL origin_url,
+    AlternativeService alternative_service,
+    NetLog* net_log)
+    : HttpStreamFactoryImpl::Job(delegate,
+                                 job_type,
+                                 session,
+                                 request_info,
+                                 priority,
+                                 server_ssl_config,
+                                 proxy_ssl_config,
+                                 destination,
+                                 origin_url,
+                                 alternative_service,
+                                 net_log) {}
+
+TestHttpStreamFactoryImplJob::~TestHttpStreamFactoryImplJob() {}
+
+TestJobFactory::TestJobFactory()
+    : main_job_(nullptr), alternative_job_(nullptr) {}
+
+TestJobFactory::~TestJobFactory() {}
+
+HttpStreamFactoryImpl::Job* TestJobFactory::CreateJob(
+    HttpStreamFactoryImpl::Job::Delegate* delegate,
+    HttpStreamFactoryImpl::JobType job_type,
+    HttpNetworkSession* session,
+    const HttpRequestInfo& request_info,
+    RequestPriority priority,
+    const SSLConfig& server_ssl_config,
+    const SSLConfig& proxy_ssl_config,
+    HostPortPair destination,
+    GURL origin_url,
+    NetLog* net_log) {
+  DCHECK(!main_job_);
+  main_job_ = new TestHttpStreamFactoryImplJob(
+      delegate, job_type, session, request_info, priority, SSLConfig(),
+      SSLConfig(), destination, origin_url, nullptr);
+
+  EXPECT_CALL(*main_job_, Start(_)).Times(1);
+
+  return main_job_;
+}
+
+HttpStreamFactoryImpl::Job* TestJobFactory::CreateJob(
+    HttpStreamFactoryImpl::Job::Delegate* delegate,
+    HttpStreamFactoryImpl::JobType job_type,
+    HttpNetworkSession* session,
+    const HttpRequestInfo& request_info,
+    RequestPriority priority,
+    const SSLConfig& server_ssl_config,
+    const SSLConfig& proxy_ssl_config,
+    HostPortPair destination,
+    GURL origin_url,
+    AlternativeService alternative_service,
+    NetLog* net_log) {
+  DCHECK(!alternative_job_);
+  alternative_job_ = new TestHttpStreamFactoryImplJob(
+      delegate, job_type, session, request_info, priority, SSLConfig(),
+      SSLConfig(), destination, origin_url, alternative_service, nullptr);
+
+  EXPECT_CALL(*alternative_job_, Start(_)).Times(1);
+
+  return alternative_job_;
+}
+
+}  // namespace net