Fix system-level detection in the mini_installer.

chrome/installer/mini_installer/configuration_test.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 "chrome/installer/mini_installer/configuration.h"
 
 #include <stddef.h>
 #include <stdlib.h>
 
+#include <memory>
+
+#include "base/environment.h"
 #include "chrome/installer/mini_installer/appid.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using mini_installer::Configuration;
 
+namespace {
+
+// A helper class to set the "GoogleUpdateIsMachine" environment variable.
+class ScopedGoogleUpdateIsMachine {
+ public:
+  explicit ScopedGoogleUpdateIsMachine(bool value)
+      : env_(base::Environment::Create()) {
+    env_->SetVar("GoogleUpdateIsMachine", value ? "1" : "0");
+  }
+
+  ~ScopedGoogleUpdateIsMachine() {
+    env_->UnSetVar("GoogleUpdateIsMachine");
+  }
+
+ private:
+  std::unique_ptr<base::Environment> env_;
+};
+
+}  // namespace
+
 class TestConfiguration : public Configuration {
  public:
   explicit TestConfiguration(const wchar_t* command_line)
       : Configuration(),
         open_registry_key_result_(false),
         read_registry_value_result_(0),
         read_registry_value_(L"") {
     Initialize(command_line);
   }
   explicit TestConfiguration(const wchar_t* command_line,
@@ skipping 128 matching lines @@
   EXPECT_TRUE(TestConfiguration(L"spam.exe --multi-install --chrome-frame")
                   .is_multi_install());
   EXPECT_TRUE(TestConfiguration(L"spam.exe --multi-install")
                   .is_multi_install());
 }
 
 TEST(MiniInstallerConfigurationTest, IsSystemLevel) {
   EXPECT_FALSE(TestConfiguration(L"spam.exe").is_system_level());
   EXPECT_FALSE(TestConfiguration(L"spam.exe --chrome").is_system_level());
   EXPECT_TRUE(TestConfiguration(L"spam.exe --system-level").is_system_level());
+
+  {
+    ScopedGoogleUpdateIsMachine env_setter(false);
+    EXPECT_FALSE(TestConfiguration(L"spam.exe").is_system_level());
+  }
+
+  {
+    ScopedGoogleUpdateIsMachine env_setter(true);
+    EXPECT_TRUE(TestConfiguration(L"spam.exe").is_system_level());
+  }
 }