chromium/chrome/updater/update_usage_stats_task_unittest.cc

// Copyright 2022 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/updater/update_usage_stats_task.h"

#include <memory>
#include <optional>
#include <string>
#include <vector>

#include "base/files/file_path.h"
#include "base/files/file_util.h"
#include "base/functional/callback_helpers.h"
#include "base/strings/strcat.h"
#include "base/test/bind.h"
#include "build/branding_buildflags.h"
#include "build/build_config.h"
#include "build/buildflag.h"
#include "chrome/updater/constants.h"
#include "chrome/updater/crash_client.h"
#include "chrome/updater/test/test_scope.h"
#include "chrome/updater/updater_branding.h"
#include "chrome/updater/util/util.h"
#include "components/update_client/update_client.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/crashpad/crashpad/client/crash_report_database.h"
#include "third_party/crashpad/crashpad/client/settings.h"

#if BUILDFLAG(IS_MAC)
#include "chrome/updater/util/mac_util.h"
#elif BUILDFLAG(IS_WIN)
#include "base/strings/sys_string_conversions.h"
#include "base/win/registry.h"
#include "base/win/windows_types.h"
#include "chrome/updater/util/win_util.h"
#include "chrome/updater/win/win_constants.h"
#endif

namespace updater {

namespace {

#if BUILDFLAG(IS_MAC)
base::FilePath AppIDToPath(const std::string& app_id) {
  std::optional<base::FilePath> application_support_dir =
      GetApplicationSupportDirectory(UpdaterScope::kUser);
  EXPECT_TRUE(application_support_dir);
  return (*application_support_dir)
      .Append(FILE_PATH_LITERAL(COMPANY_SHORTNAME_STRING))
      .AppendASCII(base::StrCat({"UpdateUsageStatsTaskTest_", app_id}));
}
#endif

#if BUILDFLAG(IS_WIN)
std::vector<std::wstring> UsageStatsRegKeyPaths() {
  std::vector<std::wstring> key_paths = {CLIENT_STATE_KEY};
  if (IsSystemInstall(GetUpdaterScopeForTesting())) {
    key_paths.push_back(CLIENT_STATE_MEDIUM_KEY);
  }
  return key_paths;
}
#endif

void ClearAppUsageStats(const std::string& app_id) { … }

}  // namespace

class UpdateUsageStatsTaskTest : public testing::Test { … };

#if BUILDFLAG(IS_WIN)
TEST_F(UpdateUsageStatsTaskTest, NoApps) {
  ClearAppUsageStats("app1");
  ClearAppUsageStats("app2");
  ASSERT_FALSE(
      OtherAppUsageStatsAllowed({"app1", "app2"}, GetUpdaterScopeForTesting()));
  ASSERT_FALSE(
      AreRawUsageStatsEnabled(GetUpdaterScopeForTesting(), {"app1", "app2"}));
}

TEST_F(UpdateUsageStatsTaskTest, OneAppEnabled) {
  for (const auto& key_path : UsageStatsRegKeyPaths()) {
    ClearAppUsageStats("app1");
    ClearAppUsageStats("app2");
    SetAppUsageStats(key_path, "app1", true);
    SetAppUsageStats(key_path, "app2", false);
    ASSERT_TRUE(OtherAppUsageStatsAllowed({"app1", "app2"},
                                          GetUpdaterScopeForTesting()));
    ASSERT_TRUE(
        AreRawUsageStatsEnabled(GetUpdaterScopeForTesting(), {"app1", "app2"}));
  }
}

TEST_F(UpdateUsageStatsTaskTest, ZeroAppsEnabled) {
  for (const auto& key_path : UsageStatsRegKeyPaths()) {
    ClearAppUsageStats("app1");
    ClearAppUsageStats("app2");
    SetAppUsageStats(key_path, "app1", false);
    SetAppUsageStats(key_path, "app2", false);
    ASSERT_FALSE(OtherAppUsageStatsAllowed({"app1", "app2"},
                                           GetUpdaterScopeForTesting()));
    ASSERT_FALSE(
        AreRawUsageStatsEnabled(GetUpdaterScopeForTesting(), {"app1", "app2"}));
  }
}

TEST_F(UpdateUsageStatsTaskTest,
       SystemInstallClientStateMediumKeyShadowClientStateKey) {
  if (!IsSystemInstall(GetUpdaterScopeForTesting())) {
    return;
  }
  SetAppUsageStats(CLIENT_STATE_MEDIUM_KEY, "app1", false);
  SetAppUsageStats(CLIENT_STATE_KEY, "app1", true);
  ASSERT_FALSE(
      OtherAppUsageStatsAllowed({"app1"}, GetUpdaterScopeForTesting()));

  SetAppUsageStats(CLIENT_STATE_MEDIUM_KEY, "app1", true);
  SetAppUsageStats(CLIENT_STATE_KEY, "app1", false);
  ASSERT_TRUE(OtherAppUsageStatsAllowed({"app1"}, GetUpdaterScopeForTesting()));
  ASSERT_TRUE(AreRawUsageStatsEnabled(GetUpdaterScopeForTesting(), {"app1"}));
}
#elif !BUILDFLAG(IS_MAC) || !BUILDFLAG(GOOGLE_CHROME_BRANDING)
// Mac Google-branded builds may pick up Chrome or other Google software
// usagestat opt-ins from outside this test. Disable the test in that
// configuration.
TEST_F(UpdateUsageStatsTaskTest, NoApps) { … }

// TODO(crbug.com/40821596): Enable tests once the feature is implemented.
#if !BUILDFLAG(IS_LINUX)
TEST_F(UpdateUsageStatsTaskTest, OneAppEnabled) {
  SetAppUsageStats("app1", true);
  SetAppUsageStats("app2", false);
  ASSERT_TRUE(
      OtherAppUsageStatsAllowed({"app1", "app2"}, GetUpdaterScopeForTesting()));
}
#endif  // !BUILDFLAG(IS_LINUX)

TEST_F(UpdateUsageStatsTaskTest, ZeroAppsEnabled) { … }
#endif

}  // namespace updater