privacy_math_unittest.cc | Explore in Territory

// 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 "components/attribution_reporting/privacy_math.h"

#include <stdint.h>

#include <cmath>
#include <limits>
#include <map>
#include <optional>
#include <set>
#include <utility>
#include <vector>

#include "base/numerics/checked_math.h"
#include "base/test/gmock_expected_support.h"
#include "base/time/time.h"
#include "base/types/expected.h"
#include "base/types/expected_macros.h"
#include "build/build_config.h"
#include "components/attribution_reporting/attribution_scopes_data.h"
#include "components/attribution_reporting/attribution_scopes_set.h"
#include "components/attribution_reporting/event_report_windows.h"
#include "components/attribution_reporting/max_event_level_reports.h"
#include "components/attribution_reporting/source_type.mojom.h"
#include "components/attribution_reporting/test_utils.h"
#include "components/attribution_reporting/trigger_config.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace attribution_reporting {
namespace {

EventReportWindows;
MaxEventLevelReports;
TriggerSpecs;
SourceType;

TEST(PrivacyMathTest, BinomialCoefficient) { … }

TEST(PrivacyMathTest, GetKCombinationAtIndex) { … }

// Simple stress test to make sure that GetKCombinationAtIndex is returning
// combinations uniquely indexed by the given index, i.e. there are never any
// repeats.
// Times out on Windows debug builds; see https://crbug.com/329138754.
#if BUILDFLAG(IS_WIN) && !defined(NDEBUG)
#define MAYBE_GetKCombination_NoRepeats …
#else
#define MAYBE_GetKCombination_NoRepeats …
#endif
TEST(PrivacyMathTest, MAYBE_GetKCombination_NoRepeats) { … }

// The k-combination at a given index is the unique set of k positive integers
// a_k > a_{k-1} > ... > a_2 > a_1 >= 0 such that
// `index` = \sum_{i=1}^k {a_i}\choose{i}
// Times out on Windows debug builds; see https://crbug.com/329138754.
#if BUILDFLAG(IS_WIN) && !defined(NDEBUG)
#define MAYBE_GetKCombination_MatchesDefinition …
#else
#define MAYBE_GetKCombination_MatchesDefinition …
#endif
TEST(PrivacyMathTest, MAYBE_GetKCombination_MatchesDefinition) { … }

TEST(PrivacyMathTest, GetNumberOfStarsAndBarsSequences) { … }

TEST(PrivacyMathTest, GetStarIndices) { … }

TEST(PrivacyMathTest, GetBarsPrecedingEachStar) { … }

// Adapted from
// https://github.com/WICG/attribution-reporting-api/blob/ab43f8c989cf881ffd7a7f71801b98d649ed164a/flexible-event/privacy.test.ts#L76C1-L82C2
TEST(PrivacyMathTest, BinaryEntropy) { … }

// Adapted from
// https://github.com/WICG/attribution-reporting-api/blob/ab43f8c989cf881ffd7a7f71801b98d649ed164a/flexible-event/privacy.test.ts#L10-L31
TEST(PrivacyMathTest, GetRandomizedResponseRate) { … }

// Adapted from
// https://github.com/WICG/attribution-reporting-api/blob/ab43f8c989cf881ffd7a7f71801b98d649ed164a/flexible-event/privacy.test.ts#L38-L69
TEST(PrivacyMathTest, ComputeChannelCapacity) { … }

TEST(PrivacyMathTest, ComputeChannelCapacityWithScopes) { … }

TEST(PrivacyMathTest, GetFakeReportsForSequenceIndex) { … }

void RunRandomFakeReportsTest(const TriggerSpecs& specs,
                              const int num_samples,
                              const double tolerance) { … }

TEST(PrivacyMathTest, GetRandomFakeReports_Event_MatchesExpectedDistribution) { … }

// Times out on Windows debug builds; see https://crbug.com/329138754.
#if BUILDFLAG(IS_WIN) && !defined(NDEBUG)
#define MAYBE_GetRandomFakeReports_Navigation_MatchesExpectedDistribution …
#else
#define MAYBE_GetRandomFakeReports_Navigation_MatchesExpectedDistribution …
#endif
TEST(PrivacyMathTest,
     MAYBE_GetRandomFakeReports_Navigation_MatchesExpectedDistribution) { … }

TEST(PrivacyMathTest, GetRandomFakeReports_Custom_MatchesExpectedDistribution) { … }

const struct { … } kNumStateTestCases[] = …;

TEST(PrivacyMathTest, GetNumStates) { … }

TEST(PrivacyMathTest, NumStatesForTriggerSpecs_UniqueSampling) { … }

// Regression test for http://crbug.com/1503728 in which the optimized
// randomized-response incorrectly returned the trigger data *index* rather than
// the trigger data *value* in the fake reports.
TEST(PrivacyMathTest, NonDefaultTriggerDataForSingleSharedSpec) { … }

TEST(PrivacyMathTest, RandomizedResponse_ExceedsChannelCapacity) { … }

TEST(PrivacyMathTest, RandomizedResponse_ExceedsScopesChannelCapacity) { … }

// Regression test for http://crbug.com/1504144 in which empty specs cause an
// invalid iterator dereference and thus a crash.
TEST(PrivacyMathTest, UnaryChannel) { … }

TEST(PrivacyMathTest, IsValid) { … }

}  // namespace
}  // namespace attribution_reporting
chromium/components/attribution_reporting/privacy_math_unittest.cc
