components/translate/core/browser/mock_translate_driver.h

// Copyright 2014 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_TRANSLATE_CORE_BROWSER_MOCK_TRANSLATE_DRIVER_H_
#define COMPONENTS_TRANSLATE_CORE_BROWSER_MOCK_TRANSLATE_DRIVER_H_

#include <memory>

#include "base/macros.h"
#include "components/translate/core/browser/language_state.h"
#include "components/translate/core/browser/translate_driver.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "url/gurl.h"

namespace translate {

namespace testing {

class MockTranslateDriver : public TranslateDriver {
 public:
  MockTranslateDriver();

  void Reset();

  virtual ~MockTranslateDriver() {}

  void OnIsPageTranslatedChanged() override;

  void OnTranslateEnabledChanged() override;

  bool IsLinkNavigation() override;

  void TranslatePage(int page_seq_no,
                     const std::string& translate_script,
                     const std::string& source_lang,
                     const std::string& target_lang) override {}

  void RevertTranslation(int page_seq_no) override {}

  bool IsOffTheRecord() override;

  const std::string& GetContentsMimeType() override;

  const GURL& GetLastCommittedURL() override;

  const GURL& GetVisibleURL() override;

  bool HasCurrentPage() override;

  void OpenUrlInNewTab(const GURL& url) override {}

  bool on_is_page_translated_changed_called() const {
    return on_is_page_translated_changed_called_;
  }

  bool on_translate_enabled_changed_called() const {
    return on_translate_enabled_changed_called_;
  }

 private:
  bool on_is_page_translated_changed_called_;
  bool on_translate_enabled_changed_called_;
  LanguageState language_state_;

  DISALLOW_COPY_AND_ASSIGN(MockTranslateDriver);
};

}  // namespace testing

}  // namespace translate

#endif  // COMPONENTS_TRANSLATE_CORE_BROWSER_MOCK_TRANSLATE_DRIVER_H_