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 #include "chrome/browser/extensions/platform_app_browsertest_util.h" 6 7 #include "apps/app_window_contents.h" 8 #include "apps/native_app_window.h" 9 #include "apps/shell_window_registry.h" 10 #include "base/command_line.h" 11 #include "base/strings/stringprintf.h" 12 #include "chrome/browser/extensions/api/tabs/tabs_api.h" 13 #include "chrome/browser/extensions/extension_function_test_utils.h" 14 #include "chrome/browser/ui/apps/chrome_shell_window_delegate.h" 15 #include "chrome/browser/ui/browser.h" 16 #include "chrome/browser/ui/extensions/application_launch.h" 17 #include "chrome/common/chrome_switches.h" 18 #include "content/public/browser/notification_service.h" 19 #include "content/public/test/test_utils.h" 20 #include "extensions/common/switches.h" 21 22 using apps::ShellWindow; 23 using apps::ShellWindowRegistry; 24 using content::WebContents; 25 26 namespace utils = extension_function_test_utils; 27 28 namespace extensions { 29 30 PlatformAppBrowserTest::PlatformAppBrowserTest() { 31 ChromeShellWindowDelegate::DisableExternalOpenForTesting(); 32 } 33 34 void PlatformAppBrowserTest::SetUpCommandLine(CommandLine* command_line) { 35 // Skips ExtensionApiTest::SetUpCommandLine. 36 ExtensionBrowserTest::SetUpCommandLine(command_line); 37 38 // Make event pages get suspended quicker. 39 command_line->AppendSwitchASCII(::switches::kEventPageIdleTime, "1"); 40 command_line->AppendSwitchASCII(::switches::kEventPageSuspendingTime, "1"); 41 } 42 43 const Extension* PlatformAppBrowserTest::LoadAndLaunchPlatformApp( 44 const char* name) { 45 content::WindowedNotificationObserver app_loaded_observer( 46 content::NOTIFICATION_LOAD_COMPLETED_MAIN_FRAME, 47 content::NotificationService::AllSources()); 48 49 const Extension* extension = LoadExtension( 50 test_data_dir_.AppendASCII("platform_apps").AppendASCII(name)); 51 EXPECT_TRUE(extension); 52 53 chrome::OpenApplication(chrome::AppLaunchParams(browser()->profile(), 54 extension, 55 extension_misc::LAUNCH_NONE, 56 NEW_WINDOW)); 57 58 app_loaded_observer.Wait(); 59 60 return extension; 61 } 62 63 const Extension* PlatformAppBrowserTest::InstallAndLaunchPlatformApp( 64 const char* name) { 65 content::WindowedNotificationObserver app_loaded_observer( 66 content::NOTIFICATION_LOAD_COMPLETED_MAIN_FRAME, 67 content::NotificationService::AllSources()); 68 69 const Extension* extension = InstallExtension( 70 test_data_dir_.AppendASCII("platform_apps").AppendASCII(name), 1); 71 EXPECT_TRUE(extension); 72 73 chrome::OpenApplication(chrome::AppLaunchParams(browser()->profile(), 74 extension, 75 extension_misc::LAUNCH_NONE, 76 NEW_WINDOW)); 77 78 app_loaded_observer.Wait(); 79 80 return extension; 81 } 82 83 WebContents* PlatformAppBrowserTest::GetFirstShellWindowWebContents() { 84 ShellWindow* window = GetFirstShellWindow(); 85 if (window) 86 return window->web_contents(); 87 88 return NULL; 89 } 90 91 ShellWindow* PlatformAppBrowserTest::GetFirstShellWindow() { 92 ShellWindowRegistry* app_registry = 93 ShellWindowRegistry::Get(browser()->profile()); 94 const ShellWindowRegistry::ShellWindowList& shell_windows = 95 app_registry->shell_windows(); 96 97 ShellWindowRegistry::const_iterator iter = shell_windows.begin(); 98 if (iter != shell_windows.end()) 99 return *iter; 100 101 return NULL; 102 } 103 104 size_t PlatformAppBrowserTest::RunGetWindowsFunctionForExtension( 105 const Extension* extension) { 106 scoped_refptr<WindowsGetAllFunction> function = new WindowsGetAllFunction(); 107 function->set_extension(extension); 108 scoped_ptr<base::ListValue> result(utils::ToList( 109 utils::RunFunctionAndReturnSingleResult(function.get(), 110 "[]", 111 browser()))); 112 return result->GetSize(); 113 } 114 115 bool PlatformAppBrowserTest::RunGetWindowFunctionForExtension( 116 int window_id, 117 const Extension* extension) { 118 scoped_refptr<WindowsGetFunction> function = new WindowsGetFunction(); 119 function->set_extension(extension); 120 utils::RunFunction( 121 function.get(), 122 base::StringPrintf("[%u]", window_id), 123 browser(), 124 utils::NONE); 125 return function->GetResultList() != NULL; 126 } 127 128 size_t PlatformAppBrowserTest::GetShellWindowCount() { 129 return ShellWindowRegistry::Get(browser()->profile())-> 130 shell_windows().size(); 131 } 132 133 void PlatformAppBrowserTest::ClearCommandLineArgs() { 134 CommandLine* command_line = CommandLine::ForCurrentProcess(); 135 CommandLine::StringVector args = command_line->GetArgs(); 136 CommandLine::StringVector argv = command_line->argv(); 137 for (size_t i = 0; i < args.size(); i++) 138 argv.pop_back(); 139 command_line->InitFromArgv(argv); 140 } 141 142 void PlatformAppBrowserTest::SetCommandLineArg(const std::string& test_file) { 143 ClearCommandLineArgs(); 144 CommandLine* command_line = CommandLine::ForCurrentProcess(); 145 base::FilePath test_doc(test_data_dir_.AppendASCII(test_file)); 146 test_doc = test_doc.NormalizePathSeparators(); 147 command_line->AppendArgPath(test_doc); 148 } 149 150 ShellWindow* PlatformAppBrowserTest::CreateShellWindow( 151 const Extension* extension) { 152 return CreateShellWindowFromParams(extension, ShellWindow::CreateParams()); 153 } 154 155 ShellWindow* PlatformAppBrowserTest::CreateShellWindowFromParams( 156 const Extension* extension, const ShellWindow::CreateParams& params) { 157 ShellWindow* window = new ShellWindow(browser()->profile(), 158 new ChromeShellWindowDelegate(), 159 extension); 160 window->Init(GURL(std::string()), 161 new apps::AppWindowContents(window), 162 params); 163 return window; 164 } 165 166 void PlatformAppBrowserTest::CloseShellWindow(ShellWindow* window) { 167 content::WindowedNotificationObserver destroyed_observer( 168 content::NOTIFICATION_WEB_CONTENTS_DESTROYED, 169 content::NotificationService::AllSources()); 170 window->GetBaseWindow()->Close(); 171 destroyed_observer.Wait(); 172 } 173 174 void PlatformAppBrowserTest::CallAdjustBoundsToBeVisibleOnScreenForShellWindow( 175 ShellWindow* window, 176 const gfx::Rect& cached_bounds, 177 const gfx::Rect& cached_screen_bounds, 178 const gfx::Rect& current_screen_bounds, 179 const gfx::Size& minimum_size, 180 gfx::Rect* bounds) { 181 window->AdjustBoundsToBeVisibleOnScreen(cached_bounds, 182 cached_screen_bounds, 183 current_screen_bounds, 184 minimum_size, 185 bounds); 186 } 187 188 void ExperimentalPlatformAppBrowserTest::SetUpCommandLine( 189 CommandLine* command_line) { 190 PlatformAppBrowserTest::SetUpCommandLine(command_line); 191 command_line->AppendSwitch(switches::kEnableExperimentalExtensionApis); 192 } 193 194 } // namespace extensions 195