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
89
90
91
92
|
// Copyright 2019 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "third_party/blink/public/common/permissions_policy/document_policy.h"
#include "base/functional/callback.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/blink/public/common/permissions_policy/document_policy_features.h"
#include "third_party/blink/public/mojom/permissions_policy/document_policy_feature.mojom.h"
namespace blink {
namespace {
using DocumentPolicyTest = ::testing::Test;
// Helper function to convert literal to FeatureState.
template <class T>
DocumentPolicyFeatureState FeatureState(
std::vector<std::pair<int32_t, T>> literal,
const base::RepeatingCallback<PolicyValue(T)>& create_pv_cb) {
DocumentPolicyFeatureState result;
for (const auto& entry : literal) {
result.insert({static_cast<mojom::DocumentPolicyFeature>(entry.first),
create_pv_cb.Run(entry.second)});
}
return result;
}
TEST_F(DocumentPolicyTest, MergeFeatureState) {
base::RepeatingCallback<PolicyValue(bool)> bool_cb =
base::BindRepeating(PolicyValue::CreateBool);
base::RepeatingCallback<PolicyValue(double)> dec_double_cb =
base::BindRepeating(PolicyValue::CreateDecDouble);
base::RepeatingCallback<PolicyValue(int32_t)> enum_cb =
base::BindRepeating(PolicyValue::CreateEnum);
EXPECT_EQ(DocumentPolicy::MergeFeatureState(
FeatureState<bool>(
{{1, false}, {2, false}, {3, true}, {4, true}, {5, false}},
bool_cb),
FeatureState<bool>(
{{2, true}, {3, true}, {4, false}, {5, false}, {6, true}},
bool_cb)),
FeatureState<bool>({{1, false},
{2, false},
{3, true},
{4, false},
{5, false},
{6, true}},
bool_cb));
EXPECT_EQ(
DocumentPolicy::MergeFeatureState(
FeatureState<double>({{1, 1.0}, {2, 1.0}, {3, 1.0}, {4, 0.5}},
dec_double_cb),
FeatureState<double>({{2, 0.5}, {3, 1.0}, {4, 1.0}, {5, 1.0}},
dec_double_cb)),
FeatureState<double>({{1, 1.0}, {2, 0.5}, {3, 1.0}, {4, 0.5}, {5, 1.0}},
dec_double_cb));
EXPECT_EQ(
DocumentPolicy::MergeFeatureState(
/* base_policy */ FeatureState<int32_t>(
{{1, 1}, {2, 1}, {3, 1}, {4, 2}}, enum_cb),
/* override_policy */ FeatureState<int32_t>(
{{2, 2}, {3, 1}, {4, 1}, {5, 1}}, enum_cb)),
FeatureState<int32_t>({{1, 1}, {2, 2}, {3, 1}, {4, 1}, {5, 1}}, enum_cb));
}
// IsPolicyCompatible should use default value for incoming policy when required
// policy specifies a value for a feature and incoming policy is missing value
// for that feature.
// TODO: This is not testable as only boolean features exist currently.
// TEST_F(DocumentPolicyTest, IsPolicyCompatible) {
// mojom::DocumentPolicyFeature feature =
// mojom::DocumentPolicyFeature::kLosslessImagesMaxBpp;
// double default_policy_value =
// GetDocumentPolicyFeatureInfoMap().at(feature).default_value.DoubleValue();
// // Cap the default_policy_value, as it can be INF.
// double strict_policy_value =
// default_policy_value > 1.0 ? 1.0 : default_policy_value / 2;
//
// EXPECT_FALSE(DocumentPolicy::IsPolicyCompatible(
// DocumentPolicyFeatureState{
// {feature, PolicyValue::CreateDecDouble(
// strict_policy_value)}}, /* required policy */
// DocumentPolicyFeatureState{} /* incoming policy */
// ));
// }
} // namespace
} // namespace blink
|
