Index: chrome/installer/util/create_reg_key_work_item_unittest.cc |
=================================================================== |
--- chrome/installer/util/create_reg_key_work_item_unittest.cc (revision 71561) |
+++ chrome/installer/util/create_reg_key_work_item_unittest.cc (working copy) |
@@ -24,14 +24,15 @@ |
// 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)); |
+ ASSERT_NE(ERROR_SUCCESS, key.Open(HKEY_CURRENT_USER, test_root, KEY_READ)); |
+ ASSERT_EQ(ERROR_SUCCESS, key.Create(HKEY_CURRENT_USER, test_root, |
+ KEY_READ)); |
} |
virtual void TearDown() { |
logging::CloseLogFile(); |
// Clean up the temporary key |
RegKey key(HKEY_CURRENT_USER, L"", KEY_ALL_ACCESS); |
- ASSERT_TRUE(key.DeleteKey(test_root)); |
+ ASSERT_EQ(ERROR_SUCCESS, key.DeleteKey(test_root)); |
} |
}; |
@@ -42,8 +43,8 @@ |
FilePath parent_key(test_root); |
parent_key = parent_key.AppendASCII("a"); |
- ASSERT_TRUE(key.Create(HKEY_CURRENT_USER, parent_key.value().c_str(), |
- KEY_READ)); |
+ ASSERT_EQ(ERROR_SUCCESS, |
+ key.Create(HKEY_CURRENT_USER, parent_key.value().c_str(), KEY_READ)); |
FilePath top_key_to_create(parent_key); |
top_key_to_create = top_key_to_create.AppendASCII("b"); |
@@ -58,16 +59,16 @@ |
EXPECT_TRUE(work_item->Do()); |
- EXPECT_TRUE(key.Open(HKEY_CURRENT_USER, key_to_create.value().c_str(), |
- KEY_READ)); |
+ EXPECT_EQ(ERROR_SUCCESS, |
+ key.Open(HKEY_CURRENT_USER, key_to_create.value().c_str(), KEY_READ)); |
work_item->Rollback(); |
// Rollback should delete all the keys up to top_key_to_create. |
- EXPECT_FALSE(key.Open(HKEY_CURRENT_USER, top_key_to_create.value().c_str(), |
- KEY_READ)); |
- EXPECT_TRUE(key.Open(HKEY_CURRENT_USER, parent_key.value().c_str(), |
- KEY_READ)); |
+ EXPECT_NE(ERROR_SUCCESS, |
+ key.Open(HKEY_CURRENT_USER, top_key_to_create.value().c_str(), KEY_READ)); |
+ EXPECT_EQ(ERROR_SUCCESS, |
+ key.Open(HKEY_CURRENT_USER, parent_key.value().c_str(), KEY_READ)); |
} |
TEST_F(CreateRegKeyWorkItemTest, CreateExistingKey) { |
@@ -75,8 +76,8 @@ |
FilePath key_to_create(test_root); |
key_to_create = key_to_create.AppendASCII("aa"); |
- ASSERT_TRUE(key.Create(HKEY_CURRENT_USER, key_to_create.value().c_str(), |
- KEY_READ)); |
+ ASSERT_EQ(ERROR_SUCCESS, |
+ key.Create(HKEY_CURRENT_USER, key_to_create.value().c_str(), KEY_READ)); |
scoped_ptr<CreateRegKeyWorkItem> work_item( |
WorkItem::CreateCreateRegKeyWorkItem(HKEY_CURRENT_USER, |
@@ -84,15 +85,15 @@ |
EXPECT_TRUE(work_item->Do()); |
- EXPECT_TRUE(key.Open(HKEY_CURRENT_USER, key_to_create.value().c_str(), |
- KEY_READ)); |
+ EXPECT_EQ(ERROR_SUCCESS, |
+ key.Open(HKEY_CURRENT_USER, key_to_create.value().c_str(), KEY_READ)); |
work_item->Rollback(); |
// Rollback should not remove the key since it exists before |
// the CreateRegKeyWorkItem is called. |
- EXPECT_TRUE(key.Open(HKEY_CURRENT_USER, key_to_create.value().c_str(), |
- KEY_READ)); |
+ EXPECT_EQ(ERROR_SUCCESS, |
+ key.Open(HKEY_CURRENT_USER, key_to_create.value().c_str(), KEY_READ)); |
} |
TEST_F(CreateRegKeyWorkItemTest, CreateSharedKey) { |
@@ -112,26 +113,26 @@ |
EXPECT_TRUE(work_item->Do()); |
- EXPECT_TRUE(key.Open(HKEY_CURRENT_USER, key_to_create_3.value().c_str(), |
- KEY_READ)); |
+ EXPECT_EQ(ERROR_SUCCESS, |
+ key.Open(HKEY_CURRENT_USER, key_to_create_3.value().c_str(), KEY_READ)); |
// Create another key under key_to_create_2 |
FilePath key_to_create_4(key_to_create_2); |
key_to_create_4 = key_to_create_4.AppendASCII("ddd"); |
- ASSERT_TRUE(key.Create(HKEY_CURRENT_USER, key_to_create_4.value().c_str(), |
- KEY_READ)); |
+ ASSERT_EQ(ERROR_SUCCESS, |
+ key.Create(HKEY_CURRENT_USER, key_to_create_4.value().c_str(), KEY_READ)); |
work_item->Rollback(); |
// Rollback should delete key_to_create_3. |
- EXPECT_FALSE(key.Open(HKEY_CURRENT_USER, key_to_create_3.value().c_str(), |
- KEY_READ)); |
+ EXPECT_NE(ERROR_SUCCESS, |
+ key.Open(HKEY_CURRENT_USER, key_to_create_3.value().c_str(), KEY_READ)); |
// Rollback should not delete key_to_create_2 as it is shared. |
- EXPECT_TRUE(key.Open(HKEY_CURRENT_USER, key_to_create_2.value().c_str(), |
- KEY_READ)); |
- EXPECT_TRUE(key.Open(HKEY_CURRENT_USER, key_to_create_4.value().c_str(), |
- KEY_READ)); |
+ EXPECT_EQ(ERROR_SUCCESS, |
+ key.Open(HKEY_CURRENT_USER, key_to_create_2.value().c_str(), KEY_READ)); |
+ EXPECT_EQ(ERROR_SUCCESS, |
+ key.Open(HKEY_CURRENT_USER, key_to_create_4.value().c_str(), KEY_READ)); |
} |
TEST_F(CreateRegKeyWorkItemTest, RollbackWithMissingKey) { |
@@ -151,22 +152,22 @@ |
EXPECT_TRUE(work_item->Do()); |
- EXPECT_TRUE(key.Open(HKEY_CURRENT_USER, key_to_create_3.value().c_str(), |
- KEY_READ)); |
+ EXPECT_EQ(ERROR_SUCCESS, |
+ key.Open(HKEY_CURRENT_USER, key_to_create_3.value().c_str(), KEY_READ)); |
key.Close(); |
// now delete key_to_create_3 |
- ASSERT_TRUE(RegDeleteKey(HKEY_CURRENT_USER, |
- key_to_create_3.value().c_str()) == ERROR_SUCCESS); |
- ASSERT_FALSE(key.Open(HKEY_CURRENT_USER, key_to_create_3.value().c_str(), |
- KEY_READ)); |
+ ASSERT_EQ(ERROR_SUCCESS, |
+ RegDeleteKey(HKEY_CURRENT_USER, key_to_create_3.value().c_str())); |
+ ASSERT_NE(ERROR_SUCCESS, |
+ key.Open(HKEY_CURRENT_USER, key_to_create_3.value().c_str(), KEY_READ)); |
work_item->Rollback(); |
// key_to_create_3 has already been deleted, Rollback should delete |
// the rest. |
- ASSERT_FALSE(key.Open(HKEY_CURRENT_USER, key_to_create_1.value().c_str(), |
- KEY_READ)); |
+ ASSERT_NE(ERROR_SUCCESS, |
+ key.Open(HKEY_CURRENT_USER, key_to_create_1.value().c_str(), KEY_READ)); |
} |
TEST_F(CreateRegKeyWorkItemTest, RollbackWithSetValue) { |
@@ -182,14 +183,14 @@ |
EXPECT_TRUE(work_item->Do()); |
// Write a value under the key we just created. |
- EXPECT_TRUE(key.Open(HKEY_CURRENT_USER, key_to_create.value().c_str(), |
- KEY_READ | KEY_SET_VALUE)); |
- EXPECT_TRUE(key.WriteValue(L"name", L"value")); |
+ EXPECT_EQ(ERROR_SUCCESS, key.Open(HKEY_CURRENT_USER, |
+ key_to_create.value().c_str(), KEY_READ | KEY_SET_VALUE)); |
+ EXPECT_EQ(ERROR_SUCCESS, key.WriteValue(L"name", L"value")); |
key.Close(); |
work_item->Rollback(); |
// Rollback should not remove the key. |
- EXPECT_TRUE(key.Open(HKEY_CURRENT_USER, key_to_create.value().c_str(), |
- KEY_READ)); |
+ EXPECT_EQ(ERROR_SUCCESS, |
+ key.Open(HKEY_CURRENT_USER, key_to_create.value().c_str(), KEY_READ)); |
} |