| /* Copyright 2019 The Chromium OS Authors. All rights reserved. |
| * Use of this source code is governed by a BSD-style license that can be |
| * found in the LICENSE file. |
| */ |
| |
| #include "hardware_verifier/probe_result_getter_fake.h" |
| |
| #include <base/files/file_path.h> |
| #include <base/optional.h> |
| #include <runtime_probe/proto_bindings/runtime_probe.pb.h> |
| |
| namespace hardware_verifier { |
| |
| FakeProbeResultGetter::FakeProbeResultGetter() |
| : runtime_probe_run_success_(false), |
| runtime_probe_output_(), |
| file_probe_results_() {} |
| |
| base::Optional<runtime_probe::ProbeResult> |
| FakeProbeResultGetter::GetFromRuntimeProbe() const { |
| if (runtime_probe_run_success_) { |
| return runtime_probe_output_; |
| } |
| return base::nullopt; |
| } |
| |
| base::Optional<runtime_probe::ProbeResult> FakeProbeResultGetter::GetFromFile( |
| const base::FilePath& file_path) const { |
| const auto it = file_probe_results_.find(file_path.value()); |
| if (it == file_probe_results_.end()) { |
| return base::nullopt; |
| } |
| return it->second; |
| } |
| |
| void FakeProbeResultGetter::set_runtime_probe_fail() { |
| runtime_probe_run_success_ = false; |
| } |
| |
| void FakeProbeResultGetter::set_runtime_probe_output( |
| const runtime_probe::ProbeResult& data) { |
| runtime_probe_run_success_ = true; |
| runtime_probe_output_ = data; |
| } |
| |
| void FakeProbeResultGetter::set_file_probe_results( |
| const FileProbeResults& data) { |
| file_probe_results_ = data; |
| } |
| |
| } // namespace hardware_verifier |