OSDN Git Service

Misc fixes for items under system setting
authorFan Zhang <zhfan@google.com>
Thu, 23 Mar 2017 23:42:13 +0000 (16:42 -0700)
committerFan Zhang <zhfan@google.com>
Sat, 25 Mar 2017 02:43:23 +0000 (02:43 +0000)
- Show summary under virtual keyboard: list all avaiable keyboards
- Set about phone subtext to device model string.
- Update DND Suggestion text

Change-Id: I20acffcf691d1354c7636f5851b376099844700f
Fix: 35959817
Fix: 36395109
Test: robotests

res/values/strings.xml
res/xml/language_and_input.xml
src/com/android/settings/inputmethod/PhysicalKeyboardFragment.java
src/com/android/settings/inputmethod/PhysicalKeyboardPreferenceController.java [new file with mode: 0644]
src/com/android/settings/inputmethod/VirtualKeyboardPreferenceController.java [new file with mode: 0644]
src/com/android/settings/language/LanguageAndInputSettings.java
tests/robotests/src/com/android/settings/inputmethod/PhysicalKeyboardPreferenceControllerTest.java [new file with mode: 0644]
tests/robotests/src/com/android/settings/inputmethod/VirtualKeyboardPreferenceControllerTest.java [new file with mode: 0644]
tests/robotests/src/com/android/settings/language/LanguageAndInputSettingsTest.java
tests/robotests/src/com/android/settings/testutils/shadow/ShadowInputDevice.java [new file with mode: 0644]

index 130b9cd..b252e33 100644 (file)
     <!-- Title for the 'keyboard and input methods' preference category. [CHAR LIMIT=45] -->
     <string name="keyboard_and_input_methods_category">Keyboard &amp; inputs</string>
     <!-- Title for the 'virtual keyboard' preference sub-screen. [CHAR LIMIT=35] -->
-    <string name="virtual_keyboard_category">Virtual keyboards</string>
+    <string name="virtual_keyboard_category">Virtual keyboard</string>
     <!-- Title for the 'available virtual keyboard' preference sub-screen. [CHAR LIMIT=35] -->
     <string name="available_virtual_keyboard_category">Available virtual keyboard</string>
     <!-- Title for the button to trigger the 'Manage keyboards' preference sub-screen, where the user can turn on/off installed virtual keyboards.[CHAR LIMIT=35] -->
     <string name="zen_mode_automation_settings_title">Automatic rules</string>
 
     <!--  Do not disturb: Title for the zen mode automation option Suggestion. [CHAR LIMIT=30] -->
-    <string name="zen_mode_automation_suggestion_title">Set Do Not Disturb schedule</string>
+    <string name="zen_mode_automation_suggestion_title">Set Do Not Disturb rules</string>
 
     <!--  Do not disturb: Summary for the zen mode automation option Suggestion. [CHAR LIMIT=30] -->
-    <string name="zen_mode_automation_suggestion_summary">Silence your device at certain times</string>
+    <string name="zen_mode_automation_suggestion_summary">Limit sounds &amp; vibrations at certain times</string>
 
     <!--  Do not disturb: Zen mode option: Important interruptions [CHAR LIMIT=60] -->
     <string name="zen_mode_option_important_interruptions">Priority only</string>
index 79f441f..17bd576 100644 (file)
     <PreferenceCategory
         android:title="@string/keyboard_and_input_methods_category">
         <Preference
+            android:key="virtual_keyboard_pref"
             android:title="@string/virtual_keyboard_category"
             android:fragment="com.android.settings.inputmethod.VirtualKeyboardFragment"/>
         <Preference
+            android:key="physical_keyboard_pref"
             android:title="@string/physical_keyboard_title"
+            android:summary="@string/summary_placeholder"
             android:fragment="com.android.settings.inputmethod.PhysicalKeyboardFragment"/>
     </PreferenceCategory>
 
index ad6e041..42eff15 100644 (file)
@@ -136,7 +136,7 @@ public final class PhysicalKeyboardFragment extends SettingsPreferenceFragment
         unregisterShowVirtualKeyboardSettingsObserver();
     }
 
-    public void onLoadFinishedInternal(
+    private void onLoadFinishedInternal(
             final int loaderId, @NonNull final List<Keyboards> keyboardsList) {
         if (!mLoaderIDs.remove(loaderId)) {
             // Already destroyed loader.  Ignore.
@@ -198,8 +198,8 @@ public final class PhysicalKeyboardFragment extends SettingsPreferenceFragment
     }
 
     @NonNull
-    private static ArrayList<HardKeyboardDeviceInfo> getHardKeyboards() {
-        final ArrayList<HardKeyboardDeviceInfo> keyboards = new ArrayList<>();
+    public static List<HardKeyboardDeviceInfo> getHardKeyboards() {
+        final List<HardKeyboardDeviceInfo> keyboards = new ArrayList<>();
         final int[] devicesIds = InputDevice.getDeviceIds();
         for (int deviceId : devicesIds) {
             final InputDevice device = InputDevice.getDevice(deviceId);
@@ -211,7 +211,7 @@ public final class PhysicalKeyboardFragment extends SettingsPreferenceFragment
     }
 
     private void updateHardKeyboards() {
-        final ArrayList<HardKeyboardDeviceInfo> newHardKeyboards = getHardKeyboards();
+        final List<HardKeyboardDeviceInfo> newHardKeyboards = getHardKeyboards();
         if (!Objects.equals(newHardKeyboards, mLastHardKeyboards)) {
             clearLoader();
             mLastHardKeyboards.clear();
diff --git a/src/com/android/settings/inputmethod/PhysicalKeyboardPreferenceController.java b/src/com/android/settings/inputmethod/PhysicalKeyboardPreferenceController.java
new file mode 100644 (file)
index 0000000..aa9f5ad
--- /dev/null
@@ -0,0 +1,111 @@
+/*
+ * 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.inputmethod;
+
+import android.content.Context;
+import android.hardware.input.InputManager;
+import android.support.v7.preference.Preference;
+
+import com.android.settings.R;
+import com.android.settings.core.PreferenceController;
+import com.android.settings.core.lifecycle.Lifecycle;
+import com.android.settings.core.lifecycle.LifecycleObserver;
+import com.android.settings.core.lifecycle.events.OnPause;
+import com.android.settings.core.lifecycle.events.OnResume;
+import com.android.settings.inputmethod.PhysicalKeyboardFragment.HardKeyboardDeviceInfo;
+
+import java.util.List;
+
+public class PhysicalKeyboardPreferenceController extends PreferenceController implements
+        LifecycleObserver, OnResume, OnPause, InputManager.InputDeviceListener {
+
+    private final InputManager mIm;
+
+    private Preference mPreference;
+
+    public PhysicalKeyboardPreferenceController(Context context, Lifecycle lifecycle) {
+        super(context);
+        mIm = (InputManager) context.getSystemService(Context.INPUT_SERVICE);
+
+        if (lifecycle != null) {
+            lifecycle.addObserver(this);
+        }
+    }
+
+    @Override
+    public boolean isAvailable() {
+        return true;
+    }
+
+    @Override
+    public void updateState(Preference preference) {
+        mPreference = preference;
+        updateSummary();
+    }
+
+    @Override
+    public String getPreferenceKey() {
+        return "physical_keyboard_pref";
+    }
+
+    @Override
+    public void onPause() {
+        mIm.registerInputDeviceListener(this, null);
+    }
+
+    @Override
+    public void onResume() {
+        mIm.unregisterInputDeviceListener(this);
+    }
+
+    @Override
+    public void onInputDeviceAdded(int deviceId) {
+        updateSummary();
+    }
+
+    @Override
+    public void onInputDeviceRemoved(int deviceId) {
+        updateSummary();
+    }
+
+    @Override
+    public void onInputDeviceChanged(int deviceId) {
+        updateSummary();
+    }
+
+    private void updateSummary() {
+        if (mPreference == null) {
+            return;
+        }
+        final List<HardKeyboardDeviceInfo> keyboards =
+                PhysicalKeyboardFragment.getHardKeyboards();
+        if (keyboards.isEmpty()) {
+            mPreference.setSummary(R.string.disconnected);
+            return;
+        }
+        String summary = null;
+        for (HardKeyboardDeviceInfo info : keyboards) {
+            if (summary == null) {
+                summary = info.mDeviceName;
+            } else {
+                summary = mContext.getString(R.string.join_many_items_middle, summary,
+                        info.mDeviceName);
+            }
+        }
+        mPreference.setSummary(summary);
+    }
+}
diff --git a/src/com/android/settings/inputmethod/VirtualKeyboardPreferenceController.java b/src/com/android/settings/inputmethod/VirtualKeyboardPreferenceController.java
new file mode 100644 (file)
index 0000000..b7bf3dc
--- /dev/null
@@ -0,0 +1,89 @@
+/*
+ * 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.inputmethod;
+
+import android.app.admin.DevicePolicyManager;
+import android.content.Context;
+import android.content.pm.PackageManager;
+import android.support.v7.preference.Preference;
+import android.view.inputmethod.InputMethodInfo;
+import android.view.inputmethod.InputMethodManager;
+
+import com.android.settings.R;
+import com.android.settings.core.PreferenceController;
+
+import java.util.ArrayList;
+import java.util.List;
+
+public class VirtualKeyboardPreferenceController extends PreferenceController {
+
+    private final InputMethodManager mImm;
+    private final DevicePolicyManager mDpm;
+    private final PackageManager mPm;
+
+    public VirtualKeyboardPreferenceController(Context context) {
+        super(context);
+        mPm = mContext.getPackageManager();
+        mDpm = (DevicePolicyManager) context.getSystemService(Context.DEVICE_POLICY_SERVICE);
+        mImm = (InputMethodManager) mContext.getSystemService(Context.INPUT_METHOD_SERVICE);
+    }
+
+    @Override
+    public boolean isAvailable() {
+        return true;
+    }
+
+    @Override
+    public String getPreferenceKey() {
+        return "virtual_keyboard_pref";
+    }
+
+    @Override
+    public void updateState(Preference preference) {
+        final List<InputMethodInfo> imis = mImm.getEnabledInputMethodList();
+        if (imis == null) {
+            preference.setSummary(R.string.summary_empty);
+            return;
+        }
+
+        final List<String> permittedList = mDpm.getPermittedInputMethodsForCurrentUser();
+        final List<String> labels = new ArrayList<>();
+
+        for (InputMethodInfo imi : imis) {
+            final boolean isAllowedByOrganization = permittedList == null
+                    || permittedList.contains(imi.getPackageName());
+            if (!isAllowedByOrganization) {
+                continue;
+            }
+            labels.add(imi.loadLabel(mPm).toString());
+        }
+        if (labels.isEmpty()) {
+            preference.setSummary(R.string.summary_empty);
+            return;
+        }
+
+        String summary = null;
+        for (String label : labels) {
+            if (summary == null) {
+                summary = label;
+            } else {
+                summary = mContext.getString(R.string.join_many_items_middle, summary, label);
+            }
+        }
+        preference.setSummary(summary);
+    }
+}
index 5a2c0c7..86ce118 100644 (file)
@@ -43,7 +43,9 @@ import com.android.settings.gestures.DoubleTwistPreferenceController;
 import com.android.settings.gestures.PickupGesturePreferenceController;
 import com.android.settings.gestures.SwipeToNotificationPreferenceController;
 import com.android.settings.inputmethod.GameControllerPreferenceController;
+import com.android.settings.inputmethod.PhysicalKeyboardPreferenceController;
 import com.android.settings.inputmethod.SpellCheckerPreferenceController;
+import com.android.settings.inputmethod.VirtualKeyboardPreferenceController;
 import com.android.settings.search.BaseSearchIndexProvider;
 
 import java.util.ArrayList;
@@ -87,9 +89,11 @@ public class LanguageAndInputSettings extends DashboardFragment {
         controllers.add(new UserDictionaryPreferenceController(context));
         controllers.add(new TtsPreferenceController(context, new TtsEngines(context)));
         // Input
+        controllers.add(new VirtualKeyboardPreferenceController(context));
+        controllers.add(new PhysicalKeyboardPreferenceController(context, lifecycle));
         final GameControllerPreferenceController gameControllerPreferenceController
                 = new GameControllerPreferenceController(context);
-        getLifecycle().addObserver(gameControllerPreferenceController);
+        lifecycle.addObserver(gameControllerPreferenceController);
 
         if (mAmbientDisplayConfig == null) {
             mAmbientDisplayConfig = new AmbientDisplayConfiguration(context);
diff --git a/tests/robotests/src/com/android/settings/inputmethod/PhysicalKeyboardPreferenceControllerTest.java b/tests/robotests/src/com/android/settings/inputmethod/PhysicalKeyboardPreferenceControllerTest.java
new file mode 100644 (file)
index 0000000..5fa2c0d
--- /dev/null
@@ -0,0 +1,98 @@
+/*
+ * 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.inputmethod;
+
+import static com.google.common.truth.Truth.assertThat;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+import android.content.Context;
+import android.hardware.input.InputManager;
+import android.support.v7.preference.Preference;
+import android.view.InputDevice;
+
+import com.android.settings.R;
+import com.android.settings.SettingsRobolectricTestRunner;
+import com.android.settings.TestConfig;
+import com.android.settings.testutils.shadow.ShadowInputDevice;
+
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.Mock;
+import org.mockito.MockitoAnnotations;
+import org.robolectric.annotation.Config;
+
+@RunWith(SettingsRobolectricTestRunner.class)
+@Config(manifest = TestConfig.MANIFEST_PATH, sdk = TestConfig.SDK_VERSION)
+public class PhysicalKeyboardPreferenceControllerTest {
+
+    @Mock
+    private Context mContext;
+    @Mock
+    private InputManager mIm;
+    @Mock
+    private Preference mPreference;
+
+    private PhysicalKeyboardPreferenceController mController;
+
+    @Before
+    public void setUp() {
+        MockitoAnnotations.initMocks(this);
+        when(mContext.getSystemService(Context.INPUT_SERVICE)).thenReturn(mIm);
+        mController = new PhysicalKeyboardPreferenceController(mContext, null /* lifecycle */);
+    }
+
+    @After
+    public void tearDown() {
+        ShadowInputDevice.reset();
+    }
+
+    @Test
+    public void shouldAlwaysBeAvailable() {
+        assertThat(mController.isAvailable()).isTrue();
+    }
+
+    @Test
+    @Config(shadows = {
+            ShadowInputDevice.class,
+    })
+    public void updateState_noKeyboard_setDisconnectedSummary() {
+        ShadowInputDevice.sDeviceIds = new int[]{};
+        mController.updateState(mPreference);
+
+        verify(mPreference).setSummary(R.string.disconnected);
+    }
+
+    @Test
+    @Config(shadows = ShadowInputDevice.class)
+    public void updateState_hasKeyboard_setSummaryToKeyboardName() {
+        final InputDevice device = mock(InputDevice.class);
+        when(device.isVirtual()).thenReturn(false);
+        when(device.isFullKeyboard()).thenReturn(true);
+        when(device.getName()).thenReturn("test_keyboard");
+        ShadowInputDevice.sDeviceIds = new int[]{0};
+        ShadowInputDevice.addDevice(0, device);
+
+        mController.updateState(mPreference);
+
+        verify(mPreference).setSummary(device.getName());
+    }
+
+}
diff --git a/tests/robotests/src/com/android/settings/inputmethod/VirtualKeyboardPreferenceControllerTest.java b/tests/robotests/src/com/android/settings/inputmethod/VirtualKeyboardPreferenceControllerTest.java
new file mode 100644 (file)
index 0000000..3256950
--- /dev/null
@@ -0,0 +1,99 @@
+/*
+ * 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.inputmethod;
+
+
+import static com.google.common.truth.Truth.assertThat;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+import android.app.admin.DevicePolicyManager;
+import android.content.ComponentName;
+import android.content.Context;
+import android.content.pm.PackageManager;
+import android.support.v7.preference.Preference;
+import android.view.inputmethod.InputMethodInfo;
+import android.view.inputmethod.InputMethodManager;
+
+import com.android.settings.R;
+import com.android.settings.SettingsRobolectricTestRunner;
+import com.android.settings.TestConfig;
+
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.Mock;
+import org.mockito.MockitoAnnotations;
+import org.robolectric.annotation.Config;
+
+import java.util.ArrayList;
+import java.util.List;
+
+@RunWith(SettingsRobolectricTestRunner.class)
+@Config(manifest = TestConfig.MANIFEST_PATH, sdk = TestConfig.SDK_VERSION)
+public class VirtualKeyboardPreferenceControllerTest {
+
+    @Mock
+    private Context mContext;
+    @Mock
+    private InputMethodManager mImm;
+    @Mock
+    private DevicePolicyManager mDpm;
+    @Mock
+    private PackageManager mPm;
+    @Mock
+    private Preference mPreference;
+
+    private VirtualKeyboardPreferenceController mController;
+
+    @Before
+    public void setUp() {
+        MockitoAnnotations.initMocks(this);
+        when(mContext.getSystemService(Context.DEVICE_POLICY_SERVICE)).thenReturn(mDpm);
+        when(mContext.getSystemService(Context.INPUT_METHOD_SERVICE)).thenReturn(mImm);
+        when(mContext.getPackageManager()).thenReturn(mPm);
+        mController = new VirtualKeyboardPreferenceController(mContext);
+    }
+
+    @Test
+    public void shouldAlwaysBeAvailable() {
+        assertThat(mController.isAvailable()).isTrue();
+    }
+
+    @Test
+    public void updateState_noEnabledIMEs_setEmptySummary() {
+        mController.updateState(mPreference);
+
+        verify(mPreference).setSummary(R.string.summary_empty);
+    }
+
+    @Test
+    public void updateState_singleIme_setImeLabelToSummary() {
+        when(mDpm.getPermittedInputMethodsForCurrentUser()).thenReturn(null);
+        final ComponentName componentName = new ComponentName("pkg", "cls");
+        final List<InputMethodInfo> imis = new ArrayList<>();
+        imis.add(mock(InputMethodInfo.class));
+        when(imis.get(0).getPackageName()).thenReturn(componentName.getPackageName());
+        when(mImm.getEnabledInputMethodList()).thenReturn(imis);
+        when(imis.get(0).loadLabel(mPm)).thenReturn("label");
+
+        mController.updateState(mPreference);
+
+        verify(mPreference).setSummary("label");
+    }
+}
index 026264e..643e305 100644 (file)
 
 package com.android.settings.language;
 
+import static com.google.common.truth.Truth.assertThat;
+import static org.mockito.Matchers.any;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.times;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
 import android.app.Activity;
+import android.app.admin.DevicePolicyManager;
 import android.content.ComponentName;
 import android.content.Context;
 import android.content.pm.PackageManager;
@@ -48,13 +56,6 @@ import org.robolectric.annotation.Config;
 import java.util.ArrayList;
 import java.util.List;
 
-import static com.google.common.truth.Truth.assertThat;
-import static org.mockito.Matchers.any;
-import static org.mockito.Mockito.mock;
-import static org.mockito.Mockito.times;
-import static org.mockito.Mockito.verify;
-import static org.mockito.Mockito.when;
-
 @RunWith(SettingsRobolectricTestRunner.class)
 @Config(manifest = TestConfig.MANIFEST_PATH, sdk = TestConfig.SDK_VERSION)
 public class LanguageAndInputSettingsTest {
@@ -64,6 +65,12 @@ public class LanguageAndInputSettingsTest {
     @Mock
     private PackageManager mPackageManager;
     @Mock
+    private InputManager mIm;
+    @Mock
+    private InputMethodManager mImm;
+    @Mock
+    private DevicePolicyManager mDpm;
+    @Mock
     private InputMethodManager mInputMethodManager;
     private TestFragment mFragment;
 
@@ -72,8 +79,11 @@ public class LanguageAndInputSettingsTest {
         MockitoAnnotations.initMocks(this);
         when(mContext.getSystemService(Context.USER_SERVICE)).thenReturn(mock(UserManager.class));
         when(mContext.getSystemService(Context.INPUT_SERVICE)).thenReturn(mock(InputManager.class));
+        when(mContext.getSystemService(Context.INPUT_SERVICE)).thenReturn(mIm);
         when(mContext.getSystemService(Context.TEXT_SERVICES_MANAGER_SERVICE))
                 .thenReturn(mock(TextServicesManager.class));
+        when(mContext.getSystemService(Context.DEVICE_POLICY_SERVICE)).thenReturn(mDpm);
+        when(mContext.getSystemService(Context.INPUT_METHOD_SERVICE)).thenReturn(mImm);
         mFragment = new TestFragment(mContext);
     }
 
@@ -96,6 +106,7 @@ public class LanguageAndInputSettingsTest {
     }
 
     @Test
+
     public void testGetPreferenceControllers_shouldAllBeCreated() {
         final List<PreferenceController> controllers = mFragment.getPreferenceControllers(mContext);
 
diff --git a/tests/robotests/src/com/android/settings/testutils/shadow/ShadowInputDevice.java b/tests/robotests/src/com/android/settings/testutils/shadow/ShadowInputDevice.java
new file mode 100644 (file)
index 0000000..ecb1b71
--- /dev/null
@@ -0,0 +1,54 @@
+/*
+ * 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.testutils.shadow;
+
+import android.view.InputDevice;
+
+import org.robolectric.annotation.Implementation;
+import org.robolectric.annotation.Implements;
+import org.robolectric.annotation.Resetter;
+
+import java.util.HashMap;
+import java.util.Map;
+
+@Implements(InputDevice.class)
+public class ShadowInputDevice extends org.robolectric.shadows.ShadowInputDevice {
+
+    public static int[] sDeviceIds;
+
+    private static Map<Integer, InputDevice> sDeviceMap = new HashMap<>();
+
+    @Implementation
+    public static int[] getDeviceIds() {
+        return sDeviceIds;
+    }
+
+    @Implementation
+    public static InputDevice getDevice(int id) {
+        return sDeviceMap.get(id);
+    }
+
+    public static void addDevice(int id, InputDevice device) {
+        sDeviceMap.put(id, device);
+    }
+
+    @Resetter
+    public static void reset() {
+        sDeviceIds = null;
+        sDeviceMap.clear();
+    }
+}