OLD | NEW |
(Empty) | |
| 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 |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #include "build/build_config.h" |
| 6 |
| 7 #include <windows.h> |
| 8 |
| 9 #include "base/memory/ref_counted.h" |
| 10 #include "base/memory/scoped_ptr.h" |
| 11 #include "base/process/process.h" |
| 12 #include "ipc/attachment_broker_messages.h" |
| 13 #include "ipc/attachment_broker_win.h" |
| 14 #include "ipc/handle_attachment_win.h" |
| 15 #include "testing/gtest/include/gtest/gtest.h" |
| 16 |
| 17 namespace IPC { |
| 18 |
| 19 TEST(AttachmentBrokerWinTest, ReceiveValidMessage) { |
| 20 HANDLE handle = LongToHandle(8); |
| 21 base::ProcessId destination = base::Process::Current().Pid(); |
| 22 scoped_refptr<internal::HandleAttachmentWin> attachment( |
| 23 new internal::HandleAttachmentWin(handle)); |
| 24 AttachmentBrokerMsg_WinHandleHasBeenDuplicated msg( |
| 25 attachment->GetWireFormat(destination)); |
| 26 AttachmentBrokerWin attachment_broker; |
| 27 EXPECT_TRUE(attachment_broker.OnMessageReceived(msg)); |
| 28 EXPECT_EQ(1u, attachment_broker.attachments_.size()); |
| 29 |
| 30 scoped_refptr<BrokerableAttachment> received_attachment = |
| 31 attachment_broker.attachments_.front(); |
| 32 EXPECT_EQ(BrokerableAttachment::WIN_HANDLE, |
| 33 received_attachment->GetBrokerableType()); |
| 34 internal::HandleAttachmentWin* received_handle_attachment = |
| 35 static_cast<internal::HandleAttachmentWin*>(received_attachment.get()); |
| 36 EXPECT_EQ(handle, received_handle_attachment->get_handle()); |
| 37 } |
| 38 |
| 39 TEST(AttachmentBrokerWinTest, ReceiveInvalidMessage) { |
| 40 HANDLE handle = LongToHandle(8); |
| 41 base::ProcessId destination = base::Process::Current().Pid() + 1; |
| 42 scoped_refptr<internal::HandleAttachmentWin> attachment( |
| 43 new internal::HandleAttachmentWin(handle)); |
| 44 AttachmentBrokerMsg_WinHandleHasBeenDuplicated msg( |
| 45 attachment->GetWireFormat(destination)); |
| 46 AttachmentBrokerWin attachment_broker; |
| 47 EXPECT_TRUE(attachment_broker.OnMessageReceived(msg)); |
| 48 EXPECT_EQ(0u, attachment_broker.attachments_.size()); |
| 49 } |
| 50 |
| 51 } // namespace IPC |
OLD | NEW |