components/autofill_assistant/browser/mock_web_controller.h

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

#ifndef COMPONENTS_AUTOFILL_ASSISTANT_BROWSER_MOCK_WEB_CONTROLLER_H_
#define COMPONENTS_AUTOFILL_ASSISTANT_BROWSER_MOCK_WEB_CONTROLLER_H_

#include <string>
#include <vector>

#include "base/callback.h"
#include "components/autofill_assistant/browser/web_controller.h"
#include "testing/gmock/include/gmock/gmock.h"

namespace autofill_assistant {
struct RectF;

class MockWebController : public WebController {
 public:
  MockWebController();
  ~MockWebController() override;

  MOCK_METHOD0(GetUrl, const GURL&());

  MOCK_METHOD1(LoadURL, void(const GURL&));

  void ClickOrTapElement(const std::vector<std::string>& selectors,
                         base::OnceCallback<void(bool)> callback) override {
    // Transforming callback into a references allows using RunOnceCallback on
    // the argument.
    OnClickOrTapElement(selectors, callback);
  }
  MOCK_METHOD2(OnClickOrTapElement,
               void(const std::vector<std::string>& selectors,
                    base::OnceCallback<void(bool)>& callback));

  void FocusElement(const std::vector<std::string>& selectors,
                    base::OnceCallback<void(bool)> callback) override {
    OnFocusElement(selectors, callback);
  }
  MOCK_METHOD2(OnFocusElement,
               void(const std::vector<std::string>& selectors,
                    base::OnceCallback<void(bool)>& callback));

  void ElementCheck(ElementCheckType check_type,
                    const std::vector<std::string>& selectors,
                    base::OnceCallback<void(bool)> callback) override {
    OnElementCheck(check_type, selectors, callback);
  }
  MOCK_METHOD3(OnElementCheck,
               void(ElementCheckType check_type,
                    const std::vector<std::string>& selectors,
                    base::OnceCallback<void(bool)>& callback));

  void GetFieldValue(
      const std::vector<std::string>& selectors,
      base::OnceCallback<void(bool, const std::string&)> callback) override {
    OnGetFieldValue(selectors, callback);
  }
  MOCK_METHOD2(
      OnGetFieldValue,
      void(const std::vector<std::string>& selectors,
           base::OnceCallback<void(bool, const std::string&)>& callback));

  void GetElementPosition(
      const std::vector<std::string>& selectors,
      base::OnceCallback<void(bool, const RectF&)> callback) override {
    OnGetElementPosition(selectors, callback);
  }

  MOCK_METHOD2(OnGetElementPosition,
               void(const std::vector<std::string>& selectors,
                    base::OnceCallback<void(bool, const RectF&)>& callback));
};

}  // namespace autofill_assistant

#endif  // COMPONENTS_AUTOFILL_ASSISTANT_BROWSER_MOCK_WEB_CONTROLLER_H_