chrome/browser/ui/read_later/read_later_test_utils.h

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

#ifndef CHROME_BROWSER_UI_READ_LATER_READ_LATER_TEST_UTILS_H_
#define CHROME_BROWSER_UI_READ_LATER_READ_LATER_TEST_UTILS_H_

#include "base/run_loop.h"
#include "components/reading_list/core/reading_list_model_observer.h"

class GURL;
class ReadingListModel;

namespace test {

// ReadingListLoadObserver is used to observe the ReadingListModel passed in the
// constructor for the ReadingListModelLoaded event.
class ReadingListLoadObserver : public ReadingListModelObserver {
 public:
  explicit ReadingListLoadObserver(ReadingListModel* model);
  ReadingListLoadObserver(const ReadingListLoadObserver&) = delete;
  ReadingListLoadObserver& operator=(const ReadingListLoadObserver&) = delete;
  ~ReadingListLoadObserver() override;

  void Wait();

 private:
  // ReadingListModelObserver:
  void ReadingListModelLoaded(const ReadingListModel* model) override;
  void ReadingListModelBeganBatchUpdates(
      const ReadingListModel* model) override {}
  void ReadingListModelCompletedBatchUpdates(
      const ReadingListModel* model) override {}
  void ReadingListModelBeingShutdown(const ReadingListModel* model) override {}
  void ReadingListModelBeingDeleted(const ReadingListModel* model) override {}
  void ReadingListWillRemoveEntry(const ReadingListModel* model,
                                  const GURL& url) override {}
  void ReadingListWillMoveEntry(const ReadingListModel* model,
                                const GURL& url) override {}
  void ReadingListDidMoveEntry(const ReadingListModel* model,
                               const GURL& url) override {}
  void ReadingListWillAddEntry(const ReadingListModel* model,
                               const ReadingListEntry& entry) override {}
  void ReadingListDidAddEntry(const ReadingListModel* model,
                              const GURL& url,
                              reading_list::EntrySource source) override {}
  void ReadingListWillUpdateEntry(const ReadingListModel* model,
                                  const GURL& url) override {}
  void ReadingListDidApplyChanges(ReadingListModel* model) override {}

  ReadingListModel* const model_;
  base::RunLoop run_loop_;
};

}  // namespace test

#endif  // CHROME_BROWSER_UI_READ_LATER_READ_LATER_TEST_UTILS_H_