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/memory_mapped_file.h" | 11 #include "base/files/memory_mapped_file.h" |
12 #include "base/memory/scoped_ptr.h" | 12 #include "base/memory/scoped_ptr.h" |
13 #include "base/path_service.h" | 13 #include "base/path_service.h" |
14 #include "base/strings/string_util.h" | 14 #include "base/strings/string_util.h" |
| 15 #include "base/strings/utf_string_conversions.h" |
15 #include "chrome/installer/util/installer_util_test_common.h" | 16 #include "chrome/installer/util/installer_util_test_common.h" |
16 #include "chrome/installer/util/move_tree_work_item.h" | 17 #include "chrome/installer/util/move_tree_work_item.h" |
17 #include "chrome/installer/util/work_item.h" | 18 #include "chrome/installer/util/work_item.h" |
18 #include "testing/gtest/include/gtest/gtest.h" | 19 #include "testing/gtest/include/gtest/gtest.h" |
19 | 20 |
20 namespace { | 21 namespace { |
21 class MoveTreeWorkItemTest : public testing::Test { | 22 class MoveTreeWorkItemTest : public testing::Test { |
22 protected: | 23 protected: |
23 virtual void SetUp() { | 24 virtual void SetUp() { |
24 ASSERT_TRUE(temp_from_dir_.CreateUniqueTempDir()); | 25 ASSERT_TRUE(temp_from_dir_.CreateUniqueTempDir()); |
25 ASSERT_TRUE(temp_to_dir_.CreateUniqueTempDir()); | 26 ASSERT_TRUE(temp_to_dir_.CreateUniqueTempDir()); |
26 } | 27 } |
27 | 28 |
28 base::ScopedTempDir temp_from_dir_; | 29 base::ScopedTempDir temp_from_dir_; |
29 base::ScopedTempDir temp_to_dir_; | 30 base::ScopedTempDir temp_to_dir_; |
30 }; | 31 }; |
31 | 32 |
32 // Simple function to dump some text into a new file. | 33 // Simple function to dump some text into a new file. |
33 void CreateTextFile(const std::wstring& filename, | 34 void CreateTextFile(const std::wstring& filename, |
34 const std::wstring& contents) { | 35 const std::wstring& contents) { |
35 std::wofstream file; | 36 std::wofstream file; |
36 file.open(WideToASCII(filename).c_str()); | 37 file.open(base::UTF16ToASCII(filename).c_str()); |
37 ASSERT_TRUE(file.is_open()); | 38 ASSERT_TRUE(file.is_open()); |
38 file << contents; | 39 file << contents; |
39 file.close(); | 40 file.close(); |
40 } | 41 } |
41 | 42 |
42 // Simple function to read text from a file. | 43 // Simple function to read text from a file. |
43 std::wstring ReadTextFile(const base::FilePath& path) { | 44 std::wstring ReadTextFile(const base::FilePath& path) { |
44 WCHAR contents[64]; | 45 WCHAR contents[64]; |
45 std::wifstream file; | 46 std::wifstream file; |
46 file.open(WideToASCII(path.value()).c_str()); | 47 file.open(base::UTF16ToASCII(path.value()).c_str()); |
47 EXPECT_TRUE(file.is_open()); | 48 EXPECT_TRUE(file.is_open()); |
48 file.getline(contents, arraysize(contents)); | 49 file.getline(contents, arraysize(contents)); |
49 file.close(); | 50 file.close(); |
50 return std::wstring(contents); | 51 return std::wstring(contents); |
51 } | 52 } |
52 | 53 |
53 const wchar_t kTextContent1[] = L"Gooooooooooooooooooooogle"; | 54 const wchar_t kTextContent1[] = L"Gooooooooooooooooooooogle"; |
54 const wchar_t kTextContent2[] = L"Overwrite Me"; | 55 const wchar_t kTextContent2[] = L"Overwrite Me"; |
55 }; // namespace | 56 }; // namespace |
56 | 57 |
(...skipping 476 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
533 // the source files. | 534 // the source files. |
534 EXPECT_TRUE(base::PathExists(from_dir1)); | 535 EXPECT_TRUE(base::PathExists(from_dir1)); |
535 EXPECT_TRUE(base::PathExists(to_dir)); | 536 EXPECT_TRUE(base::PathExists(to_dir)); |
536 EXPECT_TRUE(base::PathExists(orig_to_file)); | 537 EXPECT_TRUE(base::PathExists(orig_to_file)); |
537 EXPECT_EQ(0, ReadTextFile(orig_to_file).compare(kTextContent1)); | 538 EXPECT_EQ(0, ReadTextFile(orig_to_file).compare(kTextContent1)); |
538 EXPECT_EQ(0, ReadTextFile(from_file).compare(kTextContent1)); | 539 EXPECT_EQ(0, ReadTextFile(from_file).compare(kTextContent1)); |
539 | 540 |
540 // Also, after rollback the new "to" file should be gone. | 541 // Also, after rollback the new "to" file should be gone. |
541 EXPECT_FALSE(base::PathExists(new_to_file2)); | 542 EXPECT_FALSE(base::PathExists(new_to_file2)); |
542 } | 543 } |
OLD | NEW |