1 files changed, 94 insertions, 0 deletions
diff --git a/remoting/host/url_fetcher_unittest.cc b/remoting/host/url_fetcher_unittest.cc
new file mode 100644
index 0000000..614474e
--- /dev/null
+++ b/remoting/host/url_fetcher_unittest.cc
@@ -0,0 +1,94 @@
+// Copyright (c) 2012 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 "remoting/host/url_fetcher.h"
+
+#include "base/message_loop.h"
+#include "base/threading/thread.h"
+#include "net/test/test_server.h"
+#include "net/url_request/url_request.h"
+#include "net/url_request/url_request_status.h"
+#include "remoting/host/url_request_context.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace remoting {
+
+class UrlFetcherTest : public testing::Test  {
+ public:
+  UrlFetcherTest()
+      : test_server_(
+          net::TestServer::TYPE_HTTPS,
+          net::TestServer::kLocalhost,
+          FilePath(FILE_PATH_LITERAL("net/data/url_request_unittest"))),
+        io_thread_("TestIOThread"),
+        file_thread_("TestFileThread") {
+  }
+
+ protected:
+  void SetUp() OVERRIDE {
+    ASSERT_TRUE(io_thread_.StartWithOptions(
+        base::Thread::Options(MessageLoop::TYPE_IO, 0)));
+    ASSERT_TRUE(file_thread_.StartWithOptions(
+        base::Thread::Options(MessageLoop::TYPE_IO, 0)));
+    context_getter_ = new URLRequestContextGetter(io_thread_.message_loop(),
+                                                  file_thread_.message_loop());
+    ASSERT_TRUE(test_server_.Start());
+  }
+
+ protected:
+  void OnDone(const net::URLRequestStatus& status,
+              int response_code,
+              const std::string& response) {
+    ASSERT_EQ(MessageLoop::current(), &message_loop_);
+    status_ = status;
+    response_code_ = response_code;
+    response_ = response;
+    message_loop_.PostTask(FROM_HERE, MessageLoop::QuitClosure());
+  }
+
+  net::TestServer test_server_;
+  MessageLoopForUI message_loop_;
+  base::Thread io_thread_;
+  base::Thread file_thread_;
+  scoped_refptr<URLRequestContextGetter> context_getter_;
+  net::URLRequestStatus status_;
+  std::string response_;
+  int response_code_;
+};
+
+TEST_F(UrlFetcherTest, TestGet) {
+  UrlFetcher fetcher(test_server_.GetURL("default"), UrlFetcher::GET);
+  fetcher.SetRequestContext(context_getter_);
+  fetcher.Start(base::Bind(&UrlFetcherTest_TestGet_Test::OnDone,
+                           base::Unretained(this)));
+  message_loop_.Run();
+  EXPECT_EQ(net::URLRequestStatus::SUCCESS, status_.status());
+  EXPECT_EQ("Default response given for path: /default", response_);
+  EXPECT_EQ(200, response_code_);
+}
+
+TEST_F(UrlFetcherTest, TestPost) {
+  const char kTestQueryData[] = "123qwe123qwe";
+  UrlFetcher fetcher(test_server_.GetURL("echo"), UrlFetcher::POST);
+  fetcher.SetRequestContext(context_getter_);
+  fetcher.SetUploadData("text/html", kTestQueryData);
+  fetcher.Start(base::Bind(&UrlFetcherTest_TestPost_Test::OnDone,
+                           base::Unretained(this)));
+  message_loop_.Run();
+  EXPECT_EQ(net::URLRequestStatus::SUCCESS, status_.status());
+  EXPECT_EQ(kTestQueryData, response_);
+  EXPECT_EQ(200, response_code_);
+}
+
+TEST_F(UrlFetcherTest, TestFailed) {
+  UrlFetcher fetcher(test_server_.GetURL("auth-basic"), UrlFetcher::GET);
+  fetcher.SetRequestContext(context_getter_);
+  fetcher.Start(base::Bind(&UrlFetcherTest_TestFailed_Test::OnDone,
+                           base::Unretained(this)));
+  message_loop_.Run();
+  EXPECT_EQ(net::URLRequestStatus::SUCCESS, status_.status());
+  EXPECT_EQ(401, response_code_);
+}
+
+}  // namespace remoting