| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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/message_loop/message_loop.h" |
| 5 #include "mojo/public/bindings/allocation_scope.h" | 6 #include "mojo/public/bindings/allocation_scope.h" |
| 6 #include "mojo/public/bindings/remote_ptr.h" | 7 #include "mojo/public/bindings/remote_ptr.h" |
| 7 #include "mojo/public/environment/environment.h" | 8 #include "mojo/public/environment/environment.h" |
| 8 #include "mojo/public/shell/application.h" | 9 #include "mojo/public/shell/application.h" |
| 9 #include "mojo/public/shell/shell.mojom.h" | 10 #include "mojo/public/shell/shell.mojom.h" |
| 10 #include "mojo/public/utility/run_loop.h" | |
| 11 #include "mojo/service_manager/service_loader.h" | 11 #include "mojo/service_manager/service_loader.h" |
| 12 #include "mojo/service_manager/service_manager.h" | 12 #include "mojo/service_manager/service_manager.h" |
| 13 #include "mojo/service_manager/test.mojom.h" | 13 #include "mojo/service_manager/test.mojom.h" |
| 14 #include "testing/gtest/include/gtest/gtest.h" | 14 #include "testing/gtest/include/gtest/gtest.h" |
| 15 | 15 |
| 16 namespace mojo { | 16 namespace mojo { |
| 17 namespace { | 17 namespace { |
| 18 | 18 |
| 19 const char kTestURLString[] = "test:testService"; | 19 const char kTestURLString[] = "test:testService"; |
| 20 | 20 |
| (...skipping 29 matching lines...) Expand all Loading... |
| 50 public: | 50 public: |
| 51 explicit TestClientImpl(ScopedTestServiceHandle service_handle) | 51 explicit TestClientImpl(ScopedTestServiceHandle service_handle) |
| 52 : service_(service_handle.Pass(), this), | 52 : service_(service_handle.Pass(), this), |
| 53 quit_after_ack_(false) { | 53 quit_after_ack_(false) { |
| 54 } | 54 } |
| 55 | 55 |
| 56 virtual ~TestClientImpl() {} | 56 virtual ~TestClientImpl() {} |
| 57 | 57 |
| 58 virtual void AckTest() OVERRIDE { | 58 virtual void AckTest() OVERRIDE { |
| 59 if (quit_after_ack_) | 59 if (quit_after_ack_) |
| 60 mojo::RunLoop::current()->Quit(); | 60 base::MessageLoop::current()->Quit(); |
| 61 } | 61 } |
| 62 | 62 |
| 63 void Test(std::string test_string) { | 63 void Test(std::string test_string) { |
| 64 AllocationScope scope; | 64 AllocationScope scope; |
| 65 quit_after_ack_ = true; | 65 quit_after_ack_ = true; |
| 66 service_->Test(mojo::String(test_string)); | 66 service_->Test(mojo::String(test_string)); |
| 67 } | 67 } |
| 68 | 68 |
| 69 private: | 69 private: |
| 70 RemotePtr<TestService> service_; | 70 RemotePtr<TestService> service_; |
| (...skipping 25 matching lines...) Expand all Loading... |
| 96 } | 96 } |
| 97 | 97 |
| 98 virtual void LoadService(ServiceManager* manager, | 98 virtual void LoadService(ServiceManager* manager, |
| 99 const GURL& url, | 99 const GURL& url, |
| 100 ScopedShellHandle shell_handle) OVERRIDE { | 100 ScopedShellHandle shell_handle) OVERRIDE { |
| 101 test_app_.reset(new Application(shell_handle.Pass())); | 101 test_app_.reset(new Application(shell_handle.Pass())); |
| 102 test_app_->AddServiceFactory( | 102 test_app_->AddServiceFactory( |
| 103 new ServiceFactory<TestServiceImpl, TestContext>(&context_)); | 103 new ServiceFactory<TestServiceImpl, TestContext>(&context_)); |
| 104 } | 104 } |
| 105 | 105 |
| 106 virtual void OnServiceError(ServiceManager* manager, |
| 107 const GURL& url) OVERRIDE { |
| 108 base::MessageLoop::current()->PostTask(FROM_HERE, |
| 109 base::MessageLoop::QuitClosure()); |
| 110 } |
| 111 |
| 106 bool HasFactoryForTestURL() { | 112 bool HasFactoryForTestURL() { |
| 107 ServiceManager::TestAPI manager_test_api(service_manager_.get()); | 113 ServiceManager::TestAPI manager_test_api(service_manager_.get()); |
| 108 return manager_test_api.HasFactoryForURL(GURL(kTestURLString)); | 114 return manager_test_api.HasFactoryForURL(GURL(kTestURLString)); |
| 109 } | 115 } |
| 110 | 116 |
| 111 protected: | 117 protected: |
| 112 mojo::Environment env_; | 118 mojo::Environment env_; |
| 113 mojo::RunLoop loop_; | 119 base::MessageLoop loop_; |
| 114 TestContext context_; | 120 TestContext context_; |
| 115 scoped_ptr<Application> test_app_; | 121 scoped_ptr<Application> test_app_; |
| 116 scoped_ptr<TestClientImpl> test_client_; | 122 scoped_ptr<TestClientImpl> test_client_; |
| 117 scoped_ptr<ServiceManager> service_manager_; | 123 scoped_ptr<ServiceManager> service_manager_; |
| 118 DISALLOW_COPY_AND_ASSIGN(ServiceManagerTest); | 124 DISALLOW_COPY_AND_ASSIGN(ServiceManagerTest); |
| 119 }; | 125 }; |
| 120 | 126 |
| 121 TEST_F(ServiceManagerTest, Basic) { | 127 TEST_F(ServiceManagerTest, Basic) { |
| 122 test_client_->Test("test"); | 128 test_client_->Test("test"); |
| 123 loop_.Run(); | 129 loop_.Run(); |
| 124 EXPECT_EQ(std::string("test"), context_.last_test_string); | 130 EXPECT_EQ(std::string("test"), context_.last_test_string); |
| 125 } | 131 } |
| 126 | 132 |
| 127 TEST_F(ServiceManagerTest, ClientError) { | 133 TEST_F(ServiceManagerTest, ClientError) { |
| 128 test_client_->Test("test"); | 134 test_client_->Test("test"); |
| 129 EXPECT_TRUE(HasFactoryForTestURL()); | 135 EXPECT_TRUE(HasFactoryForTestURL()); |
| 130 loop_.Run(); | 136 loop_.Run(); |
| 131 EXPECT_EQ(1, context_.num_impls); | 137 EXPECT_EQ(1, context_.num_impls); |
| 132 test_client_.reset(NULL); | 138 test_client_.reset(NULL); |
| 133 loop_.Run(); | 139 loop_.Run(); |
| 134 EXPECT_EQ(0, context_.num_impls); | 140 EXPECT_EQ(0, context_.num_impls); |
| 135 EXPECT_FALSE(HasFactoryForTestURL()); | 141 EXPECT_FALSE(HasFactoryForTestURL()); |
| 136 } | 142 } |
| 137 } // namespace mojo | 143 } // namespace mojo |
| OLD | NEW |