Move API flows for cloud print into common class CloudPrintBaseApiFlow

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"
@@ skipping 35 matching lines @@
   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));
+  MOCK_METHOD1(ConfirmCallback, void(CloudPrintBaseApiFlow::Status));
 
   PrivetConfirmApiCallFlow::ResponseCallback callback() {
     return base::Bind(&MockableConfirmCallback::ConfirmCallback,
                       base::Unretained(this));
   }
 };
 
 class PrivetConfirmApiFlowTest : public testing::Test {
  public:
   PrivetConfirmApiFlowTest()
@@ skipping 15 matching lines @@
   net::TestURLFetcherFactory fetcher_factory_;
   TestOAuth2TokenService token_service_;
   MockableConfirmCallback callback_;
 };
 
 TEST_F(PrivetConfirmApiFlowTest, SuccessOAuth2) {
   PrivetConfirmApiCallFlow confirm_flow(request_context_.get(),
                                         &token_service_,
                                         GURL("http://SoMeUrL.com"),
                                         callback_.callback());
+  CloudPrintBaseApiFlow* cloudprint_flow =
+      confirm_flow.GetBaseApiFlowForTests();
 
   confirm_flow.Start();
 
-  confirm_flow.OnGetTokenSuccess(NULL, "SomeToken", base::Time());
+  cloudprint_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));
+  EXPECT_CALL(callback_, ConfirmCallback(CloudPrintBaseApiFlow::SUCCESS));
 
   fetcher->delegate()->OnURLFetchComplete(fetcher);
 }
 
 TEST_F(PrivetConfirmApiFlowTest, SuccessCookies) {
   PrivetConfirmApiCallFlow confirm_flow(request_context_.get(),
                                         1,
                                         "SomeToken",
                                         GURL("http://SoMeUrL.com?token=tkn"),
                                         callback_.callback());
 
   confirm_flow.Start();
 
   net::TestURLFetcher* fetcher = fetcher_factory_.GetFetcherByID(0);
 
   EXPECT_EQ(GURL("http://SoMeUrL.com?token=tkn&xsrf=SomeToken&user=1"),
             fetcher->GetOriginalURL());
 
   net::HttpRequestHeaders headers;
   fetcher->GetExtraRequestHeaders(&headers);
   std::string proxy;
   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));
+  EXPECT_CALL(callback_, ConfirmCallback(CloudPrintBaseApiFlow::SUCCESS));
 
   fetcher->delegate()->OnURLFetchComplete(fetcher);
 }
 
 TEST_F(PrivetConfirmApiFlowTest, BadToken) {
   PrivetConfirmApiCallFlow confirm_flow(request_context_.get(),
                                         &token_service_,
                                         GURL("http://SoMeUrL.com"),
                                         callback_.callback());
 
   confirm_flow.Start();
 
+  CloudPrintBaseApiFlow* cloudprint_flow =
+      confirm_flow.GetBaseApiFlowForTests();
+
   EXPECT_CALL(callback_,
-              ConfirmCallback(PrivetConfirmApiCallFlow::ERROR_TOKEN));
-  confirm_flow.OnGetTokenFailure(NULL, GoogleServiceAuthError(
+              ConfirmCallback(CloudPrintBaseApiFlow::ERROR_TOKEN));
+  cloudprint_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.Start();
 
-  confirm_flow.OnGetTokenSuccess(NULL, "SomeToken", base::Time());
+  CloudPrintBaseApiFlow* cloudprint_flow =
+      confirm_flow.GetBaseApiFlowForTests();
+
+  cloudprint_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));
+              ConfirmCallback(CloudPrintBaseApiFlow::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.Start();
 
-  confirm_flow.OnGetTokenSuccess(NULL, "SomeToken", base::Time());
+  CloudPrintBaseApiFlow* cloudprint_flow =
+      confirm_flow.GetBaseApiFlowForTests();
+
+  cloudprint_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));
+              (CloudPrintBaseApiFlow::ERROR_MALFORMED_RESPONSE));
 
   fetcher->delegate()->OnURLFetchComplete(fetcher);
 }
 
 }  // namespace
 
 }  // namespace local_discovery