OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include <windows.h> | 5 #include <windows.h> |
6 | 6 |
7 #include <fstream> | 7 #include <fstream> |
8 | 8 |
9 #include "base/base_paths.h" | 9 #include "base/base_paths.h" |
10 #include "base/file_util.h" | 10 #include "base/file_util.h" |
| 11 #include "base/files/scoped_temp_dir.h" |
11 #include "base/logging.h" | 12 #include "base/logging.h" |
12 #include "base/memory/scoped_ptr.h" | 13 #include "base/memory/scoped_ptr.h" |
13 #include "base/path_service.h" | 14 #include "base/path_service.h" |
14 #include "base/process_util.h" | 15 #include "base/process_util.h" |
15 #include "base/scoped_temp_dir.h" | |
16 #include "base/string_util.h" | 16 #include "base/string_util.h" |
17 #include "chrome/installer/util/delete_tree_work_item.h" | 17 #include "chrome/installer/util/delete_tree_work_item.h" |
18 #include "chrome/installer/util/work_item.h" | 18 #include "chrome/installer/util/work_item.h" |
19 #include "testing/gtest/include/gtest/gtest.h" | 19 #include "testing/gtest/include/gtest/gtest.h" |
20 | 20 |
21 namespace { | 21 namespace { |
22 class DeleteTreeWorkItemTest : public testing::Test { | 22 class DeleteTreeWorkItemTest : public testing::Test { |
23 protected: | 23 protected: |
24 virtual void SetUp() { | 24 virtual void SetUp() { |
25 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); | 25 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); |
26 } | 26 } |
27 | 27 |
28 // The temporary directory used to contain the test operations. | 28 // The temporary directory used to contain the test operations. |
29 ScopedTempDir temp_dir_; | 29 base::ScopedTempDir temp_dir_; |
30 }; | 30 }; |
31 | 31 |
32 // Simple function to dump some text into a new file. | 32 // Simple function to dump some text into a new file. |
33 void CreateTextFile(const std::wstring& filename, | 33 void CreateTextFile(const std::wstring& filename, |
34 const std::wstring& contents) { | 34 const std::wstring& contents) { |
35 std::ofstream file; | 35 std::ofstream file; |
36 file.open(filename.c_str()); | 36 file.open(filename.c_str()); |
37 ASSERT_TRUE(file.is_open()); | 37 ASSERT_TRUE(file.is_open()); |
38 file << contents; | 38 file << contents; |
39 file.close(); | 39 file.close(); |
(...skipping 25 matching lines...) Expand all Loading... |
65 file_name_delete_1 = file_name_delete_1.AppendASCII("File_1.txt"); | 65 file_name_delete_1 = file_name_delete_1.AppendASCII("File_1.txt"); |
66 CreateTextFile(file_name_delete_1.value(), text_content_1); | 66 CreateTextFile(file_name_delete_1.value(), text_content_1); |
67 ASSERT_TRUE(file_util::PathExists(file_name_delete_1)); | 67 ASSERT_TRUE(file_util::PathExists(file_name_delete_1)); |
68 | 68 |
69 FilePath file_name_delete_2(dir_name_delete_2); | 69 FilePath file_name_delete_2(dir_name_delete_2); |
70 file_name_delete_2 = file_name_delete_2.AppendASCII("File_2.txt"); | 70 file_name_delete_2 = file_name_delete_2.AppendASCII("File_2.txt"); |
71 CreateTextFile(file_name_delete_2.value(), text_content_1); | 71 CreateTextFile(file_name_delete_2.value(), text_content_1); |
72 ASSERT_TRUE(file_util::PathExists(file_name_delete_2)); | 72 ASSERT_TRUE(file_util::PathExists(file_name_delete_2)); |
73 | 73 |
74 // Test Do(). | 74 // Test Do(). |
75 ScopedTempDir temp_dir; | 75 base::ScopedTempDir temp_dir; |
76 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); | 76 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); |
77 | 77 |
78 std::vector<FilePath> key_files; | 78 std::vector<FilePath> key_files; |
79 scoped_ptr<DeleteTreeWorkItem> work_item( | 79 scoped_ptr<DeleteTreeWorkItem> work_item( |
80 WorkItem::CreateDeleteTreeWorkItem(dir_name_delete, temp_dir.path(), | 80 WorkItem::CreateDeleteTreeWorkItem(dir_name_delete, temp_dir.path(), |
81 key_files)); | 81 key_files)); |
82 EXPECT_TRUE(work_item->Do()); | 82 EXPECT_TRUE(work_item->Do()); |
83 | 83 |
84 // everything should be gone | 84 // everything should be gone |
85 EXPECT_FALSE(file_util::PathExists(file_name_delete_1)); | 85 EXPECT_FALSE(file_util::PathExists(file_name_delete_1)); |
(...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
117 file_name_delete_1 = file_name_delete_1.AppendASCII("File_1.txt"); | 117 file_name_delete_1 = file_name_delete_1.AppendASCII("File_1.txt"); |
118 CreateTextFile(file_name_delete_1.value(), text_content_1); | 118 CreateTextFile(file_name_delete_1.value(), text_content_1); |
119 ASSERT_TRUE(file_util::PathExists(file_name_delete_1)); | 119 ASSERT_TRUE(file_util::PathExists(file_name_delete_1)); |
120 | 120 |
121 FilePath file_name_delete_2(dir_name_delete_2); | 121 FilePath file_name_delete_2(dir_name_delete_2); |
122 file_name_delete_2 = file_name_delete_2.AppendASCII("File_2.txt"); | 122 file_name_delete_2 = file_name_delete_2.AppendASCII("File_2.txt"); |
123 CreateTextFile(file_name_delete_2.value(), text_content_1); | 123 CreateTextFile(file_name_delete_2.value(), text_content_1); |
124 ASSERT_TRUE(file_util::PathExists(file_name_delete_2)); | 124 ASSERT_TRUE(file_util::PathExists(file_name_delete_2)); |
125 | 125 |
126 // test Do() | 126 // test Do() |
127 ScopedTempDir temp_dir; | 127 base::ScopedTempDir temp_dir; |
128 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); | 128 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); |
129 | 129 |
130 std::vector<FilePath> key_files(1, file_name_delete_1); | 130 std::vector<FilePath> key_files(1, file_name_delete_1); |
131 scoped_ptr<DeleteTreeWorkItem> work_item( | 131 scoped_ptr<DeleteTreeWorkItem> work_item( |
132 WorkItem::CreateDeleteTreeWorkItem(dir_name_delete, temp_dir.path(), | 132 WorkItem::CreateDeleteTreeWorkItem(dir_name_delete, temp_dir.path(), |
133 key_files)); | 133 key_files)); |
134 EXPECT_TRUE(work_item->Do()); | 134 EXPECT_TRUE(work_item->Do()); |
135 | 135 |
136 // everything should be gone | 136 // everything should be gone |
137 EXPECT_FALSE(file_util::PathExists(file_name_delete_1)); | 137 EXPECT_FALSE(file_util::PathExists(file_name_delete_1)); |
(...skipping 52 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
190 // Run the key path file to keep it in use. | 190 // Run the key path file to keep it in use. |
191 STARTUPINFOW si = {sizeof(si)}; | 191 STARTUPINFOW si = {sizeof(si)}; |
192 PROCESS_INFORMATION pi = {0}; | 192 PROCESS_INFORMATION pi = {0}; |
193 ASSERT_TRUE( | 193 ASSERT_TRUE( |
194 ::CreateProcessW(NULL, const_cast<wchar_t*>(key_path.value().c_str()), | 194 ::CreateProcessW(NULL, const_cast<wchar_t*>(key_path.value().c_str()), |
195 NULL, NULL, FALSE, CREATE_NO_WINDOW | CREATE_SUSPENDED, | 195 NULL, NULL, FALSE, CREATE_NO_WINDOW | CREATE_SUSPENDED, |
196 NULL, NULL, &si, &pi)); | 196 NULL, NULL, &si, &pi)); |
197 | 197 |
198 // test Do(). | 198 // test Do(). |
199 { | 199 { |
200 ScopedTempDir temp_dir; | 200 base::ScopedTempDir temp_dir; |
201 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); | 201 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); |
202 | 202 |
203 std::vector<FilePath> key_paths(1, key_path); | 203 std::vector<FilePath> key_paths(1, key_path); |
204 scoped_ptr<DeleteTreeWorkItem> work_item( | 204 scoped_ptr<DeleteTreeWorkItem> work_item( |
205 WorkItem::CreateDeleteTreeWorkItem(dir_name_delete, temp_dir.path(), | 205 WorkItem::CreateDeleteTreeWorkItem(dir_name_delete, temp_dir.path(), |
206 key_paths)); | 206 key_paths)); |
207 | 207 |
208 // delete should fail as file in use. | 208 // delete should fail as file in use. |
209 EXPECT_FALSE(work_item->Do()); | 209 EXPECT_FALSE(work_item->Do()); |
210 } | 210 } |
211 | 211 |
212 // verify everything is still there. | 212 // verify everything is still there. |
213 EXPECT_TRUE(file_util::PathExists(key_path)); | 213 EXPECT_TRUE(file_util::PathExists(key_path)); |
214 EXPECT_TRUE(file_util::PathExists(file_name_delete_1)); | 214 EXPECT_TRUE(file_util::PathExists(file_name_delete_1)); |
215 EXPECT_TRUE(file_util::PathExists(file_name_delete_2)); | 215 EXPECT_TRUE(file_util::PathExists(file_name_delete_2)); |
216 | 216 |
217 TerminateProcess(pi.hProcess, 0); | 217 TerminateProcess(pi.hProcess, 0); |
218 // make sure the handle is closed. | 218 // make sure the handle is closed. |
219 WaitForSingleObject(pi.hProcess, INFINITE); | 219 WaitForSingleObject(pi.hProcess, INFINITE); |
220 } | 220 } |
OLD | NEW |