Implement server-side confirm request for privet registration

chrome/browser/local_discovery/privet_confirm_api_flow_unittest.cc

+// Copyright 2013 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/bind.h"
+#include "base/message_loop/message_loop.h"
+#include "chrome/browser/local_discovery/privet_confirm_api_flow.h"
+#include "content/public/test/test_browser_thread.h"
+#include "google_apis/gaia/google_service_auth_error.h"
+#include "net/base/host_port_pair.h"
+#include "net/base/net_errors.h"
+#include "net/http/http_request_headers.h"
+#include "net/url_request/test_url_fetcher_factory.h"
+#include "net/url_request/url_request_test_util.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+using testing::NiceMock;
+
+namespace local_discovery {
+
+namespace {
+
+const char kSampleConfirmResponse[] = "{"
+    "   \"success\": true"
+    "}";
+
+const char kFailedConfirmResponse[] = "{"
+    "   \"success\": false"
+    "}";
+
+
+const char kFailedConfirmResponseBadJson[] = "["
+    "   \"success\""
+    "]";
+
+class TestOAuth2TokenService : public OAuth2TokenService {
+ public:
+  explicit TestOAuth2TokenService(net::URLRequestContextGetter* request_context)
+      : request_context_(request_context) {
+  }
+ protected:
+  virtual std::string GetRefreshToken() OVERRIDE {
+    return "SampleToken";
+  }
+
+  virtual net::URLRequestContextGetter* GetRequestContext() OVERRIDE {
+    return request_context_.get();
+  }
+
+ private:
+  scoped_refptr<net::URLRequestContextGetter> request_context_;
+};
+
+class MockableConfirmCallback {
+ public:
+  MOCK_METHOD1(ConfirmCallback, void(PrivetConfirmApiCallFlow::Status));
+
+  PrivetConfirmApiCallFlow::ResponseCallback callback() {
+    return base::Bind(&MockableConfirmCallback::ConfirmCallback,
+                      base::Unretained(this));
+  }
+};
+
+class PrivetConfirmApiFlowTest : public testing::Test {
+ public:
+  PrivetConfirmApiFlowTest()
+      : ui_thread_(content::BrowserThread::UI,
+                   &loop_),
+        request_context_(new net::TestURLRequestContextGetter(
+            base::MessageLoopProxy::current())),
+        token_service_(request_context_.get()) {
+    ui_thread_.Stop();  // HACK: Fake being on the UI thread
+  }
+
+  virtual ~PrivetConfirmApiFlowTest() {
+  }
+
+ protected:
+  base::MessageLoopForUI loop_;
+  content::TestBrowserThread ui_thread_;
+  scoped_refptr<net::TestURLRequestContextGetter> request_context_;
+  net::TestURLFetcherFactory fetcher_factory_;
+  TestOAuth2TokenService token_service_;
+  MockableConfirmCallback callback_;
+};
+
+TEST_F(PrivetConfirmApiFlowTest, Success) {
+  PrivetConfirmApiCallFlow confirm_flow(request_context_.get(),
+                                        &token_service_,
+                                        GURL("http://SoMeUrL.com"),
+                                        callback_.callback());
+
+  confirm_flow.OnGetTokenSuccess(NULL, "SomeToken", base::Time());
+  net::TestURLFetcher* fetcher = fetcher_factory_.GetFetcherByID(0);
+
+  EXPECT_EQ(GURL("http://SoMeUrL.com"), fetcher->GetOriginalURL());
+
+  net::HttpRequestHeaders headers;
+  fetcher->GetExtraRequestHeaders(&headers);
+  std::string oauth_header;
+  std::string proxy;
+  EXPECT_TRUE(headers.GetHeader("Authorization", &oauth_header));
+  EXPECT_EQ("Bearer SomeToken", oauth_header);
+  EXPECT_TRUE(headers.GetHeader("X-Cloudprint-Proxy", &proxy));
+  EXPECT_EQ("Chrome", proxy);
+
+  fetcher->SetResponseString(kSampleConfirmResponse);
+  fetcher->set_status(net::URLRequestStatus(net::URLRequestStatus::SUCCESS,
+                                            net::OK));
+  fetcher->set_response_code(200);
+
+  EXPECT_CALL(callback_, ConfirmCallback(PrivetConfirmApiCallFlow::SUCCESS));
+
+  fetcher->delegate()->OnURLFetchComplete(fetcher);
+}
+
+TEST_F(PrivetConfirmApiFlowTest, BadToken) {
+  PrivetConfirmApiCallFlow confirm_flow(request_context_.get(),
+                                        &token_service_,
+                                        GURL("http://SoMeUrL.com"),
+                                        callback_.callback());
+
+  EXPECT_CALL(callback_,
+              ConfirmCallback(PrivetConfirmApiCallFlow::ERROR_TOKEN));
+  confirm_flow.OnGetTokenFailure(NULL, GoogleServiceAuthError(
+      GoogleServiceAuthError::USER_NOT_SIGNED_UP));
+}
+
+TEST_F(PrivetConfirmApiFlowTest, ServerFailure) {
+  PrivetConfirmApiCallFlow confirm_flow(request_context_.get(),
+                                        &token_service_,
+                                        GURL("http://SoMeUrL.com"),
+                                        callback_.callback());
+
+  confirm_flow.OnGetTokenSuccess(NULL, "SomeToken", base::Time());
+  net::TestURLFetcher* fetcher = fetcher_factory_.GetFetcherByID(0);
+
+  EXPECT_EQ(GURL("http://SoMeUrL.com"), fetcher->GetOriginalURL());
+
+  fetcher->SetResponseString(kFailedConfirmResponse);
+  fetcher->set_status(net::URLRequestStatus(net::URLRequestStatus::SUCCESS,
+                                            net::OK));
+  fetcher->set_response_code(200);
+
+  EXPECT_CALL(callback_,
+              ConfirmCallback(PrivetConfirmApiCallFlow::ERROR_FROM_SERVER));
+
+  fetcher->delegate()->OnURLFetchComplete(fetcher);
+}
+
+TEST_F(PrivetConfirmApiFlowTest, BadJson) {
+  PrivetConfirmApiCallFlow confirm_flow(request_context_.get(),
+                                        &token_service_,
+                                        GURL("http://SoMeUrL.com"),
+                                        callback_.callback());
+
+  confirm_flow.OnGetTokenSuccess(NULL, "SomeToken", base::Time());
+  net::TestURLFetcher* fetcher = fetcher_factory_.GetFetcherByID(0);
+
+  EXPECT_EQ(GURL("http://SoMeUrL.com"), fetcher->GetOriginalURL());
+
+  fetcher->SetResponseString(kFailedConfirmResponseBadJson);
+  fetcher->set_status(net::URLRequestStatus(
+      net::URLRequestStatus::SUCCESS,
+      net::OK));
+  fetcher->set_response_code(200);
+
+  EXPECT_CALL(callback_, ConfirmCallback
+              (PrivetConfirmApiCallFlow::ERROR_MALFORMED_RESPONSE));
+
+  fetcher->delegate()->OnURLFetchComplete(fetcher);
+}
+
+}  // namespace
+
+}  // namespace local_discovery