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 |