| /* |
| * Copyright (c) 2013 The WebRTC project authors. All Rights Reserved. |
| * |
| * Use of this source code is governed by a BSD-style license |
| * that can be found in the LICENSE file in the root of the source |
| * tree. An additional intellectual property rights grant can be found |
| * in the file PATENTS. All contributing project authors may |
| * be found in the AUTHORS file in the root of the source tree. |
| */ |
| |
| #ifndef WEBRTC_MODULES_REMOTE_BITRATE_ESTIMATOR_TEST_BWE_TEST_H_ |
| #define WEBRTC_MODULES_REMOTE_BITRATE_ESTIMATOR_TEST_BWE_TEST_H_ |
| |
| #include <map> |
| #include <string> |
| #include <vector> |
| #include "testing/gtest/include/gtest/gtest.h" |
| #include "webrtc/base/constructormagic.h" |
| #include "webrtc/modules/remote_bitrate_estimator/include/remote_bitrate_estimator.h" |
| #include "webrtc/modules/remote_bitrate_estimator/test/bwe.h" |
| #include "webrtc/modules/remote_bitrate_estimator/test/bwe_test_framework.h" |
| |
| namespace webrtc { |
| |
| namespace testing { |
| namespace bwe { |
| |
| class BweReceiver; |
| class PacketReceiver; |
| class PacketSender; |
| |
| class PacketProcessorRunner { |
| public: |
| explicit PacketProcessorRunner(PacketProcessor* processor); |
| ~PacketProcessorRunner(); |
| |
| bool RunsProcessor(const PacketProcessor* processor) const; |
| void RunFor(int64_t time_ms, int64_t time_now_ms, Packets* in_out); |
| |
| private: |
| void FindPacketsToProcess(const FlowIds& flow_ids, Packets* in, Packets* out); |
| void QueuePackets(Packets* batch, int64_t end_of_batch_time_us); |
| |
| PacketProcessor* processor_; |
| Packets queue_; |
| }; |
| |
| class Link : public PacketProcessorListener { |
| public: |
| virtual ~Link() {} |
| |
| virtual void AddPacketProcessor(PacketProcessor* processor, |
| ProcessorType type); |
| virtual void RemovePacketProcessor(PacketProcessor* processor); |
| |
| void Run(int64_t run_for_ms, int64_t now_ms, Packets* packets); |
| |
| const std::vector<PacketSender*>& senders() { return senders_; } |
| const std::vector<PacketProcessorRunner>& processors() { return processors_; } |
| |
| private: |
| std::vector<PacketSender*> senders_; |
| std::vector<PacketReceiver*> receivers_; |
| std::vector<PacketProcessorRunner> processors_; |
| }; |
| |
| class BweTest { |
| public: |
| BweTest(); |
| ~BweTest(); |
| |
| protected: |
| void SetUp(); |
| |
| void VerboseLogging(bool enable); |
| void RunFor(int64_t time_ms); |
| std::string GetTestName() const; |
| |
| void PrintResults(double max_throughput_kbps, |
| Stats<double> throughput_kbps, |
| Stats<double> delay_ms, |
| std::vector<Stats<double>> flow_throughput_kbps); |
| |
| void RunFairnessTest(BandwidthEstimatorType bwe_type, |
| size_t num_media_flows, |
| size_t num_tcp_flows, |
| int64_t run_time_seconds, |
| int capacity_kbps, |
| int max_delay_ms); |
| |
| Link downlink_; |
| Link uplink_; |
| |
| private: |
| void FindPacketsToProcess(const FlowIds& flow_ids, Packets* in, |
| Packets* out); |
| void GiveFeedbackToAffectedSenders(PacketReceiver* receiver); |
| |
| int64_t run_time_ms_; |
| int64_t time_now_ms_; |
| int64_t simulation_interval_ms_; |
| std::vector<Link*> links_; |
| Packets packets_; |
| |
| DISALLOW_COPY_AND_ASSIGN(BweTest); |
| }; |
| } // namespace bwe |
| } // namespace testing |
| } // namespace webrtc |
| |
| #endif // WEBRTC_MODULES_REMOTE_BITRATE_ESTIMATOR_TEST_BWE_TEST_H_ |