| 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 |