Add WOW64 support to the installer registry work items

chrome/installer/util/create_reg_key_work_item_unittest.cc

 // Copyright (c) 2011 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 36 matching lines @@
       key.Create(HKEY_CURRENT_USER, parent_key.value().c_str(), KEY_READ));
 
   base::FilePath top_key_to_create(parent_key);
   top_key_to_create = top_key_to_create.AppendASCII("b");
 
   base::FilePath key_to_create(top_key_to_create);
   key_to_create = key_to_create.AppendASCII("c");
   key_to_create = key_to_create.AppendASCII("d");
 
   scoped_ptr<CreateRegKeyWorkItem> work_item(
-      WorkItem::CreateCreateRegKeyWorkItem(HKEY_CURRENT_USER,
-                                           key_to_create.value()));
+      WorkItem::CreateCreateRegKeyWorkItem(
+          HKEY_CURRENT_USER, key_to_create.value(), WorkItem::kWow64Default));
 
   EXPECT_TRUE(work_item->Do());
 
   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_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) {
   RegKey key;
 
   base::FilePath key_to_create(test_root);
   key_to_create = key_to_create.AppendASCII("aa");
   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,
-                                           key_to_create.value()));
+      WorkItem::CreateCreateRegKeyWorkItem(
+          HKEY_CURRENT_USER, key_to_create.value(), WorkItem::kWow64Default));
 
   EXPECT_TRUE(work_item->Do());
 
   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_EQ(ERROR_SUCCESS,
       key.Open(HKEY_CURRENT_USER, key_to_create.value().c_str(), KEY_READ));
 }
 
 TEST_F(CreateRegKeyWorkItemTest, CreateSharedKey) {
   RegKey key;
   base::FilePath key_to_create_1(test_root);
   key_to_create_1 = key_to_create_1.AppendASCII("aaa");
 
   base::FilePath key_to_create_2(key_to_create_1);
   key_to_create_2 = key_to_create_2.AppendASCII("bbb");
 
   base::FilePath key_to_create_3(key_to_create_2);
   key_to_create_3 = key_to_create_3.AppendASCII("ccc");
 
   scoped_ptr<CreateRegKeyWorkItem> work_item(
-      WorkItem::CreateCreateRegKeyWorkItem(HKEY_CURRENT_USER,
-                                           key_to_create_3.value()));
+      WorkItem::CreateCreateRegKeyWorkItem(
+          HKEY_CURRENT_USER, key_to_create_3.value(), WorkItem::kWow64Default));
 
   EXPECT_TRUE(work_item->Do());
 
   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
   base::FilePath key_to_create_4(key_to_create_2);
   key_to_create_4 = key_to_create_4.AppendASCII("ddd");
   ASSERT_EQ(ERROR_SUCCESS,
@@ skipping 17 matching lines @@
   base::FilePath key_to_create_1(test_root);
   key_to_create_1 = key_to_create_1.AppendASCII("aaaa");
 
   base::FilePath key_to_create_2(key_to_create_1);
   key_to_create_2 = key_to_create_2.AppendASCII("bbbb");
 
   base::FilePath key_to_create_3(key_to_create_2);
   key_to_create_3 = key_to_create_3.AppendASCII("cccc");
 
   scoped_ptr<CreateRegKeyWorkItem> work_item(
-      WorkItem::CreateCreateRegKeyWorkItem(HKEY_CURRENT_USER,
-                                           key_to_create_3.value()));
+      WorkItem::CreateCreateRegKeyWorkItem(
+          HKEY_CURRENT_USER, key_to_create_3.value(), WorkItem::kWow64Default));
 
   EXPECT_TRUE(work_item->Do());
 
   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_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_NE(ERROR_SUCCESS,
       key.Open(HKEY_CURRENT_USER, key_to_create_1.value().c_str(), KEY_READ));
 }
 
 TEST_F(CreateRegKeyWorkItemTest, RollbackWithSetValue) {
   RegKey key;
 
   base::FilePath key_to_create(test_root);
   key_to_create = key_to_create.AppendASCII("aaaaa");
 
   scoped_ptr<CreateRegKeyWorkItem> work_item(
-      WorkItem::CreateCreateRegKeyWorkItem(HKEY_CURRENT_USER,
-                                           key_to_create.value()));
+      WorkItem::CreateCreateRegKeyWorkItem(
+          HKEY_CURRENT_USER, key_to_create.value(), WorkItem::kWow64Default));
 
   EXPECT_TRUE(work_item->Do());
 
   // Write a value under the key we just created.
   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_EQ(ERROR_SUCCESS,
       key.Open(HKEY_CURRENT_USER, key_to_create.value().c_str(), KEY_READ));
 }