From 64fdef1dc7c354d55d991ca91105818dcb5e423a Mon Sep 17 00:00:00 2001
From: Joshua Mccloskey <joshmccloskey@google.com>
Date: Thu, 15 Apr 2021 14:02:51 -0700
Subject: [PATCH] Updated settings for new get/set feature change

Test: Verified e2e get/set work within settings
Bug: 184657294
Change-Id: I0240b092751eb1ca205d1cf245c005c0fe9d4338
---
 .../FaceSettingsAttentionPreferenceController.java | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/src/com/android/settings/biometrics/face/FaceSettingsAttentionPreferenceController.java b/src/com/android/settings/biometrics/face/FaceSettingsAttentionPreferenceController.java
index 200c0b949ec..0eb823bc4d9 100644
--- a/src/com/android/settings/biometrics/face/FaceSettingsAttentionPreferenceController.java
+++ b/src/com/android/settings/biometrics/face/FaceSettingsAttentionPreferenceController.java
@@ -57,15 +57,15 @@ public class FaceSettingsAttentionPreferenceController extends FaceSettingsPrefe
 
     private final GetFeatureCallback mGetFeatureCallback = new GetFeatureCallback() {
         @Override
-        public void onCompleted(boolean success, int feature, boolean value) {
-            if (feature == FaceManager.FEATURE_REQUIRE_ATTENTION && success) {
-                if (!mFaceManager.hasEnrolledTemplates(getUserId())) {
-                    mPreference.setEnabled(false);
-                } else {
-                    mPreference.setEnabled(true);
-                    mPreference.setChecked(value);
+        public void onCompleted(boolean success, int[] features, boolean[] featureState) {
+            boolean requireAttentionEnabled = false;
+            for (int i = 0; i < features.length; i++) {
+                if (features[i] == FaceManager.FEATURE_REQUIRE_ATTENTION) {
+                    requireAttentionEnabled = featureState[i];
                 }
             }
+            mPreference.setEnabled(success);
+            mPreference.setChecked(requireAttentionEnabled);
         }
     };