chrome/browser/local_discovery/gcd_api_flow_unittest.cc

   1 // Copyright 2014 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 "chrome/browser/local_discovery/gcd_api_flow.h"
   6
   7 #include "base/bind.h"
   8 #include "base/message_loop/message_loop.h"
   9 #include "base/thread_task_runner_handle.h"
  10 #include "base/values.h"
  11 #include "chrome/browser/local_discovery/gcd_api_flow_impl.h"
  12 #include "content/public/test/test_browser_thread.h"
  13 #include "google_apis/gaia/fake_oauth2_token_service.h"
  14 #include "google_apis/gaia/google_service_auth_error.h"
  15 #include "net/base/host_port_pair.h"
  16 #include "net/base/net_errors.h"
  17 #include "net/http/http_request_headers.h"
  18 #include "net/url_request/test_url_fetcher_factory.h"
  19 #include "net/url_request/url_request_test_util.h"
  20 #include "testing/gmock/include/gmock/gmock.h"
  21 #include "testing/gtest/include/gtest/gtest.h"
  22
  23 using testing::_;
  24 using testing::Return;
  25
  26 namespace local_discovery {
  27
  28 namespace {
  29
  30 const char kSampleConfirmResponse[] = "{}";
  31
  32 const char kFailedConfirmResponseBadJson[] = "[]";
  33
  34 const char kAccountId[] = "account_id";
  35
  36 class MockDelegate : public CloudPrintApiFlowRequest {
  37  public:
  38   MOCK_METHOD1(OnGCDAPIFlowError, void(GCDApiFlow::Status));
  39   MOCK_METHOD1(OnGCDAPIFlowComplete, void(const base::DictionaryValue&));
  40
  41   MOCK_METHOD0(GetURL, GURL());
  42 };
  43
  44 class GCDApiFlowTest : public testing::Test {
  45  public:
  46   GCDApiFlowTest()
  47       : ui_thread_(content::BrowserThread::UI, &loop_),
  48         request_context_(new net::TestURLRequestContextGetter(
  49             base::ThreadTaskRunnerHandle::Get())),
  50         account_id_(kAccountId) {}
  51
  52   ~GCDApiFlowTest() override {}
  53
  54  protected:
  55   void SetUp() override {
  56     token_service_.GetFakeOAuth2TokenServiceDelegate()->set_request_context(
  57         request_context_.get());
  58     token_service_.AddAccount(account_id_);
  59     ui_thread_.Stop();  // HACK: Fake being on the UI thread
  60
  61     scoped_ptr<MockDelegate> delegate(new MockDelegate);
  62     mock_delegate_ = delegate.get();
  63     EXPECT_CALL(*mock_delegate_, GetURL())
  64         .WillRepeatedly(Return(
  65             GURL("https://www.google.com/cloudprint/confirm?token=SomeToken")));
  66     gcd_flow_.reset(new GCDApiFlowImpl(
  67         request_context_.get(), &token_service_, account_id_));
  68     gcd_flow_->Start(delegate.Pass());
  69   }
  70   base::MessageLoopForUI loop_;
  71   content::TestBrowserThread ui_thread_;
  72   scoped_refptr<net::TestURLRequestContextGetter> request_context_;
  73   net::TestURLFetcherFactory fetcher_factory_;
  74   FakeOAuth2TokenService token_service_;
  75   std::string account_id_;
  76   scoped_ptr<GCDApiFlowImpl> gcd_flow_;
  77   MockDelegate* mock_delegate_;
  78 };
  79
  80 TEST_F(GCDApiFlowTest, SuccessOAuth2) {
  81   gcd_flow_->OnGetTokenSuccess(NULL, "SomeToken", base::Time());
  82   net::TestURLFetcher* fetcher = fetcher_factory_.GetFetcherByID(0);
  83
  84   EXPECT_EQ(GURL("https://www.google.com/cloudprint/confirm?token=SomeToken"),
  85             fetcher->GetOriginalURL());
  86
  87   net::HttpRequestHeaders headers;
  88   fetcher->GetExtraRequestHeaders(&headers);
  89   std::string oauth_header;
  90   std::string proxy;
  91   EXPECT_TRUE(headers.GetHeader("Authorization", &oauth_header));
  92   EXPECT_EQ("Bearer SomeToken", oauth_header);
  93   EXPECT_TRUE(headers.GetHeader("X-Cloudprint-Proxy", &proxy));
  94   EXPECT_EQ("Chrome", proxy);
  95
  96   fetcher->SetResponseString(kSampleConfirmResponse);
  97   fetcher->set_status(
  98       net::URLRequestStatus(net::URLRequestStatus::SUCCESS, net::OK));
  99   fetcher->set_response_code(200);
 100
 101   EXPECT_CALL(*mock_delegate_, OnGCDAPIFlowComplete(_));
 102
 103   fetcher->delegate()->OnURLFetchComplete(fetcher);
 104 }
 105
 106 TEST_F(GCDApiFlowTest, BadToken) {
 107   EXPECT_CALL(*mock_delegate_, OnGCDAPIFlowError(GCDApiFlow::ERROR_TOKEN));
 108   gcd_flow_->OnGetTokenFailure(
 109       NULL, GoogleServiceAuthError(GoogleServiceAuthError::USER_NOT_SIGNED_UP));
 110 }
 111
 112 TEST_F(GCDApiFlowTest, BadJson) {
 113   gcd_flow_->OnGetTokenSuccess(NULL, "SomeToken", base::Time());
 114   net::TestURLFetcher* fetcher = fetcher_factory_.GetFetcherByID(0);
 115
 116   EXPECT_EQ(GURL("https://www.google.com/cloudprint/confirm?token=SomeToken"),
 117             fetcher->GetOriginalURL());
 118
 119   fetcher->SetResponseString(kFailedConfirmResponseBadJson);
 120   fetcher->set_status(
 121       net::URLRequestStatus(net::URLRequestStatus::SUCCESS, net::OK));
 122   fetcher->set_response_code(200);
 123
 124   EXPECT_CALL(*mock_delegate_,
 125               OnGCDAPIFlowError(GCDApiFlow::ERROR_MALFORMED_RESPONSE));
 126
 127   fetcher->delegate()->OnURLFetchComplete(fetcher);
 128 }
 129
 130 }  // namespace
 131
 132 }  // namespace local_discovery