OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 "base/utf_string_conversions.h" | 5 #include "base/utf_string_conversions.h" |
6 #include "base/values.h" | 6 #include "base/values.h" |
7 #include "chrome/browser/extensions/extension_message_service.h" | 7 #include "chrome/browser/extensions/extension_message_service.h" |
8 #include "chrome/common/extensions/extension_messages.h" | 8 #include "chrome/common/extensions/extension_messages.h" |
9 #include "chrome/common/render_messages.h" | 9 #include "chrome/common/render_messages.h" |
10 #include "chrome/renderer/extensions/chrome_v8_context.h" | 10 #include "chrome/renderer/extensions/chrome_v8_context.h" |
11 #include "chrome/renderer/extensions/extension_dispatcher.h" | 11 #include "chrome/renderer/extensions/extension_dispatcher.h" |
12 #include "chrome/renderer/extensions/miscellaneous_bindings.h" | 12 #include "chrome/renderer/extensions/miscellaneous_bindings.h" |
13 #include "chrome/test/base/chrome_render_view_test.h" | 13 #include "chrome/test/base/chrome_render_view_test.h" |
14 #include "content/common/view_messages.h" | |
15 #include "testing/gtest/include/gtest/gtest.h" | 14 #include "testing/gtest/include/gtest/gtest.h" |
16 | 15 |
17 using extensions::MiscellaneousBindings; | 16 using extensions::MiscellaneousBindings; |
18 | 17 |
19 namespace { | 18 namespace { |
20 | 19 |
21 static const char kTestingExtensionId[] = "oooooooooooooooooooooooooooooooo"; | 20 static const char kTestingExtensionId[] = "oooooooooooooooooooooooooooooooo"; |
22 | 21 |
23 void DispatchOnConnect(const ChromeV8ContextSet& v8_context_set, | 22 void DispatchOnConnect(const ChromeV8ContextSet& v8_context_set, |
24 int source_port_id, const std::string& name, | 23 int source_port_id, const std::string& name, |
(...skipping 55 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
80 EXPECT_EQ("{\"message\":\"content ready\"}", post_params.b); | 79 EXPECT_EQ("{\"message\":\"content ready\"}", post_params.b); |
81 | 80 |
82 // Now simulate getting a message back from the other side. | 81 // Now simulate getting a message back from the other side. |
83 render_thread_->sink().ClearMessages(); | 82 render_thread_->sink().ClearMessages(); |
84 const int kPortId = 0; | 83 const int kPortId = 0; |
85 MiscellaneousBindings::DeliverMessage( | 84 MiscellaneousBindings::DeliverMessage( |
86 extension_dispatcher_->v8_context_set().GetAll(), | 85 extension_dispatcher_->v8_context_set().GetAll(), |
87 kPortId, "{\"val\": 42}", NULL); | 86 kPortId, "{\"val\": 42}", NULL); |
88 | 87 |
89 // Verify that we got it. | 88 // Verify that we got it. |
90 const IPC::Message* alert_msg = | 89 render_thread_->VerifyRunJavaScriptMessageSend( |
91 render_thread_->sink().GetUniqueMessageMatching( | 90 ASCIIToUTF16("content got: 42")); |
92 ViewHostMsg_RunJavaScriptMessage::ID); | |
93 ASSERT_TRUE(alert_msg); | |
94 iter = IPC::SyncMessage::GetDataIterator(alert_msg); | |
95 ViewHostMsg_RunJavaScriptMessage::SendParam alert_param; | |
96 ASSERT_TRUE(IPC::ReadParam(alert_msg, &iter, &alert_param)); | |
97 EXPECT_EQ(ASCIIToUTF16("content got: 42"), alert_param.a); | |
98 } | 91 } |
99 | 92 |
100 // Tests that the bindings for handling a new channel connection and channel | 93 // Tests that the bindings for handling a new channel connection and channel |
101 // closing all works. | 94 // closing all works. |
102 TEST_F(ChromeRenderViewTest, ExtensionMessagesOnConnect) { | 95 TEST_F(ChromeRenderViewTest, ExtensionMessagesOnConnect) { |
103 extension_dispatcher_->SetTestExtensionId(kTestingExtensionId); | 96 extension_dispatcher_->SetTestExtensionId(kTestingExtensionId); |
104 LoadHTML("<body></body>"); | 97 LoadHTML("<body></body>"); |
105 ExecuteJavaScript( | 98 ExecuteJavaScript( |
106 "chrome.extension.onConnect.addListener(function (port) {" | 99 "chrome.extension.onConnect.addListener(function (port) {" |
107 " port.test = 24;" | 100 " port.test = 24;" |
(...skipping 28 matching lines...) Expand all Loading... |
136 "{\"message\":\"onconnect from foo://bar name " + kPortName + "\"}"; | 129 "{\"message\":\"onconnect from foo://bar name " + kPortName + "\"}"; |
137 EXPECT_EQ(expected_msg, post_params.b); | 130 EXPECT_EQ(expected_msg, post_params.b); |
138 | 131 |
139 // Now simulate getting a message back from the channel opener. | 132 // Now simulate getting a message back from the channel opener. |
140 render_thread_->sink().ClearMessages(); | 133 render_thread_->sink().ClearMessages(); |
141 MiscellaneousBindings::DeliverMessage( | 134 MiscellaneousBindings::DeliverMessage( |
142 extension_dispatcher_->v8_context_set().GetAll(), | 135 extension_dispatcher_->v8_context_set().GetAll(), |
143 kPortId, "{\"val\": 42}", NULL); | 136 kPortId, "{\"val\": 42}", NULL); |
144 | 137 |
145 // Verify that we got it. | 138 // Verify that we got it. |
146 const IPC::Message* alert_msg = | 139 render_thread_->VerifyRunJavaScriptMessageSend(ASCIIToUTF16("got: 42")); |
147 render_thread_->sink().GetUniqueMessageMatching( | |
148 ViewHostMsg_RunJavaScriptMessage::ID); | |
149 ASSERT_TRUE(alert_msg); | |
150 void* iter = IPC::SyncMessage::GetDataIterator(alert_msg); | |
151 ViewHostMsg_RunJavaScriptMessage::SendParam alert_param; | |
152 ASSERT_TRUE(IPC::ReadParam(alert_msg, &iter, &alert_param)); | |
153 EXPECT_EQ(ASCIIToUTF16("got: 42"), alert_param.a); | |
154 | 140 |
155 // Now simulate the channel closing. | 141 // Now simulate the channel closing. |
156 render_thread_->sink().ClearMessages(); | 142 render_thread_->sink().ClearMessages(); |
157 DispatchOnDisconnect(extension_dispatcher_->v8_context_set(), kPortId); | 143 DispatchOnDisconnect(extension_dispatcher_->v8_context_set(), kPortId); |
158 | 144 |
159 // Verify that we got it. | 145 // Verify that we got it. |
160 alert_msg = | 146 render_thread_->VerifyRunJavaScriptMessageSend( |
161 render_thread_->sink().GetUniqueMessageMatching( | 147 ASCIIToUTF16("disconnected: 24")); |
162 ViewHostMsg_RunJavaScriptMessage::ID); | |
163 ASSERT_TRUE(alert_msg); | |
164 iter = IPC::SyncMessage::GetDataIterator(alert_msg); | |
165 ASSERT_TRUE(IPC::ReadParam(alert_msg, &iter, &alert_param)); | |
166 EXPECT_EQ(ASCIIToUTF16("disconnected: 24"), alert_param.a); | |
167 } | 148 } |
OLD | NEW |