Split the mediastream module in Blink into mediastream and peerconnection

third_party/WebKit/Source/modules/mediastream/RTCDataChannelTest.cpp

-// Copyright 2015 The Chromium Authors. All rights reserved.
-// Use of this source code is governed by a BSD-style license that can be
-// found in the LICENSE file.
-
-#include "modules/mediastream/RTCDataChannel.h"
-
-#include "bindings/core/v8/ExceptionStatePlaceholder.h"
-#include "core/dom/DOMArrayBuffer.h"
-#include "core/dom/DOMException.h"
-#include "core/events/Event.h"
-#include "platform/heap/Heap.h"
-#include "public/platform/WebRTCDataChannelHandler.h"
-#include "public/platform/WebVector.h"
-#include "testing/gtest/include/gtest/gtest.h"
-#include "wtf/PtrUtil.h"
-#include "wtf/RefPtr.h"
-#include "wtf/text/WTFString.h"
-
-namespace blink {
-namespace {
-
-class MockHandler final : public WebRTCDataChannelHandler {
-public:
-    MockHandler()
-        : m_client(0)
-        , m_state(WebRTCDataChannelHandlerClient::ReadyStateConnecting)
-        , m_bufferedAmount(0)
-    {
-    }
-    void setClient(WebRTCDataChannelHandlerClient* client) override { m_client = client; }
-    WebString label() override { return WebString(""); }
-    bool ordered() const override { return true; }
-    unsigned short maxRetransmitTime() const override { return 0; }
-    unsigned short maxRetransmits() const override { return 0; }
-    WebString protocol() const override { return WebString(""); }
-    bool negotiated() const override { return false; }
-    unsigned short id() const override { return 0; }
-
-    WebRTCDataChannelHandlerClient::ReadyState state() const override { return m_state; }
-    unsigned long bufferedAmount() override { return m_bufferedAmount; }
-    bool sendStringData(const WebString& s) override
-    {
-        m_bufferedAmount += s.length();
-        return true;
-    }
-    bool sendRawData(const char* data, size_t length) override
-    {
-        m_bufferedAmount += length;
-        return true;
-    }
-    void close() override { }
-
-    // Methods for testing.
-    void changeState(WebRTCDataChannelHandlerClient::ReadyState state)
-    {
-        m_state = state;
-        if (m_client) {
-            m_client->didChangeReadyState(m_state);
-        }
-    }
-    void drainBuffer(unsigned long bytes)
-    {
-        unsigned long oldBufferedAmount = m_bufferedAmount;
-        m_bufferedAmount -= bytes;
-        if (m_client) {
-            m_client->didDecreaseBufferedAmount(oldBufferedAmount);
-        }
-    }
-
-private:
-    WebRTCDataChannelHandlerClient* m_client;
-    WebRTCDataChannelHandlerClient::ReadyState m_state;
-    unsigned long m_bufferedAmount;
-};
-
-} // namespace
-
-TEST(RTCDataChannelTest, BufferedAmount)
-{
-    MockHandler* handler = new MockHandler();
-    RTCDataChannel* channel = RTCDataChannel::create(0, wrapUnique(handler));
-
-    handler->changeState(WebRTCDataChannelHandlerClient::ReadyStateOpen);
-    String message(std::string(100, 'A').c_str());
-    channel->send(message, IGNORE_EXCEPTION);
-    EXPECT_EQ(100U, channel->bufferedAmount());
-}
-
-TEST(RTCDataChannelTest, BufferedAmountLow)
-{
-    MockHandler* handler = new MockHandler();
-    RTCDataChannel* channel = RTCDataChannel::create(0, wrapUnique(handler));
-
-    // Add and drain 100 bytes
-    handler->changeState(WebRTCDataChannelHandlerClient::ReadyStateOpen);
-    String message(std::string(100, 'A').c_str());
-    channel->send(message, IGNORE_EXCEPTION);
-    EXPECT_EQ(100U, channel->bufferedAmount());
-    EXPECT_EQ(1U, channel->m_scheduledEvents.size());
-
-    handler->drainBuffer(100);
-    EXPECT_EQ(0U, channel->bufferedAmount());
-    EXPECT_EQ(2U, channel->m_scheduledEvents.size());
-    EXPECT_EQ("bufferedamountlow", std::string(channel->m_scheduledEvents.last()->type().utf8().data()));
-
-    // Add and drain 1 byte
-    channel->send("A", IGNORE_EXCEPTION);
-    EXPECT_EQ(1U, channel->bufferedAmount());
-    EXPECT_EQ(2U, channel->m_scheduledEvents.size());
-
-    handler->drainBuffer(1);
-    EXPECT_EQ(0U, channel->bufferedAmount());
-    EXPECT_EQ(3U, channel->m_scheduledEvents.size());
-    EXPECT_EQ("bufferedamountlow", std::string(channel->m_scheduledEvents.last()->type().utf8().data()));
-
-    // Set the threshold to 99 bytes, add 101, and drain 1 byte at a time.
-    channel->setBufferedAmountLowThreshold(99U);
-    channel->send(message, IGNORE_EXCEPTION);
-    EXPECT_EQ(100U, channel->bufferedAmount());
-
-    channel->send("A", IGNORE_EXCEPTION);
-    EXPECT_EQ(101U, channel->bufferedAmount());
-
-    handler->drainBuffer(1);
-    EXPECT_EQ(100U, channel->bufferedAmount());
-    EXPECT_EQ(3U, channel->m_scheduledEvents.size()); // No new events.
-
-    handler->drainBuffer(1);
-    EXPECT_EQ(99U, channel->bufferedAmount());
-    EXPECT_EQ(4U, channel->m_scheduledEvents.size()); // One new event.
-    EXPECT_EQ("bufferedamountlow", std::string(channel->m_scheduledEvents.last()->type().utf8().data()));
-
-    handler->drainBuffer(1);
-    EXPECT_EQ(98U, channel->bufferedAmount());
-
-    channel->setBufferedAmountLowThreshold(97U);
-    EXPECT_EQ(4U, channel->m_scheduledEvents.size()); // No new events.
-
-    handler->drainBuffer(1);
-    EXPECT_EQ(97U, channel->bufferedAmount());
-    EXPECT_EQ(5U, channel->m_scheduledEvents.size()); // New event.
-    EXPECT_EQ("bufferedamountlow", std::string(channel->m_scheduledEvents.last()->type().utf8().data()));
-}
-
-} // namespace blink