| // |
| // Copyright 2017 The Android Open Source Project |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| // |
| |
| #pragma once |
| |
| #include <chrono> |
| #include <functional> |
| #include <future> |
| #include <memory> |
| #include <string> |
| #include <vector> |
| |
| #include "model/controller/controller_properties.h" |
| #include "model/setup/async_manager.h" |
| #include "model/setup/test_channel_transport.h" |
| #include "model/setup/test_command_handler.h" |
| #include "model/setup/test_model.h" |
| #include "net/async_data_channel_server.h" |
| |
| namespace android::net { |
| class AsyncDataChannel; |
| class AsyncDataChannelConnector; |
| } // namespace android::net |
| |
| namespace rootcanal { |
| |
| using android::net::AsyncDataChannel; |
| using android::net::AsyncDataChannelConnector; |
| using android::net::AsyncDataChannelServer; |
| using android::net::ConnectCallback; |
| |
| using rootcanal::AsyncManager; |
| using rootcanal::Device; |
| using rootcanal::Phy; |
| |
| class TestEnvironment { |
| public: |
| TestEnvironment( |
| std::function<std::shared_ptr<AsyncDataChannelServer>(AsyncManager*, int)> |
| open_server, |
| std::function<std::shared_ptr<AsyncDataChannelConnector>(AsyncManager*)> |
| open_connector, |
| int test_port, int hci_port, int link_port, int link_ble_port, |
| std::string const& config_str, |
| bool enable_hci_sniffer = false, bool enable_baseband_sniffer = false, |
| bool enable_pcap_filter = false, bool disable_address_reuse = false); |
| |
| void initialize(std::promise<void> barrier); |
| void close(); |
| |
| private: |
| rootcanal::AsyncManager async_manager_; |
| rootcanal::TestChannelTransport test_channel_transport_; |
| std::shared_ptr<AsyncDataChannelServer> test_socket_server_; |
| std::vector<std::shared_ptr<AsyncDataChannelServer>> hci_socket_servers_; |
| std::shared_ptr<AsyncDataChannelServer> link_socket_server_; |
| std::shared_ptr<AsyncDataChannelServer> link_ble_socket_server_; |
| std::shared_ptr<AsyncDataChannelConnector> connector_; |
| bool enable_hci_sniffer_; |
| bool enable_baseband_sniffer_; |
| bool enable_pcap_filter_; |
| bool test_channel_open_{false}; |
| std::promise<void> barrier_; |
| rootcanal::AsyncUserId socket_user_id_{}; |
| |
| void SetUpTestChannel(); |
| void SetUpHciServer( |
| std::function<std::shared_ptr<AsyncDataChannelServer>(AsyncManager*, int)> |
| open_server, |
| int tcp_port, rootcanal::ControllerProperties properties); |
| void SetUpLinkLayerServer(); |
| void SetUpLinkBleLayerServer(); |
| |
| std::shared_ptr<Device> ConnectToRemoteServer(const std::string& server, |
| int port, Phy::Type phy_type); |
| |
| rootcanal::TestModel test_model_{ |
| [this]() { return async_manager_.GetNextUserId(); }, |
| [this](rootcanal::AsyncUserId user_id, std::chrono::milliseconds delay, |
| const rootcanal::TaskCallback& task) { |
| return async_manager_.ExecAsync(user_id, delay, task); |
| }, |
| |
| [this](rootcanal::AsyncUserId user_id, std::chrono::milliseconds delay, |
| std::chrono::milliseconds period, |
| const rootcanal::TaskCallback& task) { |
| return async_manager_.ExecAsyncPeriodically(user_id, delay, period, |
| task); |
| }, |
| |
| [this](rootcanal::AsyncUserId user) { |
| async_manager_.CancelAsyncTasksFromUser(user); |
| }, |
| |
| [this](rootcanal::AsyncTaskId task) { |
| async_manager_.CancelAsyncTask(task); |
| }, |
| |
| [this](const std::string& server, int port, Phy::Type phy_type) { |
| return ConnectToRemoteServer(server, port, phy_type); |
| }}; |
| |
| rootcanal::TestCommandHandler test_channel_{test_model_}; |
| }; |
| |
| } // namespace rootcanal |