| 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/peerconnection/RTCDataChannel.h" | 5 #include "modules/peerconnection/RTCDataChannel.h" |
| 6 | 6 |
| 7 #include "bindings/core/v8/ExceptionState.h" | 7 #include "bindings/core/v8/ExceptionState.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" |
| (...skipping 61 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 72 }; | 72 }; |
| 73 | 73 |
| 74 } // namespace | 74 } // namespace |
| 75 | 75 |
| 76 TEST(RTCDataChannelTest, BufferedAmount) { | 76 TEST(RTCDataChannelTest, BufferedAmount) { |
| 77 MockHandler* handler = new MockHandler(); | 77 MockHandler* handler = new MockHandler(); |
| 78 RTCDataChannel* channel = RTCDataChannel::create(0, WTF::wrapUnique(handler)); | 78 RTCDataChannel* channel = RTCDataChannel::create(0, WTF::wrapUnique(handler)); |
| 79 | 79 |
| 80 handler->changeState(WebRTCDataChannelHandlerClient::ReadyStateOpen); | 80 handler->changeState(WebRTCDataChannelHandlerClient::ReadyStateOpen); |
| 81 String message(std::string(100, 'A').c_str()); | 81 String message(std::string(100, 'A').c_str()); |
| 82 channel->send(message, IGNORE_EXCEPTION); | 82 channel->send(message, IGNORE_EXCEPTION_FOR_TESTING); |
| 83 EXPECT_EQ(100U, channel->bufferedAmount()); | 83 EXPECT_EQ(100U, channel->bufferedAmount()); |
| 84 } | 84 } |
| 85 | 85 |
| 86 TEST(RTCDataChannelTest, BufferedAmountLow) { | 86 TEST(RTCDataChannelTest, BufferedAmountLow) { |
| 87 MockHandler* handler = new MockHandler(); | 87 MockHandler* handler = new MockHandler(); |
| 88 RTCDataChannel* channel = RTCDataChannel::create(0, WTF::wrapUnique(handler)); | 88 RTCDataChannel* channel = RTCDataChannel::create(0, WTF::wrapUnique(handler)); |
| 89 | 89 |
| 90 // Add and drain 100 bytes | 90 // Add and drain 100 bytes |
| 91 handler->changeState(WebRTCDataChannelHandlerClient::ReadyStateOpen); | 91 handler->changeState(WebRTCDataChannelHandlerClient::ReadyStateOpen); |
| 92 String message(std::string(100, 'A').c_str()); | 92 String message(std::string(100, 'A').c_str()); |
| 93 channel->send(message, IGNORE_EXCEPTION); | 93 channel->send(message, IGNORE_EXCEPTION_FOR_TESTING); |
| 94 EXPECT_EQ(100U, channel->bufferedAmount()); | 94 EXPECT_EQ(100U, channel->bufferedAmount()); |
| 95 EXPECT_EQ(1U, channel->m_scheduledEvents.size()); | 95 EXPECT_EQ(1U, channel->m_scheduledEvents.size()); |
| 96 | 96 |
| 97 handler->drainBuffer(100); | 97 handler->drainBuffer(100); |
| 98 EXPECT_EQ(0U, channel->bufferedAmount()); | 98 EXPECT_EQ(0U, channel->bufferedAmount()); |
| 99 EXPECT_EQ(2U, channel->m_scheduledEvents.size()); | 99 EXPECT_EQ(2U, channel->m_scheduledEvents.size()); |
| 100 EXPECT_EQ( | 100 EXPECT_EQ( |
| 101 "bufferedamountlow", | 101 "bufferedamountlow", |
| 102 std::string(channel->m_scheduledEvents.back()->type().utf8().data())); | 102 std::string(channel->m_scheduledEvents.back()->type().utf8().data())); |
| 103 | 103 |
| 104 // Add and drain 1 byte | 104 // Add and drain 1 byte |
| 105 channel->send("A", IGNORE_EXCEPTION); | 105 channel->send("A", IGNORE_EXCEPTION_FOR_TESTING); |
| 106 EXPECT_EQ(1U, channel->bufferedAmount()); | 106 EXPECT_EQ(1U, channel->bufferedAmount()); |
| 107 EXPECT_EQ(2U, channel->m_scheduledEvents.size()); | 107 EXPECT_EQ(2U, channel->m_scheduledEvents.size()); |
| 108 | 108 |
| 109 handler->drainBuffer(1); | 109 handler->drainBuffer(1); |
| 110 EXPECT_EQ(0U, channel->bufferedAmount()); | 110 EXPECT_EQ(0U, channel->bufferedAmount()); |
| 111 EXPECT_EQ(3U, channel->m_scheduledEvents.size()); | 111 EXPECT_EQ(3U, channel->m_scheduledEvents.size()); |
| 112 EXPECT_EQ( | 112 EXPECT_EQ( |
| 113 "bufferedamountlow", | 113 "bufferedamountlow", |
| 114 std::string(channel->m_scheduledEvents.back()->type().utf8().data())); | 114 std::string(channel->m_scheduledEvents.back()->type().utf8().data())); |
| 115 | 115 |
| 116 // Set the threshold to 99 bytes, add 101, and drain 1 byte at a time. | 116 // Set the threshold to 99 bytes, add 101, and drain 1 byte at a time. |
| 117 channel->setBufferedAmountLowThreshold(99U); | 117 channel->setBufferedAmountLowThreshold(99U); |
| 118 channel->send(message, IGNORE_EXCEPTION); | 118 channel->send(message, IGNORE_EXCEPTION_FOR_TESTING); |
| 119 EXPECT_EQ(100U, channel->bufferedAmount()); | 119 EXPECT_EQ(100U, channel->bufferedAmount()); |
| 120 | 120 |
| 121 channel->send("A", IGNORE_EXCEPTION); | 121 channel->send("A", IGNORE_EXCEPTION_FOR_TESTING); |
| 122 EXPECT_EQ(101U, channel->bufferedAmount()); | 122 EXPECT_EQ(101U, channel->bufferedAmount()); |
| 123 | 123 |
| 124 handler->drainBuffer(1); | 124 handler->drainBuffer(1); |
| 125 EXPECT_EQ(100U, channel->bufferedAmount()); | 125 EXPECT_EQ(100U, channel->bufferedAmount()); |
| 126 EXPECT_EQ(3U, channel->m_scheduledEvents.size()); // No new events. | 126 EXPECT_EQ(3U, channel->m_scheduledEvents.size()); // No new events. |
| 127 | 127 |
| 128 handler->drainBuffer(1); | 128 handler->drainBuffer(1); |
| 129 EXPECT_EQ(99U, channel->bufferedAmount()); | 129 EXPECT_EQ(99U, channel->bufferedAmount()); |
| 130 EXPECT_EQ(4U, channel->m_scheduledEvents.size()); // One new event. | 130 EXPECT_EQ(4U, channel->m_scheduledEvents.size()); // One new event. |
| 131 EXPECT_EQ( | 131 EXPECT_EQ( |
| 132 "bufferedamountlow", | 132 "bufferedamountlow", |
| 133 std::string(channel->m_scheduledEvents.back()->type().utf8().data())); | 133 std::string(channel->m_scheduledEvents.back()->type().utf8().data())); |
| 134 | 134 |
| 135 handler->drainBuffer(1); | 135 handler->drainBuffer(1); |
| 136 EXPECT_EQ(98U, channel->bufferedAmount()); | 136 EXPECT_EQ(98U, channel->bufferedAmount()); |
| 137 | 137 |
| 138 channel->setBufferedAmountLowThreshold(97U); | 138 channel->setBufferedAmountLowThreshold(97U); |
| 139 EXPECT_EQ(4U, channel->m_scheduledEvents.size()); // No new events. | 139 EXPECT_EQ(4U, channel->m_scheduledEvents.size()); // No new events. |
| 140 | 140 |
| 141 handler->drainBuffer(1); | 141 handler->drainBuffer(1); |
| 142 EXPECT_EQ(97U, channel->bufferedAmount()); | 142 EXPECT_EQ(97U, channel->bufferedAmount()); |
| 143 EXPECT_EQ(5U, channel->m_scheduledEvents.size()); // New event. | 143 EXPECT_EQ(5U, channel->m_scheduledEvents.size()); // New event. |
| 144 EXPECT_EQ( | 144 EXPECT_EQ( |
| 145 "bufferedamountlow", | 145 "bufferedamountlow", |
| 146 std::string(channel->m_scheduledEvents.back()->type().utf8().data())); | 146 std::string(channel->m_scheduledEvents.back()->type().utf8().data())); |
| 147 } | 147 } |
| 148 | 148 |
| 149 } // namespace blink | 149 } // namespace blink |
| OLD | NEW |