chromium/chrome/browser/ash/mahi/mahi_ui_browser_test_base.h

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

#ifndef CHROME_BROWSER_ASH_MAHI_MAHI_UI_BROWSER_TEST_BASE_H_
#define CHROME_BROWSER_ASH_MAHI_MAHI_UI_BROWSER_TEST_BASE_H_

#include <memory>
#include <string>

#include "ash/constants/ash_switches.h"
#include "base/auto_reset.h"
#include "base/test/scoped_feature_list.h"
#include "chrome/browser/ash/mahi/fake_mahi_browser_delegate_ash.h"
#include "chrome/browser/ash/mahi/mahi_manager_impl.h"
#include "chrome/browser/ash/system_web_apps/test_support/system_web_app_browsertest_base.h"
#include "content/public/test/browser_test_utils.h"
#include "ui/events/test/event_generator.h"

namespace views {
class Widget;
}  // namespace views

namespace ash {

class MahiUiBrowserTestBase : public SystemWebAppBrowserTestBase {
 public:
  MahiUiBrowserTestBase();
  MahiUiBrowserTestBase(const MahiUiBrowserTestBase&) = delete;
  MahiUiBrowserTestBase& operator=(const MahiUiBrowserTestBase&) = delete;
  ~MahiUiBrowserTestBase() override;

  void SetUpCommandLine(base::CommandLine* command_line) override;

 protected:
  // SystemWebAppBrowserTestBase:
  void SetUpOnMainThread() override;

  // Mouse clicks on the disclaimer view's accept button or the declination
  // button, specified by `accept`.
  void ClickDisclaimerViewButton(bool accept);

  void TypeStringToMahiMenuTextfield(views::Widget* mahi_menu_widget,
                                     const std::u16string& input);

  void WaitForSettingsToLoad();

  ui::test::EventGenerator& event_generator() { return *event_generator_; }

 private:
  base::AutoReset<bool> ignore_mahi_secret_key_ =
      switches::SetIgnoreMahiSecretKeyForTest();
  std::unique_ptr<ui::test::EventGenerator> event_generator_;
  net::EmbeddedTestServer https_server_;
  FakeMahiBrowserDelegateAsh fake_browser_delegate_;
  ScopedMahiBrowserDelegateOverrider browser_delegate_overrider_{
      &fake_browser_delegate_};
  base::test::ScopedFeatureList feature_list_;
};

}  // namespace ash

#endif  // CHROME_BROWSER_ASH_MAHI_MAHI_UI_BROWSER_TEST_BASE_H_