Files.app add multi-profile test with maximum number of profiles.

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 "ash/session_state_delegate.h"
+#include "ash/shell.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/stringprintf.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"
@@ skipping 816 matching lines @@
   }
 
   // Loads all users to the current session and sets up necessary fields.
   // 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);
   }
 
+  // Add as many as users
+  void AddExtraUsersForStressTesting() {
+    ash::Shell* const shell = ash::Shell::GetInstance();
+    const size_t maxLogin =
+        shell->session_state_delegate()->GetMaximumNumberOfLoggedInUsers();
+
+    for (int i = 0; i + arraysize(kTestAccounts) < maxLogin; ++i) {
+      const std::string email = base::StringPrintf("user%d@invalid.domain", i);
+      const std::string hash = base::StringPrintf("hashuser%d", i);
+      const std::string name = base::StringPrintf("Additional User %d", i);
+      const TestAccountInfo info = {email.c_str(), hash.c_str(), name.c_str()};
+      AddUser(info, true);
+    }
+  }
+
   // Returns primary profile (if it is already created.)
   virtual Profile* profile() OVERRIDE {
     Profile* const profile = chromeos::ProfileHelper::GetProfileByUserIdHash(
         kTestAccounts[PRIMARY_ACCOUNT_INDEX].hash);
     return profile ? profile : FileManagerBrowserTestBase::profile();
   }
 
   // Sets the test case name (used as a function name in test_cases.js to call.)
   void set_test_case_name(const std::string& name) { test_case_name_ = name; }
 
@@ skipping 65 matching lines @@
   // Sanity check that normal operations work in multi-profile setting as well.
   set_test_case_name("keyboardCopyDrive");
   StartTest();
 }
 
 IN_PROC_BROWSER_TEST_F(MultiProfileFileManagerBrowserTest, PRE_Badge) {
   AddAllUsers();
 }
 
 IN_PROC_BROWSER_TEST_F(MultiProfileFileManagerBrowserTest, Badge) {
+  // Test the profile badge to be correctly shown and hidden.
   set_test_case_name("multiProfileBadge");
   StartTest();
 }
 
 IN_PROC_BROWSER_TEST_F(MultiProfileFileManagerBrowserTest,
                        PRE_VisitDesktopMenu) {
   AddAllUsers();
 }
 
 IN_PROC_BROWSER_TEST_F(MultiProfileFileManagerBrowserTest, VisitDesktopMenu) {
+  // Test for the menu item for visiting other profile's desktop.
   set_test_case_name("multiProfileVisitDesktopMenu");
   StartTest();
 }
 
-// TODO(kinaba) write more tests.
+IN_PROC_BROWSER_TEST_F(MultiProfileFileManagerBrowserTest, PRE_MaxUser) {
+  AddAllUsers();
+  AddExtraUsersForStressTesting();
+}
+
+IN_PROC_BROWSER_TEST_F(MultiProfileFileManagerBrowserTest, MaxUser) {
+  // Run the same test as VisitDesktopMenu with maximum number of users logged
+  // in and checks that nothing goes wrong. Here, the primary user (alice) logs
+  // in first, then the "extra" users follow, and then lastly the other users
+  // (bob and charlie) are added in the test. Thus the existing test verifies
+  // that the feature is effectively working with lastly logged in users.
+  AddExtraUsersForStressTesting();
+
+  set_test_case_name("multiProfileVisitDesktopMenu");
+  StartTest();
+}
 
 }  // namespace
 }  // namespace file_manager