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

Side by Side Diff: mojo/fetcher/data_fetcher_unittest.cc

Issue 1563413002: Move fetchers into mojo/shell/fetcher (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Created 4 years, 11 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 unified diff | Download patch
« no previous file with comments | « mojo/fetcher/data_fetcher.cc ('k') | mojo/fetcher/local_fetcher.h » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
(Empty)
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
3 // found in the LICENSE file.
4
5 #include "mojo/fetcher/data_fetcher.h"
6
7 #include <stdint.h>
8
9 #include <utility>
10
11 #include "base/auto_reset.h"
12 #include "base/bind.h"
13 #include "base/logging.h"
14 #include "base/macros.h"
15 #include "base/memory/scoped_ptr.h"
16 #include "base/message_loop/message_loop.h"
17 #include "base/run_loop.h"
18 #include "mojo/public/cpp/system/data_pipe.h"
19 #include "testing/gtest/include/gtest/gtest.h"
20
21 namespace mojo {
22 namespace fetcher {
23 namespace {
24
25 class FetchCallbackHelper {
26 public:
27 FetchCallbackHelper() : run_loop_(nullptr) {}
28 ~FetchCallbackHelper() {}
29
30 shell::Fetcher::FetchCallback GetCallback() {
31 return base::Bind(&FetchCallbackHelper::CallbackHandler,
32 base::Unretained(this));
33 }
34
35 void WaitForCallback() {
36 base::RunLoop run_loop;
37 base::AutoReset<base::RunLoop*> auto_reset(&run_loop_, &run_loop);
38 run_loop.Run();
39 }
40
41 shell::Fetcher* fetcher() const { return fetcher_.get(); }
42
43 private:
44 void CallbackHandler(scoped_ptr<shell::Fetcher> fetcher) {
45 fetcher_ = std::move(fetcher);
46 if (run_loop_)
47 run_loop_->Quit();
48 }
49
50 // If it is not null, it points to a stack-allocated base::RunLoop instance in
51 // WaitForCallback().
52 base::RunLoop* run_loop_;
53 scoped_ptr<shell::Fetcher> fetcher_;
54 DISALLOW_COPY_AND_ASSIGN(FetchCallbackHelper);
55 };
56
57 class DataFetcherTest : public testing::Test {
58 public:
59 DataFetcherTest() {}
60 ~DataFetcherTest() override {}
61
62 protected:
63 void TestFetchURL(const std::string& url,
64 uint32_t expected_status_code,
65 const std::string& expected_mime_type,
66 const std::string& expected_body) {
67 FetchCallbackHelper helper;
68 DataFetcher::Start(GURL(url), helper.GetCallback());
69 helper.WaitForCallback();
70
71 ASSERT_TRUE(helper.fetcher());
72 URLResponsePtr response = helper.fetcher()->AsURLResponse(nullptr, 0);
73 ASSERT_TRUE(response);
74 EXPECT_EQ(url, response->url);
75 EXPECT_EQ(expected_status_code, response->status_code);
76
77 if (expected_status_code != 200)
78 return;
79
80 ASSERT_TRUE(response->body.is_valid());
81 EXPECT_EQ(expected_mime_type, response->mime_type);
82
83 uint32_t num_bytes = 0;
84 Handle body_handle = response->body.release();
85
86 MojoHandleSignalsState hss;
87 ASSERT_EQ(MOJO_RESULT_OK,
88 MojoWait(body_handle.value(), MOJO_HANDLE_SIGNAL_READABLE,
89 MOJO_DEADLINE_INDEFINITE, &hss));
90
91 MojoResult result = MojoReadData(body_handle.value(), nullptr, &num_bytes,
92 MOJO_READ_DATA_FLAG_QUERY);
93 ASSERT_EQ(MOJO_RESULT_OK, result);
94
95 scoped_ptr<char[]> body(new char[num_bytes]);
96 result = MojoReadData(body_handle.value(), body.get(), &num_bytes,
97 MOJO_READ_DATA_FLAG_ALL_OR_NONE);
98 ASSERT_EQ(MOJO_RESULT_OK, result);
99 EXPECT_EQ(expected_body, std::string(body.get(), num_bytes));
100 }
101
102 private:
103 base::MessageLoop loop_;
104
105 DISALLOW_COPY_AND_ASSIGN(DataFetcherTest);
106 };
107
108 TEST_F(DataFetcherTest, BasicSuccess) {
109 TestFetchURL("data:text/html,Hello world", 200, "text/html", "Hello world");
110 }
111
112 TEST_F(DataFetcherTest, BasicFailure) {
113 TestFetchURL("about:blank", 400, std::string(), std::string());
114 TestFetchURL("data:;base64,aGVs_-_-", 400, std::string(), std::string());
115 }
116
117 } // namespace
118 } // namespace fetcher
119 } // namespace mojo
OLDNEW
« no previous file with comments | « mojo/fetcher/data_fetcher.cc ('k') | mojo/fetcher/local_fetcher.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698