chrome/browser/privacy_budget/identifiability_study_group_settings_unittest.cc

// Copyright 2021 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/privacy_budget/identifiability_study_group_settings.h"

#include "base/test/metrics/histogram_tester.h"
#include "testing/gtest/include/gtest/gtest.h"

class IdentifiabilityStudyGroupSettingsTest : public testing::Test {
 public:
  base::HistogramTester histogram_tester;
};

TEST_F(IdentifiabilityStudyGroupSettingsTest, Disabled) {
  auto settings = IdentifiabilityStudyGroupSettings::InitFrom(
      false, 0, 0, "", "", "", "", "", "", "");
  EXPECT_FALSE(settings.enabled());
  histogram_tester.ExpectUniqueSample(
      "PrivacyBudget.Identifiability.FinchConfigValidationResult", true, 1);
}

TEST_F(IdentifiabilityStudyGroupSettingsTest, DisabledWithParams) {
  auto settings = IdentifiabilityStudyGroupSettings::InitFrom(
      false, 10, 40, "", "", "", "", "", "", "");
  EXPECT_FALSE(settings.enabled());
  histogram_tester.ExpectUniqueSample(
      "PrivacyBudget.Identifiability.FinchConfigValidationResult", true, 1);
}

TEST_F(IdentifiabilityStudyGroupSettingsTest, DisabledBySurfaceCountZero) {
  auto settings = IdentifiabilityStudyGroupSettings::InitFrom(
      true, 0, 40, "", "", "", "", "", "", "");
  EXPECT_FALSE(settings.enabled());
  histogram_tester.ExpectUniqueSample(
      "PrivacyBudget.Identifiability.FinchConfigValidationResult", false, 1);
}

TEST_F(IdentifiabilityStudyGroupSettingsTest, ValidRandomSurfaceSampling) {
  auto settings = IdentifiabilityStudyGroupSettings::InitFrom(
      true, 10, 40, "", "", "1,4", "", "", "", "");
  EXPECT_TRUE(settings.enabled());
  EXPECT_FALSE(settings.IsUsingAssignedBlockSampling());
  EXPECT_TRUE(settings.IsUsingRandomSampling());
  EXPECT_TRUE(settings.IsUsingSamplingOfSurfaces());
  EXPECT_EQ(10, settings.expected_surface_count());
  EXPECT_EQ(40, settings.surface_budget());
  const std::vector<blink::IdentifiableSurface::Type> expected_allowed_types{
      blink::IdentifiableSurface::Type::kWebFeature,
      blink::IdentifiableSurface::Type::kGenericFontLookup};
  EXPECT_EQ(expected_allowed_types, settings.allowed_random_types());

  histogram_tester.ExpectUniqueSample(
      "PrivacyBudget.Identifiability.FinchConfigValidationResult", true, 1);
}

TEST_F(IdentifiabilityStudyGroupSettingsTest, ValidAssignedBlockSampling) {
  auto settings = IdentifiabilityStudyGroupSettings::InitFrom(
      true, 0, 0, "1;2,3;4,5;6", "1,1,1", "", "", "", "", "");
  EXPECT_TRUE(settings.enabled());
  EXPECT_TRUE(settings.IsUsingAssignedBlockSampling());
  EXPECT_FALSE(settings.IsUsingRandomSampling());
  EXPECT_TRUE(settings.IsUsingSamplingOfSurfaces());
  histogram_tester.ExpectUniqueSample(
      "PrivacyBudget.Identifiability.FinchConfigValidationResult", true, 1);
}

TEST_F(IdentifiabilityStudyGroupSettingsTest, InvalidNegativeWeight) {
  auto settings = IdentifiabilityStudyGroupSettings::InitFrom(
      true, 0, 0, "1;2,3;4,5;6", "-1,1,1", "", "", "", "", "");
  EXPECT_FALSE(settings.enabled());
  histogram_tester.ExpectUniqueSample(
      "PrivacyBudget.Identifiability.FinchConfigValidationResult", false, 1);
}

TEST_F(IdentifiabilityStudyGroupSettingsTest, InvalidSurfaceTooLikely) {
  auto settings = IdentifiabilityStudyGroupSettings::InitFrom(
      true, 0, 0, "1;2,1;4,5;6", "1,1,1", "", "", "", "", "");
  EXPECT_FALSE(settings.enabled());
  histogram_tester.ExpectUniqueSample(
      "PrivacyBudget.Identifiability.FinchConfigValidationResult", false, 1);
}

TEST_F(IdentifiabilityStudyGroupSettingsTest, EnableSettingsForValidReidBlock) {
  auto settings = IdentifiabilityStudyGroupSettings::InitFrom(
      true, 0, 0, "", "", "", "1;2,4;5;6", "2,2", "2,3", "0.1,0.2");
  EXPECT_TRUE(settings.IsUsingReidScoreEstimator());
  EXPECT_FALSE(settings.IsUsingSamplingOfSurfaces());
  EXPECT_TRUE(settings.enabled());
  histogram_tester.ExpectUniqueSample(
      "PrivacyBudget.Identifiability.FinchConfigValidationResult", true, 1);
}

TEST_F(IdentifiabilityStudyGroupSettingsTest, InvalidReidBlocks) {
  auto settings = IdentifiabilityStudyGroupSettings::InitFrom(
      true, 0, 0, "", "", "", "1;2,4;5;6", "2", "2,3", "0.1,0.2");
  EXPECT_FALSE(settings.enabled());
  histogram_tester.ExpectUniqueSample(
      "PrivacyBudget.Identifiability.FinchConfigValidationResult", false, 1);
}