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
|
// 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/optimization_guide/core/delivery/model_util.h"
#include <algorithm>
#include "base/base64.h"
#include "base/command_line.h"
#include "base/strings/string_util.h"
#include "base/strings/stringprintf.h"
#include "components/optimization_guide/core/optimization_guide_switches.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace optimization_guide {
namespace {
proto::ModelCacheKey CreateModelCacheKey(const std::string& locale) {
proto::ModelCacheKey model_cache_key;
model_cache_key.set_locale(locale);
return model_cache_key;
}
} // namespace
TEST(ModelUtilTest, ModelCacheKeyHash) {
EXPECT_EQ(GetModelCacheKeyHash(CreateModelCacheKey("en-US")),
GetModelCacheKeyHash(CreateModelCacheKey("en-US")));
EXPECT_NE(GetModelCacheKeyHash(CreateModelCacheKey("en-US")),
GetModelCacheKeyHash(CreateModelCacheKey("en-UK")));
EXPECT_TRUE(
std::ranges::all_of(GetModelCacheKeyHash(CreateModelCacheKey("en-US")),
[](char ch) { return base::IsHexDigit(ch); }));
}
TEST(ModelUtilTest, PredictionModelVersionInKillSwitch) {
const std::map<proto::OptimizationTarget, std::set<int64_t>>
test_killswitch_model_versions = {
{proto::OPTIMIZATION_TARGET_PAINFUL_PAGE_LOAD, {1, 3}},
{proto::OPTIMIZATION_TARGET_MODEL_VALIDATION, {5}},
};
EXPECT_FALSE(IsPredictionModelVersionInKillSwitch(
{}, proto::OPTIMIZATION_TARGET_PAINFUL_PAGE_LOAD, 1));
EXPECT_TRUE(IsPredictionModelVersionInKillSwitch(
test_killswitch_model_versions,
proto::OPTIMIZATION_TARGET_PAINFUL_PAGE_LOAD, 1));
EXPECT_TRUE(IsPredictionModelVersionInKillSwitch(
test_killswitch_model_versions,
proto::OPTIMIZATION_TARGET_PAINFUL_PAGE_LOAD, 3));
EXPECT_TRUE(IsPredictionModelVersionInKillSwitch(
test_killswitch_model_versions,
proto::OPTIMIZATION_TARGET_MODEL_VALIDATION, 5));
EXPECT_FALSE(IsPredictionModelVersionInKillSwitch(
test_killswitch_model_versions,
proto::OPTIMIZATION_TARGET_PAINFUL_PAGE_LOAD, 2));
EXPECT_FALSE(IsPredictionModelVersionInKillSwitch(
test_killswitch_model_versions,
proto::OPTIMIZATION_TARGET_MODEL_VALIDATION, 1));
EXPECT_FALSE(IsPredictionModelVersionInKillSwitch(
test_killswitch_model_versions, proto::OPTIMIZATION_TARGET_PAGE_TOPICS,
1));
}
} // namespace optimization_guide
|
