1/* 2 * libjingle 3 * Copyright 2012, Google Inc. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions are met: 7 * 8 * 1. Redistributions of source code must retain the above copyright notice, 9 * this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright notice, 11 * this list of conditions and the following disclaimer in the documentation 12 * and/or other materials provided with the distribution. 13 * 3. The name of the author may not be used to endorse or promote products 14 * derived from this software without specific prior written permission. 15 * 16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED 17 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF 18 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO 19 * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 20 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 21 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; 22 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, 23 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR 24 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF 25 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 26 */ 27 28// This file contains mock implementations of observers used in PeerConnection. 29 30#ifndef TALK_APP_WEBRTC_TEST_MOCKPEERCONNECTIONOBSERVERS_H_ 31#define TALK_APP_WEBRTC_TEST_MOCKPEERCONNECTIONOBSERVERS_H_ 32 33#include <string> 34 35#include "talk/app/webrtc/datachannelinterface.h" 36 37namespace webrtc { 38 39class MockCreateSessionDescriptionObserver 40 : public webrtc::CreateSessionDescriptionObserver { 41 public: 42 MockCreateSessionDescriptionObserver() 43 : called_(false), 44 result_(false) {} 45 virtual ~MockCreateSessionDescriptionObserver() {} 46 virtual void OnSuccess(SessionDescriptionInterface* desc) { 47 called_ = true; 48 result_ = true; 49 desc_.reset(desc); 50 } 51 virtual void OnFailure(const std::string& error) { 52 called_ = true; 53 result_ = false; 54 } 55 bool called() const { return called_; } 56 bool result() const { return result_; } 57 SessionDescriptionInterface* release_desc() { 58 return desc_.release(); 59 } 60 61 private: 62 bool called_; 63 bool result_; 64 talk_base::scoped_ptr<SessionDescriptionInterface> desc_; 65}; 66 67class MockSetSessionDescriptionObserver 68 : public webrtc::SetSessionDescriptionObserver { 69 public: 70 MockSetSessionDescriptionObserver() 71 : called_(false), 72 result_(false) {} 73 virtual ~MockSetSessionDescriptionObserver() {} 74 virtual void OnSuccess() { 75 called_ = true; 76 result_ = true; 77 } 78 virtual void OnFailure(const std::string& error) { 79 called_ = true; 80 result_ = false; 81 } 82 bool called() const { return called_; } 83 bool result() const { return result_; } 84 85 private: 86 bool called_; 87 bool result_; 88}; 89 90class MockDataChannelObserver : public webrtc::DataChannelObserver { 91 public: 92 explicit MockDataChannelObserver(webrtc::DataChannelInterface* channel) 93 : channel_(channel) { 94 channel_->RegisterObserver(this); 95 state_ = channel_->state(); 96 } 97 virtual ~MockDataChannelObserver() { 98 channel_->UnregisterObserver(); 99 } 100 101 virtual void OnStateChange() { state_ = channel_->state(); } 102 virtual void OnMessage(const DataBuffer& buffer) { 103 last_message_.assign(buffer.data.data(), buffer.data.length()); 104 } 105 106 bool IsOpen() const { return state_ == DataChannelInterface::kOpen; } 107 const std::string& last_message() const { return last_message_; } 108 109 private: 110 talk_base::scoped_refptr<webrtc::DataChannelInterface> channel_; 111 DataChannelInterface::DataState state_; 112 std::string last_message_; 113}; 114 115class MockStatsObserver : public webrtc::StatsObserver { 116 public: 117 MockStatsObserver() 118 : called_(false) {} 119 virtual ~MockStatsObserver() {} 120 virtual void OnComplete(const std::vector<webrtc::StatsReport>& reports) { 121 called_ = true; 122 reports_ = reports; 123 } 124 125 bool called() const { return called_; } 126 size_t number_of_reports() const { return reports_.size(); } 127 128 int AudioOutputLevel() { 129 return GetSsrcStatsValue( 130 webrtc::StatsReport::kStatsValueNameAudioOutputLevel); 131 } 132 133 int AudioInputLevel() { 134 return GetSsrcStatsValue( 135 webrtc::StatsReport::kStatsValueNameAudioInputLevel); 136 } 137 138 int BytesReceived() { 139 return GetSsrcStatsValue( 140 webrtc::StatsReport::kStatsValueNameBytesReceived); 141 } 142 143 int BytesSent() { 144 return GetSsrcStatsValue(webrtc::StatsReport::kStatsValueNameBytesSent); 145 } 146 147 private: 148 int GetSsrcStatsValue(const std::string name) { 149 if (reports_.empty()) { 150 return 0; 151 } 152 for (size_t i = 0; i < reports_.size(); ++i) { 153 if (reports_[i].type != StatsReport::kStatsReportTypeSsrc) 154 continue; 155 webrtc::StatsReport::Values::const_iterator it = 156 reports_[i].values.begin(); 157 for (; it != reports_[i].values.end(); ++it) { 158 if (it->name == name) { 159 return talk_base::FromString<int>(it->value); 160 } 161 } 162 } 163 return 0; 164 } 165 166 bool called_; 167 std::vector<webrtc::StatsReport> reports_; 168}; 169 170} // namespace webrtc 171 172#endif // TALK_APP_WEBRTC_TEST_MOCKPEERCONNECTIONOBSERVERS_H_ 173