diff --git a/tests/unit/src/com/android/settings/privatespace/delete/PrivateSpaceDeletionProgressFragmentTest.java b/tests/unit/src/com/android/settings/privatespace/delete/PrivateSpaceDeletionProgressFragmentTest.java index d6eadd59780..9806540d540 100644 --- a/tests/unit/src/com/android/settings/privatespace/delete/PrivateSpaceDeletionProgressFragmentTest.java +++ b/tests/unit/src/com/android/settings/privatespace/delete/PrivateSpaceDeletionProgressFragmentTest.java @@ -21,10 +21,12 @@ import static com.android.settings.privatespace.PrivateSpaceMaintainer.ErrorDele import static com.google.common.truth.Truth.assertThat; import static org.junit.Assume.assumeTrue; +import static org.mockito.ArgumentMatchers.anyInt; import static org.mockito.Mockito.doNothing; import static org.mockito.Mockito.doReturn; import static org.mockito.Mockito.spy; import static org.mockito.Mockito.verify; +import static org.mockito.Mockito.when; import android.app.settings.SettingsEnums; import android.content.Context; @@ -36,7 +38,9 @@ import androidx.test.annotation.UiThreadTest; import androidx.test.core.app.ApplicationProvider; import androidx.test.ext.junit.runners.AndroidJUnit4; +import com.android.internal.widget.LockPatternUtils; import com.android.settings.privatespace.PrivateSpaceMaintainer; +import com.android.settings.testutils.FakeFeatureFactory; import org.junit.After; import org.junit.Before; @@ -53,6 +57,7 @@ public class PrivateSpaceDeletionProgressFragmentTest { private PrivateSpaceDeletionProgressFragment mFragment; private PrivateSpaceMaintainer mPrivateSpaceMaintainer; @Mock private PrivateSpaceMaintainer mPrivateSpaceMaintainerMock; + @Mock private LockPatternUtils mLockPatternUtils; @UiThreadTest @Before @@ -69,6 +74,10 @@ public class PrivateSpaceDeletionProgressFragmentTest { }; mPrivateSpaceMaintainer = PrivateSpaceMaintainer.getInstance(mContext); mFragment.setPrivateSpaceMaintainer(injector); + final FakeFeatureFactory featureFactory = FakeFeatureFactory.setupForTest(); + when(featureFactory.securityFeatureProvider.getLockPatternUtils(mContext)) + .thenReturn(mLockPatternUtils); + doReturn(true).when(mLockPatternUtils).isSecure(anyInt()); } @After