| Index: chrome/common/net/gaia/oauth2_access_token_fetcher_unittest.cc
|
| ===================================================================
|
| --- chrome/common/net/gaia/oauth2_access_token_fetcher_unittest.cc (revision 0)
|
| +++ chrome/common/net/gaia/oauth2_access_token_fetcher_unittest.cc (revision 0)
|
| @@ -0,0 +1,180 @@
|
| +// Copyright (c) 2011 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.
|
| +//
|
| +// A complete set of unit tests for OAuth2AccessTokenFetcher.
|
| +
|
| +#include <string>
|
| +
|
| +#include "base/memory/scoped_ptr.h"
|
| +#include "base/message_loop.h"
|
| +#include "chrome/common/net/gaia/gaia_urls.h"
|
| +#include "chrome/common/net/gaia/google_service_auth_error.h"
|
| +#include "chrome/common/net/http_return.h"
|
| +#include "chrome/common/net/gaia/oauth2_access_token_consumer.h"
|
| +#include "chrome/common/net/gaia/oauth2_access_token_fetcher.h"
|
| +#include "chrome/test/base/testing_profile.h"
|
| +#include "content/public/common/url_fetcher.h"
|
| +#include "content/public/common/url_fetcher_delegate.h"
|
| +#include "content/public/common/url_fetcher_factory.h"
|
| +#include "content/test/test_browser_thread.h"
|
| +#include "content/test/test_url_fetcher_factory.h"
|
| +#include "googleurl/src/gurl.h"
|
| +#include "net/url_request/url_request.h"
|
| +#include "net/url_request/url_request_status.h"
|
| +#include "testing/gmock/include/gmock/gmock.h"
|
| +#include "testing/gtest/include/gtest/gtest.h"
|
| +
|
| +using content::BrowserThread;
|
| +using content::URLFetcher;
|
| +using content::URLFetcherDelegate;
|
| +using content::URLFetcherFactory;
|
| +using net::ResponseCookies;
|
| +using net::URLRequestStatus;
|
| +using testing::_;
|
| +using testing::Return;
|
| +
|
| +namespace {
|
| +static const char kValidTokenResponse[] =
|
| + "{"
|
| + " \"access_token\": \"at1\","
|
| + " \"expires_in\": 3600,"
|
| + " \"token_type\": \"Bearer\""
|
| + "}";
|
| +static const char kTokenResponseNoAccessToken[] =
|
| + "{"
|
| + " \"expires_in\": 3600,"
|
| + " \"token_type\": \"Bearer\""
|
| + "}";
|
| +}
|
| +
|
| +class MockUrlFetcherFactory : public ScopedURLFetcherFactory,
|
| + public URLFetcherFactory {
|
| +public:
|
| + MockUrlFetcherFactory()
|
| + : ScopedURLFetcherFactory(ALLOW_THIS_IN_INITIALIZER_LIST(this)) {
|
| + }
|
| + virtual ~MockUrlFetcherFactory() {}
|
| +
|
| + MOCK_METHOD4(
|
| + CreateURLFetcher,
|
| + URLFetcher* (int id,
|
| + const GURL& url,
|
| + URLFetcher::RequestType request_type,
|
| + URLFetcherDelegate* d));
|
| +};
|
| +
|
| +class MockOAuth2AccessTokenConsumer : public OAuth2AccessTokenConsumer {
|
| + public:
|
| + MockOAuth2AccessTokenConsumer() {}
|
| + ~MockOAuth2AccessTokenConsumer() {}
|
| +
|
| + MOCK_METHOD1(OnGetTokenSuccess, void(const std::string& access_token));
|
| + MOCK_METHOD1(OnGetTokenFailure,
|
| + void(const GoogleServiceAuthError& error));
|
| +};
|
| +
|
| +class OAuth2AccessTokenFetcherTest : public testing::Test {
|
| + public:
|
| + OAuth2AccessTokenFetcherTest()
|
| + : ui_thread_(BrowserThread::UI, &message_loop_),
|
| + fetcher_(&consumer_, profile_.GetRequestContext(), "test") {
|
| + }
|
| +
|
| + virtual ~OAuth2AccessTokenFetcherTest() { }
|
| +
|
| + virtual TestURLFetcher* SetupGetAccessToken(
|
| + bool fetch_succeeds, int response_code, const std::string& body) {
|
| + GURL url(GaiaUrls::GetInstance()->oauth2_token_url());
|
| + TestURLFetcher* url_fetcher = new TestURLFetcher(0, url, &fetcher_);
|
| + URLRequestStatus::Status status =
|
| + fetch_succeeds ? URLRequestStatus::SUCCESS : URLRequestStatus::FAILED;
|
| + url_fetcher->set_status(URLRequestStatus(status, 0));
|
| +
|
| + if (response_code != 0)
|
| + url_fetcher->set_response_code(response_code);
|
| +
|
| + if (!body.empty())
|
| + url_fetcher->SetResponseString(body);
|
| +
|
| + EXPECT_CALL(factory_, CreateURLFetcher(_, url, _, _))
|
| + .WillOnce(Return(url_fetcher));
|
| + return url_fetcher;
|
| + }
|
| +
|
| + protected:
|
| + MessageLoop message_loop_;
|
| + content::TestBrowserThread ui_thread_;
|
| + MockUrlFetcherFactory factory_;
|
| + MockOAuth2AccessTokenConsumer consumer_;
|
| + TestingProfile profile_;
|
| + OAuth2AccessTokenFetcher fetcher_;
|
| +};
|
| +
|
| +TEST_F(OAuth2AccessTokenFetcherTest, GetAccessTokenRequestFailure) {
|
| + TestURLFetcher* url_fetcher = SetupGetAccessToken(false, 0, "");
|
| + EXPECT_CALL(consumer_, OnGetTokenFailure(_)).Times(1);
|
| + fetcher_.Start("refresh_token");
|
| + fetcher_.OnURLFetchComplete(url_fetcher);
|
| +}
|
| +
|
| +TEST_F(OAuth2AccessTokenFetcherTest, GetAccessTokenResponseCodeFailure) {
|
| + TestURLFetcher* url_fetcher = SetupGetAccessToken(true, RC_FORBIDDEN, "");
|
| + EXPECT_CALL(consumer_, OnGetTokenFailure(_)).Times(1);
|
| + fetcher_.Start("refresh_token");
|
| + fetcher_.OnURLFetchComplete(url_fetcher);
|
| +}
|
| +
|
| +TEST_F(OAuth2AccessTokenFetcherTest, GetAccessTokenResponseNoAccessToken) {
|
| + TestURLFetcher* url_fetcher = SetupGetAccessToken(
|
| + true, RC_REQUEST_OK, kTokenResponseNoAccessToken);
|
| + EXPECT_CALL(consumer_, OnGetTokenFailure(_)).Times(1);
|
| + fetcher_.Start("refresh_token");
|
| + fetcher_.OnURLFetchComplete(url_fetcher);
|
| +}
|
| +
|
| +TEST_F(OAuth2AccessTokenFetcherTest, Success) {
|
| + TestURLFetcher* url_fetcher = SetupGetAccessToken(
|
| + true, RC_REQUEST_OK, kValidTokenResponse);
|
| + EXPECT_CALL(consumer_, OnGetTokenSuccess("at1")).Times(1);
|
| + fetcher_.Start("refresh_token");
|
| + fetcher_.OnURLFetchComplete(url_fetcher);
|
| +}
|
| +
|
| +TEST_F(OAuth2AccessTokenFetcherTest, ParseGetAccessTokenResponse) {
|
| + { // No body.
|
| + TestURLFetcher url_fetcher(0, GURL("www.google.com"), NULL);
|
| +
|
| + std::string at;
|
| + EXPECT_FALSE(OAuth2AccessTokenFetcher::ParseGetAccessTokenResponse(
|
| + &url_fetcher, &at));
|
| + EXPECT_TRUE(at.empty());
|
| + }
|
| + { // Bad json.
|
| + TestURLFetcher url_fetcher(0, GURL("www.google.com"), NULL);
|
| + url_fetcher.SetResponseString("foo");
|
| +
|
| + std::string at;
|
| + EXPECT_FALSE(OAuth2AccessTokenFetcher::ParseGetAccessTokenResponse(
|
| + &url_fetcher, &at));
|
| + EXPECT_TRUE(at.empty());
|
| + }
|
| + { // Valid json: access token missing.
|
| + TestURLFetcher url_fetcher(0, GURL("www.google.com"), NULL);
|
| + url_fetcher.SetResponseString(kTokenResponseNoAccessToken);
|
| +
|
| + std::string at;
|
| + EXPECT_FALSE(OAuth2AccessTokenFetcher::ParseGetAccessTokenResponse(
|
| + &url_fetcher, &at));
|
| + EXPECT_TRUE(at.empty());
|
| + }
|
| + { // Valid json: all good.
|
| + TestURLFetcher url_fetcher(0, GURL("www.google.com"), NULL);
|
| + url_fetcher.SetResponseString(kValidTokenResponse);
|
| +
|
| + std::string at;
|
| + EXPECT_TRUE(OAuth2AccessTokenFetcher::ParseGetAccessTokenResponse(
|
| + &url_fetcher, &at));
|
| + EXPECT_EQ("at1", at);
|
| + }
|
| +}
|
|
|