Home | History | Annotate | Download | only in test 
      1 // Copyright 2014 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 #include "apps/shell/test/shell_test.h"
      6 
      7 #include "apps/shell/browser/shell_content_browser_client.h"
      8 #include "apps/shell/browser/shell_desktop_controller.h"
      9 #include "apps/shell/browser/shell_extension_system.h"
     10 #include "base/command_line.h"
     11 #include "base/files/file_path.h"
     12 #include "base/logging.h"
     13 #include "content/public/common/content_switches.h"
     14 #include "extensions/browser/extension_system.h"
     15 
     16 namespace apps {
     17 
     18 AppShellTest::AppShellTest()
     19     : browser_context_(NULL), extension_system_(NULL) {}
     20 
     21 AppShellTest::~AppShellTest() {}
     22 
     23 void AppShellTest::SetUp() {
     24   base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
     25   command_line->AppendSwitchASCII(switches::kTestType, "appshell");
     26   content::BrowserTestBase::SetUp();
     27 }
     28 
     29 void AppShellTest::SetUpOnMainThread() {
     30   browser_context_ = ShellContentBrowserClient::Get()->GetBrowserContext();
     31 
     32   extension_system_ = static_cast<extensions::ShellExtensionSystem*>(
     33       extensions::ExtensionSystem::Get(browser_context_));
     34 }
     35 
     36 void AppShellTest::RunTestOnMainThreadLoop() {
     37   base::MessageLoopForUI::current()->RunUntilIdle();
     38 
     39   SetUpOnMainThread();
     40 
     41   RunTestOnMainThread();
     42 
     43   TearDownOnMainThread();
     44 
     45   // Clean up the app window.
     46   ShellDesktopController::instance()->CloseAppWindows();
     47 }
     48 
     49 bool AppShellTest::LoadAndLaunchApp(const base::FilePath& app_dir) {
     50   bool loaded = extension_system_->LoadApp(app_dir);
     51   if (loaded)
     52     extension_system_->LaunchApp();
     53   return loaded;
     54 }
     55 
     56 }  // namespace apps
     57