// 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/metrics/google_update_metrics_provider_mac.h"

#include "base/functional/callback_helpers.h"
#include "base/hash/hash.h"
#include "base/run_loop.h"
#include "base/test/bind.h"
#include "base/test/task_environment.h"
#include "base/version.h"
#include "build/build_config.h"
#include "chrome/browser/buildflags.h"
#include "chrome/browser/updater/browser_updater_client.h"
#include "chrome/browser/updater/browser_updater_client_testutils.h"
#include "chrome/updater/branded_constants.h"
#include "chrome/updater/updater_scope.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/metrics_proto/system_profile.pb.h"

#if BUILDFLAG(ENABLE_UPDATER)
TEST(GoogleUpdateMetricsProviderMacTest, SetsVersions) {
  base::test::SingleThreadTaskEnvironment task_environment;
  updater::UpdateService::AppState app1;
  app1.app_id = updater::kUpdaterAppId;
  app1.version = "1.2.3.4";
  updater::UpdateService::AppState app2;
  app2.app_id = updater::GetAppId();
  app2.ecp = updater::GetExpectedEcp();
  app2.version = "5.6.7.8";
  base::ScopedClosureRunner scoped_override = updater::OverrideService(
      updater::UpdateService::Result::kSuccess, {
                                                    app1,
                                                    app2,
                                                });
  updater::LoadAppStates();
  metrics::SystemProfileProto proto;
  GoogleUpdateMetricsProviderMac().ProvideSystemProfileMetrics(&proto);
  EXPECT_EQ(proto.google_update().client_status().version(), "5.6.7.8");
  EXPECT_EQ(proto.google_update().google_update_status().version(), "1.2.3.4");
}

TEST(GoogleUpdateMetricsProviderMacTest, SetsCohortHash) {
  base::test::SingleThreadTaskEnvironment task_environment;

  updater::UpdateService::AppState app;
  app.app_id = updater::GetAppId();
  app.cohort = "1:A:B";
  base::ScopedClosureRunner scoped_override =
      updater::OverrideService(updater::UpdateService::Result::kSuccess, {app});

  updater::LoadAppStates();

  metrics::SystemProfileProto proto;
  GoogleUpdateMetricsProviderMac().ProvideSystemProfileMetrics(&proto);
  EXPECT_EQ(proto.google_update().client_status().cohort_hash(),
            base::PersistentHash("1:A"));
  EXPECT_FALSE(proto.google_update().google_update_status().has_cohort_hash());
}
#endif