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
|
// 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 "chrome/browser/screen_ai/screen_ai_install_state.h"
#include "base/files/file_path.h"
#include "base/scoped_observation.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace {
class TestScreenAIInstallState : public screen_ai::ScreenAIInstallState {
public:
TestScreenAIInstallState() = default;
TestScreenAIInstallState(const TestScreenAIInstallState&) = delete;
TestScreenAIInstallState& operator=(const TestScreenAIInstallState&) = delete;
~TestScreenAIInstallState() override = default;
void SetLastUsageTime() override {}
void DownloadComponentInternal() override {
// The passed file path is not used and just indicates that the component
// exists.
ScreenAIInstallState::GetInstance()->SetComponentFolder(
base::FilePath(FILE_PATH_LITERAL("tmp")));
}
};
} // namespace
namespace screen_ai {
class ScreenAIInstallStateTest : public testing::Test,
ScreenAIInstallState::Observer {
public:
ScreenAIInstallStateTest() = default;
void StartObservation() {
component_downloaded_observer_.Observe(ScreenAIInstallState::GetInstance());
}
void DownloadComponent() { test_install_state_.DownloadComponentInternal(); }
void StateChanged(ScreenAIInstallState::State state) override {
if (state == ScreenAIInstallState::State::kDownloaded) {
component_downloaded_received_ = true;
}
}
bool ComponentDownloadedReceived() { return component_downloaded_received_; }
private:
TestScreenAIInstallState test_install_state_;
base::ScopedObservation<screen_ai::ScreenAIInstallState,
ScreenAIInstallState::Observer>
component_downloaded_observer_{this};
bool component_downloaded_received_ = false;
};
TEST_F(ScreenAIInstallStateTest, AddingObserverTriggersDownload) {
StartObservation();
EXPECT_TRUE(ComponentDownloadedReceived());
}
TEST_F(ScreenAIInstallStateTest, DownloadedBeforeObservation) {
DownloadComponent();
StartObservation();
EXPECT_TRUE(ComponentDownloadedReceived());
}
TEST_F(ScreenAIInstallStateTest, ObservationAfterFailure) {
ScreenAIInstallState::GetInstance()->SetStateForTesting(
screen_ai::ScreenAIInstallState::State::kDownloadFailed);
StartObservation();
EXPECT_TRUE(ComponentDownloadedReceived());
}
} // namespace screen_ai
|
