Index: webrtc/api/test/mockpeerconnectionobservers.h |
diff --git a/webrtc/api/test/mockpeerconnectionobservers.h b/webrtc/api/test/mockpeerconnectionobservers.h |
index 39a8f0134d63f42d82390cc9840e98e7734024b7..23647f6de3b5cbf3d5c8e6fda845103c77a04441 100644 |
--- a/webrtc/api/test/mockpeerconnectionobservers.h |
+++ b/webrtc/api/test/mockpeerconnectionobservers.h |
@@ -74,7 +74,7 @@ class MockSetSessionDescriptionObserver |
class MockDataChannelObserver : public webrtc::DataChannelObserver { |
public: |
explicit MockDataChannelObserver(webrtc::DataChannelInterface* channel) |
- : channel_(channel), received_message_count_(0) { |
+ : channel_(channel) { |
channel_->RegisterObserver(this); |
state_ = channel_->state(); |
} |
@@ -86,19 +86,21 @@ class MockDataChannelObserver : public webrtc::DataChannelObserver { |
void OnStateChange() override { state_ = channel_->state(); } |
void OnMessage(const DataBuffer& buffer) override { |
- last_message_.assign(buffer.data.data<char>(), buffer.data.size()); |
- ++received_message_count_; |
+ messages_.push_back( |
+ std::string(buffer.data.data<char>(), buffer.data.size())); |
} |
bool IsOpen() const { return state_ == DataChannelInterface::kOpen; } |
- const std::string& last_message() const { return last_message_; } |
- size_t received_message_count() const { return received_message_count_; } |
+ std::vector<std::string> messages() const { return messages_; } |
+ std::string last_message() const { |
+ return messages_.empty() ? std::string() : messages_.back(); |
+ } |
+ size_t received_message_count() const { return messages_.size(); } |
private: |
rtc::scoped_refptr<webrtc::DataChannelInterface> channel_; |
DataChannelInterface::DataState state_; |
- std::string last_message_; |
- size_t received_message_count_; |
+ std::vector<std::string> messages_; |
}; |
class MockStatsObserver : public webrtc::StatsObserver { |