src/chrome/browser/ui/cocoa/extensions/extension_install_prompt_test_utils.h

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef CHROME_BROWSER_UI_COCOA_EXTENSIONS_EXTENSION_INSTALL_PROMPT_TEST_UTILS_H_
   6 #define CHROME_BROWSER_UI_COCOA_EXTENSIONS_EXTENSION_INSTALL_PROMPT_TEST_UTILS_H_
   7
   8 #include "chrome/browser/extensions/extension_install_prompt.h"
   9
  10 namespace chrome {
  11
  12 // A simple delegate implementation that counts the number of times
  13 // |InstallUIProceed| and |InstallUIAbort| are called.
  14 class MockExtensionInstallPromptDelegate
  15     : public ExtensionInstallPrompt::Delegate {
  16  public:
  17   MockExtensionInstallPromptDelegate()
  18       : proceed_count_(0),
  19         abort_count_(0) {}
  20
  21   // ExtensionInstallPrompt::Delegate overrides.
  22   virtual void InstallUIProceed() OVERRIDE;
  23   virtual void InstallUIAbort(bool user_initiated) OVERRIDE;
  24
  25   int proceed_count() { return proceed_count_; }
  26   int abort_count() { return abort_count_; }
  27
  28  protected:
  29   int proceed_count_;
  30   int abort_count_;
  31 };
  32
  33 // Loads the install prompt test extension.
  34 scoped_refptr<extensions::Extension> LoadInstallPromptExtension();
  35
  36 // Loads the icon for the install prompt extension.
  37 gfx::Image LoadInstallPromptIcon();
  38
  39 // Builds a prompt using the given extension.
  40 ExtensionInstallPrompt::Prompt BuildExtensionInstallPrompt(
  41     extensions::Extension* extension);
  42
  43 ExtensionInstallPrompt::Prompt BuildExtensionPostInstallPermissionsPrompt(
  44     extensions::Extension* extension);
  45
  46 }  // namespace chrome
  47
  48 #endif  // CHROME_BROWSER_UI_COCOA_EXTENSIONS_EXTENSION_INSTALL_PROMPT_TEST_UTILS_H_