Chromium Code Reviews| Index: components/copresence/rpc/http_post_unittest.cc |
| diff --git a/components/copresence/rpc/http_post_unittest.cc b/components/copresence/rpc/http_post_unittest.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..925ee30c899d297ecec0aa30e1f285164d1b323b |
| --- /dev/null |
| +++ b/components/copresence/rpc/http_post_unittest.cc |
| @@ -0,0 +1,109 @@ |
| +// Copyright 2014 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/command_line.h" |
|
willchan no longer on Chromium
2014/08/08 01:05:59
Why is this here?
Charlie
2014/08/08 21:57:53
Outdated. Removed.
|
| +#include "base/test/test_simple_task_runner.h" |
| +#include "components/copresence/proto/data.pb.h" |
| +#include "components/copresence/rpc/http_post.h" |
|
willchan no longer on Chromium
2014/08/08 01:05:59
This goes first, as per style guide.
Charlie
2014/08/08 21:57:53
Done.
|
| +#include "net/http/http_status_code.h" |
| +#include "net/url_request/test_url_fetcher_factory.h" |
| +#include "net/url_request/url_request_test_util.h" |
| +#include "testing/gtest/include/gtest/gtest.h" |
| + |
| +namespace { |
| + |
| +const char kFakeServerHost[] = "test.server.google.com"; |
| +const char kRPCName[] = "testRpc"; |
| + |
| +} // namespace |
| + |
| +using google::protobuf::MessageLite; |
| + |
| +namespace copresence { |
| + |
| +class HttpPostTest : public testing::Test { |
| + public: |
| + HttpPostTest() |
| + : received_response_code_(0) { |
| + context_getter_ = new net::TestURLRequestContextGetter( |
| + make_scoped_refptr(new base::TestSimpleTaskRunner)); |
| + proto_.set_client("test_client"); |
| + proto_.set_version_code(123); |
| + } |
| + virtual ~HttpPostTest() {} |
| + |
| + // Record the response sent back to the client for verification. |
| + void TestResponseCallback(int response_code, |
| + const std::string& response, |
| + HttpPost* completed_post) { |
| + received_response_code_ = response_code; |
| + received_response_ = response; |
| + completed_post_ = completed_post; |
| + } |
| + |
| + protected: |
| + bool ResponsePassedThrough(int response_code, const std::string& response) { |
| + HttpPost* post = new HttpPost( |
| + context_getter_.get(), |
| + std::string("http://") + kFakeServerHost, |
| + kRPCName, |
| + proto_, |
| + base::Bind(&HttpPostTest::TestResponseCallback, |
| + base::Unretained(this))); |
| + net::TestURLFetcher* fetcher = fetcher_factory_.GetFetcherByID( |
| + HttpPost::kUrlFetcherId); |
| + fetcher->set_response_code(response_code); |
| + fetcher->SetResponseString(response); |
| + fetcher->delegate()->OnURLFetchComplete(fetcher); |
| + return received_response_code_ == response_code && |
| + received_response_ == response && |
| + completed_post_ == post; |
| + } |
| + |
| + net::TestURLFetcherFactory fetcher_factory_; |
| + scoped_refptr<net::TestURLRequestContextGetter> context_getter_; |
| + |
| + ClientVersion proto_; |
| + |
| + int received_response_code_; |
| + std::string received_response_; |
| + HttpPost* completed_post_; |
| +}; |
| + |
| +TEST_F(HttpPostTest, OKResponse) { |
| + // "Send" the proto to the "server". |
| + HttpPost* post = new HttpPost(context_getter_.get(), |
| + std::string("http://") + kFakeServerHost, |
| + kRPCName, |
| + proto_, |
| + base::Bind(&HttpPostTest::TestResponseCallback, |
| + base::Unretained(this))); |
| + |
| + // Verify that the right data got sent to the right place. |
| + net::TestURLFetcher* fetcher = fetcher_factory_.GetFetcherByID( |
| + HttpPost::kUrlFetcherId); |
| + EXPECT_EQ(kFakeServerHost, fetcher->GetOriginalURL().host()); |
| + EXPECT_EQ(std::string("/") + kRPCName, fetcher->GetOriginalURL().path()); |
| + std::string upload_data; |
| + DCHECK(proto_.SerializeToString(&upload_data)); |
| + EXPECT_EQ(upload_data, fetcher->upload_data()); |
| + |
| + // Send a response and check that it's passed along correctly. |
| + fetcher->set_response_code(net::HTTP_OK); |
| + fetcher->SetResponseString("Hello World!"); |
| + fetcher->delegate()->OnURLFetchComplete(fetcher); |
| + EXPECT_EQ(net::HTTP_OK, received_response_code_); |
| + EXPECT_EQ("Hello World!", received_response_); |
| + EXPECT_EQ(post, completed_post_); |
| +} |
| + |
| +TEST_F(HttpPostTest, ErrorResponse) { |
| + EXPECT_TRUE(ResponsePassedThrough( |
| + net::HTTP_BAD_REQUEST, "Bad client. Shame on you.")); |
| + EXPECT_TRUE(ResponsePassedThrough( |
| + net::HTTP_INTERNAL_SERVER_ERROR, "I'm dying. Forgive me.")); |
| + EXPECT_TRUE(ResponsePassedThrough(-1, "")); |
| +} |
| + |
| +} // namespace copresence |