| 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 | 6 |
| 7 #include "base/run_loop.h" | 7 #include "base/run_loop.h" |
| 8 #import "base/test/ios/wait_util.h" | 8 #import "base/test/ios/wait_util.h" |
| 9 #import "ios/web/public/navigation_manager.h" | 9 #import "ios/web/public/navigation_manager.h" |
| 10 #include "ios/web/public/web_ui_ios_data_source.h" | 10 #include "ios/web/public/web_ui_ios_data_source.h" |
| (...skipping 19 matching lines...) Expand all Loading... |
| 30 const char kTestWebUIURLHost[] = "testwebui"; | 30 const char kTestWebUIURLHost[] = "testwebui"; |
| 31 | 31 |
| 32 // UI handler class which communicates with test WebUI page as follows: | 32 // UI handler class which communicates with test WebUI page as follows: |
| 33 // - page sends "syn" message to |TestUIHandler| | 33 // - page sends "syn" message to |TestUIHandler| |
| 34 // - |TestUIHandler| replies with "ack" message | 34 // - |TestUIHandler| replies with "ack" message |
| 35 // - page replies back with "fin" | 35 // - page replies back with "fin" |
| 36 // | 36 // |
| 37 // Once "fin" is received |IsFinReceived()| call will return true, indicating | 37 // Once "fin" is received |IsFinReceived()| call will return true, indicating |
| 38 // that communication was successful. See test WebUI page code here: | 38 // that communication was successful. See test WebUI page code here: |
| 39 // ios/web/test/data/mojo_test.js | 39 // ios/web/test/data/mojo_test.js |
| 40 class TestUIHandler : public TestUIHandlerMojo, | 40 class TestUIHandler |
| 41 public shell::InterfaceFactory<TestUIHandlerMojo> { | 41 : public TestUIHandlerMojo, |
| 42 public service_manager::InterfaceFactory<TestUIHandlerMojo> { |
| 42 public: | 43 public: |
| 43 TestUIHandler() {} | 44 TestUIHandler() {} |
| 44 ~TestUIHandler() override {} | 45 ~TestUIHandler() override {} |
| 45 | 46 |
| 46 // Returns true if "fin" has been received. | 47 // Returns true if "fin" has been received. |
| 47 bool IsFinReceived() { return fin_received_; } | 48 bool IsFinReceived() { return fin_received_; } |
| 48 | 49 |
| 49 // TestUIHandlerMojo overrides. | 50 // TestUIHandlerMojo overrides. |
| 50 void SetClientPage(TestPagePtr page) override { page_ = std::move(page); } | 51 void SetClientPage(TestPagePtr page) override { page_ = std::move(page); } |
| 51 void HandleJsMessage(const mojo::String& message) override { | 52 void HandleJsMessage(const mojo::String& message) override { |
| 52 if (message.get() == "syn") { | 53 if (message.get() == "syn") { |
| 53 // Received "syn" message from WebUI page, send "ack" as reply. | 54 // Received "syn" message from WebUI page, send "ack" as reply. |
| 54 DCHECK(!syn_received_); | 55 DCHECK(!syn_received_); |
| 55 DCHECK(!fin_received_); | 56 DCHECK(!fin_received_); |
| 56 syn_received_ = true; | 57 syn_received_ = true; |
| 57 NativeMessageResultMojoPtr result(NativeMessageResultMojo::New()); | 58 NativeMessageResultMojoPtr result(NativeMessageResultMojo::New()); |
| 58 result->message = mojo::String::From("ack"); | 59 result->message = mojo::String::From("ack"); |
| 59 page_->HandleNativeMessage(std::move(result)); | 60 page_->HandleNativeMessage(std::move(result)); |
| 60 } else if (message.get() == "fin") { | 61 } else if (message.get() == "fin") { |
| 61 // Received "fin" from the WebUI page in response to "ack". | 62 // Received "fin" from the WebUI page in response to "ack". |
| 62 DCHECK(syn_received_); | 63 DCHECK(syn_received_); |
| 63 DCHECK(!fin_received_); | 64 DCHECK(!fin_received_); |
| 64 fin_received_ = true; | 65 fin_received_ = true; |
| 65 } else { | 66 } else { |
| 66 NOTREACHED(); | 67 NOTREACHED(); |
| 67 } | 68 } |
| 68 } | 69 } |
| 69 | 70 |
| 70 private: | 71 private: |
| 71 // shell::InterfaceFactory overrides. | 72 // service_manager::InterfaceFactory overrides. |
| 72 void Create(const shell::Identity& remote_identity, | 73 void Create(const service_manager::Identity& remote_identity, |
| 73 mojo::InterfaceRequest<TestUIHandlerMojo> request) override { | 74 mojo::InterfaceRequest<TestUIHandlerMojo> request) override { |
| 74 bindings_.AddBinding(this, std::move(request)); | 75 bindings_.AddBinding(this, std::move(request)); |
| 75 } | 76 } |
| 76 | 77 |
| 77 mojo::BindingSet<TestUIHandlerMojo> bindings_; | 78 mojo::BindingSet<TestUIHandlerMojo> bindings_; |
| 78 TestPagePtr page_ = nullptr; | 79 TestPagePtr page_ = nullptr; |
| 79 // |true| if "syn" has been received. | 80 // |true| if "syn" has been received. |
| 80 bool syn_received_ = false; | 81 bool syn_received_ = false; |
| 81 // |true| if "fin" has been received. | 82 // |true| if "fin" has been received. |
| 82 bool fin_received_ = false; | 83 bool fin_received_ = false; |
| (...skipping 75 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 158 web_state()->GetNavigationManager()->LoadURLWithParams(load_params); | 159 web_state()->GetNavigationManager()->LoadURLWithParams(load_params); |
| 159 | 160 |
| 160 // Wait until |TestUIHandler| receives "ack" message from WebUI page. | 161 // Wait until |TestUIHandler| receives "ack" message from WebUI page. |
| 161 base::test::ios::WaitUntilCondition(^{ | 162 base::test::ios::WaitUntilCondition(^{ |
| 162 base::RunLoop().RunUntilIdle(); | 163 base::RunLoop().RunUntilIdle(); |
| 163 return test_ui_handler()->IsFinReceived(); | 164 return test_ui_handler()->IsFinReceived(); |
| 164 }); | 165 }); |
| 165 } | 166 } |
| 166 | 167 |
| 167 } // namespace web | 168 } // namespace web |
| OLD | NEW |