| 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" | |
| 16 #include "wtf/RefPtr.h" | 15 #include "wtf/RefPtr.h" |
| 17 #include "wtf/text/WTFString.h" | 16 #include "wtf/text/WTFString.h" |
| 18 | 17 |
| 19 namespace blink { | 18 namespace blink { |
| 20 namespace { | 19 namespace { |
| 21 | 20 |
| 22 class MockHandler final : public WebRTCDataChannelHandler { | 21 class MockHandler final : public WebRTCDataChannelHandler { |
| 23 public: | 22 public: |
| 24 MockHandler() | 23 MockHandler() |
| 25 : m_client(0) | 24 : m_client(0) |
| (...skipping 45 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 71 WebRTCDataChannelHandlerClient* m_client; | 70 WebRTCDataChannelHandlerClient* m_client; |
| 72 WebRTCDataChannelHandlerClient::ReadyState m_state; | 71 WebRTCDataChannelHandlerClient::ReadyState m_state; |
| 73 unsigned long m_bufferedAmount; | 72 unsigned long m_bufferedAmount; |
| 74 }; | 73 }; |
| 75 | 74 |
| 76 } // namespace | 75 } // namespace |
| 77 | 76 |
| 78 TEST(RTCDataChannelTest, BufferedAmount) | 77 TEST(RTCDataChannelTest, BufferedAmount) |
| 79 { | 78 { |
| 80 MockHandler* handler = new MockHandler(); | 79 MockHandler* handler = new MockHandler(); |
| 81 RTCDataChannel* channel = RTCDataChannel::create(0, wrapUnique(handler)); | 80 RTCDataChannel* channel = RTCDataChannel::create(0, adoptPtr(handler)); |
| 82 | 81 |
| 83 handler->changeState(WebRTCDataChannelHandlerClient::ReadyStateOpen); | 82 handler->changeState(WebRTCDataChannelHandlerClient::ReadyStateOpen); |
| 84 String message(std::string(100, 'A').c_str()); | 83 String message(std::string(100, 'A').c_str()); |
| 85 channel->send(message, IGNORE_EXCEPTION); | 84 channel->send(message, IGNORE_EXCEPTION); |
| 86 EXPECT_EQ(100U, channel->bufferedAmount()); | 85 EXPECT_EQ(100U, channel->bufferedAmount()); |
| 87 } | 86 } |
| 88 | 87 |
| 89 TEST(RTCDataChannelTest, BufferedAmountLow) | 88 TEST(RTCDataChannelTest, BufferedAmountLow) |
| 90 { | 89 { |
| 91 MockHandler* handler = new MockHandler(); | 90 MockHandler* handler = new MockHandler(); |
| 92 RTCDataChannel* channel = RTCDataChannel::create(0, wrapUnique(handler)); | 91 RTCDataChannel* channel = RTCDataChannel::create(0, adoptPtr(handler)); |
| 93 | 92 |
| 94 // Add and drain 100 bytes | 93 // Add and drain 100 bytes |
| 95 handler->changeState(WebRTCDataChannelHandlerClient::ReadyStateOpen); | 94 handler->changeState(WebRTCDataChannelHandlerClient::ReadyStateOpen); |
| 96 String message(std::string(100, 'A').c_str()); | 95 String message(std::string(100, 'A').c_str()); |
| 97 channel->send(message, IGNORE_EXCEPTION); | 96 channel->send(message, IGNORE_EXCEPTION); |
| 98 EXPECT_EQ(100U, channel->bufferedAmount()); | 97 EXPECT_EQ(100U, channel->bufferedAmount()); |
| 99 EXPECT_EQ(1U, channel->m_scheduledEvents.size()); | 98 EXPECT_EQ(1U, channel->m_scheduledEvents.size()); |
| 100 | 99 |
| 101 handler->drainBuffer(100); | 100 handler->drainBuffer(100); |
| 102 EXPECT_EQ(0U, channel->bufferedAmount()); | 101 EXPECT_EQ(0U, channel->bufferedAmount()); |
| (...skipping 33 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 136 channel->setBufferedAmountLowThreshold(97U); | 135 channel->setBufferedAmountLowThreshold(97U); |
| 137 EXPECT_EQ(4U, channel->m_scheduledEvents.size()); // No new events. | 136 EXPECT_EQ(4U, channel->m_scheduledEvents.size()); // No new events. |
| 138 | 137 |
| 139 handler->drainBuffer(1); | 138 handler->drainBuffer(1); |
| 140 EXPECT_EQ(97U, channel->bufferedAmount()); | 139 EXPECT_EQ(97U, channel->bufferedAmount()); |
| 141 EXPECT_EQ(5U, channel->m_scheduledEvents.size()); // New event. | 140 EXPECT_EQ(5U, channel->m_scheduledEvents.size()); // New event. |
| 142 EXPECT_EQ("bufferedamountlow", std::string(channel->m_scheduledEvents.last()
->type().utf8().data())); | 141 EXPECT_EQ("bufferedamountlow", std::string(channel->m_scheduledEvents.last()
->type().utf8().data())); |
| 143 } | 142 } |
| 144 | 143 |
| 145 } // namespace blink | 144 } // namespace blink |
| OLD | NEW |