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 |