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
|
// Copyright 2023 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/chromeos/extensions/telemetry/api/routines/fake_diagnostic_routines_service.h"
#include <utility>
#include "base/notreached.h"
#include "chromeos/crosapi/mojom/telemetry_diagnostic_routine_service.mojom.h"
#include "mojo/public/cpp/bindings/pending_receiver.h"
#include "mojo/public/cpp/bindings/pending_remote.h"
namespace chromeos {
namespace {
namespace crosapi = ::crosapi::mojom;
} // namespace
FakeDiagnosticRoutinesService::FakeDiagnosticRoutinesService() = default;
FakeDiagnosticRoutinesService::~FakeDiagnosticRoutinesService() = default;
FakeDiagnosticRoutineControl*
FakeDiagnosticRoutinesService::GetCreatedRoutineControlForRoutineType(
crosapi::TelemetryDiagnosticRoutineArgument::Tag tag) {
auto it = routine_controls_.find(tag);
if (it == routine_controls_.end()) {
return nullptr;
}
return &(it->second);
}
void FakeDiagnosticRoutinesService::FlushForTesting() {
receiver_.FlushForTesting();
}
void FakeDiagnosticRoutinesService::SetIsRoutineArgumentSupportedResponse(
crosapi::TelemetryExtensionSupportStatusPtr result) {
is_routine_argument_supported_response_ = std::move(result);
}
void FakeDiagnosticRoutinesService::CreateRoutine(
crosapi::TelemetryDiagnosticRoutineArgumentPtr routine_argument,
mojo::PendingReceiver<crosapi::TelemetryDiagnosticRoutineControl>
control_remote,
mojo::PendingRemote<crosapi::TelemetryDiagnosticRoutineObserver>
observer_receiver) {
routine_controls_.emplace(
std::piecewise_construct,
std::forward_as_tuple(routine_argument->which()),
std::forward_as_tuple(std::move(control_remote),
std::move(observer_receiver)));
if (on_routine_created_) {
on_routine_created_.Run();
}
}
void FakeDiagnosticRoutinesService::IsRoutineArgumentSupported(
crosapi::TelemetryDiagnosticRoutineArgumentPtr routine_argument,
IsRoutineArgumentSupportedCallback callback) {
std::move(callback).Run(is_routine_argument_supported_response_->Clone());
}
void FakeDiagnosticRoutinesService::SetOnCreateRoutineCalled(
base::RepeatingClosure callback) {
on_routine_created_ = std::move(callback);
}
} // namespace chromeos
|
