File: surface_set_valuation_unittest.cc

package info (click to toggle)
chromium 138.0.7204.157-1
  • links: PTS, VCS
  • area: main
  • in suites: trixie
  • size: 6,071,864 kB
  • sloc: cpp: 34,936,859; ansic: 7,176,967; javascript: 4,110,704; python: 1,419,953; asm: 946,768; xml: 739,967; pascal: 187,324; sh: 89,623; perl: 88,663; objc: 79,944; sql: 50,304; cs: 41,786; fortran: 24,137; makefile: 21,806; php: 13,980; tcl: 13,166; yacc: 8,925; ruby: 7,485; awk: 3,720; lisp: 3,096; lex: 1,327; ada: 727; jsp: 228; sed: 36
file content (88 lines) | stat: -rw-r--r-- 3,649 bytes parent folder | download | duplicates (5)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
// 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/surface_set_valuation.h"

#include "chrome/browser/privacy_budget/surface_set_equivalence.h"
#include "chrome/common/privacy_budget/scoped_privacy_budget_config.h"
#include "chrome/common/privacy_budget/types.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/blink/public/common/privacy_budget/identifiable_surface.h"

namespace {

constexpr auto kSurface1 = blink::IdentifiableSurface::FromTypeAndToken(
    blink::IdentifiableSurface::Type::kCanvasRenderingContext,
    3);
constexpr auto kSurface2 = blink::IdentifiableSurface::FromTypeAndToken(
    blink::IdentifiableSurface::Type::kHTMLMediaElement_CanPlayType,
    5);
constexpr auto kSurface3 = blink::IdentifiableSurface::FromTypeAndToken(
    blink::IdentifiableSurface::Type::kWebFeature,
    3);
constexpr auto kSurface4 = blink::IdentifiableSurface::FromTypeAndToken(
    blink::IdentifiableSurface::Type::kWebFeature,
    4);

}  // namespace

TEST(SurfaceSetValuationTest, NoConfig) {
  // Verify behavior when there's no server config that defines costing
  // parameters. All surfaces should be valued at 1.0.

  test::ScopedPrivacyBudgetConfig::Parameters config_params(
      test::ScopedPrivacyBudgetConfig::Presets::kEnableRandomSampling);
  ASSERT_TRUE(config_params.per_surface_cost.empty());
  ASSERT_TRUE(config_params.per_type_cost.empty());
  test::ScopedPrivacyBudgetConfig study_configuration(config_params);

  SurfaceSetEquivalence equivalence;
  SurfaceSetValuation valuation(equivalence);

  IdentifiableSurfaceSet surfaces = {kSurface1, kSurface2, kSurface3,
                                     kSurface4};

  EXPECT_FLOAT_EQ(1.0, valuation.Cost(kSurface1));
  EXPECT_FLOAT_EQ(4.0, valuation.Cost(surfaces));
}

TEST(SurfaceSetValuationTest, PerSurface) {
  test::ScopedPrivacyBudgetConfig::Parameters config_params(
      test::ScopedPrivacyBudgetConfig::Presets::kEnableRandomSampling);
  config_params.per_surface_cost =
      IdentifiableSurfaceCostMap{{kSurface1, 0.5}, {kSurface2, 0.25}};
  test::ScopedPrivacyBudgetConfig study_configuration(config_params);

  SurfaceSetEquivalence equivalence;
  SurfaceSetValuation valuation(equivalence);

  EXPECT_FLOAT_EQ(0.5, valuation.Cost(kSurface1));
  EXPECT_FLOAT_EQ(1.0, valuation.Cost(kSurface4));
  EXPECT_FLOAT_EQ(1.75, valuation.Cost(IdentifiableSurfaceSet{
                            kSurface1, kSurface2, kSurface3}));
}

TEST(SurfaceSetValuationTest, PerType) {
  test::ScopedPrivacyBudgetConfig::Parameters config_params(
      test::ScopedPrivacyBudgetConfig::Presets::kEnableRandomSampling);
  config_params.per_type_cost[blink::IdentifiableSurface::Type::kWebFeature] =
      0.25;
  test::ScopedPrivacyBudgetConfig study_configuration(config_params);

  SurfaceSetEquivalence equivalence;
  SurfaceSetValuation valuation(equivalence);

  EXPECT_FLOAT_EQ(1.0, valuation.Cost(kSurface1));
  EXPECT_FLOAT_EQ(0.25, valuation.Cost(kSurface4));
  EXPECT_FLOAT_EQ(2.25, valuation.Cost(IdentifiableSurfaceSet{
                            kSurface1, kSurface2, kSurface3}));
}

TEST(SurfaceSetValuationTest, ExpectedSurfaceCountForCost) {
  EXPECT_EQ(0u, SurfaceSetValuation::ExpectedSurfaceCountForCost(0.0));
  EXPECT_EQ(1u, SurfaceSetValuation::ExpectedSurfaceCountForCost(0.5));
  EXPECT_EQ(1u, SurfaceSetValuation::ExpectedSurfaceCountForCost(1.0));
  EXPECT_EQ(2u, SurfaceSetValuation::ExpectedSurfaceCountForCost(1.4));
  EXPECT_EQ(3u, SurfaceSetValuation::ExpectedSurfaceCountForCost(2.1));
}