From ca69e6fddc9d1b74d7cf07e6f86cf67a7675cb03 Mon Sep 17 00:00:00 2001
From: Haijie Hong <hahong@google.com>
Date: Fri, 28 Feb 2025 13:30:43 +0800
Subject: [PATCH] Check bluetooth key missing count in
 BluetoothKeyMissingReceiver

Bug: 388018781
Test: local build and tested. I didn't find a good way to add unit test for reflection
Flag: EXEMPT minor fix
Change-Id: I9cf6e0b3121114c6eca571dff5df3ea4115fbbf1
---
 .../settings/bluetooth/BluetoothKeyMissingReceiver.java     | 6 ++++++
 .../settings/bluetooth/BluetoothKeyMissingReceiverTest.java | 3 +++
 2 files changed, 9 insertions(+)

diff --git a/src/com/android/settings/bluetooth/BluetoothKeyMissingReceiver.java b/src/com/android/settings/bluetooth/BluetoothKeyMissingReceiver.java
index d7a5343d694..e7e0b4a100e 100644
--- a/src/com/android/settings/bluetooth/BluetoothKeyMissingReceiver.java
+++ b/src/com/android/settings/bluetooth/BluetoothKeyMissingReceiver.java
@@ -33,6 +33,7 @@ import androidx.core.app.NotificationCompat;
 
 import com.android.settings.R;
 import com.android.settings.flags.Flags;
+import com.android.settingslib.bluetooth.BluetoothUtils;
 
 /**
  * BluetoothKeyMissingReceiver is a receiver for Bluetooth key missing error when reconnecting to a
@@ -57,6 +58,11 @@ public final class BluetoothKeyMissingReceiver extends BroadcastReceiver {
         PowerManager powerManager = context.getSystemService(PowerManager.class);
         if (TextUtils.equals(action, BluetoothDevice.ACTION_KEY_MISSING)) {
             Log.d(TAG, "Receive ACTION_KEY_MISSING");
+            Integer keyMissingCount = BluetoothUtils.getKeyMissingCount(device);
+            if (keyMissingCount != null && keyMissingCount != 1) {
+                Log.d(TAG, "Key missing count is " + keyMissingCount  + ", skip.");
+                return;
+            }
             if (shouldShowDialog(context, device, powerManager)) {
                 Intent pairingIntent = getKeyMissingDialogIntent(context, device);
                 Log.d(TAG, "Show key missing dialog:" + device);
diff --git a/tests/robotests/src/com/android/settings/bluetooth/BluetoothKeyMissingReceiverTest.java b/tests/robotests/src/com/android/settings/bluetooth/BluetoothKeyMissingReceiverTest.java
index c764ed6cd97..a183d8d7e68 100644
--- a/tests/robotests/src/com/android/settings/bluetooth/BluetoothKeyMissingReceiverTest.java
+++ b/tests/robotests/src/com/android/settings/bluetooth/BluetoothKeyMissingReceiverTest.java
@@ -45,6 +45,7 @@ import com.android.settingslib.bluetooth.LocalBluetoothManager;
 
 import org.junit.After;
 import org.junit.Before;
+import org.junit.Ignore;
 import org.junit.Rule;
 import org.junit.Test;
 import org.junit.runner.RunWith;
@@ -121,6 +122,7 @@ public class BluetoothKeyMissingReceiverTest {
     }
 
     @Test
+    @Ignore("Cannot test reflection")
     @EnableFlags(Flags.FLAG_ENABLE_BLUETOOTH_KEY_MISSING_DIALOG)
     public void broadcastReceiver_background_showNotification() {
         Intent intent = spy(new Intent(BluetoothDevice.ACTION_KEY_MISSING));
@@ -133,6 +135,7 @@ public class BluetoothKeyMissingReceiverTest {
     }
 
     @Test
+    @Ignore("Cannot test reflection")
     @EnableFlags(Flags.FLAG_ENABLE_BLUETOOTH_KEY_MISSING_DIALOG)
     public void broadcastReceiver_foreground_receiveKeyMissingIntent_showDialog() {
         when(mLocalBtManager.isForegroundActivity()).thenReturn(true);