blob: 62e995e3b094d6642d73df078011082449bbb0d6 [file] [log] [blame]
/*
* Copyright 2016 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_PC_TEST_RTCSTATSOBTAINER_H_
#define WEBRTC_PC_TEST_RTCSTATSOBTAINER_H_
#include "webrtc/api/stats/rtcstatsreport.h"
#include "webrtc/rtc_base/gunit.h"
namespace webrtc {
class RTCStatsObtainer : public RTCStatsCollectorCallback {
public:
static rtc::scoped_refptr<RTCStatsObtainer> Create(
rtc::scoped_refptr<const RTCStatsReport>* report_ptr = nullptr) {
return rtc::scoped_refptr<RTCStatsObtainer>(
new rtc::RefCountedObject<RTCStatsObtainer>(report_ptr));
}
void OnStatsDelivered(
const rtc::scoped_refptr<const RTCStatsReport>& report) override {
EXPECT_TRUE(thread_checker_.CalledOnValidThread());
report_ = report;
if (report_ptr_)
*report_ptr_ = report_;
}
rtc::scoped_refptr<const RTCStatsReport> report() const {
EXPECT_TRUE(thread_checker_.CalledOnValidThread());
return report_;
}
protected:
explicit RTCStatsObtainer(
rtc::scoped_refptr<const RTCStatsReport>* report_ptr)
: report_ptr_(report_ptr) {}
private:
rtc::ThreadChecker thread_checker_;
rtc::scoped_refptr<const RTCStatsReport> report_;
rtc::scoped_refptr<const RTCStatsReport>* report_ptr_;
};
} // namespace webrtc
#endif // WEBRTC_PC_TEST_RTCSTATSOBTAINER_H_