| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 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/at_exit.h" | 5 #include "base/at_exit.h" |
| 6 #include "base/bind.h" | 6 #include "base/bind.h" |
| 7 #include "base/macros.h" | 7 #include "base/macros.h" |
| 8 #include "base/memory/scoped_vector.h" | 8 #include "base/memory/scoped_vector.h" |
| 9 #include "base/message_loop/message_loop.h" | 9 #include "base/message_loop/message_loop.h" |
| 10 #include "base/stl_util.h" | 10 #include "base/stl_util.h" |
| 11 #include "base/strings/stringprintf.h" | 11 #include "base/strings/stringprintf.h" |
| 12 #include "mojo/application/public/cpp/application_connection.h" | 12 #include "mojo/application/public/cpp/application_connection.h" |
| 13 #include "mojo/application/public/cpp/application_delegate.h" | 13 #include "mojo/application/public/cpp/application_delegate.h" |
| 14 #include "mojo/application/public/cpp/application_impl.h" | 14 #include "mojo/application/public/cpp/application_impl.h" |
| 15 #include "mojo/application/public/cpp/connect.h" | 15 #include "mojo/application/public/cpp/connect.h" |
| 16 #include "mojo/application/public/cpp/interface_factory.h" | 16 #include "mojo/application/public/cpp/interface_factory.h" |
| 17 #include "mojo/application/public/cpp/service_provider_impl.h" | 17 #include "mojo/application/public/cpp/service_provider_impl.h" |
| 18 #include "mojo/application/public/interfaces/content_handler.mojom.h" | 18 #include "mojo/application/public/interfaces/content_handler.mojom.h" |
| 19 #include "mojo/common/weak_binding_set.h" | 19 #include "mojo/common/weak_binding_set.h" |
| 20 #include "mojo/public/cpp/bindings/strong_binding.h" | 20 #include "mojo/public/cpp/bindings/strong_binding.h" |
| 21 #include "mojo/shell/application_fetcher.h" |
| 21 #include "mojo/shell/application_loader.h" | 22 #include "mojo/shell/application_loader.h" |
| 22 #include "mojo/shell/application_manager.h" | 23 #include "mojo/shell/application_manager.h" |
| 23 #include "mojo/shell/capability_filter_unittest.mojom.h" | 24 #include "mojo/shell/capability_filter_unittest.mojom.h" |
| 24 #include "mojo/shell/test_package_manager.h" | |
| 25 #include "testing/gtest/include/gtest/gtest.h" | 25 #include "testing/gtest/include/gtest/gtest.h" |
| 26 | 26 |
| 27 namespace mojo { | 27 namespace mojo { |
| 28 namespace shell { | 28 namespace shell { |
| 29 namespace { | 29 namespace { |
| 30 | 30 |
| 31 const char kTestMimeType[] = "test/mime-type"; | 31 const char kTestMimeType[] = "test/mime-type"; |
| 32 | 32 |
| 33 // Lives on the main thread of the test. | 33 // Lives on the main thread of the test. |
| 34 // Listens for services exposed/blocked and for application connections being | 34 // Listens for services exposed/blocked and for application connections being |
| (...skipping 243 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 278 base::Callback<void(const base::FilePath&, bool)> callback) override {} | 278 base::Callback<void(const base::FilePath&, bool)> callback) override {} |
| 279 std::string MimeType() override { return kTestMimeType; } | 279 std::string MimeType() override { return kTestMimeType; } |
| 280 bool HasMojoMagic() override { return false; } | 280 bool HasMojoMagic() override { return false; } |
| 281 bool PeekFirstLine(std::string* line) override { return false; } | 281 bool PeekFirstLine(std::string* line) override { return false; } |
| 282 | 282 |
| 283 const GURL url_; | 283 const GURL url_; |
| 284 | 284 |
| 285 DISALLOW_COPY_AND_ASSIGN(TestFetcher); | 285 DISALLOW_COPY_AND_ASSIGN(TestFetcher); |
| 286 }; | 286 }; |
| 287 | 287 |
| 288 class CFTestPackageManager : public TestPackageManager { | 288 class TestApplicationFetcher : public ApplicationFetcher { |
| 289 public: | 289 public: |
| 290 CFTestPackageManager() {} | 290 TestApplicationFetcher() {} |
| 291 ~CFTestPackageManager() override {} | 291 ~TestApplicationFetcher() override {} |
| 292 | 292 |
| 293 void set_use_test_fetcher(bool use_test_fetcher) { | 293 void set_use_test_fetcher(bool use_test_fetcher) { |
| 294 use_test_fetcher_ = use_test_fetcher; | 294 use_test_fetcher_ = use_test_fetcher; |
| 295 } | 295 } |
| 296 | 296 |
| 297 private: | 297 private: |
| 298 // Overridden from TestPackageManager: | 298 // Overridden from ApplicationFetcher: |
| 299 void SetApplicationManager(ApplicationManager* manager) override {} |
| 300 GURL ResolveURL(const GURL& url) override { |
| 301 return url; |
| 302 } |
| 299 void FetchRequest(URLRequestPtr request, | 303 void FetchRequest(URLRequestPtr request, |
| 300 const Fetcher::FetchCallback& loader_callback) override { | 304 const Fetcher::FetchCallback& loader_callback) override { |
| 301 if (use_test_fetcher_) | 305 if (use_test_fetcher_) |
| 302 new TestFetcher(GURL(request->url), loader_callback); | 306 new TestFetcher(GURL(request->url), loader_callback); |
| 303 } | 307 } |
| 304 | 308 |
| 305 bool use_test_fetcher_; | 309 bool use_test_fetcher_; |
| 306 | 310 |
| 307 DISALLOW_COPY_AND_ASSIGN(CFTestPackageManager); | 311 DISALLOW_COPY_AND_ASSIGN(TestApplicationFetcher); |
| 308 }; | 312 }; |
| 309 | 313 |
| 310 class TestLoader : public ApplicationLoader { | 314 class TestLoader : public ApplicationLoader { |
| 311 public: | 315 public: |
| 312 explicit TestLoader(ApplicationDelegate* delegate) : delegate_(delegate) {} | 316 explicit TestLoader(ApplicationDelegate* delegate) : delegate_(delegate) {} |
| 313 ~TestLoader() override {} | 317 ~TestLoader() override {} |
| 314 | 318 |
| 315 private: | 319 private: |
| 316 // Overridden from ApplicationLoader: | 320 // Overridden from ApplicationLoader: |
| 317 void Load(const GURL& url, InterfaceRequest<Application> request) override { | 321 void Load(const GURL& url, InterfaceRequest<Application> request) override { |
| 318 app_.reset(new ApplicationImpl(delegate_.get(), request.Pass())); | 322 app_.reset(new ApplicationImpl(delegate_.get(), request.Pass())); |
| 319 } | 323 } |
| 320 | 324 |
| 321 scoped_ptr<ApplicationDelegate> delegate_; | 325 scoped_ptr<ApplicationDelegate> delegate_; |
| 322 scoped_ptr<ApplicationImpl> app_; | 326 scoped_ptr<ApplicationImpl> app_; |
| 323 | 327 |
| 324 DISALLOW_COPY_AND_ASSIGN(TestLoader); | 328 DISALLOW_COPY_AND_ASSIGN(TestLoader); |
| 325 }; | 329 }; |
| 326 | 330 |
| 327 class CapabilityFilterTest : public testing::Test { | 331 class CapabilityFilterTest : public testing::Test { |
| 328 public: | 332 public: |
| 329 CapabilityFilterTest() | 333 CapabilityFilterTest() |
| 330 : test_package_manager_(nullptr), | 334 : test_application_fetcher_(nullptr), |
| 331 validator_(nullptr) {} | 335 validator_(nullptr) {} |
| 332 ~CapabilityFilterTest() override {} | 336 ~CapabilityFilterTest() override {} |
| 333 | 337 |
| 334 protected: | 338 protected: |
| 335 void RunApplication(const std::string& url, const CapabilityFilter& filter) { | 339 void RunApplication(const std::string& url, const CapabilityFilter& filter) { |
| 336 ServiceProviderPtr services; | 340 ServiceProviderPtr services; |
| 337 | 341 |
| 338 // We expose Validator to the test application via ConnectToApplication | 342 // We expose Validator to the test application via ConnectToApplication |
| 339 // because we don't allow the test application to connect to test:validator. | 343 // because we don't allow the test application to connect to test:validator. |
| 340 // Adding it to the CapabilityFilter would interfere with the test. | 344 // Adding it to the CapabilityFilter would interfere with the test. |
| (...skipping 24 matching lines...) Expand all Loading... |
| 365 loop()->Run(); | 369 loop()->Run(); |
| 366 EXPECT_TRUE(validator_->expectations_met()); | 370 EXPECT_TRUE(validator_->expectations_met()); |
| 367 if (!validator_->expectations_met()) | 371 if (!validator_->expectations_met()) |
| 368 validator_->PrintUnmetExpectations(); | 372 validator_->PrintUnmetExpectations(); |
| 369 } | 373 } |
| 370 | 374 |
| 371 void RunContentHandlerTest() { | 375 void RunContentHandlerTest() { |
| 372 set_use_test_fetcher(); | 376 set_use_test_fetcher(); |
| 373 | 377 |
| 374 GURL content_handler_url("test:content_handler"); | 378 GURL content_handler_url("test:content_handler"); |
| 375 test_package_manager_->RegisterContentHandler(kTestMimeType, | 379 application_manager()->RegisterContentHandler(kTestMimeType, |
| 376 content_handler_url); | 380 content_handler_url); |
| 377 | 381 |
| 378 CreateLoader<TestContentHandler>(content_handler_url.spec()); | 382 CreateLoader<TestContentHandler>(content_handler_url.spec()); |
| 379 RunTest(); | 383 RunTest(); |
| 380 } | 384 } |
| 381 | 385 |
| 382 base::MessageLoop* loop() { return &loop_; } | 386 base::MessageLoop* loop() { return &loop_; } |
| 383 ApplicationManager* application_manager() { | 387 ApplicationManager* application_manager() { |
| 384 return application_manager_.get(); | 388 return application_manager_.get(); |
| 385 } | 389 } |
| 386 ConnectionValidator* validator() { return validator_; } | 390 ConnectionValidator* validator() { return validator_; } |
| 387 void set_use_test_fetcher() { | 391 void set_use_test_fetcher() { |
| 388 test_package_manager_->set_use_test_fetcher(true); | 392 test_application_fetcher_->set_use_test_fetcher(true); |
| 389 } | 393 } |
| 390 | 394 |
| 391 // Overridden from testing::Test: | 395 // Overridden from testing::Test: |
| 392 void SetUp() override { | 396 void SetUp() override { |
| 393 test_package_manager_ = new CFTestPackageManager; | 397 test_application_fetcher_ = new TestApplicationFetcher; |
| 394 application_manager_.reset( | 398 application_manager_.reset( |
| 395 new ApplicationManager(make_scoped_ptr(test_package_manager_))); | 399 new ApplicationManager(make_scoped_ptr(test_application_fetcher_))); |
| 396 CreateLoader<ServiceApplication>("test:service"); | 400 CreateLoader<ServiceApplication>("test:service"); |
| 397 CreateLoader<ServiceApplication>("test:service2"); | 401 CreateLoader<ServiceApplication>("test:service2"); |
| 398 } | 402 } |
| 399 void TearDown() override { | 403 void TearDown() override { |
| 400 application_manager_.reset(); | 404 application_manager_.reset(); |
| 401 test_package_manager_->set_use_test_fetcher(false); | 405 test_application_fetcher_->set_use_test_fetcher(false); |
| 402 } | 406 } |
| 403 | 407 |
| 404 private: | 408 private: |
| 405 template<class T> | 409 template<class T> |
| 406 scoped_ptr<ApplicationDelegate> CreateApplicationDelegate() { | 410 scoped_ptr<ApplicationDelegate> CreateApplicationDelegate() { |
| 407 return scoped_ptr<ApplicationDelegate>(new T); | 411 return scoped_ptr<ApplicationDelegate>(new T); |
| 408 } | 412 } |
| 409 | 413 |
| 410 CFTestPackageManager* test_package_manager_; | 414 TestApplicationFetcher* test_application_fetcher_; |
| 411 base::ShadowingAtExitManager at_exit_; | 415 base::ShadowingAtExitManager at_exit_; |
| 412 base::MessageLoop loop_; | 416 base::MessageLoop loop_; |
| 413 scoped_ptr<ApplicationManager> application_manager_; | 417 scoped_ptr<ApplicationManager> application_manager_; |
| 414 ConnectionValidator* validator_; | 418 ConnectionValidator* validator_; |
| 415 | 419 |
| 416 DISALLOW_COPY_AND_ASSIGN(CapabilityFilterTest); | 420 DISALLOW_COPY_AND_ASSIGN(CapabilityFilterTest); |
| 417 }; | 421 }; |
| 418 | 422 |
| 419 class CapabilityFilter_BlockingTest : public CapabilityFilterTest { | 423 class CapabilityFilter_BlockingTest : public CapabilityFilterTest { |
| 420 public: | 424 public: |
| (...skipping 120 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 541 RunTest(); | 545 RunTest(); |
| 542 } | 546 } |
| 543 | 547 |
| 544 TEST_F(CapabilityFilter_WildcardsTest, ContentHandler) { | 548 TEST_F(CapabilityFilter_WildcardsTest, ContentHandler) { |
| 545 RunContentHandlerTest(); | 549 RunContentHandlerTest(); |
| 546 } | 550 } |
| 547 | 551 |
| 548 } // namespace | 552 } // namespace |
| 549 } // namespace shell | 553 } // namespace shell |
| 550 } // namespace mojo | 554 } // namespace mojo |
| OLD | NEW |