| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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/websockets/DocumentWebSocketChannel.h" | 5 #include "modules/websockets/DocumentWebSocketChannel.h" |
| 6 | 6 |
| 7 #include <stdint.h> | 7 #include <stdint.h> |
| 8 #include <memory> | 8 #include <memory> |
| 9 #include "core/dom/DOMArrayBuffer.h" | 9 #include "core/dom/DOMArrayBuffer.h" |
| 10 #include "core/dom/Document.h" | 10 #include "core/dom/Document.h" |
| 11 #include "core/fileapi/Blob.h" | 11 #include "core/fileapi/Blob.h" |
| 12 #include "core/testing/DummyPageHolder.h" | 12 #include "core/testing/DummyPageHolder.h" |
| 13 #include "modules/websockets/WebSocketChannel.h" | 13 #include "modules/websockets/WebSocketChannel.h" |
| 14 #include "modules/websockets/WebSocketChannelClient.h" | 14 #include "modules/websockets/WebSocketChannelClient.h" |
| 15 #include "modules/websockets/WebSocketHandle.h" | 15 #include "modules/websockets/WebSocketHandle.h" |
| 16 #include "modules/websockets/WebSocketHandleClient.h" | 16 #include "modules/websockets/WebSocketHandleClient.h" |
| 17 #include "platform/heap/Handle.h" | 17 #include "platform/heap/Handle.h" |
| 18 #include "platform/weborigin/KURL.h" | 18 #include "platform/weborigin/KURL.h" |
| 19 #include "platform/wtf/PtrUtil.h" | 19 #include "platform/wtf/PtrUtil.h" |
| 20 #include "platform/wtf/Vector.h" | 20 #include "platform/wtf/Vector.h" |
| 21 #include "platform/wtf/text/WTFString.h" | 21 #include "platform/wtf/text/WTFString.h" |
| 22 #include "public/platform/WebCallbacks.h" | 22 #include "public/platform/WebCallbacks.h" |
| 23 #include "public/platform/WebSocketHandshakeThrottle.h" | 23 #include "public/platform/WebSocketHandshakeThrottle.h" |
| 24 #include "public/platform/WebURL.h" | 24 #include "public/platform/WebURL.h" |
| 25 #include "public/web/WebLocalFrame.h" | 25 #include "public/web/WebLocalFrame.h" |
| 26 #include "testing/gmock/include/gmock/gmock.h" | 26 #include "testing/gmock/include/gmock/gmock.h" |
| 27 #include "testing/gtest/include/gtest/gtest.h" | 27 #include "testing/gtest/include/gtest/gtest.h" |
| 28 | 28 |
| 29 using testing::_; | 29 using ::testing::_; |
| 30 using testing::InSequence; | 30 using ::testing::InSequence; |
| 31 using testing::PrintToString; | 31 using ::testing::PrintToString; |
| 32 using testing::AnyNumber; | 32 using ::testing::AnyNumber; |
| 33 using testing::SaveArg; | 33 using ::testing::SaveArg; |
| 34 | 34 |
| 35 namespace blink { | 35 namespace blink { |
| 36 | 36 |
| 37 namespace { | 37 namespace { |
| 38 | 38 |
| 39 typedef testing::StrictMock<testing::MockFunction<void(int)>> Checkpoint; | 39 typedef ::testing::StrictMock<::testing::MockFunction<void(int)>> Checkpoint; |
| 40 | 40 |
| 41 class MockWebSocketChannelClient | 41 class MockWebSocketChannelClient |
| 42 : public GarbageCollectedFinalized<MockWebSocketChannelClient>, | 42 : public GarbageCollectedFinalized<MockWebSocketChannelClient>, |
| 43 public WebSocketChannelClient { | 43 public WebSocketChannelClient { |
| 44 USING_GARBAGE_COLLECTED_MIXIN(MockWebSocketChannelClient); | 44 USING_GARBAGE_COLLECTED_MIXIN(MockWebSocketChannelClient); |
| 45 | 45 |
| 46 public: | 46 public: |
| 47 static MockWebSocketChannelClient* Create() { | 47 static MockWebSocketChannelClient* Create() { |
| 48 return new testing::StrictMock<MockWebSocketChannelClient>(); | 48 return new ::testing::StrictMock<MockWebSocketChannelClient>(); |
| 49 } | 49 } |
| 50 | 50 |
| 51 MockWebSocketChannelClient() {} | 51 MockWebSocketChannelClient() {} |
| 52 | 52 |
| 53 ~MockWebSocketChannelClient() override {} | 53 ~MockWebSocketChannelClient() override {} |
| 54 | 54 |
| 55 MOCK_METHOD2(DidConnect, void(const String&, const String&)); | 55 MOCK_METHOD2(DidConnect, void(const String&, const String&)); |
| 56 MOCK_METHOD1(DidReceiveTextMessage, void(const String&)); | 56 MOCK_METHOD1(DidReceiveTextMessage, void(const String&)); |
| 57 void DidReceiveBinaryMessage(std::unique_ptr<Vector<char>> payload) override { | 57 void DidReceiveBinaryMessage(std::unique_ptr<Vector<char>> payload) override { |
| 58 DidReceiveBinaryMessageMock(*payload); | 58 DidReceiveBinaryMessageMock(*payload); |
| 59 } | 59 } |
| 60 MOCK_METHOD1(DidReceiveBinaryMessageMock, void(const Vector<char>&)); | 60 MOCK_METHOD1(DidReceiveBinaryMessageMock, void(const Vector<char>&)); |
| 61 MOCK_METHOD0(DidError, void()); | 61 MOCK_METHOD0(DidError, void()); |
| 62 MOCK_METHOD1(DidConsumeBufferedAmount, void(uint64_t)); | 62 MOCK_METHOD1(DidConsumeBufferedAmount, void(uint64_t)); |
| 63 MOCK_METHOD0(DidStartClosingHandshake, void()); | 63 MOCK_METHOD0(DidStartClosingHandshake, void()); |
| 64 MOCK_METHOD3(DidClose, | 64 MOCK_METHOD3(DidClose, |
| 65 void(ClosingHandshakeCompletionStatus, | 65 void(ClosingHandshakeCompletionStatus, |
| 66 unsigned short, | 66 unsigned short, |
| 67 const String&)); | 67 const String&)); |
| 68 | 68 |
| 69 DEFINE_INLINE_VIRTUAL_TRACE() { WebSocketChannelClient::Trace(visitor); } | 69 DEFINE_INLINE_VIRTUAL_TRACE() { WebSocketChannelClient::Trace(visitor); } |
| 70 }; | 70 }; |
| 71 | 71 |
| 72 class MockWebSocketHandle : public WebSocketHandle { | 72 class MockWebSocketHandle : public WebSocketHandle { |
| 73 public: | 73 public: |
| 74 static MockWebSocketHandle* Create() { | 74 static MockWebSocketHandle* Create() { |
| 75 return new testing::StrictMock<MockWebSocketHandle>(); | 75 return new ::testing::StrictMock<MockWebSocketHandle>(); |
| 76 } | 76 } |
| 77 | 77 |
| 78 MockWebSocketHandle() {} | 78 MockWebSocketHandle() {} |
| 79 | 79 |
| 80 ~MockWebSocketHandle() override {} | 80 ~MockWebSocketHandle() override {} |
| 81 | 81 |
| 82 MOCK_METHOD1(DoInitialize, void(mojom::blink::WebSocketPtr*)); | 82 MOCK_METHOD1(DoInitialize, void(mojom::blink::WebSocketPtr*)); |
| 83 void Initialize(mojom::blink::WebSocketPtr websocket) override { | 83 void Initialize(mojom::blink::WebSocketPtr websocket) override { |
| 84 DoInitialize(&websocket); | 84 DoInitialize(&websocket); |
| 85 } | 85 } |
| 86 | 86 |
| 87 MOCK_METHOD6(Connect, | 87 MOCK_METHOD6(Connect, |
| 88 void(const KURL&, | 88 void(const KURL&, |
| 89 const Vector<String>&, | 89 const Vector<String>&, |
| 90 SecurityOrigin*, | 90 SecurityOrigin*, |
| 91 const KURL&, | 91 const KURL&, |
| 92 const String&, | 92 const String&, |
| 93 WebSocketHandleClient*)); | 93 WebSocketHandleClient*)); |
| 94 MOCK_METHOD4(Send, | 94 MOCK_METHOD4(Send, |
| 95 void(bool, WebSocketHandle::MessageType, const char*, size_t)); | 95 void(bool, WebSocketHandle::MessageType, const char*, size_t)); |
| 96 MOCK_METHOD1(FlowControl, void(int64_t)); | 96 MOCK_METHOD1(FlowControl, void(int64_t)); |
| 97 MOCK_METHOD2(Close, void(unsigned short, const String&)); | 97 MOCK_METHOD2(Close, void(unsigned short, const String&)); |
| 98 }; | 98 }; |
| 99 | 99 |
| 100 class MockWebSocketHandshakeThrottle : public WebSocketHandshakeThrottle { | 100 class MockWebSocketHandshakeThrottle : public WebSocketHandshakeThrottle { |
| 101 public: | 101 public: |
| 102 static MockWebSocketHandshakeThrottle* Create() { | 102 static MockWebSocketHandshakeThrottle* Create() { |
| 103 return new testing::StrictMock<MockWebSocketHandshakeThrottle>(); | 103 return new ::testing::StrictMock<MockWebSocketHandshakeThrottle>(); |
| 104 } | 104 } |
| 105 MockWebSocketHandshakeThrottle() {} | 105 MockWebSocketHandshakeThrottle() {} |
| 106 ~MockWebSocketHandshakeThrottle() override { Destructor(); } | 106 ~MockWebSocketHandshakeThrottle() override { Destructor(); } |
| 107 | 107 |
| 108 MOCK_METHOD3(ThrottleHandshake, | 108 MOCK_METHOD3(ThrottleHandshake, |
| 109 void(const WebURL&, | 109 void(const WebURL&, |
| 110 WebLocalFrame*, | 110 WebLocalFrame*, |
| 111 WebCallbacks<void, const WebString&>*)); | 111 WebCallbacks<void, const WebString&>*)); |
| 112 | 112 |
| 113 // This method is used to allow us to require that the destructor is called at | 113 // This method is used to allow us to require that the destructor is called at |
| (...skipping 908 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1022 } | 1022 } |
| 1023 Channel()->Connect(url(), ""); | 1023 Channel()->Connect(url(), ""); |
| 1024 HandleClient()->DidClose(Handle(), false, | 1024 HandleClient()->DidClose(Handle(), false, |
| 1025 WebSocketChannel::kCloseEventCodeProtocolError, | 1025 WebSocketChannel::kCloseEventCodeProtocolError, |
| 1026 "connect error"); | 1026 "connect error"); |
| 1027 } | 1027 } |
| 1028 | 1028 |
| 1029 } // namespace | 1029 } // namespace |
| 1030 | 1030 |
| 1031 } // namespace blink | 1031 } // namespace blink |
| OLD | NEW |