| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "modules/mediastream/RTCDataChannel.h" | 5 #include "modules/mediastream/RTCDataChannel.h" |
| 6 | 6 |
| 7 #include "bindings/core/v8/ExceptionStatePlaceholder.h" | 7 #include "bindings/core/v8/ExceptionStatePlaceholder.h" |
| 8 #include "core/dom/DOMArrayBuffer.h" | 8 #include "core/dom/DOMArrayBuffer.h" |
| 9 #include "core/dom/DOMException.h" | 9 #include "core/dom/DOMException.h" |
| 10 #include "core/events/Event.h" | 10 #include "core/events/Event.h" |
| 11 #include "platform/heap/Heap.h" | 11 #include "platform/heap/Heap.h" |
| 12 #include "public/platform/WebRTCDataChannelHandler.h" | 12 #include "public/platform/WebRTCDataChannelHandler.h" |
| 13 #include "public/platform/WebVector.h" | 13 #include "public/platform/WebVector.h" |
| 14 #include "testing/gtest/include/gtest/gtest.h" | 14 #include "testing/gtest/include/gtest/gtest.h" |
| 15 #include "wtf/PtrUtil.h" |
| 15 #include "wtf/RefPtr.h" | 16 #include "wtf/RefPtr.h" |
| 16 #include "wtf/text/WTFString.h" | 17 #include "wtf/text/WTFString.h" |
| 17 | 18 |
| 18 namespace blink { | 19 namespace blink { |
| 19 namespace { | 20 namespace { |
| 20 | 21 |
| 21 class MockHandler final : public WebRTCDataChannelHandler { | 22 class MockHandler final : public WebRTCDataChannelHandler { |
| 22 public: | 23 public: |
| 23 MockHandler() | 24 MockHandler() |
| 24 : m_client(0) | 25 : m_client(0) |
| (...skipping 45 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 70 WebRTCDataChannelHandlerClient* m_client; | 71 WebRTCDataChannelHandlerClient* m_client; |
| 71 WebRTCDataChannelHandlerClient::ReadyState m_state; | 72 WebRTCDataChannelHandlerClient::ReadyState m_state; |
| 72 unsigned long m_bufferedAmount; | 73 unsigned long m_bufferedAmount; |
| 73 }; | 74 }; |
| 74 | 75 |
| 75 } // namespace | 76 } // namespace |
| 76 | 77 |
| 77 TEST(RTCDataChannelTest, BufferedAmount) | 78 TEST(RTCDataChannelTest, BufferedAmount) |
| 78 { | 79 { |
| 79 MockHandler* handler = new MockHandler(); | 80 MockHandler* handler = new MockHandler(); |
| 80 RTCDataChannel* channel = RTCDataChannel::create(0, adoptPtr(handler)); | 81 RTCDataChannel* channel = RTCDataChannel::create(0, wrapUnique(handler)); |
| 81 | 82 |
| 82 handler->changeState(WebRTCDataChannelHandlerClient::ReadyStateOpen); | 83 handler->changeState(WebRTCDataChannelHandlerClient::ReadyStateOpen); |
| 83 String message(std::string(100, 'A').c_str()); | 84 String message(std::string(100, 'A').c_str()); |
| 84 channel->send(message, IGNORE_EXCEPTION); | 85 channel->send(message, IGNORE_EXCEPTION); |
| 85 EXPECT_EQ(100U, channel->bufferedAmount()); | 86 EXPECT_EQ(100U, channel->bufferedAmount()); |
| 86 } | 87 } |
| 87 | 88 |
| 88 TEST(RTCDataChannelTest, BufferedAmountLow) | 89 TEST(RTCDataChannelTest, BufferedAmountLow) |
| 89 { | 90 { |
| 90 MockHandler* handler = new MockHandler(); | 91 MockHandler* handler = new MockHandler(); |
| 91 RTCDataChannel* channel = RTCDataChannel::create(0, adoptPtr(handler)); | 92 RTCDataChannel* channel = RTCDataChannel::create(0, wrapUnique(handler)); |
| 92 | 93 |
| 93 // Add and drain 100 bytes | 94 // Add and drain 100 bytes |
| 94 handler->changeState(WebRTCDataChannelHandlerClient::ReadyStateOpen); | 95 handler->changeState(WebRTCDataChannelHandlerClient::ReadyStateOpen); |
| 95 String message(std::string(100, 'A').c_str()); | 96 String message(std::string(100, 'A').c_str()); |
| 96 channel->send(message, IGNORE_EXCEPTION); | 97 channel->send(message, IGNORE_EXCEPTION); |
| 97 EXPECT_EQ(100U, channel->bufferedAmount()); | 98 EXPECT_EQ(100U, channel->bufferedAmount()); |
| 98 EXPECT_EQ(1U, channel->m_scheduledEvents.size()); | 99 EXPECT_EQ(1U, channel->m_scheduledEvents.size()); |
| 99 | 100 |
| 100 handler->drainBuffer(100); | 101 handler->drainBuffer(100); |
| 101 EXPECT_EQ(0U, channel->bufferedAmount()); | 102 EXPECT_EQ(0U, channel->bufferedAmount()); |
| (...skipping 33 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 135 channel->setBufferedAmountLowThreshold(97U); | 136 channel->setBufferedAmountLowThreshold(97U); |
| 136 EXPECT_EQ(4U, channel->m_scheduledEvents.size()); // No new events. | 137 EXPECT_EQ(4U, channel->m_scheduledEvents.size()); // No new events. |
| 137 | 138 |
| 138 handler->drainBuffer(1); | 139 handler->drainBuffer(1); |
| 139 EXPECT_EQ(97U, channel->bufferedAmount()); | 140 EXPECT_EQ(97U, channel->bufferedAmount()); |
| 140 EXPECT_EQ(5U, channel->m_scheduledEvents.size()); // New event. | 141 EXPECT_EQ(5U, channel->m_scheduledEvents.size()); // New event. |
| 141 EXPECT_EQ("bufferedamountlow", std::string(channel->m_scheduledEvents.last()
->type().utf8().data())); | 142 EXPECT_EQ("bufferedamountlow", std::string(channel->m_scheduledEvents.last()
->type().utf8().data())); |
| 142 } | 143 } |
| 143 | 144 |
| 144 } // namespace blink | 145 } // namespace blink |
| OLD | NEW |