chrome/browser/extensions/webstore_installer_test.h

   1 // Copyright 2013 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_EXTENSIONS_WEBSTORE_INSTALLER_TEST_H_
   6 #define CHROME_BROWSER_EXTENSIONS_WEBSTORE_INSTALLER_TEST_H_
   7
   8 #include <string>
   9
  10 #include "base/files/scoped_temp_dir.h"
  11 #include "chrome/browser/extensions/extension_browsertest.h"
  12 #include "extensions/browser/extension_dialog_auto_confirm.h"
  13 #include "url/gurl.h"
  14
  15 namespace base {
  16 class CommandLine;
  17 }
  18
  19 namespace contents {
  20 class WebContents;
  21 }
  22
  23 class WebstoreInstallerTest : public ExtensionBrowserTest {
  24  public:
  25   WebstoreInstallerTest(const std::string& webstore_domain,
  26                         const std::string& test_data_path,
  27                         const std::string& crx_filename,
  28                         const std::string& verified_domain,
  29                         const std::string& unverified_domain);
  30   ~WebstoreInstallerTest() override;
  31
  32   void SetUpCommandLine(base::CommandLine* command_line) override;
  33   void SetUpInProcessBrowserTestFixture() override;
  34   void SetUpOnMainThread() override;
  35
  36  protected:
  37   GURL GenerateTestServerUrl(const std::string& domain,
  38                              const std::string& page_filename);
  39
  40   void RunTest(const std::string& test_function_name);
  41
  42   void RunTest(content::WebContents* web_contents,
  43                const std::string& test_function_name);
  44
  45   // Passes |i| to |test_function_name|, and expects that function to
  46   // return one of "FAILED", "KEEPGOING" or "DONE". KEEPGOING should be
  47   // returned if more tests remain to be run and the current test succeeded,
  48   // FAILED is returned when a test fails, and DONE is returned by the last
  49   // test if it succeeds.
  50   // This methods returns true iff there are more tests that need to be run.
  51   bool RunIndexedTest(const std::string& test_function_name, int i);
  52
  53   // Runs a test without waiting for any results from the renderer.
  54   void RunTestAsync(const std::string& test_function_name);
  55
  56   // Configures command line switches to simulate a user accepting the install
  57   // prompt.
  58   void AutoAcceptInstall();
  59
  60   // Configures command line switches to simulate a user cancelling the install
  61   // prompt.
  62   void AutoCancelInstall();
  63
  64   std::string webstore_domain_;
  65   std::string test_data_path_;
  66   std::string crx_filename_;
  67   std::string verified_domain_;
  68   std::string unverified_domain_;
  69   std::string test_gallery_url_;
  70
  71   base::ScopedTempDir download_directory_;
  72
  73   scoped_ptr<extensions::ScopedTestDialogAutoConfirm> install_auto_confirm_;
  74
  75   DISALLOW_COPY_AND_ASSIGN(WebstoreInstallerTest);
  76 };
  77
  78 #endif  // CHROME_BROWSER_EXTENSIONS_WEBSTORE_INSTALLER_TEST_H_