1package com.android.settings.deviceinfo.storage; 2 3import static org.mockito.Matchers.any; 4import static org.mockito.Mockito.mock; 5import static org.mockito.Mockito.when; 6 7import android.os.storage.VolumeInfo; 8 9import com.android.settings.testutils.SettingsRobolectricTestRunner; 10import com.android.settingslib.deviceinfo.PrivateStorageInfo; 11import com.android.settingslib.deviceinfo.StorageVolumeProvider; 12 13import static com.google.common.truth.Truth.assertThat; 14 15import org.junit.Test; 16import org.junit.runner.RunWith; 17 18@RunWith(SettingsRobolectricTestRunner.class) 19public class VolumeSizesLoaderTest { 20 @Test 21 public void getVolumeSize_getsValidSizes() throws Exception { 22 VolumeInfo info = mock(VolumeInfo.class); 23 StorageVolumeProvider storageVolumeProvider = mock(StorageVolumeProvider.class); 24 when(storageVolumeProvider.getTotalBytes(any(), any())).thenReturn(10000L); 25 when(storageVolumeProvider.getFreeBytes(any(), any())).thenReturn(1000L); 26 27 PrivateStorageInfo storageInfo = 28 VolumeSizesLoader.getVolumeSize(storageVolumeProvider, null, info); 29 30 assertThat(storageInfo.freeBytes).isEqualTo(1000L); 31 assertThat(storageInfo.totalBytes).isEqualTo(10000L); 32 } 33} 34