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

#include <optional>

#include "base/memory/raw_ptr.h"
#include "base/run_loop.h"
#include "base/time/time.h"
#include "chrome/browser/metrics/ukm_background_recorder_service.h"
#include "chrome/browser/ui/browser_window/public/browser_window_interface.h"
#include "chrome/test/base/in_process_browser_test.h"
#include "components/history/core/browser/history_service.h"
#include "content/public/test/browser_test.h"
#include "services/metrics/public/cpp/ukm_builders.h"
#include "services/metrics/public/cpp/ukm_source_id.h"
#include "url/gurl.h"
#include "url/origin.h"

namespace {

constexpr char kVisitedUrl[] = "https://foobar.com/baz";

void DidGetRecordResult(base::OnceClosure quit_closure,
                        std::optional<ukm::SourceId>* out_result,
                        std::optional<ukm::SourceId> result) {
  *out_result = std::move(result);
  std::move(quit_closure).Run();
}

}  // namespace

class UkmBackgroundRecorderBrowserTest : public InProcessBrowserTest {
 public:
  UkmBackgroundRecorderBrowserTest() = default;

  UkmBackgroundRecorderBrowserTest(const UkmBackgroundRecorderBrowserTest&) =
      delete;
  UkmBackgroundRecorderBrowserTest& operator=(
      const UkmBackgroundRecorderBrowserTest&) = delete;

  ~UkmBackgroundRecorderBrowserTest() override = default;

  void SetUpOnMainThread() override {
    background_recorder_service_ =
        ukm::UkmBackgroundRecorderFactory::GetForProfile(
            browser()->GetProfile());
    DCHECK(background_recorder_service_);

    // Adds the URL to the history so that UKM events for this origin are
    // recorded.
    background_recorder_service_->history_service_->AddPage(
        GURL(kVisitedUrl), base::Time::Now(), history::SOURCE_BROWSED);
  }

  void TearDownOnMainThread() override {
    background_recorder_service_ = nullptr;
  }

 protected:
  std::optional<ukm::SourceId> GetSourceId(const url::Origin& origin) {
    std::optional<ukm::SourceId> result;

    base::RunLoop run_loop;
    background_recorder_service_->GetBackgroundSourceIdIfAllowed(
        origin,
        base::BindOnce(&DidGetRecordResult, run_loop.QuitClosure(), &result));
    run_loop.Run();

    return result;
  }

 private:
  raw_ptr<ukm::UkmBackgroundRecorderService> background_recorder_service_ =
      nullptr;
};

IN_PROC_BROWSER_TEST_F(UkmBackgroundRecorderBrowserTest,
                       SourceIdReturnedWhenOriginInHistory) {
  // Check visited origin.
  auto source_id = GetSourceId(url::Origin::Create(GURL(kVisitedUrl)));
  ASSERT_TRUE(source_id);
  EXPECT_NE(*source_id, ukm::kInvalidSourceId);
  EXPECT_EQ(ukm::GetSourceIdType(*source_id), ukm::SourceIdType::HISTORY_ID);

  // Check unvisited origin.
  EXPECT_FALSE(
      GetSourceId(url::Origin::Create(GURL("https://notvisited.com"))));
}