Adds resource_provider::ResourceProvider

components/resource_provider/resource_provider_apptest.cc

+// 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 <stdint.h>
+
+#include "base/bind.h"
+#include "base/containers/scoped_ptr_hash_map.h"
+#include "base/run_loop.h"
+#include "components/resource_provider/public/interfaces/resource_provider.mojom.h"
+#include "mojo/application/application_test_base_chromium.h"
+#include "mojo/common/common_type_converters.h"
+#include "third_party/mojo/src/mojo/public/cpp/application/application_delegate.h"
+#include "third_party/mojo/src/mojo/public/cpp/application/application_impl.h"
+#include "third_party/mojo/src/mojo/public/cpp/application/service_provider_impl.h"
+#include "third_party/mojo/src/mojo/public/cpp/bindings/array.h"
+#include "third_party/mojo/src/mojo/public/cpp/system/macros.h"
+
+namespace resource_provider {
+namespace {
+
+// ResourceFetcher fetches resources from a ResourceProvider, storing the
+// results as well as running
+// a message loop until the resource has been obtained.
+class ResourceFetcher {
+ public:
+  using ResourceMap =
+      base::ScopedPtrHashMap<std::string, scoped_ptr<mojo::Array<uint8_t>>>;
+
+  explicit ResourceFetcher(ResourceProvider* provider)
+      : provider_(provider), waiting_count_(0u) {}
+  ~ResourceFetcher() {}
+
+  ResourceMap* resources() { return &resources_; }
+
+  void WaitForResults() {
+    ASSERT_NE(0u, waiting_count_);
+    run_loop_.Run();
+  }
+
+  void GetResource(const std::string& path) {
+    waiting_count_++;
+    provider_->GetResource(path, base::Bind(&ResourceFetcher::OnGotResource,
+                                            base::Unretained(this), path));
+  }
+
+ private:
+  // Callback when a resource has been fetched.
+  void OnGotResource(const std::string& path, mojo::Array<uint8_t> data) {
+    scoped_ptr<mojo::Array<uint8_t>> resource;
+    EXPECT_FALSE(resources_.contains(path));
+    if (!data.is_null())
+      resource.reset(new mojo::Array<uint8_t>(data.Pass()));
+    resources_.add(path, resource.Pass());
+
+    if (waiting_count_ > 0) {
+      waiting_count_--;
+      if (waiting_count_ == 0)
+        run_loop_.Quit();
+    }
+  }
+
+  ResourceProvider* provider_;
+  ResourceMap resources_;
+  // Number of resources we're waiting on.
+  size_t waiting_count_;
+  base::RunLoop run_loop_;
+
+  DISALLOW_COPY_AND_ASSIGN(ResourceFetcher);
+};
+
+class ResourceProviderApplicationTest : public mojo::test::ApplicationTestBase {
+ public:
+  ResourceProviderApplicationTest() {}
+  ~ResourceProviderApplicationTest() override {}
+
+ protected:
+  // ApplicationTestBase:
+  void SetUp() override {
+    ApplicationTestBase::SetUp();
+    application_impl()->ConnectToService("mojo:resource_provider",
+                                         &resource_provider_);
+  }
+
+  ResourceProviderPtr resource_provider_;
+
+ private:
+  MOJO_DISALLOW_COPY_AND_ASSIGN(ResourceProviderApplicationTest);
+};
+
+TEST_F(ResourceProviderApplicationTest, FetchOneResource) {
+  ResourceFetcher fetcher(resource_provider_.get());
+  fetcher.GetResource("sample");
+  fetcher.WaitForResults();
+  ASSERT_TRUE(fetcher.resources()->contains("sample"));
+  ASSERT_FALSE(fetcher.resources()->get("sample")->is_null());
+  EXPECT_EQ("test data\n",
+            fetcher.resources()->get("sample")->To<std::string>());
+}
+
+TEST_F(ResourceProviderApplicationTest, FetchTwoResources) {
+  ResourceFetcher fetcher(resource_provider_.get());
+  fetcher.GetResource("sample");
+  fetcher.GetResource("dir/sample2");
+  fetcher.WaitForResults();
+  ASSERT_TRUE(fetcher.resources()->contains("sample"));
+  ASSERT_TRUE(fetcher.resources()->get("sample"));
+  ASSERT_FALSE(fetcher.resources()->get("sample")->is_null());
+  EXPECT_EQ("test data\n",
+            fetcher.resources()->get("sample")->To<std::string>());
+
+  ASSERT_TRUE(fetcher.resources()->contains("dir/sample2"));
+  ASSERT_TRUE(fetcher.resources()->get("dir/sample2"));
+  ASSERT_FALSE(fetcher.resources()->get("dir/sample2")->is_null());
+  EXPECT_EQ("xxyy\n",
+            fetcher.resources()->get("dir/sample2")->To<std::string>());
+}
+
+}  // namespace
+}  // namespace resource_provider