UserLocationStoreTest.java

@RunWith(PowerMockRunner.class)
@PrepareForTest(ContextCompat.class)
public class UserLocationStoreTest {
  private UserLocationStore userLocationStore;

  @Before public void setUp() throws Exception {
    userLocationStore = new UserLocationStore(PowerMockito.mock(Context.class));
  }

  @Test public void getLocation_shouldReturnDeviceLocation() throws Exception {
    PowerMockito.mockStatic(ContextCompat.class);
    PowerMockito.when(ContextCompat.checkSelfPermission(any(Context.class), anyString()))
        .thenReturn(PackageManager.PERMISSION_GRANTED);
    assertThat(userLocationStore.getLocation().getSource()).isEqualTo("device");
  }

  @Test public void getLocation_shouldReturnWebLocation() throws Exception {
    PowerMockito.mockStatic(ContextCompat.class);
    PowerMockito.when(ContextCompat.checkSelfPermission(any(Context.class), anyString()))
        .thenReturn(PackageManager.PERMISSION_DENIED);
    assertThat(userLocationStore.getLocation().getSource()).isEqualTo("web");
  }
}