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 test extension from the given test directory and manifest file.
  34 scoped_refptr<extensions::Extension> LoadInstallPromptExtension(
  35     const char* extension_dir_name,
  36     const char* manifest_file);
  37
  38 // Loads the default install_prompt test extension.
  39 scoped_refptr<extensions::Extension> LoadInstallPromptExtension();
  40
  41 // Loads the icon for the install prompt extension.
  42 gfx::Image LoadInstallPromptIcon();
  43
  44 // Builds a prompt using the given extension.
  45 ExtensionInstallPrompt::Prompt BuildExtensionInstallPrompt(
  46     extensions::Extension* extension);
  47
  48 ExtensionInstallPrompt::Prompt BuildExtensionPostInstallPermissionsPrompt(
  49     extensions::Extension* extension);
  50
  51 }  // namespace chrome
  52
  53 #endif  // CHROME_BROWSER_UI_COCOA_EXTENSIONS_EXTENSION_INSTALL_PROMPT_TEST_UTILS_H_