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