OLD | NEW |
| (Empty) |
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 | |
5 #include "chrome/browser/google_apis/base_requests.h" | |
6 | |
7 #include "base/bind.h" | |
8 #include "base/file_util.h" | |
9 #include "base/files/scoped_temp_dir.h" | |
10 #include "base/memory/scoped_ptr.h" | |
11 #include "base/message_loop/message_loop.h" | |
12 #include "base/run_loop.h" | |
13 #include "chrome/browser/google_apis/dummy_auth_service.h" | |
14 #include "chrome/browser/google_apis/request_sender.h" | |
15 #include "chrome/browser/google_apis/task_util.h" | |
16 #include "chrome/browser/google_apis/test_util.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/url_request/url_request_test_util.h" | |
21 #include "testing/gtest/include/gtest/gtest.h" | |
22 | |
23 namespace google_apis { | |
24 | |
25 namespace { | |
26 | |
27 const char kTestUserAgent[] = "test-user-agent"; | |
28 | |
29 } // namespace | |
30 | |
31 class BaseRequestsServerTest : public testing::Test { | |
32 protected: | |
33 BaseRequestsServerTest() { | |
34 } | |
35 | |
36 virtual void SetUp() OVERRIDE { | |
37 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); | |
38 | |
39 request_context_getter_ = new net::TestURLRequestContextGetter( | |
40 message_loop_.message_loop_proxy()); | |
41 | |
42 request_sender_.reset(new RequestSender( | |
43 new DummyAuthService, | |
44 request_context_getter_.get(), | |
45 message_loop_.message_loop_proxy(), | |
46 kTestUserAgent)); | |
47 | |
48 ASSERT_TRUE(test_server_.InitializeAndWaitUntilReady()); | |
49 test_server_.RegisterRequestHandler( | |
50 base::Bind(&test_util::HandleDownloadFileRequest, | |
51 test_server_.base_url(), | |
52 base::Unretained(&http_request_))); | |
53 } | |
54 | |
55 // Returns a temporary file path suitable for storing the cache file. | |
56 base::FilePath GetTestCachedFilePath(const base::FilePath& file_name) { | |
57 return temp_dir_.path().Append(file_name); | |
58 } | |
59 | |
60 base::MessageLoopForIO message_loop_; // Test server needs IO thread. | |
61 net::test_server::EmbeddedTestServer test_server_; | |
62 scoped_ptr<RequestSender> request_sender_; | |
63 scoped_refptr<net::TestURLRequestContextGetter> request_context_getter_; | |
64 base::ScopedTempDir temp_dir_; | |
65 | |
66 // The incoming HTTP request is saved so tests can verify the request | |
67 // parameters like HTTP method (ex. some requests should use DELETE | |
68 // instead of GET). | |
69 net::test_server::HttpRequest http_request_; | |
70 }; | |
71 | |
72 TEST_F(BaseRequestsServerTest, DownloadFileRequest_ValidFile) { | |
73 GDataErrorCode result_code = GDATA_OTHER_ERROR; | |
74 base::FilePath temp_file; | |
75 { | |
76 base::RunLoop run_loop; | |
77 DownloadFileRequestBase* request = new DownloadFileRequestBase( | |
78 request_sender_.get(), | |
79 test_util::CreateQuitCallback( | |
80 &run_loop, | |
81 test_util::CreateCopyResultCallback(&result_code, &temp_file)), | |
82 GetContentCallback(), | |
83 ProgressCallback(), | |
84 test_server_.GetURL("/files/gdata/testfile.txt"), | |
85 GetTestCachedFilePath( | |
86 base::FilePath::FromUTF8Unsafe("cached_testfile.txt"))); | |
87 request_sender_->StartRequestWithRetry(request); | |
88 run_loop.Run(); | |
89 } | |
90 | |
91 std::string contents; | |
92 base::ReadFileToString(temp_file, &contents); | |
93 base::DeleteFile(temp_file, false); | |
94 | |
95 EXPECT_EQ(HTTP_SUCCESS, result_code); | |
96 EXPECT_EQ(net::test_server::METHOD_GET, http_request_.method); | |
97 EXPECT_EQ("/files/gdata/testfile.txt", http_request_.relative_url); | |
98 | |
99 const base::FilePath expected_path = | |
100 test_util::GetTestFilePath("gdata/testfile.txt"); | |
101 std::string expected_contents; | |
102 base::ReadFileToString(expected_path, &expected_contents); | |
103 EXPECT_EQ(expected_contents, contents); | |
104 } | |
105 | |
106 TEST_F(BaseRequestsServerTest, DownloadFileRequest_NonExistentFile) { | |
107 GDataErrorCode result_code = GDATA_OTHER_ERROR; | |
108 base::FilePath temp_file; | |
109 { | |
110 base::RunLoop run_loop; | |
111 DownloadFileRequestBase* request = new DownloadFileRequestBase( | |
112 request_sender_.get(), | |
113 test_util::CreateQuitCallback( | |
114 &run_loop, | |
115 test_util::CreateCopyResultCallback(&result_code, &temp_file)), | |
116 GetContentCallback(), | |
117 ProgressCallback(), | |
118 test_server_.GetURL("/files/gdata/no-such-file.txt"), | |
119 GetTestCachedFilePath( | |
120 base::FilePath::FromUTF8Unsafe("cache_no-such-file.txt"))); | |
121 request_sender_->StartRequestWithRetry(request); | |
122 run_loop.Run(); | |
123 } | |
124 EXPECT_EQ(HTTP_NOT_FOUND, result_code); | |
125 EXPECT_EQ(net::test_server::METHOD_GET, http_request_.method); | |
126 EXPECT_EQ("/files/gdata/no-such-file.txt", | |
127 http_request_.relative_url); | |
128 // Do not verify the not found message. | |
129 } | |
130 | |
131 } // namespace google_apis | |
OLD | NEW |