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
|
// 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/omnibox/browser/autocomplete_grouper_groups.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/omnibox_proto/groups.pb.h"
TEST(AutocompleteGrouperGroupsTest, Group) {
Group group(4, {
{omnibox::GROUP_SEARCH, 2},
{omnibox::GROUP_DOCUMENT, 4},
});
AutocompleteMatch search_match{};
search_match.suggestion_group_id = omnibox::GROUP_SEARCH;
AutocompleteMatch doc_match{};
doc_match.suggestion_group_id = omnibox::GROUP_DOCUMENT;
AutocompleteMatch other_match{};
other_match.suggestion_group_id = omnibox::GROUP_OTHER_NAVS;
EXPECT_TRUE(group.CanAdd(search_match));
EXPECT_TRUE(group.CanAdd(doc_match));
EXPECT_FALSE(group.CanAdd(other_match));
// Verify the group limit with 2/2 searches, 0/4 docs, and 2/4 total.
group.Add(search_match); // 1/2 searches, 0/4 docs, 1/4 total.
EXPECT_TRUE(group.CanAdd(search_match));
group.Add(search_match); // 2/2 searches, 0/4 docs, 2/4 total.
EXPECT_FALSE(group.CanAdd(search_match));
EXPECT_TRUE(group.CanAdd(doc_match));
EXPECT_FALSE(group.CanAdd(other_match));
// Verify the total `Group` limit with 2/2 searches, 2/4 docs, and 4/4 total.
group.Add(doc_match); // 2/2 searches, 1/4 docs, 3/4 total.
EXPECT_TRUE(group.CanAdd(doc_match));
group.Add(doc_match); // 2/2 searches, 2/4 docs, 4/4 total.
EXPECT_FALSE(group.CanAdd(search_match));
EXPECT_FALSE(group.CanAdd(doc_match));
EXPECT_FALSE(group.CanAdd(other_match));
// Verify with 0/2 searches, 4/4 docs, and 4/4 total.
Group group2(4, {
{omnibox::GROUP_SEARCH, 2},
{omnibox::GROUP_DOCUMENT, 4},
});
group2.Add(doc_match); // 0/2 searches, 1/4 docs, 1/4 total.
group2.Add(doc_match); // 0/2 searches, 2/4 docs, 2/4 total.
group2.Add(doc_match); // 0/2 searches, 3/4 docs, 3/4 total.
EXPECT_TRUE(group2.CanAdd(search_match));
EXPECT_TRUE(group2.CanAdd(doc_match));
group2.Add(doc_match); // 0/2 searches, 4/4 docs, 4/4 total.
EXPECT_FALSE(group2.CanAdd(search_match));
EXPECT_FALSE(group2.CanAdd(doc_match));
EXPECT_FALSE(group2.CanAdd(other_match));
}
TEST(AutocompleteGrouperGroupsTest, DefaultGroup) {
Group default_group(1,
{
{omnibox::GROUP_STARTER_PACK, 1},
{omnibox::GROUP_SEARCH, 1},
{omnibox::GROUP_OTHER_NAVS, 1},
},
/*is_zps=*/false,
/*is_default=*/true);
ACMatches matches{{}, {}, {}};
// Can't be added because `allowed_to_be_default` is false.
AutocompleteMatch non_default_match{};
non_default_match.suggestion_group_id = omnibox::GROUP_STARTER_PACK;
// Can't be added because `GROUP_DOCUMENT` is not allowed.
AutocompleteMatch default_match_doc{};
default_match_doc.suggestion_group_id = omnibox::GROUP_DOCUMENT;
default_match_doc.allowed_to_be_default_match = true;
// Can be added.
AutocompleteMatch default_match_search{};
default_match_search.suggestion_group_id = omnibox::GROUP_SEARCH;
default_match_search.allowed_to_be_default_match = true;
EXPECT_FALSE(default_group.CanAdd(non_default_match));
EXPECT_TRUE(default_group.CanAdd(default_match_search));
EXPECT_FALSE(default_group.CanAdd(default_match_doc));
default_group.Add(default_match_search);
EXPECT_FALSE(default_group.CanAdd(non_default_match));
EXPECT_FALSE(default_group.CanAdd(default_match_search));
EXPECT_FALSE(default_group.CanAdd(default_match_doc));
}
|
