Add WOW64 support to the installer registry work items

chrome/installer/util/set_reg_value_work_item_unittest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include <windows.h>
 
 #include "base/files/file_path.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/strings/string_util.h"
 #include "base/win/registry.h"
@@ skipping 40 matching lines @@
 
   std::wstring parent_key(kTestRoot);
   parent_key.append(&base::FilePath::kSeparators[0], 1);
   parent_key.append(L"WriteNewNonOverwrite");
   ASSERT_EQ(ERROR_SUCCESS,
       key.Create(HKEY_CURRENT_USER, parent_key.c_str(), KEY_READ));
 
   std::wstring name_str(kNameStr);
   std::wstring data_str(kDataStr1);
   scoped_ptr<SetRegValueWorkItem> work_item1(
-      WorkItem::CreateSetRegValueWorkItem(HKEY_CURRENT_USER, parent_key,
-                                          name_str, data_str, false));
+      WorkItem::CreateSetRegValueWorkItem(HKEY_CURRENT_USER,
+                                          parent_key,
+                                          WorkItem::kWow64Default,
+                                          name_str,
+                                          data_str,
+                                          false));
 
   std::wstring name_dword(kNameDword);
   scoped_ptr<SetRegValueWorkItem> work_item2(
-      WorkItem::CreateSetRegValueWorkItem(HKEY_CURRENT_USER, parent_key,
-                                          name_dword, dword1, false));
+      WorkItem::CreateSetRegValueWorkItem(HKEY_CURRENT_USER,
+                                          parent_key,
+                                          WorkItem::kWow64Default,
+                                          name_dword,
+                                          dword1,
+                                          false));
 
   EXPECT_TRUE(work_item1->Do());
   EXPECT_TRUE(work_item2->Do());
 
   std::wstring read_out;
   DWORD 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, kDataStr1);
   EXPECT_EQ(read_dword, dword1);
@@ skipping 12 matching lines @@
 
   std::wstring parent_key(kTestRoot);
   parent_key.append(&base::FilePath::kSeparators[0], 1);
   parent_key.append(L"WriteNewOverwrite");
   ASSERT_EQ(ERROR_SUCCESS,
       key.Create(HKEY_CURRENT_USER, parent_key.c_str(), KEY_READ));
 
   std::wstring name_str(kNameStr);
   std::wstring data_str(kDataStr1);
   scoped_ptr<SetRegValueWorkItem> work_item1(
-      WorkItem::CreateSetRegValueWorkItem(HKEY_CURRENT_USER, parent_key,
-                                          name_str, data_str, true));
+      WorkItem::CreateSetRegValueWorkItem(HKEY_CURRENT_USER,
+                                          parent_key,
+                                          WorkItem::kWow64Default,
+                                          name_str,
+                                          data_str,
+                                          true));
 
   std::wstring name_dword(kNameDword);
   scoped_ptr<SetRegValueWorkItem> work_item2(
-      WorkItem::CreateSetRegValueWorkItem(HKEY_CURRENT_USER, parent_key,
-                                          name_dword, dword1, true));
+      WorkItem::CreateSetRegValueWorkItem(HKEY_CURRENT_USER,
+                                          parent_key,
+                                          WorkItem::kWow64Default,
+                                          name_dword,
+                                          dword1,
+                                          true));
 
   EXPECT_TRUE(work_item1->Do());
   EXPECT_TRUE(work_item2->Do());
 
   std::wstring read_out;
   DWORD 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, kDataStr1);
   EXPECT_EQ(read_dword, dword1);
@@ skipping 18 matching lines @@
       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(kNameStr);
   ASSERT_EQ(ERROR_SUCCESS, key.WriteValue(name.c_str(), kDataStr1));
 
   std::wstring data(kDataStr2);
   scoped_ptr<SetRegValueWorkItem> work_item(
-      WorkItem::CreateSetRegValueWorkItem(HKEY_CURRENT_USER, parent_key,
-                                          name, data, false));
+      WorkItem::CreateSetRegValueWorkItem(HKEY_CURRENT_USER,
+                                          parent_key,
+                                          WorkItem::kWow64Default,
+                                          name,
+                                          data,
+                                          false));
   EXPECT_TRUE(work_item->Do());
 
   std::wstring read_out;
   EXPECT_EQ(ERROR_SUCCESS, key.ReadValue(name.c_str(), &read_out));
   EXPECT_EQ(0, read_out.compare(kDataStr1));
 
   work_item->Rollback();
   EXPECT_TRUE(key.HasValue(name.c_str()));
   EXPECT_EQ(ERROR_SUCCESS, key.ReadValue(name.c_str(), &read_out));
   EXPECT_EQ(read_out, kDataStr1);
 
   // Now test REG_DWORD value.
   // Write data to the value we are going to set.
   name.assign(kNameDword);
   ASSERT_EQ(ERROR_SUCCESS, key.WriteValue(name.c_str(), dword1));
   work_item.reset(WorkItem::CreateSetRegValueWorkItem(HKEY_CURRENT_USER,
-      parent_key, name, dword2, false));
+                                                      parent_key,
+                                                      WorkItem::kWow64Default,
+                                                      name,
+                                                      dword2,
+                                                      false));
   EXPECT_TRUE(work_item->Do());
 
   DWORD read_dword;
   EXPECT_EQ(ERROR_SUCCESS, key.ReadValueDW(name.c_str(), &read_dword));
   EXPECT_EQ(read_dword, dword1);
 
   work_item->Rollback();
   EXPECT_TRUE(key.HasValue(name.c_str()));
   EXPECT_EQ(ERROR_SUCCESS, key.ReadValueDW(name.c_str(), &read_dword));
   EXPECT_EQ(read_dword, dword1);
@@ skipping 15 matching lines @@
   // Write data to the value we are going to set.
   std::wstring name(kNameStr);
   ASSERT_EQ(ERROR_SUCCESS, key.WriteValue(name.c_str(), kDataStr1));
 
   std::wstring name_empty(L"name_empty");
   ASSERT_EQ(ERROR_SUCCESS, RegSetValueEx(key.Handle(), name_empty.c_str(), NULL,
                                          REG_SZ, NULL, 0));
 
   std::wstring data(kDataStr2);
   scoped_ptr<SetRegValueWorkItem> work_item1(
-      WorkItem::CreateSetRegValueWorkItem(HKEY_CURRENT_USER, parent_key,
-                                          name, data, true));
+      WorkItem::CreateSetRegValueWorkItem(HKEY_CURRENT_USER,
+                                          parent_key,
+                                          WorkItem::kWow64Default,
+                                          name,
+                                          data,
+                                          true));
   scoped_ptr<SetRegValueWorkItem> work_item2(
-      WorkItem::CreateSetRegValueWorkItem(HKEY_CURRENT_USER, parent_key,
-                                          name_empty, data, true));
+      WorkItem::CreateSetRegValueWorkItem(HKEY_CURRENT_USER,
+                                          parent_key,
+                                          WorkItem::kWow64Default,
+                                          name_empty,
+                                          data,
+                                          true));
 
   EXPECT_TRUE(work_item1->Do());
   EXPECT_TRUE(work_item2->Do());
 
   std::wstring read_out;
   EXPECT_EQ(ERROR_SUCCESS, key.ReadValue(name.c_str(), &read_out));
   EXPECT_EQ(0, read_out.compare(kDataStr2));
 
   EXPECT_EQ(ERROR_SUCCESS, key.ReadValue(name_empty.c_str(), &read_out));
   EXPECT_EQ(0, read_out.compare(kDataStr2));
@@ skipping 10 matching lines @@
   EXPECT_EQ(ERROR_SUCCESS, key.ReadValue(name_empty.c_str(), NULL, &size,
                                          &type));
   EXPECT_EQ(REG_SZ, type);
   EXPECT_EQ(0, size);
 
   // Now test REG_DWORD value.
   // Write data to the value we are going to set.
   name.assign(kNameDword);
   ASSERT_EQ(ERROR_SUCCESS, key.WriteValue(name.c_str(), dword1));
   scoped_ptr<SetRegValueWorkItem> work_item3(
-      WorkItem::CreateSetRegValueWorkItem(HKEY_CURRENT_USER, parent_key, name,
-                                          dword2, true));
+      WorkItem::CreateSetRegValueWorkItem(HKEY_CURRENT_USER,
+                                          parent_key,
+                                          WorkItem::kWow64Default,
+                                          name,
+                                          dword2,
+                                          true));
   EXPECT_TRUE(work_item3->Do());
 
   DWORD read_dword;
   EXPECT_EQ(ERROR_SUCCESS, key.ReadValueDW(name.c_str(), &read_dword));
   EXPECT_EQ(read_dword, dword2);
 
   work_item3->Rollback();
   EXPECT_TRUE(key.HasValue(name.c_str()));
   EXPECT_EQ(ERROR_SUCCESS, key.ReadValueDW(name.c_str(), &read_dword));
   EXPECT_EQ(read_dword, dword1);
 }
 
 // Write a value to a non-existing key. This should fail.
 TEST_F(SetRegValueWorkItemTest, WriteNonExistingKey) {
   RegKey key;
 
   std::wstring parent_key(kTestRoot);
   parent_key.append(&base::FilePath::kSeparators[0], 1);
   parent_key.append(L"WriteNonExistingKey");
 
   std::wstring name(L"name");
   std::wstring data(kDataStr1);
   scoped_ptr<SetRegValueWorkItem> work_item(
-      WorkItem::CreateSetRegValueWorkItem(HKEY_CURRENT_USER, parent_key,
-                                          name, data, false));
+      WorkItem::CreateSetRegValueWorkItem(HKEY_CURRENT_USER,
+                                          parent_key,
+                                          WorkItem::kWow64Default,
+                                          name,
+                                          data,
+                                          false));
   EXPECT_FALSE(work_item->Do());
 
   work_item.reset(WorkItem::CreateSetRegValueWorkItem(HKEY_CURRENT_USER,
-      parent_key, name, dword1, false));
+                                                      parent_key,
+                                                      WorkItem::kWow64Default,
+                                                      name,
+                                                      dword1,
+                                                      false));
   EXPECT_FALSE(work_item->Do());
 }