drive_base_requests_server_unittest.cc 6.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169
  1. // Copyright (c) 2013 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 "google_apis/drive/drive_base_requests.h"
  5. #include <memory>
  6. #include "base/bind.h"
  7. #include "base/files/file_util.h"
  8. #include "base/files/scoped_temp_dir.h"
  9. #include "base/run_loop.h"
  10. #include "base/test/task_environment.h"
  11. #include "google_apis/common/dummy_auth_service.h"
  12. #include "google_apis/common/request_sender.h"
  13. #include "google_apis/common/task_util.h"
  14. #include "google_apis/common/test_util.h"
  15. #include "mojo/public/cpp/bindings/pending_remote.h"
  16. #include "mojo/public/cpp/bindings/remote.h"
  17. #include "net/test/embedded_test_server/embedded_test_server.h"
  18. #include "net/test/embedded_test_server/http_request.h"
  19. #include "net/test/embedded_test_server/http_response.h"
  20. #include "net/traffic_annotation/network_traffic_annotation_test_helper.h"
  21. #include "services/network/network_service.h"
  22. #include "services/network/public/cpp/weak_wrapper_shared_url_loader_factory.h"
  23. #include "services/network/public/mojom/network_context.mojom.h"
  24. #include "services/network/public/mojom/url_loader_factory.mojom.h"
  25. #include "services/network/test/fake_test_cert_verifier_params_factory.h"
  26. #include "testing/gtest/include/gtest/gtest.h"
  27. namespace google_apis {
  28. namespace {
  29. const char kTestUserAgent[] = "test-user-agent";
  30. } // namespace
  31. class BaseRequestsServerTest : public testing::Test {
  32. protected:
  33. BaseRequestsServerTest() {
  34. mojo::Remote<network::mojom::NetworkService> network_service_remote;
  35. network_service_ = network::NetworkService::Create(
  36. network_service_remote.BindNewPipeAndPassReceiver());
  37. network::mojom::NetworkContextParamsPtr context_params =
  38. network::mojom::NetworkContextParams::New();
  39. // Use a dummy CertVerifier that always passes cert verification, since
  40. // these unittests don't need to test CertVerifier behavior.
  41. context_params->cert_verifier_params =
  42. network::FakeTestCertVerifierParamsFactory::GetCertVerifierParams();
  43. network_service_remote->CreateNetworkContext(
  44. network_context_.BindNewPipeAndPassReceiver(),
  45. std::move(context_params));
  46. mojo::PendingReceiver<network::mojom::URLLoaderNetworkServiceObserver>
  47. default_observer_receiver;
  48. network::mojom::NetworkServiceParamsPtr network_service_params =
  49. network::mojom::NetworkServiceParams::New();
  50. network_service_params->default_observer =
  51. default_observer_receiver.InitWithNewPipeAndPassRemote();
  52. network_service_remote->SetParams(std::move(network_service_params));
  53. network::mojom::URLLoaderFactoryParamsPtr params =
  54. network::mojom::URLLoaderFactoryParams::New();
  55. params->process_id = network::mojom::kBrowserProcessId;
  56. params->is_corb_enabled = false;
  57. network_context_->CreateURLLoaderFactory(
  58. url_loader_factory_.BindNewPipeAndPassReceiver(), std::move(params));
  59. test_shared_loader_factory_ =
  60. base::MakeRefCounted<network::WeakWrapperSharedURLLoaderFactory>(
  61. url_loader_factory_.get());
  62. }
  63. void SetUp() override {
  64. ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
  65. request_sender_ = std::make_unique<RequestSender>(
  66. std::make_unique<DummyAuthService>(), test_shared_loader_factory_,
  67. task_environment_.GetMainThreadTaskRunner(), kTestUserAgent,
  68. TRAFFIC_ANNOTATION_FOR_TESTS);
  69. ASSERT_TRUE(test_server_.InitializeAndListen());
  70. test_server_.RegisterRequestHandler(base::BindRepeating(
  71. &test_util::HandleDownloadFileRequest, test_server_.base_url(),
  72. base::Unretained(&http_request_)));
  73. test_server_.StartAcceptingConnections();
  74. }
  75. // Returns a temporary file path suitable for storing the cache file.
  76. base::FilePath GetTestCachedFilePath(const base::FilePath& file_name) {
  77. return temp_dir_.GetPath().Append(file_name);
  78. }
  79. base::test::TaskEnvironment task_environment_{
  80. base::test::TaskEnvironment::MainThreadType::IO};
  81. net::EmbeddedTestServer test_server_;
  82. std::unique_ptr<RequestSender> request_sender_;
  83. std::unique_ptr<network::mojom::NetworkService> network_service_;
  84. mojo::Remote<network::mojom::NetworkContext> network_context_;
  85. mojo::Remote<network::mojom::URLLoaderFactory> url_loader_factory_;
  86. scoped_refptr<network::WeakWrapperSharedURLLoaderFactory>
  87. test_shared_loader_factory_;
  88. base::ScopedTempDir temp_dir_;
  89. // The incoming HTTP request is saved so tests can verify the request
  90. // parameters like HTTP method (ex. some requests should use DELETE
  91. // instead of GET).
  92. net::test_server::HttpRequest http_request_;
  93. };
  94. TEST_F(BaseRequestsServerTest, DownloadFileRequest_ValidFile) {
  95. ApiErrorCode result_code = OTHER_ERROR;
  96. base::FilePath temp_file;
  97. {
  98. base::RunLoop run_loop;
  99. std::unique_ptr<DownloadFileRequestBase> request =
  100. std::make_unique<DownloadFileRequestBase>(
  101. request_sender_.get(),
  102. test_util::CreateQuitCallback(
  103. &run_loop,
  104. test_util::CreateCopyResultCallback(&result_code, &temp_file)),
  105. GetContentCallback(), ProgressCallback(),
  106. test_server_.GetURL("/files/drive/testfile.txt"),
  107. GetTestCachedFilePath(
  108. base::FilePath::FromUTF8Unsafe("cached_testfile.txt")));
  109. request_sender_->StartRequestWithAuthRetry(std::move(request));
  110. run_loop.Run();
  111. }
  112. std::string contents;
  113. base::ReadFileToString(temp_file, &contents);
  114. base::DeleteFile(temp_file);
  115. EXPECT_EQ(HTTP_SUCCESS, result_code);
  116. EXPECT_EQ(net::test_server::METHOD_GET, http_request_.method);
  117. EXPECT_EQ("/files/drive/testfile.txt", http_request_.relative_url);
  118. const base::FilePath expected_path =
  119. test_util::GetTestFilePath("drive/testfile.txt");
  120. std::string expected_contents;
  121. base::ReadFileToString(expected_path, &expected_contents);
  122. EXPECT_EQ(expected_contents, contents);
  123. }
  124. TEST_F(BaseRequestsServerTest, DownloadFileRequest_NonExistentFile) {
  125. ApiErrorCode result_code = OTHER_ERROR;
  126. base::FilePath temp_file;
  127. {
  128. base::RunLoop run_loop;
  129. std::unique_ptr<DownloadFileRequestBase> request =
  130. std::make_unique<DownloadFileRequestBase>(
  131. request_sender_.get(),
  132. test_util::CreateQuitCallback(
  133. &run_loop,
  134. test_util::CreateCopyResultCallback(&result_code, &temp_file)),
  135. GetContentCallback(), ProgressCallback(),
  136. test_server_.GetURL("/files/gdata/no-such-file.txt"),
  137. GetTestCachedFilePath(
  138. base::FilePath::FromUTF8Unsafe("cache_no-such-file.txt")));
  139. request_sender_->StartRequestWithAuthRetry(std::move(request));
  140. run_loop.Run();
  141. }
  142. EXPECT_EQ(HTTP_NOT_FOUND, result_code);
  143. EXPECT_EQ(net::test_server::METHOD_GET, http_request_.method);
  144. EXPECT_EQ("/files/gdata/no-such-file.txt", http_request_.relative_url);
  145. // Do not verify the not found message.
  146. }
  147. } // namespace google_apis