components/zoom/test/zoom_test_utils.h

// Copyright 2015 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 COMPONENTS_ZOOM_TEST_ZOOM_TEST_UTILS_H_
#define COMPONENTS_ZOOM_TEST_ZOOM_TEST_UTILS_H_

#include "base/callback_forward.h"
#include "base/macros.h"
#include "components/zoom/zoom_controller.h"
#include "components/zoom/zoom_observer.h"

namespace content {
class MessageLoopRunner;
}

namespace zoom {

bool operator==(const ZoomController::ZoomChangedEventData& lhs,
                const ZoomController::ZoomChangedEventData& rhs);

class ZoomChangedWatcher : public zoom::ZoomObserver {
 public:
  using ZoomEventPred = base::RepeatingCallback<bool(
      const ZoomController::ZoomChangedEventData&)>;

  // Used to wait until we see a zoom changed event that satisfies the
  // given |predicate|.
  ZoomChangedWatcher(ZoomController* zoom_controller, ZoomEventPred predicate);
  ZoomChangedWatcher(content::WebContents* web_contents,
                     ZoomEventPred predicate);

  // Used to wait until we see a zoom changed event equal to the given
  // |expected_event_data|.
  ZoomChangedWatcher(
      ZoomController* zoom_controller,
      const ZoomController::ZoomChangedEventData& expected_event_data);
  ZoomChangedWatcher(
      content::WebContents* web_contents,
      const ZoomController::ZoomChangedEventData& expected_event_data);

  ~ZoomChangedWatcher() override;

  void Wait();

  // zoom::ZoomObserver:
  void OnZoomChanged(
      const ZoomController::ZoomChangedEventData& event_data) override;

 private:
  ZoomController* zoom_controller_;
  ZoomEventPred predicate_;
  scoped_refptr<content::MessageLoopRunner> message_loop_runner_;
  bool change_received_ = false;

  DISALLOW_COPY_AND_ASSIGN(ZoomChangedWatcher);
};

}  // namespace zoom
#endif  // COMPONENTS_ZOOM_TEST_ZOOM_TEST_UTILS_H_