// Copyright 2019 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 WEBLAYER_TEST_TEST_NAVIGATION_OBSERVER_H_
#define WEBLAYER_TEST_TEST_NAVIGATION_OBSERVER_H_

#include "base/macros.h"
#include "base/optional.h"
#include "base/run_loop.h"
#include "url/gurl.h"
#include "weblayer/public/navigation_observer.h"

namespace weblayer {

class Shell;
class Tab;

// A helper that waits for a navigation to finish.
class TestNavigationObserver : public NavigationObserver {
 public:
  enum class NavigationEvent { kStart, kCompletion, kFailure };

  // Creates an instance that begins waiting for a Navigation within |shell| and
  // to |url| to reach the specified |target_event|.
  TestNavigationObserver(const GURL& url,
                         NavigationEvent target_event,
                         Shell* shell);
  TestNavigationObserver(const GURL& url,
                         NavigationEvent target_event,
                         Tab* tab);
  ~TestNavigationObserver() override;

  // Spins a RunLoop until the requested type of navigation event is observed.
  void Wait();

 private:
  // NavigationObserver implementation:
  void NavigationStarted(Navigation* navigation) override;
  void NavigationCompleted(Navigation* navigation) override;
  void NavigationFailed(Navigation* navigation) override;
  void LoadStateChanged(bool is_loading, bool to_different_document) override;

  void CheckNavigationCompleted();

  const GURL url_;
  base::Optional<NavigationEvent> observed_event_;
  NavigationEvent target_event_;
  Tab* tab_;
  bool done_loading_ = false;
  base::RunLoop run_loop_;

  DISALLOW_COPY_AND_ASSIGN(TestNavigationObserver);
};

}  // namespace weblayer

#endif  // WEBLAYER_TEST_TEST_NAVIGATION_OBSERVER_H_