chrome/browser/sessions/session_restore_test_helper.h

// Copyright 2015 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CHROME_BROWSER_SESSIONS_SESSION_RESTORE_TEST_HELPER_H_
#define CHROME_BROWSER_SESSIONS_SESSION_RESTORE_TEST_HELPER_H_

#include "base/memory/weak_ptr.h"
#include "chrome/browser/sessions/session_restore.h"

class Profile;

namespace content {
class MessageLoopRunner;
}

// This class waits on the SessionRestore callback.
class SessionRestoreTestHelper {
 public:
  SessionRestoreTestHelper();

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

  ~SessionRestoreTestHelper();

  // Blocks until OnSessionRestore() is called.
  void Wait();

 private:
  // Callback for session restore notifications.
  void OnSessionRestoreDone(Profile* profile, int /* num_tabs_restored */);

  // Indicates whether a session restore notification has been received.
  bool restore_notification_seen_;

  // Indicates whether |message_loop_runner_| is running.
  bool loop_is_running_;

  // Loop that runs while waiting for notification callback.
  scoped_refptr<content::MessageLoopRunner> message_loop_runner_;

  // For automatically unsubscribing from callback-based notifications.
  base::CallbackListSubscription callback_subscription_;

  // For safely binding pointers to callbacks.
  base::WeakPtrFactory<SessionRestoreTestHelper> weak_ptr_factory{this};
};

#endif  // CHROME_BROWSER_SESSIONS_SESSION_RESTORE_TEST_HELPER_H_