[ios] WebStateList owns all WebState it manages.

ios/chrome/browser/tabs/tab_model_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 <objc/runtime.h>
 
 #include "base/files/file_path.h"
 #include "base/mac/scoped_nsautorelease_pool.h"
 #include "base/memory/ptr_util.h"
 #include "base/run_loop.h"
 #include "base/strings/sys_string_conversions.h"
 #include "ios/chrome/browser/browser_state/test_chrome_browser_state.h"
 #include "ios/chrome/browser/browser_state/test_chrome_browser_state_manager.h"
 #include "ios/chrome/browser/chrome_url_constants.h"
 #include "ios/chrome/browser/infobars/infobar_manager_impl.h"
 #include "ios/chrome/browser/sessions/ios_chrome_session_tab_helper.h"
 #import "ios/chrome/browser/sessions/session_window_ios.h"
 #import "ios/chrome/browser/sessions/test_session_service.h"
+#import "ios/chrome/browser/tabs/legacy_tab_helper.h"
 #import "ios/chrome/browser/tabs/tab.h"
+#import "ios/chrome/browser/tabs/tab_helper_util.h"
 #import "ios/chrome/browser/tabs/tab_model.h"
 #import "ios/chrome/browser/tabs/tab_model_observer.h"
 #import "ios/chrome/browser/tabs/tab_private.h"
 #import "ios/chrome/browser/web/chrome_web_client.h"
 #include "ios/chrome/test/ios_chrome_scoped_testing_chrome_browser_state_manager.h"
+#import "ios/shared/chrome/browser/tabs/web_state_list.h"
 #import "ios/web/navigation/crw_session_controller.h"
 #import "ios/web/navigation/navigation_manager_impl.h"
 #import "ios/web/public/crw_session_storage.h"
 #import "ios/web/public/navigation_manager.h"
 #include "ios/web/public/referrer.h"
 #import "ios/web/public/serializable_user_data_manager.h"
 #include "ios/web/public/test/scoped_testing_web_client.h"
 #include "ios/web/public/test/test_web_thread_bundle.h"
 #include "ios/web/public/web_thread.h"
 #import "ios/web/web_state/ui/crw_web_controller.h"
 #import "ios/web/web_state/web_state_impl.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "testing/gtest_mac.h"
 #include "testing/platform_test.h"
 #import "third_party/ocmock/OCMock/OCMock.h"
 #include "third_party/ocmock/gtest_support.h"
 
-using web::WebStateImpl;
-
-
 @interface TabModel (VisibleForTesting)
 - (SessionWindowIOS*)windowForSavingSession;
 @end
 
 // Defines a TabModelObserver for use in unittests.  This class can be used to
 // test if an observer method was called or not.
 @interface TabModelObserverPong : NSObject<TabModelObserver> {
   // TODO(crbug.com/661989): Add tests for the other observer methods.
   BOOL tabMovedWasCalled_;
 }
@@ skipping 1002 matching lines @@
   EXPECT_NSEQ(tab1, [tab_model_ tabAtIndex:0]);
   EXPECT_NSEQ(tab2, [tab_model_ tabAtIndex:1]);
   EXPECT_NSEQ(tab0, [tab_model_ tabAtIndex:2]);
   EXPECT_FALSE([tab_model_observer tabMovedWasCalled]);
 
   // TabModel asserts that there are no observer when it is deallocated,
   // so remove the observer before the end of the method.
   [tab_model_ removeObserver:tab_model_observer.get()];
 }
 
-TEST_F(TabModelTest, SetParentModel) {
-  // Create a tab without a parent model and make sure it doesn't crash.  Then
-  // set its parent TabModel and make sure that works as well.
-  base::scoped_nsobject<Tab> tab([[Tab alloc]
-      initWithBrowserState:chrome_browser_state_.get()
-                    opener:nil
-               openedByDOM:NO
-                     model:nil]);
-  EXPECT_TRUE([tab parentTabModel] == nil);
-  [tab_model_ insertTab:tab atIndex:0];
-  [tab setParentTabModel:tab_model_.get()];
-  EXPECT_FALSE([tab parentTabModel] == nil);
-  [tab_model_ closeTabAtIndex:0];
+TEST_F(TabModelTest, ParentTabModel) {
+  std::unique_ptr<web::WebState> web_state = web::WebState::Create(
+      web::WebState::CreateParams(chrome_browser_state_.get()));
+  AttachTabHelpers(web_state.get());
+
+  Tab* tab = LegacyTabHelper::GetTabForWebState(web_state.get());
+  EXPECT_NSEQ(nil, [tab parentTabModel]);
+
+  [tab_model_ webStateList]->InsertWebState(0, std::move(web_state));
+  EXPECT_NSEQ(tab_model_.get(), [tab parentTabModel]);
+}
+
+TEST_F(TabModelTest, TabCreatedOnInsertion) {
+  std::unique_ptr<web::WebState> web_state = web::WebState::Create(
+      web::WebState::CreateParams(chrome_browser_state_.get()));
+
+  EXPECT_NSEQ(nil, LegacyTabHelper::GetTabForWebState(web_state.get()));
+
+  web::WebState* web_state_ptr = web_state.get();
+  [tab_model_ webStateList]->InsertWebState(0, std::move(web_state));
+  EXPECT_NSNE(nil, LegacyTabHelper::GetTabForWebState(web_state_ptr));
 }
 
 TEST_F(TabModelTest, PersistSelectionChange) {
   NSString* stashPath =
       base::SysUTF8ToNSString(chrome_browser_state_->GetStatePath().value());
 
   [tab_model_ insertTabWithURL:GURL(kURL1)
                       referrer:web::Referrer()
                     transition:ui::PAGE_TRANSITION_TYPED
                         opener:nil
@@ skipping 55 matching lines @@
             [tab_model_ openerOfTab:[tab_model_ tabAtIndex:2]]);
   EXPECT_EQ([tab_model_ tabAtIndex:2],
             [tab_model_ openerOfTab:[tab_model_ tabAtIndex:0]]);
 
   // Clean up.
   EXPECT_TRUE([[NSFileManager defaultManager] removeItemAtPath:stashPath
                                                          error:nullptr]);
 }
 
 }  // anonymous namespace