The chrome.lockScreen.data API implementation

extensions/browser/test_extensions_browser_client.h

 // Copyright 2014 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.
 
 #ifndef EXTENSIONS_BROWSER_TEST_EXTENSIONS_BROWSER_CLIENT_H_
 #define EXTENSIONS_BROWSER_TEST_EXTENSIONS_BROWSER_CLIENT_H_
 
+#include <memory>
 #include <string>
 #include <utility>
 #include <vector>
 
 #include "base/callback.h"
 #include "base/compiler_specific.h"
 #include "base/macros.h"
 #include "base/memory/ref_counted.h"
 #include "build/build_config.h"
 #include "components/update_client/update_client.h"
@@ skipping 15 matching lines @@
   void set_process_manager_delegate(ProcessManagerDelegate* delegate) {
     process_manager_delegate_ = delegate;
   }
   void set_extension_system_factory(ExtensionSystemProvider* factory) {
     extension_system_factory_ = factory;
   }
   void set_extension_cache(std::unique_ptr<ExtensionCache> extension_cache) {
     extension_cache_ = std::move(extension_cache);
   }
 
+  void set_lock_screen_context(content::BrowserContext* context) {
+    lock_screen_context_ = context;
+  }
+
   // Sets a factory to respond to calls of the CreateUpdateClient method.
   void SetUpdateClientFactory(
       const base::Callback<update_client::UpdateClient*(void)>& factory);
 
   // Associates an incognito context with |main_context_|.
   void SetIncognitoContext(content::BrowserContext* incognito_context);
 
   // ExtensionsBrowserClient overrides:
   bool IsShuttingDown() override;
   bool AreExtensionsDisabled(const base::CommandLine& command_line,
@@ skipping 60 matching lines @@
   KioskDelegate* GetKioskDelegate() override;
   scoped_refptr<update_client::UpdateClient> CreateUpdateClient(
       content::BrowserContext* context) override;
   bool IsLockScreenContext(content::BrowserContext* context) override;
 
   ExtensionSystemProvider* extension_system_factory() {
     return extension_system_factory_;
   }
 
  private:
-  content::BrowserContext* main_context_;       // Not owned.
-  content::BrowserContext* incognito_context_;  // Not owned, defaults to NULL.
+  // Not owned.
+  content::BrowserContext* main_context_;
+  // Not owned, defaults to nullptr.
+  content::BrowserContext* incognito_context_;
+  // Not owned, defaults to nullptr.
+  content::BrowserContext* lock_screen_context_;
 
-  // Not owned, defaults to NULL.
+  // Not owned, defaults to nullptr.
   ProcessManagerDelegate* process_manager_delegate_;
 
-  // Not owned, defaults to NULL.
+  // Not owned, defaults to nullptr.
   ExtensionSystemProvider* extension_system_factory_;
 
   std::unique_ptr<ExtensionCache> extension_cache_;
 
   base::Callback<update_client::UpdateClient*(void)> update_client_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(TestExtensionsBrowserClient);
 };
 
 }  // namespace extensions
 
 #endif  // EXTENSIONS_BROWSER_TEST_EXTENSIONS_BROWSER_CLIENT_H_