Index: chrome/installer/util/work_item_list_unittest.cc |
diff --git a/chrome/installer/util/work_item_list_unittest.cc b/chrome/installer/util/work_item_list_unittest.cc |
index 857987f6b0ba270345129ba51a8b072fe144bd34..3e4a77afab5c6af150641a1ff2b2cc2a57386615 100644 |
--- a/chrome/installer/util/work_item_list_unittest.cc |
+++ b/chrome/installer/util/work_item_list_unittest.cc |
@@ -10,6 +10,7 @@ |
#include "base/memory/scoped_ptr.h" |
#include "base/path_service.h" |
#include "base/strings/string_util.h" |
+#include "base/test/test_reg_util_win.h" |
#include "base/win/registry.h" |
#include "chrome/installer/util/conditional_work_item_list.h" |
#include "chrome/installer/util/work_item.h" |
@@ -27,32 +28,22 @@ const wchar_t kName[] = L"name"; |
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(kTestRoot); |
- ASSERT_NE(ERROR_SUCCESS, key.Open(HKEY_CURRENT_USER, kTestRoot, KEY_READ)); |
- ASSERT_EQ(ERROR_SUCCESS, |
- key.Create(HKEY_CURRENT_USER, kTestRoot, KEY_READ)); |
- |
ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); |
+ registry_override_manager_.OverrideRegistry(HKEY_CURRENT_USER, L"root_pit"); |
grt (UTC plus 2)
2014/07/23 22:03:59
this is so much better. thank you for doing the ri
|
} |
virtual void TearDown() { |
logging::CloseLogFile(); |
- |
- // Clean up the temporary key |
- RegKey key(HKEY_CURRENT_USER, L"", KEY_ALL_ACCESS); |
- ASSERT_EQ(ERROR_SUCCESS, key.DeleteKey(kTestRoot)); |
} |
base::ScopedTempDir temp_dir_; |
+ registry_util::RegistryOverrideManager registry_override_manager_; |
}; |
} // namespace |
// Execute a WorkItem list successfully and then rollback. |
-// http://crbug.com/396405 |
-TEST_F(WorkItemListTest, DISABLED_ExecutionSuccess) { |
+TEST_F(WorkItemListTest, ExecutionSuccess) { |
scoped_ptr<WorkItemList> work_item_list(WorkItem::CreateWorkItemList()); |
scoped_ptr<WorkItem> work_item; |
@@ -233,8 +224,7 @@ TEST_F(WorkItemListTest, ConditionalExecutionSuccess) { |
EXPECT_FALSE(base::PathExists(top_dir_to_create)); |
} |
-// http://crbug.com/396405 |
-TEST_F(WorkItemListTest, DISABLED_ConditionalExecutionConditionFailure) { |
+TEST_F(WorkItemListTest, ConditionalExecutionConditionFailure) { |
scoped_ptr<WorkItemList> work_item_list(WorkItem::CreateWorkItemList()); |
scoped_ptr<WorkItem> work_item; |