url_request_job_factory_unittest.cc 3.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192
  1. // Copyright (c) 2012 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "net/url_request/url_request_job_factory.h"
  5. #include "base/bind.h"
  6. #include "base/location.h"
  7. #include "base/memory/weak_ptr.h"
  8. #include "base/run_loop.h"
  9. #include "base/task/single_thread_task_runner.h"
  10. #include "base/test/task_environment.h"
  11. #include "base/threading/thread_task_runner_handle.h"
  12. #include "net/base/request_priority.h"
  13. #include "net/test/gtest_util.h"
  14. #include "net/traffic_annotation/network_traffic_annotation_test_helper.h"
  15. #include "net/url_request/url_request.h"
  16. #include "net/url_request/url_request_context.h"
  17. #include "net/url_request/url_request_context_builder.h"
  18. #include "net/url_request/url_request_job.h"
  19. #include "net/url_request/url_request_test_util.h"
  20. #include "testing/gmock/include/gmock/gmock.h"
  21. #include "testing/gtest/include/gtest/gtest.h"
  22. using net::test::IsError;
  23. using net::test::IsOk;
  24. namespace net {
  25. namespace {
  26. class MockURLRequestJob : public URLRequestJob {
  27. public:
  28. explicit MockURLRequestJob(URLRequest* request) : URLRequestJob(request) {}
  29. ~MockURLRequestJob() override = default;
  30. void Start() override {
  31. // Start reading asynchronously so that all error reporting and data
  32. // callbacks happen as they would for network requests.
  33. base::ThreadTaskRunnerHandle::Get()->PostTask(
  34. FROM_HERE, base::BindOnce(&MockURLRequestJob::StartAsync,
  35. weak_factory_.GetWeakPtr()));
  36. }
  37. private:
  38. void StartAsync() { NotifyHeadersComplete(); }
  39. base::WeakPtrFactory<MockURLRequestJob> weak_factory_{this};
  40. };
  41. class DummyProtocolHandler : public URLRequestJobFactory::ProtocolHandler {
  42. public:
  43. std::unique_ptr<URLRequestJob> CreateJob(URLRequest* request) const override {
  44. return std::make_unique<MockURLRequestJob>(request);
  45. }
  46. };
  47. TEST(URLRequestJobFactoryTest, NoProtocolHandler) {
  48. base::test::TaskEnvironment task_environment(
  49. base::test::TaskEnvironment::MainThreadType::IO);
  50. TestDelegate delegate;
  51. auto request_context = CreateTestURLRequestContextBuilder()->Build();
  52. std::unique_ptr<URLRequest> request(
  53. request_context->CreateRequest(GURL("foo://bar"), DEFAULT_PRIORITY,
  54. &delegate, TRAFFIC_ANNOTATION_FOR_TESTS));
  55. request->Start();
  56. base::RunLoop().Run();
  57. EXPECT_EQ(ERR_UNKNOWN_URL_SCHEME, delegate.request_status());
  58. }
  59. TEST(URLRequestJobFactoryTest, BasicProtocolHandler) {
  60. base::test::TaskEnvironment task_environment(
  61. base::test::TaskEnvironment::MainThreadType::IO);
  62. TestDelegate delegate;
  63. auto context_builder = CreateTestURLRequestContextBuilder();
  64. context_builder->SetProtocolHandler("foo",
  65. std::make_unique<DummyProtocolHandler>());
  66. auto request_context = context_builder->Build();
  67. std::unique_ptr<URLRequest> request(
  68. request_context->CreateRequest(GURL("foo://bar"), DEFAULT_PRIORITY,
  69. &delegate, TRAFFIC_ANNOTATION_FOR_TESTS));
  70. request->Start();
  71. base::RunLoop().Run();
  72. EXPECT_EQ(OK, delegate.request_status());
  73. }
  74. } // namespace
  75. } // namespace net