Index: chrome/installer/util/work_item_list_unittest.cc |
=================================================================== |
--- chrome/installer/util/work_item_list_unittest.cc (revision 62848) |
+++ chrome/installer/util/work_item_list_unittest.cc (working copy) |
@@ -7,49 +7,53 @@ |
#include "base/base_paths.h" |
#include "base/file_util.h" |
#include "base/path_service.h" |
-#include "base/registry.h" |
#include "base/scoped_ptr.h" |
#include "base/string_util.h" |
+#include "base/win/registry.h" |
#include "chrome/installer/util/work_item.h" |
#include "chrome/installer/util/work_item_list.h" |
#include "testing/gtest/include/gtest/gtest.h" |
+using base::win::RegKey; |
+ |
namespace { |
- wchar_t test_root[] = L"ListList"; |
- wchar_t data_str[] = L"data_111"; |
- class WorkItemListTest : public testing::Test { |
- protected: |
- virtual void SetUp() { |
- // Create a temporary key for testing |
- RegKey key(HKEY_CURRENT_USER, L"", KEY_ALL_ACCESS); |
- key.DeleteKey(test_root); |
- ASSERT_FALSE(key.Open(HKEY_CURRENT_USER, test_root, KEY_READ)); |
- ASSERT_TRUE(key.Create(HKEY_CURRENT_USER, test_root, KEY_READ)); |
+wchar_t test_root[] = L"ListList"; |
+wchar_t data_str[] = L"data_111"; |
- // Create a temp directory for test. |
- ASSERT_TRUE(PathService::Get(base::DIR_TEMP, &test_dir_)); |
- test_dir_ = test_dir_.AppendASCII("WorkItemListTest"); |
- file_util::Delete(test_dir_, true); |
- ASSERT_FALSE(file_util::PathExists(test_dir_)); |
- file_util::CreateDirectoryW(test_dir_); |
- ASSERT_TRUE(file_util::PathExists(test_dir_)); |
- } |
+class WorkItemListTest : public testing::Test { |
+ protected: |
+ virtual void SetUp() { |
+ // Create a temporary key for testing |
+ RegKey key(HKEY_CURRENT_USER, L"", KEY_ALL_ACCESS); |
+ key.DeleteKey(test_root); |
+ ASSERT_FALSE(key.Open(HKEY_CURRENT_USER, test_root, KEY_READ)); |
+ ASSERT_TRUE(key.Create(HKEY_CURRENT_USER, test_root, KEY_READ)); |
- virtual void TearDown() { |
- logging::CloseLogFile(); |
- // Clean up test directory |
- ASSERT_TRUE(file_util::Delete(test_dir_, true)); |
- ASSERT_FALSE(file_util::PathExists(test_dir_)); |
- // Clean up the temporary key |
- RegKey key(HKEY_CURRENT_USER, L"", KEY_ALL_ACCESS); |
- ASSERT_TRUE(key.DeleteKey(test_root)); |
- } |
+ // Create a temp directory for test. |
+ ASSERT_TRUE(PathService::Get(base::DIR_TEMP, &test_dir_)); |
+ test_dir_ = test_dir_.AppendASCII("WorkItemListTest"); |
+ file_util::Delete(test_dir_, true); |
+ ASSERT_FALSE(file_util::PathExists(test_dir_)); |
+ file_util::CreateDirectoryW(test_dir_); |
+ ASSERT_TRUE(file_util::PathExists(test_dir_)); |
+ } |
- FilePath test_dir_; |
- }; |
+ virtual void TearDown() { |
+ logging::CloseLogFile(); |
+ // Clean up test directory |
+ ASSERT_TRUE(file_util::Delete(test_dir_, true)); |
+ ASSERT_FALSE(file_util::PathExists(test_dir_)); |
+ // Clean up the temporary key |
+ RegKey key(HKEY_CURRENT_USER, L"", KEY_ALL_ACCESS); |
+ ASSERT_TRUE(key.DeleteKey(test_root)); |
+ } |
+ |
+ FilePath test_dir_; |
}; |
+} // namespace |
+ |
// Execute a WorkItem list successfully and then rollback. |
TEST_F(WorkItemListTest, ExecutionSuccess) { |
scoped_ptr<WorkItemList> work_item_list(WorkItem::CreateWorkItemList()); |