Settings: Users: Make isCurrentUserOwner async

chrome/browser/chromeos/extensions/users_private/users_private_apitest.cc

 // Copyright 2015 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 <vector>
 
 #include "base/command_line.h"
 #include "base/macros.h"
 #include "base/memory/ptr_util.h"
 #include "base/values.h"
 #include "build/build_config.h"
 #include "chrome/browser/chromeos/extensions/users_private/users_private_delegate.h"
 #include "chrome/browser/chromeos/extensions/users_private/users_private_delegate_factory.h"
 #include "chrome/browser/extensions/api/settings_private/prefs_util.h"
 #include "chrome/browser/extensions/extension_apitest.h"
 #include "chrome/common/extensions/api/users_private.h"
 #include "components/keyed_service/core/keyed_service.h"
 #include "components/prefs/pref_service.h"
 #include "content/public/test/test_utils.h"
 #include "extensions/common/switches.h"
 
 #if defined(OS_CHROMEOS)
+#include "chrome/browser/chromeos/ownership/owner_settings_service_chromeos_factory.h"
 #include "chromeos/chromeos_switches.h"
+#include "components/ownership/mock_owner_key_util.h"
+#include "crypto/rsa_private_key.h"
 #endif
 
 namespace extensions {
 
 namespace {
 
 class TestPrefsUtil : public PrefsUtil {
  public:
   explicit TestPrefsUtil(Profile* profile) : PrefsUtil(profile) {}
 
@@ skipping 65 matching lines @@
 
  private:
   Profile* profile_;  // weak
   std::unique_ptr<TestPrefsUtil> prefs_util_;
 
   DISALLOW_COPY_AND_ASSIGN(TestDelegate);
 };
 
 class UsersPrivateApiTest : public ExtensionApiTest {
  public:
-  UsersPrivateApiTest() {}
+  UsersPrivateApiTest() {
+#if defined(OS_CHROMEOS)
+    // Mock owner key pairs. Note this needs to happen before
+    // OwnerSettingsServiceChromeOS is created.
+    scoped_refptr<ownership::MockOwnerKeyUtil> owner_key_util =
+        new ownership::MockOwnerKeyUtil();
+    owner_key_util->SetPrivateKey(crypto::RSAPrivateKey::Create(512));
+
+    chromeos::OwnerSettingsServiceChromeOSFactory::GetInstance()
+        ->SetOwnerKeyUtilForTesting(owner_key_util);
+#endif
+  }
   ~UsersPrivateApiTest() override {}
 
   static std::unique_ptr<KeyedService> GetUsersPrivateDelegate(
       content::BrowserContext* profile) {
     CHECK(s_test_delegate_);
     return base::WrapUnique(s_test_delegate_);
   }
 
   void SetUpCommandLine(base::CommandLine* command_line) override {
     ExtensionApiTest::SetUpCommandLine(command_line);
@@ skipping 39 matching lines @@
 IN_PROC_BROWSER_TEST_F(UsersPrivateApiTest, AddAndRemoveUsers) {
   EXPECT_TRUE(RunSubtest("addAndRemoveUsers")) << message_;
 }
 
 IN_PROC_BROWSER_TEST_F(UsersPrivateApiTest, IsOwner) {
   EXPECT_TRUE(RunSubtest("isOwner")) << message_;
 }
 #endif
 
 }  // namespace extensions