Add OAuth2 revocation fetcher that is right now used for notifications.

chrome/common/net/gaia/oauth2_revocation_fetcher_unittest.cc

+// 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 OAuth2RevocationFetcher.
+
+#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_revocation_consumer.h"
+#include "chrome/common/net/gaia/oauth2_revocation_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 {
+
+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 MockOAuth2RevocationConsumer : public OAuth2RevocationConsumer {
+ public:
+  MockOAuth2RevocationConsumer() {}
+  ~MockOAuth2RevocationConsumer() {}
+
+  MOCK_METHOD1(OnRevocationSuccess, void());
+  MOCK_METHOD1(OnRevocationFailure,
+               void(const GoogleServiceAuthError& error));
+};
+
+class OAuth2RevocationFetcherTest : public testing::Test {
+ public:
+  OAuth2RevocationFetcherTest()
+    : ui_thread_(BrowserThread::UI, &message_loop_),
+      fetcher_(&consumer_, profile_.GetRequestContext()) {
+  }
+
+  virtual ~OAuth2RevocationFetcherTest() { }
+
+  virtual TestURLFetcher* SetupRevocation(
+      bool fetch_succeeds, int response_code) {
+    GURL url = OAuth2RevocationFetcher::MakeRevocationUrl();
+    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);
+
+    EXPECT_CALL(factory_, CreateURLFetcher(_, url, _, _))
+        .WillOnce(Return(url_fetcher));
+    return url_fetcher;
+  }
+
+ protected:
+  MessageLoop message_loop_;
+  content::TestBrowserThread ui_thread_;
+  MockUrlFetcherFactory factory_;
+  MockOAuth2RevocationConsumer consumer_;
+  TestingProfile profile_;
+  OAuth2RevocationFetcher fetcher_;
+};
+
+TEST_F(OAuth2RevocationFetcherTest, RequestFailure) {
+  TestURLFetcher* url_fetcher = SetupRevocation(false, 0);
+  EXPECT_CALL(consumer_, OnRevocationFailure(_)).Times(1);
+  fetcher_.Start("access_token", "client_id", "origin");
+  fetcher_.OnURLFetchComplete(url_fetcher);
+}
+
+TEST_F(OAuth2RevocationFetcherTest, ResponseCodeFailure) {
+  TestURLFetcher* url_fetcher = SetupRevocation(true, RC_FORBIDDEN);
+  EXPECT_CALL(consumer_, OnRevocationFailure(_)).Times(1);
+  fetcher_.Start("access_token", "client_id", "origin");
+  fetcher_.OnURLFetchComplete(url_fetcher);
+}
+
+TEST_F(OAuth2RevocationFetcherTest, Success) {
+  TestURLFetcher* url_fetcher = SetupRevocation(true, RC_REQUEST_OK);
+  EXPECT_CALL(consumer_, OnRevocationSuccess()).Times(1);
+  fetcher_.Start("access_token", "client_id", "origin");
+  fetcher_.OnURLFetchComplete(url_fetcher);
+}