Index: chrome/browser/webdata/web_data_service_test_util.h |
diff --git a/chrome/browser/webdata/web_data_service_test_util.h b/chrome/browser/webdata/web_data_service_test_util.h |
index 06e4cc81542f0d0a23797782636a75ba999b87f7..8fa9a854a8326515abfa43e44fa68081d7f7ac3a 100644 |
--- a/chrome/browser/webdata/web_data_service_test_util.h |
+++ b/chrome/browser/webdata/web_data_service_test_util.h |
@@ -8,6 +8,7 @@ |
#include "base/basictypes.h" |
#include "base/message_loop.h" |
#include "chrome/browser/webdata/web_data_service.h" |
+#include "chrome/browser/webdata/web_data_service_factory.h" |
#include "content/public/browser/browser_thread.h" |
template <class T> |
@@ -37,4 +38,33 @@ class AutofillWebDataServiceConsumer: public WebDataServiceConsumer { |
DISALLOW_COPY_AND_ASSIGN(AutofillWebDataServiceConsumer); |
}; |
+// Base class for mocks of WebDataService, that does nothing in |
+// Shutdown(). |
+class MockWebDataServiceWrapperBase : public WebDataServiceWrapper { |
+ public: |
+ MockWebDataServiceWrapperBase(); |
+ virtual ~MockWebDataServiceWrapperBase(); |
+ |
+ virtual void Shutdown() OVERRIDE; |
+ |
+ private: |
+ DISALLOW_COPY_AND_ASSIGN(MockWebDataServiceWrapperBase); |
+}; |
+ |
+// Pass your fake WebDataService in the constructor and this will |
+// serve it up via GetWebData(). |
+class MockWebDataServiceWrapper : public MockWebDataServiceWrapperBase { |
+ public: |
+ MockWebDataServiceWrapper(scoped_refptr<WebDataService> fake_service); |
+ virtual ~MockWebDataServiceWrapper(); |
+ |
+ virtual scoped_refptr<WebDataService> GetWebData() OVERRIDE; |
+ |
+ protected: |
+ scoped_refptr<WebDataService> fake_web_data_service_; |
+ |
+ private: |
+ DISALLOW_COPY_AND_ASSIGN(MockWebDataServiceWrapper); |
+}; |
+ |
#endif // CHROME_BROWSER_WEBDATA_WEB_DATA_SERVICE_TEST_UTIL_H__ |