[ABANDONED] Use new Websocket SendBlob IPC from renderer.

third_party/WebKit/Source/modules/websockets/DocumentWebSocketChannelTest.cpp

 // Copyright 2014 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/websockets/WebSocketChannel.h"
 
 #include "core/dom/DOMArrayBuffer.h"
 #include "core/dom/Document.h"
 #include "core/fileapi/Blob.h"
 #include "core/frame/ConsoleTypes.h"
@@ skipping 64 matching lines @@
     {
         return new testing::StrictMock<MockWebSocketHandle>();
     }
 
     MockWebSocketHandle() { }
 
     ~MockWebSocketHandle() override { }
 
     MOCK_METHOD4(connect, void(const WebURL&, const WebVector<WebString>&, const WebSecurityOrigin&, WebSocketHandleClient*));
     MOCK_METHOD4(send, void(bool, WebSocketHandle::MessageType, const char*, size_t));
+    MOCK_METHOD2(sendBlob, void(const blink::WebString&, uint64_t));
     MOCK_METHOD1(flowControl, void(int64_t));
     MOCK_METHOD2(close, void(unsigned short, const WebString&));
 };
 
 class DocumentWebSocketChannelTest : public ::testing::Test {
 public:
     DocumentWebSocketChannelTest()
         : m_pageHolder(DummyPageHolder::create())
         , m_channelClient(MockWebSocketChannelClient::create())
         , m_handle(MockWebSocketHandle::create())
@@ skipping 46 matching lines @@
         ::testing::Mock::VerifyAndClearExpectations(this);
     }
 
     OwnPtr<DummyPageHolder> m_pageHolder;
     Persistent<MockWebSocketChannelClient> m_channelClient;
     MockWebSocketHandle* m_handle;
     Persistent<DocumentWebSocketChannel> m_channel;
     unsigned long m_sumOfConsumedBufferedAmount;
 };
 
+// The WebSocket protocol and API ignores the type of the Blob, so put a
+// default value in that field.
+PassRefPtr<BlobDataHandle> createBlob(const String& uuid, long long size)
+{
+    return BlobDataHandle::create(uuid, "unknown/unknown", size);
+}
+
 MATCHER_P2(MemEq, p, len,
     std::string("pointing to memory")
     + (negation ? " not" : "")
     + " equal to \""
     + std::string(p, len) + "\" (length=" + PrintToString(len) + ")"
 )
 {
     return memcmp(arg, p, len) == 0;
 }
 
@@ skipping 295 matching lines @@
 
     RefPtr<DOMArrayBuffer> b = DOMArrayBuffer::create("\xe7\x8b\x90\xe7\x8b\x90\xe7\x8b\x90\xe7\x8b\x90\xe7\x8b\x90\xe7\x8b\x90", 18);
     channel()->send(*b, 0, 18);
     checkpoint.Call(1);
 
     handleClient()->didReceiveFlowControl(handle(), 16);
 
     EXPECT_EQ(18ul, m_sumOfConsumedBufferedAmount);
 }
 
-// FIXME: Add tests for WebSocketChannel::send(PassRefPtr<BlobDataHandle>)
+TEST_F(DocumentWebSocketChannelTest, sendSingleBlob)
+{
+    connect();
+    EXPECT_CALL(*handle(), sendBlob(WebString("uuid"), 10));
+    handleClient()->didReceiveFlowControl(handle(), 16);
+    channel()->send(createBlob("uuid", 10));
+}
+
+TEST_F(DocumentWebSocketChannelTest, sendTwoBlobs)
+{
+    connect();
+    Checkpoint checkpoint;
+    {
+        InSequence s;
+        EXPECT_CALL(*handle(), sendBlob(WebString("uuid1"), 11));
+        EXPECT_CALL(checkpoint, Call(1));
+        EXPECT_CALL(*handle(), sendBlob(WebString("uuid2"), 12));
+    }
+    EXPECT_CALL(*channelClient(), didConsumeBufferedAmount(_))
+        .Times(AnyNumber());
+    handleClient()->didReceiveFlowControl(handle(), 32);
+    channel()->send(createBlob("uuid1", 11));
+    channel()->send(createBlob("uuid2", 12));
+    checkpoint.Call(1);
+    handleClient()->didCompleteSendingBlob(handle());
+}
+
+TEST_F(DocumentWebSocketChannelTest, flowControlReflectedInBufferedAmount)
+{
+    connect();
+    EXPECT_CALL(*handle(), sendBlob(WebString("uuid"), 256));
+    EXPECT_CALL(*channelClient(), didConsumeBufferedAmount(_))
+        .Times(AnyNumber());
+    handleClient()->didReceiveFlowControl(handle(), 128);
+    channel()->send(createBlob("uuid", 256));
+    EXPECT_EQ(0u, m_sumOfConsumedBufferedAmount);
+    handleClient()->didReceiveFlowControl(handle(), 128);
+    EXPECT_EQ(128u, m_sumOfConsumedBufferedAmount);
+    handleClient()->didCompleteSendingBlob(handle());
+    EXPECT_EQ(256u, m_sumOfConsumedBufferedAmount);
+}
+
+// Verifies that Blobs respect queue ordering.
+TEST_F(DocumentWebSocketChannelTest, sendTextThenBlob)
+{
+    connect();
+    Checkpoint checkpoint;
+    {
+        InSequence s;
+        EXPECT_CALL(checkpoint, Call(1));
+        EXPECT_CALL(*handle(), send(true, WebSocketHandle::MessageTypeText, MemEq("foo", 3), 3));
+        EXPECT_CALL(*handle(), sendBlob(WebString("uuid"), 16));
+    }
+    EXPECT_CALL(*channelClient(), didConsumeBufferedAmount(_))
+        .Times(AnyNumber());
+    channel()->send("foo");
+    channel()->send(createBlob("uuid", 16));
+    checkpoint.Call(1);
+    handleClient()->didReceiveFlowControl(handle(), 128);
+}
+
+// Verifies that "Blob sending mode" is exited correctly.
+TEST_F(DocumentWebSocketChannelTest, sendBlobThenText)
+{
+    connect();
+    Checkpoint checkpoint;
+    {
+        InSequence s;
+        EXPECT_CALL(*handle(), sendBlob(WebString("uuid"), 16));
+        EXPECT_CALL(checkpoint, Call(1));
+        EXPECT_CALL(*handle(), send(true, WebSocketHandle::MessageTypeText, MemEq("foo", 3), 3));
+    }
+    EXPECT_CALL(*channelClient(), didConsumeBufferedAmount(_))
+        .Times(AnyNumber());
+    handleClient()->didReceiveFlowControl(handle(), 128);
+    channel()->send(createBlob("uuid", 16));
+    channel()->send("foo");
+    checkpoint.Call(1);
+    // Exits "Blob sending mode" and allows text message "foo" to be sent.
+    handleClient()->didCompleteSendingBlob(handle());
+}
+
+// Verifies that other messages are queued during "Blob sending mode", and that
+// flow control messages do not cause them to be sent prematurely.
+TEST_F(DocumentWebSocketChannelTest, sendLargeBlobThenText)
+{
+    connect();
+    Checkpoint checkpoint;
+    {
+        InSequence s;
+        EXPECT_CALL(*handle(), sendBlob(WebString("uuid"), 256));
+        EXPECT_CALL(checkpoint, Call(1));
+        EXPECT_CALL(*handle(), send(true, WebSocketHandle::MessageTypeText, MemEq("foo", 3), 3));
+    }
+    EXPECT_CALL(*channelClient(), didConsumeBufferedAmount(_))
+        .Times(AnyNumber());
+    handleClient()->didReceiveFlowControl(handle(), 128);
+    channel()->send(createBlob("uuid", 256));
+    handleClient()->didReceiveFlowControl(handle(), 128);
+    handleClient()->didReceiveFlowControl(handle(), 128);
+    channel()->send("foo");
+    checkpoint.Call(1);
+    handleClient()->didCompleteSendingBlob(handle());
+}
 
 TEST_F(DocumentWebSocketChannelTest, receiveText)
 {
     connect();
     {
         InSequence s;
         EXPECT_CALL(*channelClient(), didReceiveTextMessage(String("FOO")));
         EXPECT_CALL(*channelClient(), didReceiveTextMessage(String("BAR")));
     }
 
@@ skipping 202 matching lines @@
         EXPECT_CALL(*channelClient(), didError());
         EXPECT_CALL(*channelClient(), didClose(WebSocketChannelClient::ClosingHandshakeIncomplete, WebSocketChannel::CloseEventCodeAbnormalClosure, String()));
     }
 
     channel()->fail("fail message from WebSocket", ErrorMessageLevel, "sourceURL", 1234);
 }
 
 } // namespace
 
 } // namespace blink