| Index: chrome/installer/util/set_reg_value_work_item_unittest.cc
|
| ===================================================================
|
| --- chrome/installer/util/set_reg_value_work_item_unittest.cc (revision 70917)
|
| +++ chrome/installer/util/set_reg_value_work_item_unittest.cc (working copy)
|
| @@ -27,14 +27,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));
|
| }
|
| };
|
|
|
| @@ -46,7 +47,8 @@
|
|
|
| std::wstring parent_key(test_root);
|
| file_util::AppendToPath(&parent_key, L"WriteNewNonOverwrite");
|
| - ASSERT_TRUE(key.Create(HKEY_CURRENT_USER, parent_key.c_str(), KEY_READ));
|
| + ASSERT_EQ(ERROR_SUCCESS,
|
| + key.Create(HKEY_CURRENT_USER, parent_key.c_str(), KEY_READ));
|
|
|
| std::wstring name_str(L"name_str");
|
| std::wstring data_str(data_str_1);
|
| @@ -64,8 +66,8 @@
|
|
|
| std::wstring read_out;
|
| DWORD read_dword;
|
| - EXPECT_TRUE(key.ReadValue(name_str.c_str(), &read_out));
|
| - EXPECT_TRUE(key.ReadValueDW(name_dword.c_str(), &read_dword));
|
| + EXPECT_EQ(ERROR_SUCCESS, key.ReadValue(name_str.c_str(), &read_out));
|
| + EXPECT_EQ(ERROR_SUCCESS, key.ReadValueDW(name_dword.c_str(), &read_dword));
|
| EXPECT_EQ(read_out, data_str_1);
|
| EXPECT_EQ(read_dword, dword1);
|
|
|
| @@ -83,7 +85,8 @@
|
|
|
| std::wstring parent_key(test_root);
|
| file_util::AppendToPath(&parent_key, L"WriteNewOverwrite");
|
| - ASSERT_TRUE(key.Create(HKEY_CURRENT_USER, parent_key.c_str(), KEY_READ));
|
| + ASSERT_EQ(ERROR_SUCCESS,
|
| + key.Create(HKEY_CURRENT_USER, parent_key.c_str(), KEY_READ));
|
|
|
| std::wstring name_str(L"name_str");
|
| std::wstring data_str(data_str_1);
|
| @@ -101,8 +104,8 @@
|
|
|
| std::wstring read_out;
|
| DWORD read_dword;
|
| - EXPECT_TRUE(key.ReadValue(name_str.c_str(), &read_out));
|
| - EXPECT_TRUE(key.ReadValueDW(name_dword.c_str(), &read_dword));
|
| + EXPECT_EQ(ERROR_SUCCESS, key.ReadValue(name_str.c_str(), &read_out));
|
| + EXPECT_EQ(ERROR_SUCCESS, key.ReadValueDW(name_dword.c_str(), &read_dword));
|
| EXPECT_EQ(read_out, data_str_1);
|
| EXPECT_EQ(read_dword, dword1);
|
|
|
| @@ -121,13 +124,14 @@
|
|
|
| std::wstring parent_key(test_root);
|
| file_util::AppendToPath(&parent_key, L"WriteExistingNonOverwrite");
|
| - ASSERT_TRUE(key.Create(HKEY_CURRENT_USER, parent_key.c_str(),
|
| - KEY_READ | KEY_SET_VALUE));
|
| + ASSERT_EQ(ERROR_SUCCESS,
|
| + key.Create(HKEY_CURRENT_USER, parent_key.c_str(),
|
| + KEY_READ | KEY_SET_VALUE));
|
|
|
| // First test REG_SZ value.
|
| // Write data to the value we are going to set.
|
| std::wstring name(L"name_str");
|
| - ASSERT_TRUE(key.WriteValue(name.c_str(), data_str_1));
|
| + ASSERT_EQ(ERROR_SUCCESS, key.WriteValue(name.c_str(), data_str_1));
|
|
|
| std::wstring data(data_str_2);
|
| scoped_ptr<SetRegValueWorkItem> work_item(
|
| @@ -136,29 +140,29 @@
|
| EXPECT_TRUE(work_item->Do());
|
|
|
| std::wstring read_out;
|
| - EXPECT_TRUE(key.ReadValue(name.c_str(), &read_out));
|
| + EXPECT_EQ(ERROR_SUCCESS, key.ReadValue(name.c_str(), &read_out));
|
| EXPECT_EQ(0, read_out.compare(data_str_1));
|
|
|
| work_item->Rollback();
|
| EXPECT_TRUE(key.ValueExists(name.c_str()));
|
| - EXPECT_TRUE(key.ReadValue(name.c_str(), &read_out));
|
| + EXPECT_EQ(ERROR_SUCCESS, key.ReadValue(name.c_str(), &read_out));
|
| EXPECT_EQ(read_out, data_str_1);
|
|
|
| // Now test REG_DWORD value.
|
| // Write data to the value we are going to set.
|
| name.assign(L"name_dword");
|
| - ASSERT_TRUE(key.WriteValue(name.c_str(), dword1));
|
| + ASSERT_EQ(ERROR_SUCCESS, key.WriteValue(name.c_str(), dword1));
|
| work_item.reset(WorkItem::CreateSetRegValueWorkItem(HKEY_CURRENT_USER,
|
| parent_key, name, dword2, false));
|
| EXPECT_TRUE(work_item->Do());
|
|
|
| DWORD read_dword;
|
| - EXPECT_TRUE(key.ReadValueDW(name.c_str(), &read_dword));
|
| + EXPECT_EQ(ERROR_SUCCESS, key.ReadValueDW(name.c_str(), &read_dword));
|
| EXPECT_EQ(read_dword, dword1);
|
|
|
| work_item->Rollback();
|
| EXPECT_TRUE(key.ValueExists(name.c_str()));
|
| - EXPECT_TRUE(key.ReadValueDW(name.c_str(), &read_dword));
|
| + EXPECT_EQ(ERROR_SUCCESS, key.ReadValueDW(name.c_str(), &read_dword));
|
| EXPECT_EQ(read_dword, dword1);
|
| }
|
|
|
| @@ -169,13 +173,14 @@
|
|
|
| std::wstring parent_key(test_root);
|
| file_util::AppendToPath(&parent_key, L"WriteExistingOverwrite");
|
| - ASSERT_TRUE(key.Create(HKEY_CURRENT_USER, parent_key.c_str(),
|
| - KEY_READ | KEY_SET_VALUE));
|
| + ASSERT_EQ(ERROR_SUCCESS,
|
| + key.Create(HKEY_CURRENT_USER, parent_key.c_str(),
|
| + KEY_READ | KEY_SET_VALUE));
|
|
|
| // First test REG_SZ value.
|
| // Write data to the value we are going to set.
|
| std::wstring name(L"name_str");
|
| - ASSERT_TRUE(key.WriteValue(name.c_str(), data_str_1));
|
| + ASSERT_EQ(ERROR_SUCCESS, key.WriteValue(name.c_str(), data_str_1));
|
|
|
| std::wstring data(data_str_2);
|
| scoped_ptr<SetRegValueWorkItem> work_item(
|
| @@ -184,29 +189,29 @@
|
| EXPECT_TRUE(work_item->Do());
|
|
|
| std::wstring read_out;
|
| - EXPECT_TRUE(key.ReadValue(name.c_str(), &read_out));
|
| + EXPECT_EQ(ERROR_SUCCESS, key.ReadValue(name.c_str(), &read_out));
|
| EXPECT_EQ(0, read_out.compare(data_str_2));
|
|
|
| work_item->Rollback();
|
| EXPECT_TRUE(key.ValueExists(name.c_str()));
|
| - EXPECT_TRUE(key.ReadValue(name.c_str(), &read_out));
|
| + EXPECT_EQ(ERROR_SUCCESS, key.ReadValue(name.c_str(), &read_out));
|
| EXPECT_EQ(read_out, data_str_1);
|
|
|
| // Now test REG_DWORD value.
|
| // Write data to the value we are going to set.
|
| name.assign(L"name_dword");
|
| - ASSERT_TRUE(key.WriteValue(name.c_str(), dword1));
|
| + ASSERT_EQ(ERROR_SUCCESS, key.WriteValue(name.c_str(), dword1));
|
| work_item.reset(WorkItem::CreateSetRegValueWorkItem(HKEY_CURRENT_USER,
|
| parent_key, name, dword2, true));
|
| EXPECT_TRUE(work_item->Do());
|
|
|
| DWORD read_dword;
|
| - EXPECT_TRUE(key.ReadValueDW(name.c_str(), &read_dword));
|
| + EXPECT_EQ(ERROR_SUCCESS, key.ReadValueDW(name.c_str(), &read_dword));
|
| EXPECT_EQ(read_dword, dword2);
|
|
|
| work_item->Rollback();
|
| EXPECT_TRUE(key.ValueExists(name.c_str()));
|
| - EXPECT_TRUE(key.ReadValueDW(name.c_str(), &read_dword));
|
| + EXPECT_EQ(ERROR_SUCCESS, key.ReadValueDW(name.c_str(), &read_dword));
|
| EXPECT_EQ(read_dword, dword1);
|
| }
|
|
|
|
|