Remove more wrappers in favor of new Robolectric support

Bug: 76167422
Test: robotests
Change-Id: Ic71a181bee6f90f25f9fa2a748cf708887495394

src/com/android/settings/connecteddevice/usb/UsbBackend.java

@@ -25,7 +25,6 @@ import android.net.ConnectivityManager;
 import android.os.UserManager;
 import android.support.annotation.VisibleForTesting;
 
-import com.android.settings.wrapper.UsbManagerWrapper;
 import com.android.settings.wrapper.UserManagerWrapper;
 
 /**
@@ -44,7 +43,6 @@ public class UsbBackend {
     private final boolean mTetheringSupported;
 
     private UsbManager mUsbManager;
-    private UsbManagerWrapper mUsbManagerWrapper;
 
     @Nullable
     private UsbPort mPort;
@@ -52,19 +50,13 @@ public class UsbBackend {
     private UsbPortStatus mPortStatus;
 
     public UsbBackend(Context context) {
-        this(context, new UserManagerWrapper(UserManager.get(context)), null);
+        this(context, new UserManagerWrapper(UserManager.get(context)));
     }
 
     @VisibleForTesting
-    public UsbBackend(Context context, UserManagerWrapper userManagerWrapper,
+    public UsbBackend(Context context, UserManagerWrapper userManagerWrapper) {
-            UsbManagerWrapper usbManagerWrapper) {
         mUsbManager = context.getSystemService(UsbManager.class);
 
-        mUsbManagerWrapper = usbManagerWrapper;
-        if (mUsbManagerWrapper == null) {
-            mUsbManagerWrapper = new UsbManagerWrapper(mUsbManager);
-        }
-
         mFileTransferRestricted = userManagerWrapper.isUsbFileTransferRestricted();
         mFileTransferRestrictedBySystem = userManagerWrapper.isUsbFileTransferRestrictedBySystem();
         mTetheringRestricted = userManagerWrapper.isUsbTetheringRestricted();
@@ -79,7 +71,7 @@ public class UsbBackend {
     }
 
     public long getCurrentFunctions() {
-        return mUsbManagerWrapper.getCurrentFunctions();
+        return mUsbManager.getCurrentFunctions();
     }
 
     public void setCurrentFunctions(long functions) {

src/com/android/settings/display/BrightnessLevelPreferenceController.java

@@ -31,7 +31,6 @@ import android.support.v7.preference.PreferenceScreen;
 import android.util.Log;
 
 import com.android.settings.core.PreferenceControllerMixin;
-import com.android.settings.wrapper.PowerManagerWrapper;
 import com.android.settingslib.core.AbstractPreferenceController;
 import com.android.settingslib.core.lifecycle.Lifecycle;
 import com.android.settingslib.core.lifecycle.LifecycleObserver;
@@ -72,21 +71,15 @@ public class BrightnessLevelPreferenceController extends AbstractPreferenceContr
         };
 
     public BrightnessLevelPreferenceController(Context context, Lifecycle lifecycle) {
-        this(context, lifecycle, new PowerManagerWrapper(
-                (PowerManager) context.getSystemService(Context.POWER_SERVICE)));
-    }
-
-    @VisibleForTesting
-    public BrightnessLevelPreferenceController(Context context, Lifecycle lifecycle,
-            PowerManagerWrapper powerManagerWrapper) {
         super(context);
         if (lifecycle != null) {
             lifecycle.addObserver(this);
         }
-        mMinBrightness = powerManagerWrapper.getMinimumScreenBrightnessSetting();
+        PowerManager powerManager = (PowerManager) context.getSystemService(Context.POWER_SERVICE);
-        mMaxBrightness = powerManagerWrapper.getMaximumScreenBrightnessSetting();
+        mMinBrightness = powerManager.getMinimumScreenBrightnessSetting();
-        mMinVrBrightness = powerManagerWrapper.getMinimumScreenBrightnessForVrSetting();
+        mMaxBrightness = powerManager.getMaximumScreenBrightnessSetting();
-        mMaxVrBrightness = powerManagerWrapper.getMaximumScreenBrightnessForVrSetting();
+        mMinVrBrightness = powerManager.getMinimumScreenBrightnessForVrSetting();
+        mMaxVrBrightness = powerManager.getMaximumScreenBrightnessForVrSetting();
         mContentResolver = mContext.getContentResolver();
     }
 

src/com/android/settings/print/PrintSettingPreferenceController.java

@@ -29,7 +29,6 @@ import android.support.v7.preference.PreferenceScreen;
 
 import com.android.settings.R;
 import com.android.settings.core.BasePreferenceController;
-import com.android.settings.wrapper.PrintManagerWrapper;
 import com.android.settingslib.RestrictedPreference;
 import com.android.settingslib.core.lifecycle.LifecycleObserver;
 import com.android.settingslib.core.lifecycle.events.OnStart;
@@ -46,13 +45,15 @@ public class PrintSettingPreferenceController extends BasePreferenceController i
     private static final String KEY_PRINTING_SETTINGS = "connected_device_printing";
 
     private final PackageManager mPackageManager;
-    private PrintManagerWrapper mPrintManager;
+    private final PrintManager mPrintManager;
+
     private Preference mPreference;
 
     public PrintSettingPreferenceController(Context context) {
         super(context, KEY_PRINTING_SETTINGS);
         mPackageManager = context.getPackageManager();
-        mPrintManager = new PrintManagerWrapper(context);
+        mPrintManager = ((PrintManager) context.getSystemService(Context.PRINT_SERVICE))
+                .getGlobalPrintManagerForUser(context.getUserId());
     }
 
     @Override
@@ -69,7 +70,7 @@ public class PrintSettingPreferenceController extends BasePreferenceController i
 
     @Override
     public void onStart() {
-        mPrintManager.addPrintJobStateChanegListener(this);
+        mPrintManager.addPrintJobStateChangeListener(this);
     }
 
     @Override

src/com/android/settings/wifi/SavedAccessPointsWifiSettings.java

@@ -28,14 +28,13 @@ import android.support.annotation.VisibleForTesting;
 import android.support.v7.preference.Preference;
 import android.support.v7.preference.PreferenceScreen;
 import android.util.Log;
-
 import android.widget.Toast;
+
 import com.android.internal.logging.nano.MetricsProto;
 import com.android.internal.logging.nano.MetricsProto.MetricsEvent;
 import com.android.settings.R;
 import com.android.settings.SettingsPreferenceFragment;
 import com.android.settings.search.Indexable;
-import com.android.settings.wrapper.WifiManagerWrapper;
 import com.android.settingslib.wifi.AccessPoint;
 import com.android.settingslib.wifi.AccessPointPreference;
 import com.android.settingslib.wifi.WifiSavedConfigUtils;
@@ -107,7 +106,7 @@ public class SavedAccessPointsWifiSettings extends SettingsPreferenceFragment
     };
 
     private WifiDialog mDialog;
-    private WifiManagerWrapper mWifiManager;
+    private WifiManager mWifiManager;
     private AccessPoint mDlgAccessPoint;
     private Bundle mAccessPointSavedState;
     private AccessPoint mSelectedAccessPoint;
@@ -139,7 +138,7 @@ public class SavedAccessPointsWifiSettings extends SettingsPreferenceFragment
     @Override
     public void onActivityCreated(Bundle savedInstanceState) {
         super.onActivityCreated(savedInstanceState);
-        mWifiManager = new WifiManagerWrapper((WifiManager) getSystemService(Context.WIFI_SERVICE));
+        mWifiManager = (WifiManager) getSystemService(Context.WIFI_SERVICE);
 
         if (savedInstanceState != null) {
             if (savedInstanceState.containsKey(SAVE_DIALOG_ACCESS_POINT_STATE)) {
@@ -154,7 +153,7 @@ public class SavedAccessPointsWifiSettings extends SettingsPreferenceFragment
         final Context context = getPrefContext();
 
         final List<AccessPoint> accessPoints =
-                WifiSavedConfigUtils.getAllConfigs(context, mWifiManager.getWifiManager());
+                WifiSavedConfigUtils.getAllConfigs(context, mWifiManager);
         Collections.sort(accessPoints, SAVED_NETWORK_COMPARATOR);
         cacheRemoveAllPrefs(preferenceScreen);
 

src/com/android/settings/wifi/WifiSettings.java

@@ -65,7 +65,6 @@ import com.android.settings.widget.SummaryUpdater.OnSummaryChangeListener;
 import com.android.settings.widget.SwitchBarController;
 import com.android.settings.wifi.details.WifiNetworkDetailsFragment;
 import com.android.settings.wrapper.ConnectivityManagerWrapper;
-import com.android.settings.wrapper.WifiManagerWrapper;
 import com.android.settingslib.RestrictedLockUtils;
 import com.android.settingslib.wifi.AccessPoint;
 import com.android.settingslib.wifi.AccessPoint.AccessPointListener;
@@ -136,7 +135,7 @@ public class WifiSettings extends RestrictedSettingsFragment
     private boolean mIsRestricted;
 
     private WifiEnabler mWifiEnabler;
-    // An access point being editted is stored here.
+    // An access point being edited is stored here.
     private AccessPoint mSelectedAccessPoint;
 
     private WifiDialog mDialog;
@@ -608,11 +607,10 @@ public class WifiSettings extends RestrictedSettingsFragment
                 if (mSelectedAccessPoint != null) {
                     mWifiToNfcDialog = new WriteWifiConfigToNfcDialog(
                             getActivity(),
-                            mSelectedAccessPoint.getSecurity(),
+                            mSelectedAccessPoint.getSecurity());
-                            new WifiManagerWrapper(mWifiManager));
                 } else if (mWifiNfcDialogSavedState != null) {
                     mWifiToNfcDialog = new WriteWifiConfigToNfcDialog(getActivity(),
-                            mWifiNfcDialogSavedState, new WifiManagerWrapper(mWifiManager));
+                            mWifiNfcDialogSavedState);
                 }
 
                 return mWifiToNfcDialog;

src/com/android/settings/wifi/WriteWifiConfigToNfcDialog.java

@@ -20,6 +20,7 @@ import android.app.Activity;
 import android.app.AlertDialog;
 import android.content.Context;
 import android.content.DialogInterface;
+import android.net.wifi.WifiManager;
 import android.nfc.FormatException;
 import android.nfc.NdefMessage;
 import android.nfc.NdefRecord;
@@ -27,7 +28,6 @@ import android.nfc.NfcAdapter;
 import android.nfc.Tag;
 import android.nfc.tech.Ndef;
 import android.os.Bundle;
-import android.os.Handler;
 import android.os.PowerManager;
 import android.text.Editable;
 import android.text.InputType;
@@ -42,7 +42,6 @@ import android.widget.ProgressBar;
 import android.widget.TextView;
 
 import com.android.settings.R;
-import com.android.settings.wrapper.WifiManagerWrapper;
 import com.android.settingslib.wifi.AccessPoint;
 
 import java.io.IOException;
@@ -67,29 +66,29 @@ class WriteWifiConfigToNfcDialog extends AlertDialog
     private TextView mLabelView;
     private CheckBox mPasswordCheckBox;
     private ProgressBar mProgressBar;
-    private WifiManagerWrapper mWifiManager;
+    private WifiManager mWifiManager;
     private String mWpsNfcConfigurationToken;
     private Context mContext;
     private int mSecurity;
 
-    WriteWifiConfigToNfcDialog(Context context, int security, WifiManagerWrapper wifiManager) {
+    WriteWifiConfigToNfcDialog(Context context, int security) {
         super(context);
 
         mContext = context;
         mWakeLock = ((PowerManager) context.getSystemService(Context.POWER_SERVICE))
                 .newWakeLock(PowerManager.PARTIAL_WAKE_LOCK, "WriteWifiConfigToNfcDialog:wakeLock");
         mSecurity = security;
-        mWifiManager = wifiManager;
+        mWifiManager = (WifiManager) context.getSystemService(Context.WIFI_SERVICE);
     }
 
-    WriteWifiConfigToNfcDialog(Context context, Bundle savedState, WifiManagerWrapper wifiManager) {
+    WriteWifiConfigToNfcDialog(Context context, Bundle savedState) {
         super(context);
 
         mContext = context;
         mWakeLock = ((PowerManager) context.getSystemService(Context.POWER_SERVICE))
                 .newWakeLock(PowerManager.PARTIAL_WAKE_LOCK, "WriteWifiConfigToNfcDialog:wakeLock");
         mSecurity = savedState.getInt(SECURITY);
-        mWifiManager = wifiManager;
+        mWifiManager = (WifiManager) context.getSystemService(Context.WIFI_SERVICE);
     }
 
     @Override

src/com/android/settings/wrapper/AccessibilityServiceInfoWrapper.java

@@ -1,47 +0,0 @@
-/*
- * Copyright (C) 2017 The Android Open Source Project
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *      http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-package com.android.settings.wrapper;
-
-import android.accessibilityservice.AccessibilityServiceInfo;
-import android.content.ComponentName;
-
-/**
- * This class replicates a subset of the
- * {@link android.accessibilityservice.AccessibilityServiceInfo}. The class
- * exists so that we can use a thin wrapper around it in production code and a mock in tests.
- * We cannot directly mock or shadow it, because some of the methods we rely on are newer than
- * the API version supported by Robolectric.
- */
-public class AccessibilityServiceInfoWrapper {
-
-    private final AccessibilityServiceInfo mServiceInfo;
-
-    public AccessibilityServiceInfoWrapper(AccessibilityServiceInfo serviceInfo) {
-        mServiceInfo = serviceInfo;
-    }
-
-    /**
-     * Returns the real {@code AccessibilityServiceInfo} object.
-     */
-    public AccessibilityServiceInfo getAccessibilityServiceInfo() {
-        return mServiceInfo;
-    }
-
-    public ComponentName getComponentName() {
-        return mServiceInfo.getComponentName();
-    }
-}

src/com/android/settings/wrapper/PowerManagerWrapper.java

@@ -1,49 +0,0 @@
-/*
- * Copyright (C) 2017 The Android Open Source Project
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *      http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-package com.android.settings.wrapper;
-
-import android.os.PowerManager;
-
-/**
- * This class replicates a subset of the android.os.PowerManager. The class exists so that we can
- * use a thin wrapper around the PowerManager in production code and a mock in tests. We cannot
- * directly mock or shadow the PowerManager, because some of the methods we rely on are newer than
- * the API version supported by Robolectric or are hidden.
- */
-public class PowerManagerWrapper {
-    private final PowerManager mPowerManager;
-
-    public PowerManagerWrapper(PowerManager powerManager) {
-        mPowerManager = powerManager;
-    }
-
-    public int getMinimumScreenBrightnessSetting() {
-        return mPowerManager.getMinimumScreenBrightnessSetting();
-    }
-
-    public int getMaximumScreenBrightnessSetting() {
-        return mPowerManager.getMaximumScreenBrightnessSetting();
-    }
-
-    public int getMinimumScreenBrightnessForVrSetting() {
-        return mPowerManager.getMinimumScreenBrightnessForVrSetting();
-    }
-
-    public int getMaximumScreenBrightnessForVrSetting() {
-        return mPowerManager.getMaximumScreenBrightnessForVrSetting();
-    }
-}

src/com/android/settings/wrapper/PrintManagerWrapper.java

@@ -1,54 +0,0 @@
-/*
- * Copyright (C) 2018 The Android Open Source Project
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *      http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-package com.android.settings.wrapper;
-
-import android.content.Context;
-import android.print.PrintJob;
-import android.print.PrintManager;
-import android.printservice.PrintServiceInfo;
-
-import java.util.List;
-
-/**
- * Wrapper class for {@link PrintManager}. This is necessary to increase testability in Robolectric.
- */
-public class PrintManagerWrapper {
-
-    private final PrintManager mPrintManager;
-
-    public PrintManagerWrapper(Context context) {
-        mPrintManager = ((PrintManager) context.getSystemService(Context.PRINT_SERVICE))
-                .getGlobalPrintManagerForUser(context.getUserId());
-    }
-
-    public List<PrintServiceInfo> getPrintServices(int selectionFlags) {
-        return mPrintManager.getPrintServices(selectionFlags);
-    }
-
-    public void addPrintJobStateChanegListener(PrintManager.PrintJobStateChangeListener listener) {
-        mPrintManager.addPrintJobStateChangeListener(listener);
-    }
-
-    public void removePrintJobStateChangeListener(
-            PrintManager.PrintJobStateChangeListener listener) {
-        mPrintManager.removePrintJobStateChangeListener(listener);
-    }
-
-    public List<PrintJob> getPrintJobs() {
-        return mPrintManager.getPrintJobs();
-    }
-}

src/com/android/settings/wrapper/UsbManagerWrapper.java

@@ -1,30 +0,0 @@
-/*
- * Copyright (C) 2018 The Android Open Source Project
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *      http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-package com.android.settings.wrapper;
-
-import android.hardware.usb.UsbManager;
-
-public class UsbManagerWrapper {
-    private UsbManager mUsbManager;
-
-    public UsbManagerWrapper(UsbManager manager) {
-        mUsbManager = manager;
-    }
-
-    public long getCurrentFunctions() {
-        return mUsbManager.getCurrentFunctions();
-    }
-}

src/com/android/settings/wrapper/WifiManagerWrapper.java

@@ -1,53 +0,0 @@
-package com.android.settings.wrapper;
-
-import android.net.wifi.WifiConfiguration;
-import android.net.wifi.WifiManager;
-
-/**
- * Wrapper around {@link WifiManager} to facilitate unit testing.
- *
- * TODO: delete this class once robolectric supports Android O
- */
-public class WifiManagerWrapper {
-    private final WifiManager mWifiManager;
-
-    public WifiManagerWrapper(WifiManager wifiManager) {
-        mWifiManager = wifiManager;
-    }
-
-    /**
-     * Gets the real WifiManager
-     * @return the real WifiManager
-     */
-    public WifiManager getWifiManager() {
-        return mWifiManager;
-    }
-
-    /**
-     * {@link WifiManager#getCurrentNetworkWpsNfcConfigurationToken}
-     */
-    public String getCurrentNetworkWpsNfcConfigurationToken() {
-        return mWifiManager.getCurrentNetworkWpsNfcConfigurationToken();
-    }
-
-    /**
-     * {@link WifiManager#removePasspointConfiguration}
-     */
-    public void removePasspointConfiguration(String fqdn) {
-        mWifiManager.removePasspointConfiguration(fqdn);
-    }
-
-    /**
-     * {@link WifiManager#forget}
-     */
-    public void forget(int netId, WifiManager.ActionListener listener) {
-        mWifiManager.forget(netId, listener);
-    }
-
-    /**
-     * {@link WifiManager#save}
-     */
-    public void save(WifiConfiguration config, WifiManager.ActionListener listener) {
-        mWifiManager.save(config, listener);
-    }
-}

tests/robotests/src/com/android/settings/DisplaySettingsTest.java

@@ -6,7 +6,7 @@ import android.content.Context;
 
 import com.android.settings.testutils.SettingsRobolectricTestRunner;
 import com.android.settings.testutils.XmlTestUtils;
-import com.android.settings.testutils.shadow.ShadowPowerManagerWrapper;
+import com.android.settings.testutils.shadow.ShadowPowerManager;
 import com.android.settingslib.core.AbstractPreferenceController;
 
 import org.junit.Test;
@@ -21,7 +21,7 @@ import java.util.List;
 public class DisplaySettingsTest {
 
     @Test
-    @Config(shadows = ShadowPowerManagerWrapper.class)
+    @Config(shadows = ShadowPowerManager.class)
     public void testPreferenceControllers_getPreferenceKeys_existInPreferenceScreen() {
         final Context context = RuntimeEnvironment.application;
         final DisplaySettings fragment = new DisplaySettings();

tests/robotests/src/com/android/settings/connecteddevice/usb/UsbBackendTest.java

@@ -17,7 +17,6 @@
 package com.android.settings.connecteddevice.usb;
 
 import static com.google.common.truth.Truth.assertThat;
-
 import static org.mockito.Answers.RETURNS_DEEP_STUBS;
 import static org.mockito.Mockito.verify;
 import static org.mockito.Mockito.when;
@@ -30,8 +29,6 @@ import android.hardware.usb.UsbPortStatus;
 import android.net.ConnectivityManager;
 
 import com.android.settings.testutils.SettingsRobolectricTestRunner;
-import com.android.settings.testutils.SettingsRobolectricTestRunner;
-import com.android.settings.wrapper.UsbManagerWrapper;
 import com.android.settings.wrapper.UserManagerWrapper;
 
 import org.junit.Before;
@@ -50,8 +47,6 @@ public class UsbBackendTest {
     @Mock
     private UserManagerWrapper mUserManagerWrapper;
     @Mock
-    private UsbManagerWrapper mUsbManagerWrapper;
-    @Mock
     private ConnectivityManager mConnectivityManager;
     @Mock
     private UsbPort mUsbPort;
@@ -71,17 +66,9 @@ public class UsbBackendTest {
         when(mUsbManager.getPortStatus(mUsbPort)).thenReturn(mUsbPortStatus);
     }
 
-    @Test
-    public void constructor_noUsbPort_shouldNotCrash() {
-        final UsbBackend usbBackend =
-                new UsbBackend(mContext, mUserManagerWrapper, mUsbManagerWrapper);
-        // Should not crash
-    }
-
     @Test
     public void setDataRole_allRolesSupported_shouldSetDataRole() {
-        final UsbBackend usbBackend =
+        final UsbBackend usbBackend = new UsbBackend(mContext, mUserManagerWrapper);
-                new UsbBackend(mContext, mUserManagerWrapper, mUsbManagerWrapper);
 
         when(mUsbPortStatus
                 .isRoleCombinationSupported(UsbPort.POWER_ROLE_SINK, UsbPort.DATA_ROLE_DEVICE))
@@ -104,8 +91,7 @@ public class UsbBackendTest {
 
     @Test
     public void setDataRole_notAllRolesSupported_shouldSetDataAndPowerRole() {
-        final UsbBackend usbBackend =
+        final UsbBackend usbBackend = new UsbBackend(mContext, mUserManagerWrapper);
-                new UsbBackend(mContext, mUserManagerWrapper, mUsbManagerWrapper);
 
         when(mUsbPortStatus
                 .isRoleCombinationSupported(UsbPort.POWER_ROLE_SINK, UsbPort.DATA_ROLE_DEVICE))
@@ -123,8 +109,7 @@ public class UsbBackendTest {
 
     @Test
     public void setPowerRole_allRolesSupported_shouldSetPowerRole() {
-        final UsbBackend usbBackend =
+        final UsbBackend usbBackend = new UsbBackend(mContext, mUserManagerWrapper);
-                new UsbBackend(mContext, mUserManagerWrapper, mUsbManagerWrapper);
 
         when(mUsbPortStatus
                 .isRoleCombinationSupported(UsbPort.POWER_ROLE_SINK, UsbPort.DATA_ROLE_DEVICE))
@@ -148,8 +133,7 @@ public class UsbBackendTest {
 
     @Test
     public void setPowerRole_notAllRolesSupported_shouldSetDataAndPowerRole() {
-        final UsbBackend usbBackend =
+        final UsbBackend usbBackend = new UsbBackend(mContext, mUserManagerWrapper);
-                new UsbBackend(mContext, mUserManagerWrapper, mUsbManagerWrapper);
 
         when(mUsbPortStatus
                 .isRoleCombinationSupported(UsbPort.POWER_ROLE_SINK, UsbPort.DATA_ROLE_DEVICE))
@@ -170,8 +154,7 @@ public class UsbBackendTest {
         when(mUserManagerWrapper.isUsbFileTransferRestricted()).thenReturn(true);
         when(mUserManagerWrapper.isUsbFileTransferRestrictedBySystem()).thenReturn(true);
 
-        final UsbBackend usbBackend =
+        final UsbBackend usbBackend = new UsbBackend(mContext, mUserManagerWrapper);
-                new UsbBackend(mContext, mUserManagerWrapper, mUsbManagerWrapper);
 
         assertThat(usbBackend.areFunctionsSupported(UsbManager.FUNCTION_MTP)).isFalse();
     }
@@ -181,8 +164,7 @@ public class UsbBackendTest {
         when(mUserManagerWrapper.isUsbFileTransferRestricted()).thenReturn(false);
         when(mUserManagerWrapper.isUsbFileTransferRestrictedBySystem()).thenReturn(false);
 
-        final UsbBackend usbBackend =
+        final UsbBackend usbBackend = new UsbBackend(mContext, mUserManagerWrapper);
-                new UsbBackend(mContext, mUserManagerWrapper, mUsbManagerWrapper);
 
         assertThat(usbBackend.areFunctionsSupported(UsbManager.FUNCTION_MTP)).isTrue();
     }

tests/robotests/src/com/android/settings/display/BrightnessLevelPreferenceControllerTest.java

@@ -16,6 +16,7 @@
 
 package com.android.settings.display;
 
+import static android.content.Context.POWER_SERVICE;
 import static com.google.common.truth.Truth.assertThat;
 import static org.mockito.Matchers.anyString;
 import static org.mockito.Mockito.doReturn;
@@ -26,12 +27,12 @@ import static org.mockito.Mockito.when;
 
 import android.content.ContentResolver;
 import android.content.Context;
+import android.os.PowerManager;
 import android.provider.Settings.System;
 import android.support.v7.preference.Preference;
 import android.support.v7.preference.PreferenceScreen;
 
 import com.android.settings.testutils.SettingsRobolectricTestRunner;
-import com.android.settings.wrapper.PowerManagerWrapper;
 
 import org.junit.Before;
 import org.junit.Test;
@@ -40,13 +41,14 @@ import org.mockito.Mock;
 import org.mockito.MockitoAnnotations;
 import org.robolectric.RuntimeEnvironment;
 import org.robolectric.shadow.api.Shadow;
+import org.robolectric.shadows.ShadowApplication;
 import org.robolectric.shadows.ShadowContentResolver;
 
 @RunWith(SettingsRobolectricTestRunner.class)
 public class BrightnessLevelPreferenceControllerTest {
 
     @Mock
-    private PowerManagerWrapper mPowerManager;
+    private PowerManager mPowerManager;
     @Mock
     private PreferenceScreen mScreen;
     @Mock
@@ -67,8 +69,10 @@ public class BrightnessLevelPreferenceControllerTest {
         when(mPowerManager.getMaximumScreenBrightnessSetting()).thenReturn(100);
         when(mPowerManager.getMinimumScreenBrightnessForVrSetting()).thenReturn(0);
         when(mPowerManager.getMaximumScreenBrightnessForVrSetting()).thenReturn(100);
+        ShadowApplication.getInstance().setSystemService(POWER_SERVICE,
+                mPowerManager);
         when(mScreen.findPreference(anyString())).thenReturn(mPreference);
-        mController = spy(new BrightnessLevelPreferenceController(mContext, null, mPowerManager));
+        mController = spy(new BrightnessLevelPreferenceController(mContext, null));
         doReturn(false).when(mController).isInVrMode();
     }
 
@@ -80,7 +84,7 @@ public class BrightnessLevelPreferenceControllerTest {
     @Test
     public void onStart_shouldRegisterObserver() {
         BrightnessLevelPreferenceController controller =
-            new BrightnessLevelPreferenceController(mContext, null, mPowerManager);
+                new BrightnessLevelPreferenceController(mContext, null);
         ShadowContentResolver shadowContentResolver = Shadow.extract(mContentResolver);
 
         controller.onStart();
@@ -96,7 +100,7 @@ public class BrightnessLevelPreferenceControllerTest {
     @Test
     public void onStop_shouldUnregisterObserver() {
         BrightnessLevelPreferenceController controller =
-            new BrightnessLevelPreferenceController(mContext, null, mPowerManager);
+                new BrightnessLevelPreferenceController(mContext, null);
         ShadowContentResolver shadowContentResolver = Shadow.extract(mContext.getContentResolver());
 
         controller.displayPreference(mScreen);

tests/robotests/src/com/android/settings/print/PrintSettingsPreferenceControllerTest.java

@@ -34,14 +34,12 @@ import android.printservice.PrintServiceInfo;
 
 import com.android.settings.R;
 import com.android.settings.testutils.SettingsRobolectricTestRunner;
-import com.android.settings.wrapper.PrintManagerWrapper;
 import com.android.settingslib.RestrictedPreference;
 import com.android.settingslib.core.lifecycle.Lifecycle;
 
 import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
-import org.mockito.Answers;
 import org.mockito.Mock;
 import org.mockito.Mockito;
 import org.mockito.MockitoAnnotations;
@@ -55,12 +53,13 @@ import java.util.List;
 public class PrintSettingsPreferenceControllerTest {
 
     @Mock
-    private PrintManagerWrapper mPrintManager;
+    private PrintManager mPrintManager;
     @Mock
     private UserManager mUserManager;
-    private Context mContext;
     @Mock
     private RestrictedPreference mPreference;
+
+    private Context mContext;
     private PrintSettingPreferenceController mController;
     private LifecycleOwner mLifecycleOwner;
     private Lifecycle mLifecycle;
@@ -83,7 +82,7 @@ public class PrintSettingsPreferenceControllerTest {
         mLifecycle.handleLifecycleEvent(ON_START);
         mLifecycle.handleLifecycleEvent(ON_STOP);
 
-        verify(mPrintManager).addPrintJobStateChanegListener(mController);
+        verify(mPrintManager).addPrintJobStateChangeListener(mController);
         verify(mPrintManager).removePrintJobStateChangeListener(mController);
     }
 

tests/robotests/src/com/android/settings/testutils/shadow/ShadowAccessibilityServiceInfo.java

@@ -16,18 +16,18 @@
 
 package com.android.settings.testutils.shadow;
 
+import android.accessibilityservice.AccessibilityServiceInfo;
 import android.content.ComponentName;
 
-import com.android.settings.wrapper.AccessibilityServiceInfoWrapper;
-
 import org.robolectric.annotation.Implementation;
 import org.robolectric.annotation.Implements;
-@Implements(AccessibilityServiceInfoWrapper.class)
+
-public class ShadowAccessibilityServiceInfoWrapperImpl {
+@Implements(AccessibilityServiceInfo.class)
+public class ShadowAccessibilityServiceInfo {
     private static ComponentName sComponentName;
 
     public static void setComponentName(String componentName) {
-        sComponentName = ComponentName.unflattenFromString(componentName);;
+        sComponentName = ComponentName.unflattenFromString(componentName);
     }
 
     @Implementation

tests/robotests/src/com/android/settings/testutils/shadow/ShadowPowerManager.java

@@ -17,12 +17,13 @@
 
 package com.android.settings.testutils.shadow;
 
-import com.android.settings.wrapper.PowerManagerWrapper;
+import android.os.PowerManager;
+
 import org.robolectric.annotation.Implementation;
 import org.robolectric.annotation.Implements;
 
-@Implements(PowerManagerWrapper.class)
+@Implements(PowerManager.class)
-public class ShadowPowerManagerWrapper {
+public class ShadowPowerManager {
 
     @Implementation
     public int getMinimumScreenBrightnessSetting() {

tests/robotests/src/com/android/settings/wifi/SavedAccessPointsWifiSettingsTest.java

@@ -23,11 +23,11 @@ import static org.mockito.Mockito.verify;
 import static org.mockito.Mockito.when;
 
 import android.net.wifi.WifiConfiguration;
+import android.net.wifi.WifiManager;
 import android.net.wifi.WifiManager.ActionListener;
 import android.os.Handler;
 
 import com.android.settings.testutils.SettingsRobolectricTestRunner;
-import com.android.settings.wrapper.WifiManagerWrapper;
 import com.android.settingslib.wifi.AccessPoint;
 
 import org.junit.Before;
@@ -40,12 +40,18 @@ import org.robolectric.util.ReflectionHelpers;
 @RunWith(SettingsRobolectricTestRunner.class)
 public class SavedAccessPointsWifiSettingsTest {
 
-    @Mock private WifiManagerWrapper mockWifiManager;
+    @Mock
-    @Mock private WifiDialog mockWifiDialog;
+    private WifiManager mockWifiManager;
-    @Mock private WifiConfigController mockConfigController;
+    @Mock
-    @Mock private WifiConfiguration mockWifiConfiguration;
+    private WifiDialog mockWifiDialog;
-    @Mock private AccessPoint mockAccessPoint;
+    @Mock
-    @Mock private Handler mHandler;
+    private WifiConfigController mockConfigController;
+    @Mock
+    private WifiConfiguration mockWifiConfiguration;
+    @Mock
+    private AccessPoint mockAccessPoint;
+    @Mock
+    private Handler mHandler;
 
     private SavedAccessPointsWifiSettings mSettings;
 

tests/robotests/src/com/android/settings/wifi/WriteWifiConfigToNfcDialogTest.java

@@ -21,11 +21,11 @@ import static org.mockito.Mockito.when;
 
 import android.app.Activity;
 import android.content.Context;
+import android.net.wifi.WifiManager;
 import android.view.inputmethod.InputMethodManager;
 
 import com.android.settings.testutils.SettingsRobolectricTestRunner;
 import com.android.settings.testutils.shadow.ShadowNfcAdapter;
-import com.android.settings.wrapper.WifiManagerWrapper;
 
 import org.junit.After;
 import org.junit.Before;
@@ -42,9 +42,9 @@ import org.robolectric.util.ReflectionHelpers;
 public class WriteWifiConfigToNfcDialogTest {
 
     @Mock
-    Activity mActivity;
+    private Activity mActivity;
     @Mock
-    WifiManagerWrapper mWifiManager;
+    private WifiManager mWifiManager;
 
     private WriteWifiConfigToNfcDialog mWriteWifiConfigToNfcDialog;
 
@@ -56,7 +56,8 @@ public class WriteWifiConfigToNfcDialogTest {
                 .thenReturn(ReflectionHelpers.newInstance(InputMethodManager.class));
 
         mWriteWifiConfigToNfcDialog = new WriteWifiConfigToNfcDialog(RuntimeEnvironment.application,
-                0 /* security */, mWifiManager);
+                0 /* security */);
+        ReflectionHelpers.setField(mWriteWifiConfigToNfcDialog, "mWifiManager", mWifiManager);
         mWriteWifiConfigToNfcDialog.setOwnerActivity(mActivity);
         mWriteWifiConfigToNfcDialog.onCreate(null /* savedInstanceState */);
     }