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

#include "base/command_line.h"
#include "base/test/scoped_feature_list.h"
#include "build/build_config.h"
#include "build/config/coverage/buildflags.h"
#include "chrome/browser/ui/omnibox/omnibox_next_features.h"
#include "chrome/common/webui_url_constants.h"
#include "chrome/test/base/test_switches.h"
#include "chrome/test/base/web_ui_mocha_browser_test.h"
#include "components/omnibox/browser/aim_eligibility_service_features.h"
#include "components/omnibox/common/omnibox_features.h"
#include "content/public/test/browser_test.h"

class OmniboxPopupTest : public WebUIMochaBrowserTest {
 protected:
  OmniboxPopupTest() {
    set_test_loader_host(chrome::kChromeUIOmniboxPopupHost);
    scoped_feature_list_.InitWithFeatures(
        {omnibox::internal::kWebUIOmniboxPopup},
        {omnibox::kWebUIOmniboxFullPopup, omnibox::kAimUsePecApi});
  }

 private:
  base::test::ScopedFeatureList scoped_feature_list_;
};

IN_PROC_BROWSER_TEST_F(OmniboxPopupTest, App) {
  RunTest("omnibox_popup/app_test.js", "mocha.run();");
}

IN_PROC_BROWSER_TEST_F(OmniboxPopupTest, Searchbox) {
  RunTest("omnibox_popup/omnibox_popup_searchbox_test.js", "mocha.run();");
}

IN_PROC_BROWSER_TEST_F(OmniboxPopupTest, TextfieldModel) {
  RunTest("omnibox_popup/textfield_model_test.js", "mocha.run();");
}

class OmniboxPopupFullTest : public WebUIMochaBrowserTest {
 protected:
  OmniboxPopupFullTest() {
    set_test_loader_host(chrome::kChromeUIOmniboxPopupHost);
  }

 private:
  base::test::ScopedFeatureList scoped_feature_list_{
      omnibox::kWebUIOmniboxFullPopup};
};

IN_PROC_BROWSER_TEST_F(OmniboxPopupFullTest, App) {
  RunTest("omnibox_popup/full_app_test.js", "mocha.run();");
}

class OmniboxPopupAimTest : public WebUIMochaBrowserTest {
 protected:
  OmniboxPopupAimTest() {
    set_test_loader_host(chrome::kChromeUIOmniboxPopupHost);
  }

 private:
  base::test::ScopedFeatureList scoped_feature_list_{
      omnibox::internal::kWebUIOmniboxAimPopup};
};

IN_PROC_BROWSER_TEST_F(OmniboxPopupAimTest, App) {
  RunTest("omnibox_popup/aim_app_test.js", "mocha.run();");
}

IN_PROC_BROWSER_TEST_F(OmniboxPopupAimTest, PopupContextualEntrypointButton) {
  RunTest("omnibox_popup/omnibox_popup_contextual_entrypoint_button_test.js",
          "mocha.run();");
}

IN_PROC_BROWSER_TEST_F(OmniboxPopupAimTest, PopupContextualEntrypoint) {
  RunTest("omnibox_popup/omnibox_popup_contextual_entrypoint_test.js",
          "mocha.run();");
}

// TODO(crbug.com/519692372): Investigate why this fails on ChromeOS.
// TODO(crbug.com/532163956): Flaky on Linux Debug.
#if BUILDFLAG(IS_CHROMEOS) || (BUILDFLAG(IS_LINUX) && !defined(NDEBUG))
#define MAYBE_Composebox DISABLED_Composebox
#else
#define MAYBE_Composebox Composebox
#endif
IN_PROC_BROWSER_TEST_F(OmniboxPopupAimTest, MAYBE_Composebox) {
  // TODO(crbug.com/486707998): Remove this skip once the coverage harness
  // crash in DevToolsListener::RetrieveMissingScripts is fixed.
  if (base::CommandLine::ForCurrentProcess()->HasSwitch(
          switches::kDevtoolsCodeCoverage)) {
    GTEST_SKIP() << "Skipping test on coverage builders due to harness crash.";
  }

  RunTest("omnibox_popup/omnibox_composebox_test.js", "mocha.run();");
}