src/components/autofill/content/browser/wallet/mock_wallet_client.h

   1 // Copyright 2013 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 #ifndef COMPONENTS_AUTOFILL_CONTENT_BROWSER_WALLET_MOCK_WALLET_CLIENT_H_
   6 #define COMPONENTS_AUTOFILL_CONTENT_BROWSER_WALLET_MOCK_WALLET_CLIENT_H_
   7
   8 #include "base/basictypes.h"
   9 #include "base/compiler_specific.h"
  10 #include "base/memory/scoped_ptr.h"
  11 #include "components/autofill/content/browser/wallet/instrument.h"
  12 #include "components/autofill/content/browser/wallet/wallet_address.h"
  13 #include "components/autofill/content/browser/wallet/wallet_client.h"
  14 #include "components/autofill/content/browser/wallet/wallet_items.h"
  15 #include "testing/gmock/include/gmock/gmock.h"
  16
  17 namespace autofill {
  18 namespace wallet {
  19
  20 // A mock version of WalletClient that never issues real requests, just records
  21 // mock calls to each entry point.
  22 class MockWalletClient : public WalletClient {
  23  public:
  24   MockWalletClient(net::URLRequestContextGetter* context,
  25                    WalletClientDelegate* delegate,
  26                    const GURL& source_url);
  27   virtual ~MockWalletClient();
  28
  29   MOCK_METHOD2(GetWalletItems,
  30       void(const base::string16&, const base::string16&));
  31
  32   MOCK_METHOD2(AcceptLegalDocuments,
  33       void(const std::vector<WalletItems::LegalDocument*>& documents,
  34            const std::string& google_transaction_id));
  35
  36   MOCK_METHOD2(AuthenticateInstrument,
  37       void(const std::string& instrument_id,
  38            const std::string& card_verification_number));
  39
  40   MOCK_METHOD1(GetFullWallet,
  41       void(const WalletClient::FullWalletRequest& request));
  42
  43   // Methods with scoped_ptrs can't be mocked but by using the implementation
  44   // below the same effect can be achieved.
  45   virtual void SaveToWallet(
  46       scoped_ptr<Instrument> instrument,
  47       scoped_ptr<Address> address,
  48       const WalletItems::MaskedInstrument* reference_instrument,
  49       const Address* reference_address) override;
  50
  51   MOCK_METHOD4(SaveToWalletMock,
  52       void(Instrument* instrument,
  53            Address* address,
  54            const WalletItems::MaskedInstrument* reference_instrument,
  55            const Address* reference_address));
  56
  57  private:
  58   DISALLOW_COPY_AND_ASSIGN(MockWalletClient);
  59 };
  60
  61 }  // namespace wallet
  62 }  // namespace autofill
  63
  64 #endif  // COMPONENTS_AUTOFILL_CONTENT_BROWSER_WALLET_MOCK_WALLET_CLIENT_H_