Merge "Add package name check for the calling app" into sc-v2-dev am: 12b7fd54ec
Original change: https://googleplex-android-review.googlesource.com/c/platform/packages/apps/Settings/+/16062219 Change-Id: I248616eaed0c9c3d19612d702b97e851f1847e2a
This commit is contained in:
@@ -18,27 +18,70 @@ package com.android.settings.wifi.addappnetworks;
|
||||
|
||||
import static com.google.common.truth.Truth.assertThat;
|
||||
|
||||
import static org.robolectric.Shadows.shadowOf;
|
||||
import android.annotation.Nullable;
|
||||
import android.app.IActivityManager;
|
||||
import android.os.RemoteException;
|
||||
|
||||
import static org.mockito.ArgumentMatchers.any;
|
||||
import static org.mockito.Mockito.when;
|
||||
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
import org.junit.runner.RunWith;
|
||||
import org.mockito.Mock;
|
||||
import org.mockito.MockitoAnnotations;
|
||||
import org.robolectric.Robolectric;
|
||||
import org.robolectric.RobolectricTestRunner;
|
||||
import org.robolectric.RuntimeEnvironment;
|
||||
|
||||
@RunWith(RobolectricTestRunner.class)
|
||||
public class AddAppNetworksActivityTest {
|
||||
|
||||
@Mock
|
||||
private IActivityManager mIActivityManager;
|
||||
|
||||
private AddAppNetworksActivity mActivity;
|
||||
|
||||
@Before
|
||||
public void setUp() {
|
||||
MockitoAnnotations.initMocks(this);
|
||||
|
||||
mActivity = Robolectric.buildActivity(AddAppNetworksActivity.class).create().get();
|
||||
mActivity.mActivityManager = mIActivityManager;
|
||||
}
|
||||
|
||||
@Test
|
||||
public void startActivity_withPackageName_bundleShouldHaveRightPackageName() {
|
||||
final String packageName = RuntimeEnvironment.application.getPackageName();
|
||||
final AddAppNetworksActivity activity =
|
||||
Robolectric.buildActivity(AddAppNetworksActivity.class).create().get();
|
||||
shadowOf(activity).setCallingPackage(packageName);
|
||||
public void getCallingAppPackageName_nullPackageName_returnNotNull() {
|
||||
fakeCallingPackage("com.android.settings");
|
||||
|
||||
activity.showAddNetworksFragment();
|
||||
assertThat(mActivity.getCallingAppPackageName()).isNotNull();
|
||||
}
|
||||
|
||||
assertThat(activity.mBundle.getString(AddAppNetworksActivity.KEY_CALLING_PACKAGE_NAME))
|
||||
.isEqualTo(packageName);
|
||||
@Test
|
||||
public void getCallingAppPackageName_withPackageName_returnNull() {
|
||||
fakeCallingPackage(null);
|
||||
|
||||
assertThat(mActivity.getCallingAppPackageName()).isNull();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void showAddNetworksFragment_nullPackageName_returnFalse() {
|
||||
fakeCallingPackage(null);
|
||||
|
||||
assertThat(mActivity.showAddNetworksFragment()).isFalse();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void showAddNetworksFragment_withPackageName_returnTrue() {
|
||||
fakeCallingPackage("com.android.settings");
|
||||
|
||||
assertThat(mActivity.showAddNetworksFragment()).isTrue();
|
||||
}
|
||||
|
||||
private void fakeCallingPackage(@Nullable String packageName) {
|
||||
try {
|
||||
when(mIActivityManager.getLaunchedFromPackage(any())).thenReturn(packageName);
|
||||
} catch (RemoteException e) {
|
||||
// Do nothing.
|
||||
}
|
||||
}
|
||||
}
|
||||
|
Reference in New Issue
Block a user