[ios] Allow having multiple session windows in a session.

ios/chrome/browser/sessions/session_service_ios_unittest.mm

 // Copyright 2012 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.
 
 #import <Foundation/Foundation.h>
 
 #include <memory>
 
 #include "base/files/file_path.h"
 #include "base/files/file_util.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/path_service.h"
 #include "base/run_loop.h"
 #include "base/strings/sys_string_conversions.h"
 #include "base/test/scoped_task_environment.h"
 #include "base/threading/thread_task_runner_handle.h"
 #include "ios/chrome/browser/chrome_paths.h"
+#import "ios/chrome/browser/sessions/session_ios.h"
 #import "ios/chrome/browser/sessions/session_service_ios.h"
 #import "ios/chrome/browser/sessions/session_window_ios.h"
 #import "ios/web/public/crw_session_storage.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "testing/gtest_mac.h"
 #include "testing/platform_test.h"
 
 #if !defined(__has_feature) || !__has_feature(objc_arc)
 #error "This file requires ARC support."
 #endif
@@ skipping 33 matching lines @@
       return nil;
 
     session_path = session_path.Append(FILE_PATH_LITERAL("sessions"));
     session_path = session_path.Append(filename);
     if (!base::PathExists(session_path))
       return nil;
 
     return base::SysUTF8ToNSString(session_path.AsUTF8Unsafe());
   }
 
-  // Create a SessionWindowIOS with |tab_count| tabs.
-  SessionWindowIOS* CreateSessionWindow(NSUInteger tab_count) {
-    NSMutableArray<CRWSessionStorage*>* tabs = [NSMutableArray array];
-    while (tabs.count < tab_count) {
-      [tabs addObject:[[CRWSessionStorage alloc] init]];
+  // Create a SessionIOS corresponding to |window_count| windows each with
+  // |tab_count| tabs.
+  SessionIOS* CreateSession(NSUInteger window_count, NSUInteger tab_count) {
+    NSMutableArray<SessionWindowIOS*>* windows = [NSMutableArray array];
+    while (windows.count < window_count) {
+      NSMutableArray<CRWSessionStorage*>* tabs = [NSMutableArray array];
+      while (tabs.count < tab_count) {
+        [tabs addObject:[[CRWSessionStorage alloc] init]];
+      }
+      [windows addObject:[[SessionWindowIOS alloc]
+                             initWithSessions:[tabs copy]
+                                selectedIndex:(tabs.count ? tabs.count - 1
+                                                          : NSNotFound)]];
     }
-    return [[SessionWindowIOS alloc]
-        initWithSessions:[tabs copy]
-           selectedIndex:(tabs.count ? tabs.count - 1 : NSNotFound)];
+    return [[SessionIOS alloc] initWithWindows:[windows copy]];
   }
 
   SessionServiceIOS* session_service() { return session_service_; }
 
   NSString* directory() { return directory_; }
 
  private:
   base::ScopedTempDir scoped_temp_directory_;
   base::test::ScopedTaskEnvironment scoped_task_environment_;
   SessionServiceIOS* session_service_;
   NSString* directory_;
 
   DISALLOW_COPY_AND_ASSIGN(SessionServiceTest);
 };
 
 TEST_F(SessionServiceTest, SessionPathForDirectory) {
   EXPECT_NSEQ(@"session.plist",
               [SessionServiceIOS sessionPathForDirectory:@""]);
 }
 
-TEST_F(SessionServiceTest, SaveSessionWindowToDirectory) {
-  [session_service() saveSessionWindow:CreateSessionWindow(0u)
-                             directory:directory()
-                           immediately:YES];
+TEST_F(SessionServiceTest, SaveSessionWindowToPath) {
+  [session_service() saveSession:CreateSession(0u, 0u)
+                       directory:directory()
+                     immediately:YES];
 
-  // Even if immediately is YES, the file is created by a task on the task
+  // Even if |immediately| is YES, the file is created by a task on the task
   // runner passed to SessionServiceIOS initializer (which is the current
   // thread task runner during test). Wait for the task to complete.
   base::RunLoop().RunUntilIdle();
 
   NSFileManager* file_manager = [NSFileManager defaultManager];
   EXPECT_TRUE([file_manager removeItemAtPath:directory() error:nullptr]);
 }
 
-TEST_F(SessionServiceTest, SaveSessionWindowToExistingDirectory) {
+TEST_F(SessionServiceTest, SaveSessionWindowToPathDirectoryExists) {
   ASSERT_TRUE([[NSFileManager defaultManager] createDirectoryAtPath:directory()
                                         withIntermediateDirectories:YES
                                                          attributes:nil
                                                               error:nullptr]);
 
-  [session_service() saveSessionWindow:CreateSessionWindow(0u)
-                             directory:directory()
-                           immediately:YES];
+  [session_service() saveSession:CreateSession(0u, 0u)
+                       directory:directory()
+                     immediately:YES];
 
-  // Even if immediately is YES, the file is created by a task on the task
+  // Even if |immediately| is YES, the file is created by a task on the task
   // runner passed to SessionServiceIOS initializer (which is the current
   // thread task runner during test). Wait for the task to complete.
   base::RunLoop().RunUntilIdle();
 
   NSFileManager* file_manager = [NSFileManager defaultManager];
   EXPECT_TRUE([file_manager removeItemAtPath:directory() error:nullptr]);
 }
 
-TEST_F(SessionServiceTest, LoadSessionWindowFromDirectoryNoFile) {
-  SessionWindowIOS* session_window =
-      [session_service() loadSessionWindowFromDirectory:directory()];
-  EXPECT_TRUE(session_window == nil);
+TEST_F(SessionServiceTest, LoadSessionFromDirectoryNoFile) {
+  SessionIOS* session =
+      [session_service() loadSessionFromDirectory:directory()];
+  EXPECT_TRUE(session == nil);
 }
 
-TEST_F(SessionServiceTest, LoadSessionWindowFromDirectory) {
-  [session_service() saveSessionWindow:CreateSessionWindow(1u)
-                             directory:directory()
-                           immediately:YES];
+TEST_F(SessionServiceTest, LoadSessionFromDirectory) {
+  [session_service() saveSession:CreateSession(2u, 1u)
+                       directory:directory()
+                     immediately:YES];
 
-  // Even if immediately is YES, the file is created by a task on the task
+  // Even if |immediately| is YES, the file is created by a task on the task
   // runner passed to SessionServiceIOS initializer (which is the current
   // thread task runner during test). Wait for the task to complete.
   base::RunLoop().RunUntilIdle();
 
-  SessionWindowIOS* session_window =
-      [session_service() loadSessionWindowFromDirectory:directory()];
-  EXPECT_EQ(1u, session_window.sessions.count);
-  EXPECT_EQ(0u, session_window.selectedIndex);
+  SessionIOS* session =
+      [session_service() loadSessionFromDirectory:directory()];
+  EXPECT_EQ(2u, session.sessionWindows.count);
+  for (SessionWindowIOS* sessionWindow in session.sessionWindows) {
+    EXPECT_EQ(1u, sessionWindow.sessions.count);
+    EXPECT_EQ(0u, sessionWindow.selectedIndex);
+  }
 }
 
-TEST_F(SessionServiceTest, LoadSessionWindowFromPath) {
-  [session_service() saveSessionWindow:CreateSessionWindow(1u)
-                             directory:directory()
-                           immediately:YES];
+TEST_F(SessionServiceTest, LoadSessionFromPath) {
+  [session_service() saveSession:CreateSession(2u, 1u)
+                       directory:directory()
+                     immediately:YES];
 
-  // Even if immediately is YES, the file is created by a task on the task
+  // Even if |immediately| is YES, the file is created by a task on the task
   // runner passed to SessionServiceIOS initializer (which is the current
   // thread task runner during test). Wait for the task to complete.
   base::RunLoop().RunUntilIdle();
 
   NSString* session_path =
       [SessionServiceIOS sessionPathForDirectory:directory()];
   NSString* renamed_path = [session_path stringByAppendingPathExtension:@"bak"];
   ASSERT_NSNE(session_path, renamed_path);
 
   // Rename the file.
   ASSERT_TRUE([[NSFileManager defaultManager] moveItemAtPath:session_path
                                                       toPath:renamed_path
                                                        error:nil]);
 
-  SessionWindowIOS* session_window =
-      [session_service() loadSessionWindowFromPath:renamed_path];
-  EXPECT_EQ(1u, session_window.sessions.count);
-  EXPECT_EQ(0u, session_window.selectedIndex);
+  SessionIOS* session = [session_service() loadSessionFromPath:renamed_path];
+  EXPECT_EQ(2u, session.sessionWindows.count);
+  for (SessionWindowIOS* sessionWindow in session.sessionWindows) {
+    EXPECT_EQ(1u, sessionWindow.sessions.count);
+    EXPECT_EQ(0u, sessionWindow.selectedIndex);
+  }
 }
 
-TEST_F(SessionServiceTest, LoadCorruptedSessionWindow) {
+TEST_F(SessionServiceTest, LoadCorruptedSession) {
   NSString* session_path =
       SessionPathForTestData(FILE_PATH_LITERAL("corrupted.plist"));
   ASSERT_NSNE(nil, session_path);
-  SessionWindowIOS* session_window =
-      [session_service() loadSessionWindowFromPath:session_path];
-  EXPECT_TRUE(session_window == nil);
+  SessionIOS* session = [session_service() loadSessionFromPath:session_path];
+  EXPECT_TRUE(session == nil);
 }
 
 // TODO(crbug.com/661633): remove this once M67 has shipped (i.e. once more
 // than a year has passed since the introduction of the compatibility code).
 TEST_F(SessionServiceTest, LoadM57Session) {
   NSString* session_path =
       SessionPathForTestData(FILE_PATH_LITERAL("session_m57.plist"));
   ASSERT_NSNE(nil, session_path);
-  SessionWindowIOS* session_window =
-      [session_service() loadSessionWindowFromPath:session_path];
-  EXPECT_TRUE(session_window != nil);
+  SessionIOS* session = [session_service() loadSessionFromPath:session_path];
+  EXPECT_EQ(1u, session.sessionWindows.count);
 }
 
 // TODO(crbug.com/661633): remove this once M68 has shipped (i.e. once more
 // than a year has passed since the introduction of the compatibility code).
 TEST_F(SessionServiceTest, LoadM58Session) {
   NSString* session_path =
       SessionPathForTestData(FILE_PATH_LITERAL("session_m58.plist"));
   ASSERT_NSNE(nil, session_path);
-  SessionWindowIOS* session_window =
-      [session_service() loadSessionWindowFromPath:session_path];
-  EXPECT_TRUE(session_window != nil);
+  SessionIOS* session = [session_service() loadSessionFromPath:session_path];
+  EXPECT_EQ(1u, session.sessionWindows.count);
 }
 
 }  // anonymous namespace