components/webdata_services/web_data_service_test_util.h

   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 #ifndef COMPONENTS_WEBDATA_SERVICES_WEB_DATA_SERVICE_TEST_UTIL_H__
   6 #define COMPONENTS_WEBDATA_SERVICES_WEB_DATA_SERVICE_TEST_UTIL_H__
   7
   8 #include "base/macros.h"
   9 #include "components/signin/core/browser/webdata/token_web_data.h"
  10 #include "components/webdata_services/web_data_service_wrapper.h"
  11
  12 // Base class for mocks of WebDataService, that does nothing in
  13 // Shutdown().
  14 class MockWebDataServiceWrapperBase : public WebDataServiceWrapper {
  15  public:
  16   MockWebDataServiceWrapperBase();
  17   ~MockWebDataServiceWrapperBase() override;
  18
  19   void Shutdown() override;
  20
  21  private:
  22   DISALLOW_COPY_AND_ASSIGN(MockWebDataServiceWrapperBase);
  23 };
  24
  25 // Pass your fake WebDataService in the constructor and this will
  26 // serve it up via GetWebData().
  27 class MockWebDataServiceWrapper : public MockWebDataServiceWrapperBase {
  28  public:
  29   MockWebDataServiceWrapper(
  30       scoped_refptr<autofill::AutofillWebDataService> fake_autofill,
  31       scoped_refptr<TokenWebData> fake_token);
  32
  33   ~MockWebDataServiceWrapper() override;
  34
  35   scoped_refptr<autofill::AutofillWebDataService> GetProfileAutofillWebData()
  36       override;
  37
  38   scoped_refptr<autofill::AutofillWebDataService> GetAccountAutofillWebData()
  39       override;
  40
  41   scoped_refptr<TokenWebData> GetTokenWebData() override;
  42
  43  protected:
  44   scoped_refptr<autofill::AutofillWebDataService> fake_autofill_web_data_;
  45   scoped_refptr<TokenWebData> fake_token_web_data_;
  46
  47  private:
  48   DISALLOW_COPY_AND_ASSIGN(MockWebDataServiceWrapper);
  49 };
  50
  51 #endif  // COMPONENTS_WEBDATA_SERVICES_WEB_DATA_SERVICE_TEST_UTIL_H__