Autofill Webdata Split TRY ONLY

chrome/browser/webdata/web_data_service_test_util.cc

Index: chrome/browser/webdata/web_data_service_test_util.cc
diff --git a/chrome/browser/webdata/web_data_service_test_util.cc b/chrome/browser/webdata/web_data_service_test_util.cc
index 521a9b891d68aeae25168ba8660e3aaa26ee5e32..e54b38425fb6727e1285624bd777f7f9e18295d3 100644
--- a/chrome/browser/webdata/web_data_service_test_util.cc
+++ b/chrome/browser/webdata/web_data_service_test_util.cc
@@ -13,14 +13,24 @@ MockWebDataServiceWrapperBase::~MockWebDataServiceWrapperBase() {
 void MockWebDataServiceWrapperBase::Shutdown() {
 }
 
+// TODO(caitkp): This won't scale well. As we get more WebData subclasses, we
+// will probably need a better way to create these mocks rather than passing
+// all the webdatas in.
 MockWebDataServiceWrapper::MockWebDataServiceWrapper(
-    scoped_refptr<WebDataService> fake_service)
-    : fake_web_data_service_(fake_service) {
+    scoped_refptr<WebDataService> fake_service,
+    scoped_refptr<AutofillWebDataService> fake_autofill)
+    : fake_autofill_web_data_(fake_autofill),
+      fake_web_data_(fake_service) {
 }
 
 MockWebDataServiceWrapper::~MockWebDataServiceWrapper() {
 }
 
+scoped_refptr<AutofillWebDataService>
+    MockWebDataServiceWrapper::GetAutofillWebData() {
+  return fake_autofill_web_data_;
+}
+
 scoped_refptr<WebDataService> MockWebDataServiceWrapper::GetWebData() {
-  return fake_web_data_service_;
+  return fake_web_data_;
 }