// Copyright 2023 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/strings/string_util.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/actions/chrome_action_id.h"
#include "chrome/browser/ui/commerce/mock_commerce_ui_tab_helper.h"
#include "chrome/browser/ui/page_action/page_action_icon_type.h"
#include "chrome/browser/ui/tabs/public/tab_features.h"
#include "chrome/browser/ui/test/test_browser_ui.h"
#include "chrome/browser/ui/views/commerce/price_insights_page_action_view_controller.h"
#include "chrome/browser/ui/views/frame/browser_view.h"
#include "chrome/browser/ui/views/frame/toolbar_button_provider.h"
#include "chrome/browser/ui/views/page_action/page_action_view_interface.h"
#include "chrome/browser/ui/views/page_action/test_support/page_action_test_accessor.h"
#include "chrome/browser/ui/views/toolbar/toolbar_view.h"
#include "chrome/test/base/ui_test_utils.h"
#include "components/commerce/core/commerce_feature_list.h"
#include "components/commerce/core/test_utils.h"
#include "components/feature_engagement/public/feature_constants.h"
#include "components/strings/grit/components_strings.h"
#include "components/tabs/public/tab_interface.h"
#include "content/public/test/browser_test.h"

namespace {
const char kTestURL[] = "about:blank";
}  // namespace

class PriceInsightsIconViewBaseBrowserTest : public UiBrowserTest {
 public:
  PriceInsightsIconViewBaseBrowserTest() {
    commerce_ui_override_ = MockCommerceUiTabHelper::ReplaceFactory();
  }

  // UiBrowserTest:
  void PreShow() override {
    MockCommerceUiTabHelper* mock_tab_helper = getTabHelper();
    EXPECT_CALL(*mock_tab_helper, ShouldShowPriceInsightsIconView)
        .Times(testing::AnyNumber());
    ON_CALL(*mock_tab_helper, ShouldShowPriceInsightsIconView)
        .WillByDefault(testing::Return(true));

    EXPECT_CALL(*mock_tab_helper, GetPriceInsightsInfo)
        .Times(testing::AnyNumber());
    ON_CALL(*mock_tab_helper, GetPriceInsightsInfo)
        .WillByDefault(testing::ReturnRef(price_insights_info_));
    EXPECT_CALL(*mock_tab_helper, ShouldExpandPageActionIcon)
        .WillRepeatedly(testing::Return(true));

    PriceInsightsIconLabelType label_type = PriceInsightsIconLabelType::kNone;
    std::string test_name =
        testing::UnitTest::GetInstance()->current_test_info()->name();
    if (test_name == "InvokeUi_show_price_insights_icon_with_low_price_label") {
      label_type = PriceInsightsIconLabelType::kPriceIsLow;
    } else if (test_name ==
               "InvokeUi_show_price_insights_icon_with_high_price_label") {
      label_type = PriceInsightsIconLabelType::kPriceIsHigh;
    }

    EXPECT_CALL(*mock_tab_helper, GetPriceInsightsIconLabelTypeForPage)
        .WillRepeatedly(testing::Return(label_type));
  }

  MockCommerceUiTabHelper* getTabHelper() {
    return static_cast<MockCommerceUiTabHelper*>(
        browser()
            ->GetActiveTabInterface()
            ->GetTabFeatures()
            ->commerce_ui_tab_helper());
  }

  void ShowUi(const std::string& name) override {
    ASSERT_TRUE(ui_test_utils::NavigateToURL(browser(), GURL(kTestURL)));

    MockCommerceUiTabHelper* mock_tab_helper = getTabHelper();
    commerce::PriceInsightsPageActionViewController::From(
        *browser()->GetActiveTabInterface())
        ->UpdatePageActionIcon(
            mock_tab_helper->ShouldShowPriceInsightsIconView(),
            mock_tab_helper->ShouldExpandPageActionIcon(
                PageActionIconType::kPriceInsights),
            mock_tab_helper->GetPriceInsightsIconLabelTypeForPage());
  }

  bool VerifyUi() override {
    auto* price_insights_chip = GetChip();
    if (!price_insights_chip) {
      return false;
    }
    EXPECT_EQ(base::ToLowerASCII(price_insights_chip->GetAccessibleName()),
              base::ToLowerASCII(l10n_util::GetStringUTF16(
                  IDS_SHOPPING_INSIGHTS_ICON_TOOLTIP_TEXT)));

    // TODO(meiliang): call VerifyPixelUi here after PriceInsightsIconView is
    // finished implementing.
    return true;
  }

  void WaitForUserDismissal() override {
    // Consider closing the browser to be dismissal.
    ui_test_utils::BrowserDestroyedObserver().Wait();
  }

 protected:
  std::optional<commerce::PriceInsightsInfo> price_insights_info_;

  page_actions::PageActionViewInterface* GetChip() {
    auto* provider = BrowserView::GetBrowserViewForBrowser(browser())
                         ->toolbar_button_provider();
    return provider ? provider->GetPageActionViewInterface(
                          kActionCommercePriceInsights)
                    : nullptr;
  }

 private:

  ui::UserDataFactory::ScopedOverride commerce_ui_override_;
};

class PriceInsightsIconViewBrowserTest
    : public PriceInsightsIconViewBaseBrowserTest {
 public:
  PriceInsightsIconViewBrowserTest() {
    test_features_.InitWithFeaturesAndParameters(
        /*enabled_features=*/
        {
            {commerce::kPriceInsights, {}},
        },
        /*disabled_features=*/{});
  }

 private:
  base::test::ScopedFeatureList test_features_;
};

IN_PROC_BROWSER_TEST_F(PriceInsightsIconViewBrowserTest,
                       InvokeUi_show_price_insights_icon) {
  ShowAndVerifyUi();
}

class PriceInsightsIconViewWithLabelBrowserTest
    : public PriceInsightsIconViewBaseBrowserTest {
 public:
  PriceInsightsIconViewWithLabelBrowserTest() {
    test_features_.InitAndEnableFeaturesWithParameters(
        /*allow_and_enable_features=*/
        {
            {commerce::kPriceInsights,
             {{commerce::kPriceInsightsChipLabelExpandOnHighPriceParam,
               "true"}}},
            {feature_engagement::kIPHPriceInsightsPageActionIconLabelFeature,
             {}},
        },
        /*disable_features=*/{});
  }

  // UiBrowserTest:
  void PreShow() override {
    PriceInsightsIconViewBaseBrowserTest::PreShow();

    std::string test_name =
        testing::UnitTest::GetInstance()->current_test_info()->name();
    if (test_name == "InvokeUi_show_price_insights_icon_with_low_price_label") {
      price_insights_info_ = commerce::CreateValidPriceInsightsInfo(
          true, true, commerce::PriceBucket::kLowPrice);
    } else if (test_name ==
               "InvokeUi_show_price_insights_icon_with_high_price_label") {
      price_insights_info_ = commerce::CreateValidPriceInsightsInfo(
          true, true, commerce::PriceBucket::kHighPrice);
    } else {
      price_insights_info_ = commerce::CreateValidPriceInsightsInfo(
          true, true, commerce::PriceBucket::kTypicalPrice);
    }
  }

  bool VerifyUi() override {
    auto* price_insights_chip = GetChip();
    if (!price_insights_chip) {
      return false;
    }

    page_actions::PageActionTestAccessor accessor(browser(),
                                                  kActionCommercePriceInsights);
    std::string test_name =
        testing::UnitTest::GetInstance()->current_test_info()->name();
    if (test_name == "InvokeUi_show_price_insights_icon_with_low_price_label") {
      EXPECT_TRUE(accessor.IsChipVisible());
      EXPECT_EQ(base::ToLowerASCII(accessor.GetText()), u"price is low");

      // TODO(meiliang): Add pixel test.
    } else if (test_name ==
               "InvokeUi_show_price_insights_icon_with_high_price_label") {
      EXPECT_TRUE(accessor.IsChipVisible());
      EXPECT_EQ(base::ToLowerASCII(accessor.GetText()), u"price is high");

      // TODO(meiliang): Add pixel test.
    }
    EXPECT_EQ(base::ToLowerASCII(price_insights_chip->GetAccessibleName()),
              base::ToLowerASCII(l10n_util::GetStringUTF16(
                  IDS_SHOPPING_INSIGHTS_ICON_TOOLTIP_TEXT)));
    return true;
  }

 private:
  feature_engagement::test::ScopedIphFeatureList test_features_;
};

IN_PROC_BROWSER_TEST_F(PriceInsightsIconViewWithLabelBrowserTest,
                       InvokeUi_show_price_insights_icon_with_low_price_label) {
  ShowAndVerifyUi();
}

IN_PROC_BROWSER_TEST_F(
    PriceInsightsIconViewWithLabelBrowserTest,
    InvokeUi_show_price_insights_icon_with_high_price_label) {
  ShowAndVerifyUi();
}