From 5d69ff508a2fe47e7f425b97a0d206306a0dcd70 Mon Sep 17 00:00:00 2001
From: Yuri Ufimtsev <yufimtsev@google.com>
Date: Wed, 23 Feb 2022 09:58:54 +0000
Subject: [PATCH] Use FakeFeatureFactory in SafetySourceBroadcastReceiverTest

Test: atest SettingsUnitTests

Bug: 215517420
Change-Id: I5beefa766440dd86a5b159160699c3bc0ca46e0c
---
 .../safetycenter/SafetySourceBroadcastReceiverTest.java  | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/tests/unit/src/com/android/settings/safetycenter/SafetySourceBroadcastReceiverTest.java b/tests/unit/src/com/android/settings/safetycenter/SafetySourceBroadcastReceiverTest.java
index d81e8afed3a..f2a28ff04c7 100644
--- a/tests/unit/src/com/android/settings/safetycenter/SafetySourceBroadcastReceiverTest.java
+++ b/tests/unit/src/com/android/settings/safetycenter/SafetySourceBroadcastReceiverTest.java
@@ -34,6 +34,9 @@ import android.safetycenter.SafetySourceData;
 import androidx.test.core.app.ApplicationProvider;
 import androidx.test.ext.junit.runners.AndroidJUnit4;
 
+import com.android.internal.widget.LockPatternUtils;
+import com.android.settings.testutils.FakeFeatureFactory;
+
 import org.junit.After;
 import org.junit.Before;
 import org.junit.Test;
@@ -50,10 +53,16 @@ public class SafetySourceBroadcastReceiverTest {
     @Mock
     private SafetyCenterManagerWrapper mSafetyCenterManagerWrapper;
 
+    @Mock
+    private LockPatternUtils mLockPatternUtils;
+
     @Before
     public void setUp() {
         MockitoAnnotations.initMocks(this);
         mApplicationContext = ApplicationProvider.getApplicationContext();
+        final FakeFeatureFactory featureFactory = FakeFeatureFactory.setupForTest();
+        when(featureFactory.securityFeatureProvider.getLockPatternUtils(mApplicationContext))
+                .thenReturn(mLockPatternUtils);
         SafetyCenterManagerWrapper.sInstance = mSafetyCenterManagerWrapper;
     }