ios/chrome/browser/web/chrome_web_test.mm

// Copyright 2014 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 "ios/chrome/browser/web/chrome_web_test.h"

#include "components/password_manager/core/browser/mock_password_store.h"
#include "components/password_manager/core/browser/password_manager_test_utils.h"
#include "ios/chrome/browser/browser_state/test_chrome_browser_state.h"
#include "ios/chrome/browser/passwords/ios_chrome_password_store_factory.h"
#import "ios/chrome/browser/ui/fullscreen/legacy_fullscreen_controller.h"
#import "ios/web/public/web_client.h"
#import "ios/web/public/web_state/web_state.h"

#if !defined(__has_feature) || !__has_feature(objc_arc)
#error "This file requires ARC support."
#endif

ChromeWebTest::~ChromeWebTest() {}

ChromeWebTest::ChromeWebTest(std::unique_ptr<web::WebClient> web_client)
    : web::WebTestWithWebState(std::move(web_client)),
      chrome_browser_state_(TestChromeBrowserState::Builder().Build()) {}

ChromeWebTest::ChromeWebTest()
    : chrome_browser_state_(TestChromeBrowserState::Builder().Build()) {}

void ChromeWebTest::SetUp() {
  web::WebTestWithWebState::SetUp();
  IOSChromePasswordStoreFactory::GetInstance()->SetTestingFactory(
      chrome_browser_state_.get(),
      &password_manager::BuildPasswordStore<
          web::BrowserState, password_manager::MockPasswordStore>);
  [LegacyFullscreenController setEnabledForTests:NO];
}

void ChromeWebTest::TearDown() {
  WaitForBackgroundTasks();
  [LegacyFullscreenController setEnabledForTests:YES];
  web::WebTestWithWebState::TearDown();
}

web::BrowserState* ChromeWebTest::GetBrowserState() {
  return chrome_browser_state_.get();
}