components/autofill/content/renderer/autofill_agent_test_api.h

// Copyright 2021 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef COMPONENTS_AUTOFILL_CONTENT_RENDERER_AUTOFILL_AGENT_TEST_API_H_
#define COMPONENTS_AUTOFILL_CONTENT_RENDERER_AUTOFILL_AGENT_TEST_API_H_

#include "base/memory/raw_ref.h"
#include "base/types/optional_ref.h"
#include "components/autofill/content/renderer/autofill_agent.h"

namespace autofill {

class AutofillAgentTestApi {
 public:
  explicit AutofillAgentTestApi(AutofillAgent* agent) : agent_(*agent) {}

  bool is_dom_content_loaded() const { return agent_->is_dom_content_loaded_; }

  FormTracker& form_tracker() { return *agent_->form_tracker_; }
  void set_form_tracker(std::unique_ptr<FormTracker> form_tracker) {
    agent_->form_tracker_ = std::move(form_tracker);
  }

  std::optional<FormData> provisionally_saved_form() {
    return agent_->provisionally_saved_form();
  }

  void FocusedElementChanged(blink::WebElement new_focused_element) {
    agent_->FocusedElementChanged(new_focused_element);
  }

  void ShowSuggestions(
      const blink::WebFormControlElement& element,
      AutofillSuggestionTriggerSource trigger_source,
      const SynchronousFormCache& form_cache,
      std::optional<PasswordSuggestionRequest> password_request) {
    agent_->ShowSuggestions(element, trigger_source, form_cache,
                            password_request);
  }

  void ShowSuggestionsForContentEditable(
      const blink::WebElement& element,
      AutofillSuggestionTriggerSource trigger_source) {
    agent_->ShowSuggestionsForContentEditable(element, trigger_source);
  }

  void OnFormNoLongerSubmittable() { agent_->OnFormNoLongerSubmittable(); }

  const FormCache& form_cache() { return agent_->form_cache_; }

  const base::OneShotTimer& process_forms_after_dynamic_change_timer() {
    return agent_->process_forms_after_dynamic_change_timer_;
  }

 private:
  const raw_ref<AutofillAgent> agent_;
};

inline AutofillAgentTestApi test_api(AutofillAgent& agent) {
  return AutofillAgentTestApi(&agent);
}

}  // namespace autofill

#endif  // COMPONENTS_AUTOFILL_CONTENT_RENDERER_AUTOFILL_AGENT_TEST_API_H_