Mock LockPatternUtils.isSecure() method in test setup

Added mockd for LockPatternUtils class in test setup as per the comment
in b/323649600#comment7 to handle for the test failure bug.
The tests are passing locally without this.

Bug: 336518132
Test: atest com.android.settings.privatespace
Change-Id: Ie042bb7dd6edefef094d3b658946ff6eaaa6f801
This commit is contained in:
josephpv
2024-04-25 08:58:49 +00:00
parent c44d4f3f6f
commit 6655fcd04e

View File

@@ -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