Home | History | Annotate | Download | only in gaia 
      1 // Copyright (c) 2011 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 // A collection of classes that are useful when testing things that use a
      6 // GaiaAuthFetcher.
      7 
      8 #ifndef CHROME_COMMON_NET_GAIA_GAIA_AUTH_FETCHER_UNITTEST_H_
      9 #define CHROME_COMMON_NET_GAIA_GAIA_AUTH_FETCHER_UNITTEST_H_
     10 #pragma once
     11 
     12 #include <string>
     13 
     14 #include "chrome/common/net/gaia/gaia_auth_fetcher.h"
     15 #include "chrome/common/net/url_fetcher.h"
     16 #include "chrome/common/net/http_return.h"
     17 #include "net/url_request/url_request_status.h"
     18 
     19 // Responds as though ClientLogin returned from the server.
     20 class MockFetcher : public URLFetcher {
     21  public:
     22   MockFetcher(bool success,
     23               const GURL& url,
     24               const std::string& results,
     25               URLFetcher::RequestType request_type,
     26               URLFetcher::Delegate* d);
     27 
     28   virtual ~MockFetcher();
     29 
     30   virtual void Start();
     31 
     32  private:
     33   bool success_;
     34   GURL url_;
     35   std::string results_;
     36   DISALLOW_COPY_AND_ASSIGN(MockFetcher);
     37 };
     38 
     39 template<typename T>
     40 class MockFactory : public URLFetcher::Factory {
     41  public:
     42   MockFactory()
     43       : success_(true) {}
     44   ~MockFactory() {}
     45   URLFetcher* CreateURLFetcher(int id,
     46                                const GURL& url,
     47                                URLFetcher::RequestType request_type,
     48                                URLFetcher::Delegate* d) {
     49     return new T(success_, url, results_, request_type, d);
     50   }
     51   void set_success(bool success) {
     52     success_ = success;
     53   }
     54   void set_results(const std::string& results) {
     55     results_ = results;
     56   }
     57  private:
     58   bool success_;
     59   std::string results_;
     60   DISALLOW_COPY_AND_ASSIGN(MockFactory);
     61 };
     62 
     63 #endif  // CHROME_COMMON_NET_GAIA_GAIA_AUTH_FETCHER_UNITTEST_H_
     64