| // Copyright 2021 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_CONNECTIVITY_WEAVE_ADAPTATION_TESTS_TEST_THREAD_STACK_MANAGER_H_ |
| #define SRC_CONNECTIVITY_WEAVE_ADAPTATION_TESTS_TEST_THREAD_STACK_MANAGER_H_ |
| |
| #include <gtest/gtest.h> |
| |
| #include "thread_stack_manager_stub_impl.h" |
| |
| namespace weave::adaptation::testing { |
| |
| class TestThreadStackManager final : public nl::Weave::DeviceLayer::ThreadStackManagerStubImpl { |
| public: |
| static constexpr char kThreadInterfaceName[] = "lowpan0"; |
| |
| // Access underlying delegate implementation. |
| using Impl = nl::Weave::DeviceLayer::ThreadStackManagerStubImpl; |
| |
| // Returns whether thread is provisioned. |
| bool IsThreadProvisioned() override { |
| return is_thread_provisioned_.value_or(Impl::IsThreadProvisioned()); |
| } |
| |
| // Returns whether thread is supported. |
| bool IsThreadSupported() const override { |
| return is_thread_supported_.value_or(Impl::IsThreadSupported()); |
| } |
| |
| // Simulate clearing Thread provision when overridden. |
| void ClearThreadProvision() override { |
| if (!is_thread_provisioned_.has_value()) { |
| Impl::ClearThreadProvision(); |
| } else { |
| is_thread_provisioned_ = false; |
| } |
| } |
| |
| // Returns the thread interface name. |
| std::string GetInterfaceName() const override { return kThreadInterfaceName; } |
| |
| // Set whether thread is provisioned. |
| TestThreadStackManager& set_is_thread_provisioned(std::optional<bool> is_thread_provisioned) { |
| is_thread_provisioned_ = is_thread_provisioned; |
| return *this; |
| } |
| |
| // Set whether thread is supported. |
| TestThreadStackManager& set_is_thread_supported(std::optional<bool> is_thread_supported) { |
| is_thread_supported_ = is_thread_supported; |
| return *this; |
| } |
| |
| private: |
| std::optional<bool> is_thread_provisioned_; |
| std::optional<bool> is_thread_supported_; |
| }; |
| |
| } // namespace weave::adaptation::testing |
| |
| #endif // SRC_CONNECTIVITY_WEAVE_ADAPTATION_TESTS_TEST_THREAD_STACK_MANAGER_H_ |