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
113
114
115
116
117
118
|
// Copyright 2025 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "base/test/metrics/histogram_tester.h"
#include "base/test/task_environment.h"
#include "chrome/browser/screen_ai/screen_ai_service_handler_base.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace screen_ai {
constexpr char kIsSuspendedMetric[] = "Accessibility.OCR.Service.IsSuspended";
constexpr char kCrashCountBeforeResumeMetric[] =
"Accessibility.OCR.Service.CrashCountBeforeResume";
class TestScreenAIServiceHandler : public ScreenAIServiceHandlerBase {
public:
std::string GetServiceName() const override { return "OCR"; }
void LoadModelFilesAndInitialize(
base::TimeTicks request_start_time) override {
service_connected_ = true;
}
bool IsConnectionBound() const override { return service_connected_; }
bool IsServiceEnabled() const override { return true; }
void ResetConnection() override { service_connected_ = false; }
private:
bool service_connected_ = false;
};
class ScreenAIServiceShutdownHandlerTest : public ::testing::Test {
public:
ScreenAIServiceShutdownHandlerTest() : handler() {}
bool IsSuspended() { return handler.GetAndRecordSuspendedState(); }
void DisconnectService() { handler.OnScreenAIServiceDisconnected(); }
void SendShuttingdownMessage() { handler.ShuttingDownOnIdle(); }
bool IsServiceAvailable() {
std::optional<bool> state = handler.GetServiceState();
if (state.has_value()) {
// A true value means that the service is already running which is not
// possible in unittest.
EXPECT_FALSE(state.value());
return false;
} else {
// An empty result means that the service is not banned, and can be used.
return true;
}
}
protected:
TestScreenAIServiceHandler handler;
base::HistogramTester histogram_tester_;
base::test::TaskEnvironment task_environment_{
base::test::TaskEnvironment::TimeSource::MOCK_TIME};
};
TEST_F(ScreenAIServiceShutdownHandlerTest, SuspendedAfterCrash) {
EXPECT_FALSE(IsSuspended());
DisconnectService();
EXPECT_TRUE(IsSuspended());
EXPECT_FALSE(IsServiceAvailable());
histogram_tester_.ExpectBucketCount(kIsSuspendedMetric, false, 1);
// Expect two since it's queried twice in `IsSuspended` and
// `IsServiceAvailable`.
histogram_tester_.ExpectBucketCount(kIsSuspendedMetric, true, 2);
}
TEST_F(ScreenAIServiceShutdownHandlerTest, NotSuspendedAfterShutdown) {
EXPECT_FALSE(IsSuspended());
SendShuttingdownMessage();
DisconnectService();
EXPECT_FALSE(IsSuspended());
EXPECT_TRUE(IsServiceAvailable());
// Expect 3 since it's queried twice in `IsSuspended` and once in
// `IsServiceAvailable`.
histogram_tester_.ExpectUniqueSample(kIsSuspendedMetric, false, 3);
}
TEST_F(ScreenAIServiceShutdownHandlerTest, CrashCountBeforeResume) {
EXPECT_FALSE(IsSuspended());
DisconnectService();
EXPECT_TRUE(IsSuspended());
EXPECT_FALSE(IsServiceAvailable());
task_environment_.FastForwardBy(base::Minutes(1));
EXPECT_FALSE(IsSuspended());
EXPECT_TRUE(IsServiceAvailable());
// Revive after crash is only recorded after shutdown message is received and
// service disconnects.
SendShuttingdownMessage();
DisconnectService();
histogram_tester_.ExpectUniqueSample(kCrashCountBeforeResumeMetric, 1, 1);
}
TEST_F(ScreenAIServiceShutdownHandlerTest, SecondCrashLongerSuspend) {
EXPECT_FALSE(IsSuspended());
DisconnectService();
EXPECT_TRUE(IsSuspended());
task_environment_.FastForwardBy(base::Minutes(1));
EXPECT_FALSE(IsSuspended());
// Crash again.
DisconnectService();
EXPECT_TRUE(IsSuspended());
task_environment_.FastForwardBy(base::Minutes(1));
// Still suspended as the second crash results in longer suspend.
EXPECT_TRUE(IsSuspended());
task_environment_.FastForwardBy(base::Minutes(3));
EXPECT_FALSE(IsSuspended());
SendShuttingdownMessage();
DisconnectService();
histogram_tester_.ExpectUniqueSample(kCrashCountBeforeResumeMetric, 2, 1);
}
} // namespace screen_ai
|
