X-Git-Url: http://git.osdn.net/view?a=blobdiff_plain;f=tests%2Frobotests%2Fsrc%2Fcom%2Fandroid%2Fsettings%2Ftestutils%2FFakeFeatureFactory.java;h=a9c1764129a5dc85af494e429e70cb5f144aa0d8;hb=1e0f8ff5e549636e98b7fa0f12761be934f8f36b;hp=f702cdce11b4da2c28ae36e98f2bc682a0ab2b8f;hpb=ab4ccb2c4955b5704fc963be203cb75f1a54958d;p=android-x86%2Fpackages-apps-Settings.git

diff --git a/tests/robotests/src/com/android/settings/testutils/FakeFeatureFactory.java b/tests/robotests/src/com/android/settings/testutils/FakeFeatureFactory.java
index f702cdce11..a9c1764129 100644
--- a/tests/robotests/src/com/android/settings/testutils/FakeFeatureFactory.java
+++ b/tests/robotests/src/com/android/settings/testutils/FakeFeatureFactory.java
@@ -26,6 +26,7 @@ import com.android.settings.localepicker.LocaleFeatureProvider;
 import com.android.settings.overlay.FeatureFactory;
 import com.android.settings.overlay.SupportFeatureProvider;
 import com.android.settings.search2.SearchFeatureProvider;
+import com.android.settings.overlay.SurveyFeatureProvider;
 
 import static org.mockito.Matchers.anyString;
 import static org.mockito.Mockito.mock;
@@ -45,6 +46,7 @@ public class FakeFeatureFactory extends FeatureFactory {
     public final ApplicationFeatureProvider applicationFeatureProvider;
     public final EnterprisePrivacyFeatureProvider enterprisePrivacyFeatureProvider;
     public final SearchFeatureProvider searchFeatureProvider;
+    public final SurveyFeatureProvider surveyFeatureProvider;
 
     /**
      * Call this in {@code @Before} method of the test class to use fake factory.
@@ -75,6 +77,7 @@ public class FakeFeatureFactory extends FeatureFactory {
         applicationFeatureProvider = mock(ApplicationFeatureProvider.class);
         enterprisePrivacyFeatureProvider = mock(EnterprisePrivacyFeatureProvider.class);
         searchFeatureProvider = mock(SearchFeatureProvider.class);
+        surveyFeatureProvider = mock(SurveyFeatureProvider.class);
     }
 
     @Override
@@ -88,7 +91,7 @@ public class FakeFeatureFactory extends FeatureFactory {
     }
 
     @Override
-    public PowerUsageFeatureProvider getPowerUsageFeatureProvider() {
+    public PowerUsageFeatureProvider getPowerUsageFeatureProvider(Context context) {
         return powerUsageFeatureProvider;
     }
 
@@ -116,4 +119,9 @@ public class FakeFeatureFactory extends FeatureFactory {
     public SearchFeatureProvider getSearchFeatureProvider(Context context) {
         return searchFeatureProvider;
     }
+
+    @Override
+    public SurveyFeatureProvider getSurveyFeatureProvider(Context context) {
+        return surveyFeatureProvider;
+    }
 }