// 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 "chrome/browser/ui/pdf/infobar/pdf_infobar_delegate.h"

#include "base/test/metrics/histogram_tester.h"
#include "base/test/scoped_feature_list.h"
#include "chrome/browser/ui/ui_features.h"
#include "chrome/browser/ui/views/chrome_layout_provider.h"
#include "chrome/test/base/testing_profile.h"
#include "components/infobars/content/content_infobar_manager.h"
#include "components/infobars/core/infobar.h"
#include "content/public/test/browser_task_environment.h"
#include "content/public/test/test_renderer_host.h"
#include "content/public/test/web_contents_tester.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace pdf::infobar {

class PdfInfoBarDelegateTest : public testing::Test {
 protected:
  PdfInfoBarDelegateTest() {
    feature_list_.InitAndDisableFeature(features::kPdfInfoBar);
  }

  void SetUp() override {
    infobar_manager_ =
        std::make_unique<infobars::ContentInfoBarManager>(web_contents_.get());
  }

  infobars::ContentInfoBarManager* infobar_manager() {
    return infobar_manager_.get();
  }

  base::HistogramTester& histogram_tester() { return histogram_tester_; }

 private:
  // Must be the first member.
  content::BrowserTaskEnvironment task_environment_;

  base::test::ScopedFeatureList feature_list_;
  base::HistogramTester histogram_tester_;

  std::unique_ptr<infobars::ContentInfoBarManager> infobar_manager_;
  ChromeLayoutProvider layout_provider_;
  TestingProfile profile_;
  content::RenderViewHostTestEnabler render_view_host_test_enabler_;
  std::unique_ptr<content::WebContents> web_contents_{
      content::WebContentsTester::CreateTestWebContents(
          content::WebContents::CreateParams(&profile_))};
};

// Executes the code to ensure that creating the infobar doesn't crash. When the
// infobar is created, the "shown" histogram should be recorded.
TEST_F(PdfInfoBarDelegateTest, Create) {
  EXPECT_TRUE(PdfInfoBarDelegate::Create(infobar_manager()));
  histogram_tester().ExpectUniqueSample("PDF.InfoBar.Shown", true, 1);
}

// When the infobar is dismissed, the "dismissed" histogram should be recorded.
// "Accept" is not tested here because it opens an OS dialog.
TEST_F(PdfInfoBarDelegateTest, DismissedHistogram) {
  infobars::InfoBar* infobar = PdfInfoBarDelegate::Create(infobar_manager());
  static_cast<PdfInfoBarDelegate*>(infobar->delegate())->InfoBarDismissed();
  histogram_tester().ExpectUniqueSample(
      "PDF.InfoBar.UserInteraction", PdfInfoBarUserInteraction::kDismissed, 1);
}

// When the infobar is destroyed without being accepted or dismissed, the
// "ignored" histogram should be recorded.
TEST_F(PdfInfoBarDelegateTest, IgnoredHistogram) {
  infobars::InfoBar* infobar = PdfInfoBarDelegate::Create(infobar_manager());
  infobar_manager()->RemoveInfoBar(infobar);
  histogram_tester().ExpectUniqueSample("PDF.InfoBar.UserInteraction",
                                        PdfInfoBarUserInteraction::kIgnored, 1);
}

// When the infobar is destroyed after being dismissed, the "dismissed"
// histogram (not the "ignored" histogram) should be recorded.
TEST_F(PdfInfoBarDelegateTest, DismissedHistogramInfoBarDestroyed) {
  infobars::InfoBar* infobar = PdfInfoBarDelegate::Create(infobar_manager());
  static_cast<PdfInfoBarDelegate*>(infobar->delegate())->InfoBarDismissed();
  infobar_manager()->RemoveInfoBar(infobar);
  histogram_tester().ExpectUniqueSample(
      "PDF.InfoBar.UserInteraction", PdfInfoBarUserInteraction::kDismissed, 1);
}

}  // namespace pdf::infobar