Index: components/autofill/content/browser/wallet/mock_wallet_client.h |
diff --git a/components/autofill/content/browser/wallet/mock_wallet_client.h b/components/autofill/content/browser/wallet/mock_wallet_client.h |
index b6751c030e33759f553bd1f0c71c1ba27d2eaec8..629a9df03859adef5fae316460fa00c83a7fbeb5 100644 |
--- a/components/autofill/content/browser/wallet/mock_wallet_client.h |
+++ b/components/autofill/content/browser/wallet/mock_wallet_client.h |
@@ -11,6 +11,7 @@ |
#include "components/autofill/content/browser/wallet/instrument.h" |
#include "components/autofill/content/browser/wallet/wallet_address.h" |
#include "components/autofill/content/browser/wallet/wallet_client.h" |
+#include "components/autofill/content/browser/wallet/wallet_items.h" |
#include "testing/gmock/include/gmock/gmock.h" |
namespace autofill { |
@@ -40,11 +41,17 @@ class MockWalletClient : public WalletClient { |
// Methods with scoped_ptrs can't be mocked but by using the implementation |
// below the same effect can be achieved. |
- virtual void SaveToWallet(scoped_ptr<wallet::Instrument> instrument, |
- scoped_ptr<wallet::Address> address) OVERRIDE; |
- |
- MOCK_METHOD2(SaveToWalletMock, |
- void(Instrument* instrument, Address* address)); |
+ virtual void SaveToWallet( |
+ scoped_ptr<Instrument> instrument, |
+ scoped_ptr<Address> address, |
+ const WalletItems::MaskedInstrument* reference_instrument, |
+ const Address* reference_address) OVERRIDE; |
+ |
+ MOCK_METHOD4(SaveToWalletMock, |
+ void(Instrument* instrument, |
+ Address* address, |
+ const WalletItems::MaskedInstrument* reference_instrument, |
+ const Address* reference_address)); |
private: |
DISALLOW_COPY_AND_ASSIGN(MockWalletClient); |