ios/chrome/browser/prerender/prerender_service_impl_unittest.mm

// Copyright 2017 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/prerender/prerender_service_impl.h"

#include <memory>

#include "ios/chrome/browser/browser_state/test_chrome_browser_state.h"
#import "ios/web/public/test/fakes/fake_web_state.h"
#include "ios/web/public/test/web_task_environment.h"
#include "testing/platform_test.h"

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

namespace {

class PrerenderServiceImplTest : public PlatformTest {
 public:
  PrerenderServiceImplTest() {
    TestChromeBrowserState::Builder builder;
    browser_state_ = builder.Build();
    service_ = std::make_unique<PrerenderServiceImpl>(browser_state_.get());
  }

  PrerenderServiceImplTest(const PrerenderServiceImplTest&) = delete;
  PrerenderServiceImplTest& operator=(const PrerenderServiceImplTest&) = delete;

  ~PrerenderServiceImplTest() override = default;

 protected:
  web::WebTaskEnvironment task_environment_;
  std::unique_ptr<TestChromeBrowserState> browser_state_;
  std::unique_ptr<PrerenderService> service_;
  web::FakeWebState web_state_;
};

}  // namespace

TEST_F(PrerenderServiceImplTest, NoPrerender) {
  GURL test_url("https://www.google.com");
  EXPECT_FALSE(service_->HasPrerenderForUrl(test_url));

  EXPECT_FALSE(service_->IsWebStatePrerendered(&web_state_));
}