Merge "Clean up PhoneTypeHelperTest" into androidx-main

diff --git a/wear/wear-phone-interactions/src/test/java/androidx/wear/phone/interactions/PhoneTypeHelperTest.kt b/wear/wear-phone-interactions/src/test/java/androidx/wear/phone/interactions/PhoneTypeHelperTest.kt
index fec1f8f..130e3a8 100644
--- a/wear/wear-phone-interactions/src/test/java/androidx/wear/phone/interactions/PhoneTypeHelperTest.kt
+++ b/wear/wear-phone-interactions/src/test/java/androidx/wear/phone/interactions/PhoneTypeHelperTest.kt

@@ -60,16 +60,7 @@
 
     @Test
     fun testGetDeviceType_returnsIosWhenAltMode() {
-        Mockito.`when`(
-            mockContentProvider!!.query(
-                ArgumentMatchers.eq(bluetoothModeUri),
-                ArgumentMatchers.any(),
-                ArgumentMatchers.any(),
-                ArgumentMatchers.any(),
-                ArgumentMatchers.any()
-            )
-        )
-            .thenReturn(createFakeBluetoothModeCursor(PhoneTypeHelper.IOS_MODE))
+        createFakePhoneTypeQuery(PhoneTypeHelper.IOS_MODE)
         assertEquals(
             getPhoneDeviceType(ApplicationProvider.getApplicationContext()).toLong(),
             PhoneTypeHelper.DEVICE_TYPE_IOS.toLong()
@@ -78,16 +69,7 @@
 
     @Test
     fun testGetDeviceType_returnsAndroidWhenNonAltMode() {
-        Mockito.`when`(
-            mockContentProvider!!.query(
-                ArgumentMatchers.eq(bluetoothModeUri),
-                ArgumentMatchers.any(),
-                ArgumentMatchers.any(),
-                ArgumentMatchers.any(),
-                ArgumentMatchers.any()
-            )
-        )
-            .thenReturn(createFakeBluetoothModeCursor(PhoneTypeHelper.ANDROID_MODE))
+        createFakePhoneTypeQuery(PhoneTypeHelper.ANDROID_MODE)
         assertEquals(
             getPhoneDeviceType(ApplicationProvider.getApplicationContext()).toLong(),
             PhoneTypeHelper.DEVICE_TYPE_ANDROID.toLong()
@@ -96,16 +78,7 @@
 
     @Test
     fun testGetDeviceType_returnsErrorWhenModeUnknown() {
-        Mockito.`when`(
-            mockContentProvider!!.query(
-                ArgumentMatchers.eq(bluetoothModeUri),
-                ArgumentMatchers.any(),
-                ArgumentMatchers.any(),
-                ArgumentMatchers.any(),
-                ArgumentMatchers.any()
-            )
-        )
-            .thenReturn(createFakeBluetoothModeCursor(PhoneTypeHelper.DEVICE_TYPE_UNKNOWN))
+        createFakePhoneTypeQuery(PhoneTypeHelper.DEVICE_TYPE_UNKNOWN)
         assertEquals(
             getPhoneDeviceType(ApplicationProvider.getApplicationContext()).toLong(),
             PhoneTypeHelper.DEVICE_TYPE_UNKNOWN.toLong()
@@ -184,4 +157,17 @@
             return cursor
         }
     }
+
+    private fun createFakePhoneTypeQuery(phoneType: Int) {
+        Mockito.`when`(
+            mockContentProvider!!.query(
+                ArgumentMatchers.eq(bluetoothModeUri),
+                ArgumentMatchers.any(),
+                ArgumentMatchers.any(),
+                ArgumentMatchers.any(),
+                ArgumentMatchers.any()
+            )
+        )
+            .thenReturn(createFakeBluetoothModeCursor(phoneType))
+    }
 }
\ No newline at end of file