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
|
// Copyright 2020 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "ash/assistant/test/assistant_ash_test_base.h"
#include "ash/assistant/test/test_assistant_setup.h"
#include "ash/public/cpp/assistant/assistant_setup.h"
#include "ui/views/view.h"
namespace ash {
namespace {
class AssistantSetupControllerTest : public AssistantAshTestBase {
protected:
// Invoke to finish opt-in flow with the desired state of completion. Note
// that this API may only be called while opt-in flow is in progress.
void FinishAssistantOptInFlow(bool completed) {
DCHECK(AssistantSetup::GetInstance());
static_cast<TestAssistantSetup*>(AssistantSetup::GetInstance())
->FinishAssistantOptInFlow(completed);
}
};
} // namespace
TEST_F(AssistantSetupControllerTest, ShouldCloseAssistantUiWhenOnboarding) {
ShowAssistantUi(AssistantEntryPoint::kUnspecified);
EXPECT_TRUE(IsVisible());
// When Launcher Search IPH is enabled and it is not in zero state view, we
// show the opt-in chips as needed.
MockTextInteraction().WithTextResponse("The response");
SetConsentStatus(assistant::prefs::ConsentStatus::kUnknown);
EXPECT_TRUE(opt_in_view()->GetVisible());
ClickOnAndWait(opt_in_view());
EXPECT_FALSE(IsVisible());
}
TEST_F(AssistantSetupControllerTest,
ShouldCloseAssistantUiWhenOnboardingInTabletMode) {
SetTabletMode(true);
ShowAssistantUi(AssistantEntryPoint::kUnspecified);
EXPECT_TRUE(IsVisible());
// When Launcher Search IPH is enabled and it is not in zero state view, we
// show the opt-in chips as needed.
// Show Assistant UI in text mode, which is required to set text query.
TapOnAndWait(keyboard_input_toggle());
MockTextInteraction().WithTextResponse("The response");
SetConsentStatus(assistant::prefs::ConsentStatus::kUnknown);
EXPECT_TRUE(opt_in_view()->GetVisible());
ClickOnAndWait(opt_in_view());
EXPECT_FALSE(IsVisible());
}
TEST_F(AssistantSetupControllerTest,
ShouldNotRelaunchAssistantIfOptInFlowAborted) {
ShowAssistantUi(AssistantEntryPoint::kUnspecified);
EXPECT_TRUE(IsVisible());
// When Launcher Search IPH is enabled and it is not in zero state view, we
// show the opt-in chips as needed.
MockTextInteraction().WithTextResponse("The response");
SetConsentStatus(assistant::prefs::ConsentStatus::kUnknown);
EXPECT_TRUE(opt_in_view()->GetVisible());
ClickOnAndWait(opt_in_view());
EXPECT_FALSE(IsVisible());
FinishAssistantOptInFlow(/*completed=*/false);
EXPECT_FALSE(IsVisible());
}
TEST_F(AssistantSetupControllerTest,
ShouldRelaunchAssistantIfOptInFlowCompleted) {
ShowAssistantUi(AssistantEntryPoint::kUnspecified);
EXPECT_TRUE(IsVisible());
// When Launcher Search IPH is enabled and it is not in zero state view, we
// show the opt-in chips as needed.
MockTextInteraction().WithTextResponse("The response");
SetConsentStatus(assistant::prefs::ConsentStatus::kUnknown);
EXPECT_TRUE(opt_in_view()->GetVisible());
ClickOnAndWait(opt_in_view());
EXPECT_FALSE(IsVisible());
FinishAssistantOptInFlow(/*completed=*/true);
EXPECT_TRUE(IsVisible());
}
} // namespace ash
|
