// Copyright 2018 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.

#include "chrome/browser/sessions/session_restore_test_utils.h"

#include "chrome/browser/sessions/tab_loader_delegate.h"

namespace testing {

bool AlwayLoadSessionRestorePolicy::ShouldLoad(
    content::WebContents* contents) const {
  return true;
}

ScopedAlwaysLoadSessionRestoreTestPolicy::
    ScopedAlwaysLoadSessionRestoreTestPolicy() {
  TabLoaderDelegate::SetSessionRestorePolicyForTesting(&policy_);
}

ScopedAlwaysLoadSessionRestoreTestPolicy::
    ~ScopedAlwaysLoadSessionRestoreTestPolicy() {
  TabLoaderDelegate::SetSessionRestorePolicyForTesting(nullptr);
}

}  // namespace testing