Report unavailable data usage.
- when showing the data usage graph, if the first data point is equal to the first available data, insert a 0 usage point on the first day of available data, so that it will show a flat line for the days that do not have available data instead of showing a non-zero slope. Change-Id: Ibf305e292a65ab3310ea12efbb575e3a7e1b6012 Fixes: 76155147 Test: make RunSettingsRoboTests
This commit is contained in:
@@ -0,0 +1,129 @@
|
||||
/*
|
||||
* Copyright (C) 2018 The Android Open Source Project
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License
|
||||
*/
|
||||
package com.android.settings.datausage;
|
||||
|
||||
import static com.google.common.truth.Truth.assertThat;
|
||||
import static org.mockito.Mockito.mock;
|
||||
import static org.mockito.Mockito.spy;
|
||||
import static org.mockito.Mockito.verify;
|
||||
import static org.mockito.Mockito.when;
|
||||
|
||||
import android.content.Context;
|
||||
import android.net.NetworkStatsHistory;
|
||||
import android.net.NetworkStatsHistory.Entry;
|
||||
import android.util.SparseIntArray;
|
||||
import com.android.settings.graph.UsageView;
|
||||
import com.android.settings.testutils.SettingsRobolectricTestRunner;
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
import org.junit.runner.RunWith;
|
||||
import org.mockito.ArgumentCaptor;
|
||||
import org.mockito.MockitoAnnotations;
|
||||
import org.robolectric.RuntimeEnvironment;
|
||||
|
||||
@RunWith(SettingsRobolectricTestRunner.class)
|
||||
public class ChartDataUsagePreferenceTest {
|
||||
|
||||
private static final long MILLIS_IN_ONE_HOUR = 60 * 60 * 1000;
|
||||
private static final long MILLIS_IN_ONE_DAY = 24 * MILLIS_IN_ONE_HOUR;
|
||||
|
||||
private NetworkStatsHistory mNetworkStatsHistory;
|
||||
private Context mContext;
|
||||
private ChartDataUsagePreference mPreference;
|
||||
private long mNow;
|
||||
|
||||
@Before
|
||||
public void setUp() {
|
||||
MockitoAnnotations.initMocks(this);
|
||||
|
||||
mContext = RuntimeEnvironment.application;
|
||||
mPreference = new ChartDataUsagePreference(mContext, null);
|
||||
mNow = System.currentTimeMillis();
|
||||
mNetworkStatsHistory = spy(new NetworkStatsHistory(MILLIS_IN_ONE_HOUR, 10));
|
||||
addTestNetworkEntries();
|
||||
mPreference.setNetworkStats(mNetworkStatsHistory);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void calcPoints_notStartOfData_shouldAddDataPointsOnly() {
|
||||
final long start = mNow - 20 * MILLIS_IN_ONE_DAY;
|
||||
final long end = mNow - 5 * MILLIS_IN_ONE_DAY;
|
||||
mPreference.setVisibleRange(start, end);
|
||||
when(mNetworkStatsHistory.getIndexAfter(start)).thenReturn(2);
|
||||
when(mNetworkStatsHistory.getIndexAfter(end)).thenReturn(7);
|
||||
final UsageView usageView = mock(UsageView.class);
|
||||
final ArgumentCaptor<SparseIntArray> pointsCaptor =
|
||||
ArgumentCaptor.forClass(SparseIntArray.class);
|
||||
|
||||
mPreference.calcPoints(usageView);
|
||||
|
||||
verify(usageView).addPath(pointsCaptor.capture());
|
||||
SparseIntArray points = pointsCaptor.getValue();
|
||||
// the point should be normal usage data
|
||||
assertThat(points.valueAt(1)).isNotEqualTo(-1);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void calcPoints_startOfData_shouldIndicateStartOfData() {
|
||||
final long start = mNow - 20 * MILLIS_IN_ONE_DAY;
|
||||
final long end = mNow - 5 * MILLIS_IN_ONE_DAY;
|
||||
mPreference.setVisibleRange(start, end);
|
||||
when(mNetworkStatsHistory.getIndexAfter(start)).thenReturn(0);
|
||||
when(mNetworkStatsHistory.getIndexAfter(end)).thenReturn(5);
|
||||
final UsageView usageView = mock(UsageView.class);
|
||||
final ArgumentCaptor<SparseIntArray> pointsCaptor =
|
||||
ArgumentCaptor.forClass(SparseIntArray.class);
|
||||
|
||||
mPreference.calcPoints(usageView);
|
||||
|
||||
verify(usageView).addPath(pointsCaptor.capture());
|
||||
SparseIntArray points = pointsCaptor.getValue();
|
||||
// indicator that no data is available
|
||||
assertThat(points.keyAt(1)).isEqualTo(points.keyAt(2) - 1);
|
||||
assertThat(points.valueAt(1)).isEqualTo(-1);
|
||||
}
|
||||
|
||||
private void addTestNetworkEntries() {
|
||||
// create 10 arbitary network data
|
||||
mNetworkStatsHistory.setValues(0, createEntry(1521583200000L, 743823454L, 16574289L));
|
||||
mNetworkStatsHistory.setValues(1, createEntry(1521586800000L, 64396L, 160364L));
|
||||
mNetworkStatsHistory.setValues(2, createEntry(1521590400000L, 2832L, 5299L));
|
||||
mNetworkStatsHistory.setValues(3, createEntry(1521655200000L, 83849690L, 3558238L));
|
||||
mNetworkStatsHistory.setValues(4, createEntry(1521658800000L, 1883657L, 353330L));
|
||||
mNetworkStatsHistory.setValues(5, createEntry(1521662400000L, 705259L, 279065L));
|
||||
mNetworkStatsHistory.setValues(5, createEntry(1521666000000L, 216169L, 155302L));
|
||||
mNetworkStatsHistory.setValues(5, createEntry(1521669600000L, 6069175L, 427581L));
|
||||
mNetworkStatsHistory.setValues(5, createEntry(1521673200000L, 120389L, 110807L));
|
||||
mNetworkStatsHistory.setValues(5, createEntry(1521676800000L, 29947L, 73257L));
|
||||
}
|
||||
|
||||
/**
|
||||
* Create a network entry to be used to calculate the usage chart. In the calculation, we only
|
||||
* need bucketStart, total bytes (rx + tx), and bucketDuration (which is set when we create
|
||||
* the NetworkStatsHistory object). Other fields are ignored, so we don't initialize here.
|
||||
* @param start the timestamp when this entry begins
|
||||
* @param rx the total number of received bytes
|
||||
* @param tx the total number of transmitted bytes
|
||||
* @return the network entry with the corresponding start time and data usage
|
||||
*/
|
||||
private Entry createEntry(long start, long rx, long tx) {
|
||||
Entry entry = new Entry();
|
||||
entry.bucketStart = start;
|
||||
entry.rxBytes = rx;
|
||||
entry.txBytes = tx;
|
||||
return entry;
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user