| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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 <memory> | 5 #include <memory> |
| 6 #include <string> | 6 #include <string> |
| 7 #include <vector> | 7 #include <vector> |
| 8 | 8 |
| 9 #include "base/base64.h" | 9 #include "base/base64.h" |
| 10 #include "content/public/test/browser_test.h" | 10 #include "content/public/test/browser_test.h" |
| (...skipping 172 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 183 devtools_client_->GetSecurity()->GetExperimental()->RemoveObserver(this); | 183 devtools_client_->GetSecurity()->GetExperimental()->RemoveObserver(this); |
| 184 FinishAsynchronousTest(); | 184 FinishAsynchronousTest(); |
| 185 } | 185 } |
| 186 }; | 186 }; |
| 187 | 187 |
| 188 HEADLESS_ASYNC_DEVTOOLED_TEST_F(HeadlessWebContentsSecurityTest); | 188 HEADLESS_ASYNC_DEVTOOLED_TEST_F(HeadlessWebContentsSecurityTest); |
| 189 | 189 |
| 190 class HeadlessTabSocketTest : public HeadlessAsyncDevTooledBrowserTest, | 190 class HeadlessTabSocketTest : public HeadlessAsyncDevTooledBrowserTest, |
| 191 public HeadlessTabSocket::Listener { | 191 public HeadlessTabSocket::Listener { |
| 192 public: | 192 public: |
| 193 void SetUp() override { |
| 194 options()->mojo_service_names.insert("headless::TabSocket"); |
| 195 HeadlessAsyncDevTooledBrowserTest::SetUp(); |
| 196 } |
| 197 |
| 193 void RunDevTooledTest() override { | 198 void RunDevTooledTest() override { |
| 194 devtools_client_->GetRuntime()->Evaluate( | 199 devtools_client_->GetRuntime()->Evaluate( |
| 195 R"(window.TabSocket.onmessage = | 200 R"(window.TabSocket.onmessage = |
| 196 function(event) { | 201 function(event) { |
| 197 window.TabSocket.send( | 202 window.TabSocket.send( |
| 198 'Embedder sent us: ' + event.detail.message); | 203 'Embedder sent us: ' + event.detail.message); |
| 199 }; | 204 }; |
| 200 )"); | 205 )"); |
| 201 | 206 |
| 202 HeadlessTabSocket* headless_tab_socket = | 207 HeadlessTabSocket* headless_tab_socket = |
| 203 web_contents_->GetHeadlessTabSocket(); | 208 web_contents_->GetHeadlessTabSocket(); |
| 204 DCHECK(headless_tab_socket); | 209 DCHECK(headless_tab_socket); |
| 205 | 210 |
| 206 headless_tab_socket->SendMessageToTab("Hello"); | 211 headless_tab_socket->SendMessageToTab("Hello"); |
| 207 headless_tab_socket->SetListener(this); | 212 headless_tab_socket->SetListener(this); |
| 208 } | 213 } |
| 209 | 214 |
| 210 void OnMessageFromTab(const std::string& message) override { | 215 void OnMessageFromTab(const std::string& message) override { |
| 211 EXPECT_EQ("Embedder sent us: Hello", message); | 216 EXPECT_EQ("Embedder sent us: Hello", message); |
| 212 FinishAsynchronousTest(); | 217 FinishAsynchronousTest(); |
| 213 } | 218 } |
| 214 | 219 |
| 215 bool GetCreateTabSocket() override { return true; } | 220 bool GetCreateTabSocket() override { return true; } |
| 216 }; | 221 }; |
| 217 | 222 |
| 218 HEADLESS_ASYNC_DEVTOOLED_TEST_F(HeadlessTabSocketTest); | 223 HEADLESS_ASYNC_DEVTOOLED_TEST_F(HeadlessTabSocketTest); |
| 219 | 224 |
| 220 } // namespace headless | 225 } // namespace headless |
| OLD | NEW |