Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(416)

Unified Diff: mojo/fetcher/data_fetcher_unittest.cc

Issue 1356623002: mojo: Add DataURLFetcher for processing data: urls. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: lolgyp Created 5 years, 3 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: mojo/fetcher/data_fetcher_unittest.cc
diff --git a/mojo/fetcher/data_fetcher_unittest.cc b/mojo/fetcher/data_fetcher_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..568088e5df728f23a34ca363a29adc97a50259de
--- /dev/null
+++ b/mojo/fetcher/data_fetcher_unittest.cc
@@ -0,0 +1,113 @@
+// 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 "mojo/fetcher/data_fetcher.h"
+
+#include "base/auto_reset.h"
+#include "base/bind.h"
+#include "base/logging.h"
+#include "base/macros.h"
+#include "base/memory/scoped_ptr.h"
+#include "base/message_loop/message_loop.h"
+#include "base/run_loop.h"
+#include "mojo/runner/context.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "third_party/mojo/src/mojo/public/cpp/system/data_pipe.h"
+
+namespace mojo {
+namespace fetcher {
+namespace {
+
+class FetchCallbackHelper {
+ public:
+ FetchCallbackHelper() : run_loop_(nullptr) {}
+ ~FetchCallbackHelper() {}
+
+ shell::Fetcher::FetchCallback GetCallback() {
+ return base::Bind(&FetchCallbackHelper::CallbackHandler,
+ base::Unretained(this));
+ }
+
+ void WaitForCallback() {
+ base::RunLoop run_loop;
+ base::AutoReset<base::RunLoop*> auto_reset(&run_loop_, &run_loop);
+ run_loop.Run();
+ }
+
+ shell::Fetcher* fetcher() const { return fetcher_.get(); }
+
+ private:
+ void CallbackHandler(scoped_ptr<shell::Fetcher> fetcher) {
+ fetcher_ = fetcher.Pass();
+ if (run_loop_)
+ run_loop_->Quit();
+ }
+
+ // If it is not null, it points to a stack-allocated base::RunLoop instance in
+ // WaitForCallback().
+ base::RunLoop* run_loop_;
+ scoped_ptr<shell::Fetcher> fetcher_;
+ DISALLOW_COPY_AND_ASSIGN(FetchCallbackHelper);
+};
+
+class DataFetcherTest : public testing::Test {
+ public:
+ DataFetcherTest() {}
+ ~DataFetcherTest() override {}
+
+ protected:
+ // Overridden from testing::Test:
+ void SetUp() override { runner::Context::EnsureEmbedderIsInitialized(); }
+
+ void TestFetchURL(const std::string& url,
+ uint32_t expected_status_code,
+ const std::string& expected_mime_type,
+ const std::string& expected_body) {
+ FetchCallbackHelper helper;
+ DataFetcher::Start(GURL(url), helper.GetCallback());
+ helper.WaitForCallback();
+
+ ASSERT_TRUE(helper.fetcher());
+ URLResponsePtr response = helper.fetcher()->AsURLResponse(nullptr, 0);
+ ASSERT_TRUE(response);
+ EXPECT_EQ(url, response->url);
+ EXPECT_EQ(expected_status_code, response->status_code);
+
+ if (expected_status_code != 200)
+ return;
+
+ ASSERT_TRUE(response->body.is_valid());
+ EXPECT_EQ(expected_mime_type, response->mime_type);
+
+ uint32_t num_bytes = 0;
+ Handle body_handle = response->body.release();
+ MojoResult result = MojoReadData(body_handle.value(), nullptr, &num_bytes,
+ MOJO_READ_DATA_FLAG_QUERY);
+ ASSERT_EQ(MOJO_RESULT_OK, result);
+
+ scoped_ptr<char[]> body(new char[num_bytes]);
+ result = MojoReadData(body_handle.value(), body.get(), &num_bytes,
+ MOJO_READ_DATA_FLAG_ALL_OR_NONE);
+ ASSERT_EQ(MOJO_RESULT_OK, result);
+ EXPECT_EQ(expected_body, std::string(body.get(), num_bytes));
+ }
+
+ private:
+ base::MessageLoop loop_;
+
+ DISALLOW_COPY_AND_ASSIGN(DataFetcherTest);
+};
+
+TEST_F(DataFetcherTest, BasicSuccess) {
+ TestFetchURL("data:text/html,Hello world", 200, "text/html", "Hello world");
+}
+
+TEST_F(DataFetcherTest, BasicFailure) {
+ TestFetchURL("about:blank", 400, std::string(), std::string());
+ TestFetchURL("data:;base64,aGVs_-_-", 400, std::string(), std::string());
+}
+
+} // namespace
+} // namespace fetcher
+} // namespace mojo

Powered by Google App Engine
This is Rietveld 408576698