Move more of ContentHandler handling into PackageManager.

mojo/package_manager/content_handler_unittest.cc

Index: mojo/package_manager/content_handler_unittest.cc
diff --git a/mojo/package_manager/content_handler_unittest.cc b/mojo/package_manager/content_handler_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..fa94a369b83c3ec53f932e4eeb779df739f03cb1
--- /dev/null
+++ b/mojo/package_manager/content_handler_unittest.cc
@@ -0,0 +1,318 @@
+// Copyright 2015 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 "base/at_exit.h"
+#include "base/bind.h"
+#include "base/macros.h"
+#include "base/memory/scoped_vector.h"
+#include "base/message_loop/message_loop.h"
+#include "base/path_service.h"
+#include "base/run_loop.h"
+#include "mojo/application/public/cpp/application_connection.h"
+#include "mojo/application/public/cpp/application_delegate.h"
+#include "mojo/application/public/cpp/application_impl.h"
+#include "mojo/application/public/cpp/interface_factory.h"
+#include "mojo/application/public/interfaces/content_handler.mojom.h"
+#include "mojo/application/public/interfaces/service_provider.mojom.h"
+#include "mojo/package_manager/package_manager_impl.h"
+#include "mojo/public/cpp/bindings/strong_binding.h"
+#include "mojo/shell/application_loader.h"
+#include "mojo/shell/application_manager.h"
+#include "mojo/shell/connect_util.h"
+#include "mojo/shell/fetcher.h"
+#include "mojo/shell/test_package_manager.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace mojo {
+namespace package_manager {
+namespace test {
+namespace {
+
+const char kTestMimeType[] = "test/mime-type";
+
+class TestFetcher : public shell::Fetcher {
+ public:
+  TestFetcher(const FetchCallback& fetch_callback,
+              const GURL& url,
+              const std::string& mime_type)
+      : Fetcher(fetch_callback), url_(url), mime_type_(mime_type) {
+    loader_callback_.Run(make_scoped_ptr(this));
+  }
+  ~TestFetcher() override {}
+
+  // Fetcher:
+  const GURL& GetURL() const override { return url_; }
+  GURL GetRedirectURL() const override { return GURL("yyy"); }
+  GURL GetRedirectReferer() const override { return GURL(); }
+  URLResponsePtr AsURLResponse(base::TaskRunner* task_runner,
+                               uint32_t skip) override {
+    return URLResponse::New().Pass();
+  }
+  void AsPath(
+      base::TaskRunner* task_runner,
+      base::Callback<void(const base::FilePath&, bool)> callback) override {}
+  std::string MimeType() override { return mime_type_; }
+  bool HasMojoMagic() override { return false; }
+  bool PeekFirstLine(std::string* line) override { return false; }
+
+ private:
+  const GURL url_;
+  const std::string mime_type_;
+
+  DISALLOW_COPY_AND_ASSIGN(TestFetcher);
+};
+
+void QuitClosure(bool* value) {
+  *value = true;
+  base::MessageLoop::current()->QuitWhenIdle();
+}
+
+class TestContentHandler : public ContentHandler, public ApplicationDelegate {
+ public:
+  TestContentHandler(ApplicationConnection* connection,
+                     InterfaceRequest<ContentHandler> request)
+      : binding_(this, request.Pass()) {}
+
+  // ContentHandler:
+  void StartApplication(InterfaceRequest<Application> application_request,
+                        URLResponsePtr response) override {
+    apps_.push_back(new ApplicationImpl(this, application_request.Pass()));
+  }
+
+ private:
+  StrongBinding<ContentHandler> binding_;
+  ScopedVector<ApplicationImpl> apps_;
+
+  DISALLOW_COPY_AND_ASSIGN(TestContentHandler);
+};
+
+class TestApplicationLoader : public shell::ApplicationLoader,
+                              public ApplicationDelegate,
+                              public InterfaceFactory<ContentHandler> {
+ public:
+  TestApplicationLoader() : num_loads_(0) {}
+  ~TestApplicationLoader() override {}
+
+  int num_loads() const { return num_loads_; }
+  const GURL& last_requestor_url() const { return last_requestor_url_; }
+
+ private:
+  // ApplicationLoader implementation.
+  void Load(const GURL& url,
+            InterfaceRequest<Application> application_request) override {
+    ++num_loads_;
+    test_app_.reset(new ApplicationImpl(this, application_request.Pass()));
+  }
+
+  // ApplicationDelegate implementation.
+  bool ConfigureIncomingConnection(ApplicationConnection* connection) override {
+    connection->AddService<ContentHandler>(this);
+    last_requestor_url_ = GURL(connection->GetRemoteApplicationURL());
+    return true;
+  }
+  // InterfaceFactory<ContentHandler> implementation.
+  void Create(ApplicationConnection* connection,
+              InterfaceRequest<ContentHandler> request) override {
+    new TestContentHandler(connection, request.Pass());
+  }
+
+  scoped_ptr<ApplicationImpl> test_app_;
+  int num_loads_;
+  GURL last_requestor_url_;
+
+  DISALLOW_COPY_AND_ASSIGN(TestApplicationLoader);
+};
+
+class TestPackageManager : public PackageManagerImpl {
+ public:
+  explicit TestPackageManager(const base::FilePath& package_path)
+      : PackageManagerImpl(package_path, nullptr),
+        mime_type_(kTestMimeType) {}
+  ~TestPackageManager() override {}
+
+  void set_mime_type(const std::string& mime_type) {
+    mime_type_ = mime_type;
+  }
+
+  // PackageManagerImpl:
+  void FetchRequest(
+      URLRequestPtr request,
+      const shell::Fetcher::FetchCallback& loader_callback) override {
+    new TestFetcher(loader_callback, GURL(request->url), mime_type_);
+  }
+
+ private:
+  std::string mime_type_;
+
+  DISALLOW_COPY_AND_ASSIGN(TestPackageManager);
+};
+
+}  // namespace
+
+class ContentHandlerTest : public testing::Test {
+ public:
+  ContentHandlerTest()
+      : content_handler_url_("http://test.content.handler"),
+        requestor_url_("http://requestor.url") {}
+  ~ContentHandlerTest() override {}
+
+  void SetUp() override {
+    base::FilePath shell_dir;
+    PathService::Get(base::DIR_MODULE, &shell_dir);
+    test_package_manager_ = new TestPackageManager(shell_dir);
+    test_package_manager_->RegisterContentHandler(kTestMimeType,
+                                                  content_handler_url_);
+    application_manager_.reset(new shell::ApplicationManager(
+        make_scoped_ptr(test_package_manager_)));
+  }
+
+  void TearDown() override {
+    test_package_manager_ = nullptr;
+    application_manager_.reset();
+  }
+
+ protected:
+  const GURL content_handler_url_;
+  const GURL requestor_url_;
+
+  base::MessageLoop loop_;
+  scoped_ptr<shell::ApplicationManager> application_manager_;
+  // Owned by ApplicationManager.
+  TestPackageManager* test_package_manager_;
+
+  DISALLOW_COPY_AND_ASSIGN(ContentHandlerTest);
+};
+
+TEST_F(ContentHandlerTest, ContentHandlerConnectionGetsRequestorURL) {
+  TestApplicationLoader* loader = new TestApplicationLoader;
+  application_manager_->SetLoaderForURL(
+      scoped_ptr<shell::ApplicationLoader>(loader),
+      content_handler_url_);
+
+  bool called = false;
+  scoped_ptr<shell::ConnectToApplicationParams> params(
+      new shell::ConnectToApplicationParams);
+  params->set_source(shell::Identity(requestor_url_));
+  params->SetTargetURL(GURL("test:test"));
+  params->set_on_application_end(
+      base::Bind(&QuitClosure, base::Unretained(&called)));
+  application_manager_->ConnectToApplication(params.Pass());
+  loop_.Run();
+  EXPECT_TRUE(called);
+
+  ASSERT_EQ(1, loader->num_loads());
+  EXPECT_EQ(requestor_url_, loader->last_requestor_url());
+}
+
+TEST_F(ContentHandlerTest,
+       MultipleConnectionsToContentHandlerGetSameContentHandlerId) {
+  TestApplicationLoader* content_handler_loader = new TestApplicationLoader;
+  application_manager_->SetLoaderForURL(
+      scoped_ptr<shell::ApplicationLoader>(content_handler_loader),
+      content_handler_url_);
+
+  uint32_t content_handler_id;
+  {
+    base::RunLoop run_loop;
+    scoped_ptr<shell::ConnectToApplicationParams> params(
+        new shell::ConnectToApplicationParams);
+    params->set_source(shell::Identity(requestor_url_));
+    params->SetTargetURL(GURL("test:test"));
+    params->set_connect_callback([&content_handler_id, &run_loop](uint32_t t) {
+      content_handler_id = t;
+      run_loop.Quit();
+    });
+    application_manager_->ConnectToApplication(params.Pass());
+    run_loop.Run();
+    EXPECT_NE(Shell::kInvalidContentHandlerID, content_handler_id);
+  }
+
+  uint32_t content_handler_id2;
+  {
+    base::RunLoop run_loop;
+    scoped_ptr<shell::ConnectToApplicationParams> params(
+        new shell::ConnectToApplicationParams);
+    params->set_source(shell::Identity(requestor_url_));
+    params->SetTargetURL(GURL("test:test"));
+    params->set_connect_callback([&content_handler_id2, &run_loop](uint32_t t) {
+      content_handler_id2 = t;
+      run_loop.Quit();
+    });
+    application_manager_->ConnectToApplication(params.Pass());
+    run_loop.Run();
+    EXPECT_NE(Shell::kInvalidContentHandlerID, content_handler_id2);
+  }
+  EXPECT_EQ(content_handler_id, content_handler_id2);
+}
+
+TEST_F(ContentHandlerTest, DifferedContentHandlersGetDifferentIDs) {
+  TestApplicationLoader* content_handler_loader = new TestApplicationLoader;
+  application_manager_->SetLoaderForURL(
+      scoped_ptr<shell::ApplicationLoader>(content_handler_loader),
+      content_handler_url_);
+
+  uint32_t content_handler_id;
+  {
+    base::RunLoop run_loop;
+    scoped_ptr<shell::ConnectToApplicationParams> params(
+        new shell::ConnectToApplicationParams);
+    params->set_source(shell::Identity(requestor_url_));
+    params->SetTargetURL(GURL("test:test"));
+    params->set_connect_callback([&content_handler_id, &run_loop](uint32_t t) {
+      content_handler_id = t;
+      run_loop.Quit();
+    });
+    application_manager_->ConnectToApplication(params.Pass());
+    run_loop.Run();
+    EXPECT_NE(Shell::kInvalidContentHandlerID, content_handler_id);
+  }
+
+  const std::string mime_type2 = "test/mime-type2";
+  const GURL content_handler_url2("http://test.content.handler2");
+  test_package_manager_->set_mime_type(mime_type2);
+  test_package_manager_->RegisterContentHandler(mime_type2,
+                                                content_handler_url2);
+
+  TestApplicationLoader* content_handler_loader2 = new TestApplicationLoader;
+  application_manager_->SetLoaderForURL(
+      scoped_ptr<shell::ApplicationLoader>(content_handler_loader2),
+      content_handler_url2);
+
+  uint32_t content_handler_id2;
+  {
+    base::RunLoop run_loop;
+    scoped_ptr<shell::ConnectToApplicationParams> params(
+        new shell::ConnectToApplicationParams);
+    params->set_source(shell::Identity(requestor_url_));
+    params->SetTargetURL(GURL("test2:test2"));
+    params->set_connect_callback([&content_handler_id2, &run_loop](uint32_t t) {
+      content_handler_id2 = t;
+      run_loop.Quit();
+    });
+    application_manager_->ConnectToApplication(params.Pass());
+    run_loop.Run();
+    EXPECT_NE(Shell::kInvalidContentHandlerID, content_handler_id2);
+  }
+  EXPECT_NE(content_handler_id, content_handler_id2);
+}
+
+TEST_F(ContentHandlerTest,
+       ConnectWithNoContentHandlerGetsInvalidContentHandlerId) {
+  application_manager_->SetLoaderForURL(
+      scoped_ptr<shell::ApplicationLoader>(new TestApplicationLoader),
+      GURL("test:test"));
+
+  uint32_t content_handler_id = 1u;
+  scoped_ptr<shell::ConnectToApplicationParams> params(
+      new shell::ConnectToApplicationParams);
+  params->SetTargetURL(GURL("test:test"));
+  params->set_connect_callback(
+      [&content_handler_id](uint32_t t) { content_handler_id = t; });
+  application_manager_->ConnectToApplication(params.Pass());
+  EXPECT_EQ(0u, content_handler_id);
+}
+
+}  // namespace test
+}  // namespace package_manager
+}  // namespace mojo