Index: apps/shell/test/app_shell_test.cc |
diff --git a/apps/test/app_shell_test.cc b/apps/shell/test/app_shell_test.cc |
similarity index 66% |
rename from apps/test/app_shell_test.cc |
rename to apps/shell/test/app_shell_test.cc |
index 683bdddd40ca0b4cba29a981edc150201c9c5631..ca0a4fbceeee949073b3de3df2aa3d9d905662d9 100644 |
--- a/apps/test/app_shell_test.cc |
+++ b/apps/shell/test/app_shell_test.cc |
@@ -2,12 +2,15 @@ |
// Use of this source code is governed by a BSD-style license that can be |
// found in the LICENSE file. |
-#include "apps/test/app_shell_test.h" |
+#include "apps/shell/test/app_shell_test.h" |
#include "apps/shell/browser/shell_content_browser_client.h" |
+#include "apps/shell/browser/shell_desktop_controller.h" |
#include "apps/shell/browser/shell_extension_system.h" |
+#include "base/command_line.h" |
#include "base/files/file_path.h" |
#include "base/logging.h" |
+#include "content/public/common/content_switches.h" |
#include "extensions/browser/extension_system.h" |
namespace apps { |
@@ -17,6 +20,14 @@ AppShellTest::AppShellTest() |
AppShellTest::~AppShellTest() {} |
+void AppShellTest::SetUp() { |
+ base::CommandLine* command_line = base::CommandLine::ForCurrentProcess(); |
+ command_line->AppendSwitchASCII(switches::kTestType, "appshell"); |
James Cook
2014/04/08 16:03:49
Just curious, why do we need a testtype switch?
Yoyo Zhou
2014/04/08 21:55:21
The chrome.test API checks that it's running with
|
+ content::BrowserTestBase::SetUp(); |
+} |
+ |
+void AppShellTest::TearDown() { content::BrowserTestBase::TearDown(); } |
James Cook
2014/04/08 16:03:49
Do you anticipate this function doing more than ca
Yoyo Zhou
2014/04/08 21:55:21
Done.
|
+ |
void AppShellTest::SetUpOnMainThread() { |
browser_context_ = ShellContentBrowserClient::Get()->GetBrowserContext(); |
@@ -25,7 +36,6 @@ void AppShellTest::SetUpOnMainThread() { |
} |
void AppShellTest::RunTestOnMainThreadLoop() { |
- // Pump startup related events. |
base::MessageLoopForUI::current()->RunUntilIdle(); |
SetUpOnMainThread(); |
@@ -33,6 +43,9 @@ void AppShellTest::RunTestOnMainThreadLoop() { |
RunTestOnMainThread(); |
TearDownOnMainThread(); |
+ |
+ // Clean up the app window. |
+ ShellDesktopController::instance()->CloseAppWindow(); |
} |
bool AppShellTest::LoadAndLaunchApp(const base::FilePath& app_dir) { |