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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
// Copyright 2024 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/saved_tab_groups/internal/startup_helper.h"
#include "base/uuid.h"
#include "components/pref_registry/pref_registry_syncable.h"
#include "components/prefs/testing_pref_service.h"
#include "components/saved_tab_groups/delegate/tab_group_sync_delegate.h"
#include "components/saved_tab_groups/public/pref_names.h"
#include "components/saved_tab_groups/public/saved_tab_group.h"
#include "components/saved_tab_groups/public/types.h"
#include "components/saved_tab_groups/test_support/mock_tab_group_sync_delegate.h"
#include "components/saved_tab_groups/test_support/mock_tab_group_sync_service.h"
#include "components/saved_tab_groups/test_support/saved_tab_group_test_utils.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
using ::testing::_;
using ::testing::Eq;
using ::testing::Return;
namespace tab_groups {
namespace {
MATCHER_P(UuidEq, uuid, "") {
return arg.saved_guid() == uuid;
}
} // namespace
class StartupHelperTest : public testing::Test {
public:
StartupHelperTest()
: local_group_id_1_(test::GenerateRandomTabGroupID()),
local_group_id_2_(test::GenerateRandomTabGroupID()),
group_1_(test::CreateTestSavedTabGroup()),
group_2_(test::CreateTestSavedTabGroup()) {}
~StartupHelperTest() override = default;
void SetUp() override {
pref_service_.registry()->RegisterBooleanPref(
prefs::kDidSyncTabGroupsInLastSession, true);
service_ = std::make_unique<MockTabGroupSyncService>();
delegate_ = std::make_unique<MockTabGroupSyncDelegate>();
startup_helper_ = std::make_unique<StartupHelper>(
delegate_.get(), service_.get(), &pref_service_);
}
void TearDown() override {}
protected:
TestingPrefServiceSimple pref_service_;
std::unique_ptr<MockTabGroupSyncService> service_;
std::unique_ptr<MockTabGroupSyncDelegate> delegate_;
std::unique_ptr<StartupHelper> startup_helper_;
LocalTabGroupID local_group_id_1_;
LocalTabGroupID local_group_id_2_;
SavedTabGroup group_1_;
SavedTabGroup group_2_;
};
TEST_F(StartupHelperTest, CloseDeletedGroups) {
EXPECT_CALL(*service_, GetDeletedGroupIds())
.WillOnce(Return(std::vector<LocalTabGroupID>{local_group_id_1_}));
EXPECT_CALL(*delegate_, CloseLocalTabGroup(Eq(local_group_id_1_)));
startup_helper_->CloseDeletedTabGroupsFromTabModel();
}
TEST_F(StartupHelperTest,
SaveUnsavedLocalGroupsOnStartupForFirstTimeFeatureLaunch) {
pref_service_.SetBoolean(prefs::kDidSyncTabGroupsInLastSession, false);
group_1_.SetLocalGroupId(local_group_id_1_);
EXPECT_CALL(*service_, GetGroup(local_group_id_1_))
.WillOnce(Return(std::make_optional<SavedTabGroup>(group_1_)));
EXPECT_CALL(*service_, GetGroup(local_group_id_2_))
.WillOnce(Return(std::nullopt));
EXPECT_CALL(*delegate_, GetLocalTabGroupIds())
.WillRepeatedly(Return(
std::vector<LocalTabGroupID>{local_group_id_1_, local_group_id_2_}));
auto saved_tab_group_2 = std::make_unique<SavedTabGroup>(group_2_);
EXPECT_CALL(*delegate_,
CreateSavedTabGroupFromLocalGroup(Eq(local_group_id_2_)))
.WillOnce(Return(std::move(saved_tab_group_2)));
EXPECT_CALL(*service_, AddGroup(_)).Times(1);
startup_helper_->HandleUnsavedLocalTabGroups();
EXPECT_TRUE(pref_service_.GetBoolean(prefs::kDidSyncTabGroupsInLastSession));
}
TEST_F(StartupHelperTest, CloseUnsavedLocalGroupsOnStartup) {
pref_service_.SetBoolean(prefs::kDidSyncTabGroupsInLastSession, true);
group_1_.SetLocalGroupId(local_group_id_1_);
EXPECT_CALL(*service_, GetGroup(local_group_id_1_))
.WillOnce(Return(std::make_optional<SavedTabGroup>(group_1_)));
EXPECT_CALL(*service_, GetGroup(local_group_id_2_))
.WillOnce(Return(std::nullopt));
EXPECT_CALL(*delegate_, GetLocalTabGroupIds())
.WillRepeatedly(Return(
std::vector<LocalTabGroupID>{local_group_id_1_, local_group_id_2_}));
EXPECT_CALL(*delegate_, CloseLocalTabGroup(Eq(local_group_id_2_)));
startup_helper_->HandleUnsavedLocalTabGroups();
EXPECT_TRUE(pref_service_.GetBoolean(prefs::kDidSyncTabGroupsInLastSession));
}
} // namespace tab_groups
|
