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 "extensions/shell/test/shell_test.h"
6
7#include "base/command_line.h"
8#include "base/logging.h"
9#include "content/public/common/content_switches.h"
10#include "extensions/browser/extension_system.h"
11#include "extensions/shell/browser/shell_content_browser_client.h"
12#include "extensions/shell/browser/shell_desktop_controller.h"
13#include "extensions/shell/browser/shell_extension_system.h"
14
15namespace extensions {
16
17AppShellTest::AppShellTest() : browser_context_(NULL), extension_system_(NULL) {
18}
19
20AppShellTest::~AppShellTest() {
21}
22
23void AppShellTest::SetUp() {
24  base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
25  command_line->AppendSwitchASCII(switches::kTestType, "appshell");
26  content::BrowserTestBase::SetUp();
27}
28
29void AppShellTest::SetUpOnMainThread() {
30  browser_context_ = ShellContentBrowserClient::Get()->GetBrowserContext();
31
32  extension_system_ = static_cast<ShellExtensionSystem*>(
33      ExtensionSystem::Get(browser_context_));
34}
35
36void 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}  // namespace extensions
50