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
|
// 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 "chrome/browser/ash/mahi/web_contents/mahi_tab_helper.h"
#include <memory>
#include <utility>
#include "base/test/scoped_feature_list.h"
#include "build/build_config.h"
#include "chrome/browser/ash/mahi/web_contents/test_support/mock_mahi_web_contents_manager.h"
#include "chrome/browser/ui/browser.h"
#include "chrome/browser/ui/tabs/tab_activity_simulator.h"
#include "chrome/test/base/chrome_render_view_host_test_harness.h"
#include "chrome/test/base/test_browser_window.h"
#include "chromeos/components/mahi/public/cpp/mahi_web_contents_manager.h"
#include "chromeos/constants/chromeos_features.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace mahi {
using testing::_;
using testing::Return;
class MahiTabHelperTest : public ChromeRenderViewHostTestHarness {
protected:
void SetUp() override {
ChromeRenderViewHostTestHarness::SetUp();
scoped_feature_list_.InitWithFeatures(
/*enabled_features=*/{chromeos::features::kMahi,
chromeos::features::kFeatureManagementMahi},
/*disabled_features=*/{});
scoped_mahi_web_contents_manager_ =
std::make_unique<chromeos::ScopedMahiWebContentsManagerOverride>(
&mock_mahi_web_contents_manager_);
// Initialize browser.
const Browser::CreateParams params(profile(), /*user_gesture=*/true);
browser_ = CreateBrowserWithTestWindowForParams(params);
tab_strip_model_ = browser_->tab_strip_model();
}
void TearDown() override {
tab_strip_model_->CloseAllTabs();
tab_strip_model_ = nullptr;
browser_.reset();
scoped_mahi_web_contents_manager_.reset();
ChromeRenderViewHostTestHarness::TearDown();
}
base::test::ScopedFeatureList scoped_feature_list_;
MockMahiWebContentsManager mock_mahi_web_contents_manager_;
std::unique_ptr<chromeos::ScopedMahiWebContentsManagerOverride>
scoped_mahi_web_contents_manager_;
TabActivitySimulator tab_activity_simulator_;
raw_ptr<TabStripModel> tab_strip_model_;
std::unique_ptr<Browser> browser_;
};
TEST_F(MahiTabHelperTest, FocusedTabLoadComplete) {
// Don't get notifications from unfocused tab.
EXPECT_CALL(mock_mahi_web_contents_manager_, OnFocusedPageLoadComplete(_))
.Times(0);
MahiTabHelper::CreateForWebContents(web_contents());
EXPECT_NE(nullptr, MahiTabHelper::FromWebContents(web_contents()));
NavigateAndCommit(GURL("https://example1.com"));
// When a tab gets focus, notification will be received from navigation.
EXPECT_CALL(mock_mahi_web_contents_manager_, OnFocusedPageLoadComplete(_))
.Times(2);
MahiTabHelper::FromWebContents(web_contents())->OnWebContentsFocused(nullptr);
NavigateAndCommit(GURL("https://example2.com"));
// After losing focus, the tab's notification will no longer be received.
EXPECT_CALL(mock_mahi_web_contents_manager_, OnFocusedPageLoadComplete(_))
.Times(0);
MahiTabHelper::FromWebContents(web_contents())
->OnWebContentsLostFocus(nullptr);
NavigateAndCommit(GURL("https://example3.com"));
}
TEST_F(MahiTabHelperTest, TabSwitch) {
MahiTabHelper::CreateForWebContents(web_contents());
NavigateAndCommit(GURL("https://example1.com"));
content::WebContents* web_contents2 =
tab_activity_simulator_.AddWebContentsAndNavigate(
tab_strip_model_, GURL("https://example2.com"));
EXPECT_NE(nullptr, MahiTabHelper::FromWebContents(web_contents()));
EXPECT_NE(nullptr, MahiTabHelper::FromWebContents(web_contents2));
// Switch back to a previous loaded tab.
EXPECT_CALL(mock_mahi_web_contents_manager_, OnFocusedPageLoadComplete(_))
.Times(1);
// Change active tab with `browser()->tab_strip_model()->ActivateTabAt()` or
// `AddPage()` will not trigger focus events. Fire it manually instead.
MahiTabHelper::FromWebContents(web_contents())->OnWebContentsFocused(nullptr);
}
} // namespace mahi
|
