| // Copyright 2022 The Fuchsia Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef SRC_VIRTUALIZATION_TESTS_LIB_FAKE_MEMORY_PRESSURE_PROVIDER_H_ |
| #define SRC_VIRTUALIZATION_TESTS_LIB_FAKE_MEMORY_PRESSURE_PROVIDER_H_ |
| |
| #include <fuchsia/memorypressure/cpp/fidl.h> |
| #include <lib/fidl/cpp/binding_set.h> |
| #include <lib/sys/component/cpp/testing/realm_builder.h> |
| |
| class FakeMemoryPressureProvider : public fuchsia::memorypressure::Provider { |
| public: |
| explicit FakeMemoryPressureProvider(async_dispatcher_t* dispatcher) : dispatcher_(dispatcher) {} |
| |
| void RegisterWatcher( |
| ::fidl::InterfaceHandle<::fuchsia::memorypressure::Watcher> watcher) override; |
| |
| void OnLevelChanged(::fuchsia::memorypressure::Level level); |
| |
| std::unique_ptr<component_testing::LocalComponentImpl> NewComponent(); |
| |
| fidl::InterfaceRequestHandler<fuchsia::memorypressure::Provider> GetHandler() { |
| return bindings_.GetHandler(this, dispatcher_); |
| } |
| |
| private: |
| async_dispatcher_t* dispatcher_; |
| std::vector<fuchsia::memorypressure::WatcherPtr> watchers_; |
| fidl::BindingSet<fuchsia::memorypressure::Provider> bindings_; |
| std::unique_ptr<component_testing::LocalComponentHandles> handles_; |
| }; |
| |
| #endif // SRC_VIRTUALIZATION_TESTS_LIB_FAKE_MEMORY_PRESSURE_PROVIDER_H_ |