Index: ios/web/webui/web_ui_mojo_inttest.mm |
diff --git a/ios/web/webui/web_ui_mojo_inttest.mm b/ios/web/webui/web_ui_mojo_inttest.mm |
deleted file mode 100644 |
index b7fa5450f0723ac3af507bc44ef4807c9182bf58..0000000000000000000000000000000000000000 |
--- a/ios/web/webui/web_ui_mojo_inttest.mm |
+++ /dev/null |
@@ -1,164 +0,0 @@ |
-// Copyright 2016 The Chromium Authors. All rights reserved. |
-// Use of this source code is governed by a BSD-style license that can be |
-// found in the LICENSE file. |
- |
-#include <memory> |
- |
-#include "base/run_loop.h" |
-#import "base/test/ios/wait_util.h" |
-#include "ios/public/provider/web/web_ui_ios_controller.h" |
-#include "ios/public/provider/web/web_ui_ios_controller_factory.h" |
-#import "ios/web/public/navigation_manager.h" |
-#include "ios/web/public/web_ui_ios_data_source.h" |
-#include "ios/web/test/grit/test_resources.h" |
-#include "ios/web/test/mojo_test.mojom.h" |
-#include "ios/web/test/test_url_constants.h" |
-#import "ios/web/test/web_int_test.h" |
-#import "ios/web/web_state/web_state_impl.h" |
-#include "mojo/public/cpp/bindings/binding_set.h" |
-#include "services/shell/public/cpp/interface_registry.h" |
-#include "url/gurl.h" |
-#include "url/scheme_host_port.h" |
- |
-namespace web { |
- |
-namespace { |
- |
-// Hostname for test WebUI page. |
-const char kTestWebUIURLHost[] = "testwebui"; |
- |
-// UI handler class which communicates with test WebUI page as follows: |
-// - page sends "syn" message to |TestUIHandler| |
-// - |TestUIHandler| replies with "ack" message |
-// - page replies back with "fin" |
-// |
-// Once "fin" is received |IsFinReceived()| call will return true, indicating |
-// that communication was successful. See test WebUI page code here: |
-// ios/web/test/data/mojo_test.js |
-class TestUIHandler : public TestUIHandlerMojo, |
- public shell::InterfaceFactory<TestUIHandlerMojo> { |
- public: |
- TestUIHandler() {} |
- ~TestUIHandler() override {} |
- |
- // Returns true if "fin" has been received. |
- bool IsFinReceived() { return fin_received_; } |
- |
- // TestUIHandlerMojo overrides. |
- void HandleJsMessage(const mojo::String& message, TestPagePtr page) override { |
- if (message.get() == "syn") { |
- // Received "syn" message from WebUI page, send "ack" as reply. |
- DCHECK(!syn_received_); |
- DCHECK(!fin_received_); |
- syn_received_ = true; |
- NativeMessageResultMojoPtr result(NativeMessageResultMojo::New()); |
- result->message = mojo::String::From("ack"); |
- page->HandleNativeMessage(std::move(result)); |
- } else if (message.get() == "fin") { |
- // Received "fin" from the WebUI page in response to "ack". |
- DCHECK(syn_received_); |
- DCHECK(!fin_received_); |
- fin_received_ = true; |
- } else { |
- NOTREACHED(); |
- } |
- } |
- |
- private: |
- // shell::InterfaceFactory overrides. |
- void Create(shell::Connection* connection, |
- mojo::InterfaceRequest<TestUIHandlerMojo> request) override { |
- bindings_.AddBinding(this, std::move(request)); |
- } |
- |
- mojo::BindingSet<TestUIHandlerMojo> bindings_; |
- // |true| if "syn" has been received. |
- bool syn_received_ = false; |
- // |true| if "fin" has been received. |
- bool fin_received_ = false; |
-}; |
- |
-// Controller for test WebUI. |
-class TestUI : public WebUIIOSController { |
- public: |
- // Constructs controller from |web_ui| and |ui_handler| which will communicate |
- // with test WebUI page. |
- TestUI(WebUIIOS* web_ui, TestUIHandler* ui_handler) |
- : WebUIIOSController(web_ui) { |
- web::WebUIIOSDataSource* source = |
- web::WebUIIOSDataSource::Create(kTestWebUIURLHost); |
- |
- source->AddResourcePath("mojo_test.js", IDR_MOJO_TEST_JS); |
- source->AddResourcePath("ios/web/test/mojo_test.mojom", |
- IDR_MOJO_TEST_MOJO_JS); |
- source->SetDefaultResource(IDR_MOJO_TEST_HTML); |
- |
- web::WebState* web_state = web_ui->GetWebState(); |
- web::WebUIIOSDataSource::Add(web_state->GetBrowserState(), source); |
- |
- web_state->GetMojoInterfaceRegistry()->AddInterface(ui_handler); |
- } |
-}; |
- |
-// Factory that creates TestUI controller. |
-class TestWebUIControllerFactory : public WebUIIOSControllerFactory { |
- public: |
- // Constructs a controller factory which will eventually create |ui_handler|. |
- explicit TestWebUIControllerFactory(TestUIHandler* ui_handler) |
- : ui_handler_(ui_handler) {} |
- |
- // WebUIIOSControllerFactory overrides. |
- WebUIIOSController* CreateWebUIIOSControllerForURL( |
- WebUIIOS* web_ui, |
- const GURL& url) const override { |
- DCHECK_EQ(url.scheme(), kTestWebUIScheme); |
- DCHECK_EQ(url.host(), kTestWebUIURLHost); |
- return new TestUI(web_ui, ui_handler_); |
- } |
- |
- private: |
- // UI handler class which communicates with test WebUI page. |
- TestUIHandler* ui_handler_; |
-}; |
-} // namespace |
- |
-// A test fixture for verifying mojo comminication for WebUI. |
-class WebUIMojoTest : public WebIntTest { |
- protected: |
- WebUIMojoTest() |
- : web_state_(new WebStateImpl(GetBrowserState())), |
- ui_handler_(new TestUIHandler()) { |
- web_state_->GetNavigationManagerImpl().InitializeSession(nil, nil, NO, 0); |
- WebUIIOSControllerFactory::RegisterFactory( |
- new TestWebUIControllerFactory(ui_handler_.get())); |
- } |
- |
- // Returns WebState which loads test WebUI page. |
- WebStateImpl* web_state() { return web_state_.get(); } |
- // Returns UI handler which communicates with WebUI page. |
- TestUIHandler* test_ui_handler() { return ui_handler_.get(); } |
- |
- private: |
- std::unique_ptr<WebStateImpl> web_state_; |
- std::unique_ptr<TestUIHandler> ui_handler_; |
-}; |
- |
-// Tests that JS can send messages to the native code and vice versa. |
-// TestUIHandler is used for communication and test suceeds only when |
-// |TestUIHandler| sucessfully receives "ack" message from WebUI page. |
-TEST_F(WebUIMojoTest, MessageExchange) { |
- web_state()->SetWebUsageEnabled(true); |
- web_state()->GetWebController().useMojoForWebUI = YES; |
- web_state()->GetView(); // WebState won't load URL without view. |
- NavigationManager::WebLoadParams load_params(GURL( |
- url::SchemeHostPort(kTestWebUIScheme, kTestWebUIURLHost, 0).Serialize())); |
- web_state()->GetNavigationManager()->LoadURLWithParams(load_params); |
- |
- // Wait until |TestUIHandler| receives "ack" message from WebUI page. |
- base::test::ios::WaitUntilCondition(^{ |
- base::RunLoop().RunUntilIdle(); |
- return test_ui_handler()->IsFinReceived(); |
- }); |
-} |
- |
-} // namespace web |