chrome/browser/privacy_budget/identifiability_study_group_settings_unittest.cc

// Copyright 2021 The Chromium Authors. All rights reserved.
// 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 "testing/gtest/include/gtest/gtest.h"

TEST(IdentifiabilityStudyGroupSettingsTest, Disabled) {
  auto settings =
      IdentifiabilityStudyGroupSettings::InitFrom(false, 10, 40, "", "");
  EXPECT_FALSE(settings.enabled());
}

TEST(IdentifiabilityStudyGroupSettingsTest, DisabledBySurfaceCountZero) {
  auto settings =
      IdentifiabilityStudyGroupSettings::InitFrom(true, 0, 40, "", "");
  EXPECT_FALSE(settings.enabled());
}

TEST(IdentifiabilityStudyGroupSettingsTest, ValidRandomSurfaceSampling) {
  auto settings =
      IdentifiabilityStudyGroupSettings::InitFrom(true, 10, 40, "", "");
  EXPECT_TRUE(settings.enabled());
  EXPECT_FALSE(settings.is_using_assigned_block_sampling());
  EXPECT_EQ(10, settings.expected_surface_count());
  EXPECT_EQ(40, settings.surface_budget());
}

TEST(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.is_using_assigned_block_sampling());
}

TEST(IdentifiabilityStudyGroupSettingsTest, InvalidNegativeWeight) {
  auto settings = IdentifiabilityStudyGroupSettings::InitFrom(
      true, 0, 0, "1;2,3;4,5;6", "-1,1,1");
  EXPECT_FALSE(settings.enabled());
}

TEST(IdentifiabilityStudyGroupSettingsTest, InvalidSurfaceTooLikely) {
  auto settings = IdentifiabilityStudyGroupSettings::InitFrom(
      true, 0, 0, "1;2,1;4,5;6", "1,1,1");
  EXPECT_FALSE(settings.enabled());
}