Files.app: add test for multi-profile desktop movement menu.

chrome/browser/chromeos/file_manager/file_manager_browsertest.cc

 // Copyright (c) 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 // Browser test for basic Chrome OS file manager functionality:
 //  - The file list is updated when a file is added externally to the Downloads
 //    folder.
 //  - Selecting a file and copy-pasting it with the keyboard copies the file.
 //  - Selecting a file and pressing delete deletes it.
 
 #include <deque>
 #include <string>
 
+#include "apps/shell_window.h"
+#include "apps/shell_window_registry.h"
 #include "base/bind.h"
 #include "base/callback.h"
 #include "base/file_util.h"
 #include "base/files/file_path.h"
 #include "base/json/json_reader.h"
 #include "base/json/json_value_converter.h"
 #include "base/json/json_writer.h"
 #include "base/prefs/pref_service.h"
 #include "base/strings/string_piece.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/time/time.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/chrome_notification_types.h"
 #include "chrome/browser/chromeos/drive/drive_integration_service.h"
 #include "chrome/browser/chromeos/drive/file_system_interface.h"
 #include "chrome/browser/chromeos/drive/test_util.h"
+#include "chrome/browser/chromeos/file_manager/app_id.h"
 #include "chrome/browser/chromeos/file_manager/drive_test_util.h"
 #include "chrome/browser/chromeos/file_manager/path_util.h"
 #include "chrome/browser/chromeos/file_manager/volume_manager.h"
 #include "chrome/browser/chromeos/login/user_manager.h"
 #include "chrome/browser/chromeos/profiles/profile_helper.h"
 #include "chrome/browser/drive/fake_drive_service.h"
 #include "chrome/browser/extensions/api/test/test_api.h"
 #include "chrome/browser/extensions/component_loader.h"
 #include "chrome/browser/extensions/extension_apitest.h"
 #include "chrome/browser/extensions/extension_test_message_listener.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/profiles/profile_manager.h"
+#include "chrome/browser/ui/ash/multi_user/multi_user_util.h"
+#include "chrome/browser/ui/ash/multi_user/multi_user_window_manager.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/pref_names.h"
 #include "chromeos/chromeos_switches.h"
 #include "content/public/browser/browser_context.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/test/test_utils.h"
 #include "extensions/common/extension.h"
 #include "google_apis/drive/gdata_wapi_parser.h"
 #include "google_apis/drive/test_util.h"
 #include "net/test/embedded_test_server/embedded_test_server.h"
@@ skipping 787 matching lines @@
     const TestAccountInfo& info = kTestAccounts[PRIMARY_ACCOUNT_INDEX];
 
     AddUser(info, true);
     chromeos::UserManager* const user_manager = chromeos::UserManager::Get();
     if (user_manager->GetActiveUser() != user_manager->FindUser(info.email))
       chromeos::UserManager::Get()->SwitchActiveUser(info.email);
     FileManagerBrowserTestBase::SetUpOnMainThread();
   }
 
   // Loads all users to the current session and sets up necessary fields.
-  // This is used for preparing all acounts in PRE_ test setup, and for testing
+  // This is used for preparing all accounts in PRE_ test setup, and for testing
   // actual login behavior.
   void AddAllUsers() {
     for (size_t i = 0; i < arraysize(kTestAccounts); ++i)
       AddUser(kTestAccounts[i], i >= SECONDARY_ACCOUNT_INDEX_START);
   }
 
   // Returns primary profile (if it is already created.)
   virtual Profile* profile() OVERRIDE {
     Profile* const profile = chromeos::ProfileHelper::GetProfileByUserIdHash(
         kTestAccounts[PRIMARY_ACCOUNT_INDEX].hash);
@@ skipping 21 matching lines @@
 
   virtual const char* GetTestCaseNameParam() const OVERRIDE {
     return test_case_name_.c_str();
   }
 
   virtual std::string OnMessage(const std::string& name,
                                 const base::Value* value) OVERRIDE {
     if (name == "addAllUsers") {
       AddAllUsers();
       return "true";
+    } else if (name == "getWindowOwnerId") {
+      chrome::MultiUserWindowManager* const window_manager =
+          chrome::MultiUserWindowManager::GetInstance();
+      apps::ShellWindowRegistry* const shell_window_registry =
+          apps::ShellWindowRegistry::Get(profile());
+      DCHECK(window_manager);
+      DCHECK(shell_window_registry);
+
+      const apps::ShellWindowRegistry::ShellWindowList& list =
+          shell_window_registry->GetShellWindowsForApp(
+              file_manager::kFileManagerAppId);
+      return list.size() == 1u ?
+          window_manager->GetUserPresentingWindow(
+              list.front()->GetNativeWindow()) : "";
     }
     return FileManagerBrowserTestBase::OnMessage(name, value);
   }
 
   std::string test_case_name_;
 };
 
 IN_PROC_BROWSER_TEST_F(MultiProfileFileManagerBrowserTest, PRE_BasicDownloads) {
   AddAllUsers();
 }
@@ skipping 20 matching lines @@
 
 IN_PROC_BROWSER_TEST_F(MultiProfileFileManagerBrowserTest, PRE_Badge) {
   AddAllUsers();
 }
 
 IN_PROC_BROWSER_TEST_F(MultiProfileFileManagerBrowserTest, Badge) {
   set_test_case_name("multiProfileBadge");
   StartTest();
 }
 
+IN_PROC_BROWSER_TEST_F(MultiProfileFileManagerBrowserTest,
+                       PRE_VisitDesktopMenu) {
+  AddAllUsers();
+}
+
+IN_PROC_BROWSER_TEST_F(MultiProfileFileManagerBrowserTest, VisitDesktopMenu) {
+  set_test_case_name("multiProfileVisitDesktopMenu");
+  StartTest();
+}
+
 // TODO(kinaba) write more tests.
 
 }  // namespace
 }  // namespace file_manager