1 /* 2 * Copyright (C) 2020 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 package com.android.settings.datausage; 18 19 import static com.google.common.truth.Truth.assertThat; 20 21 import android.net.NetworkPolicy; 22 import android.net.NetworkTemplate; 23 24 import androidx.test.ext.junit.runners.AndroidJUnit4; 25 26 import com.android.settingslib.net.DataUsageController.DataUsageInfo; 27 28 import org.junit.Before; 29 import org.junit.Test; 30 import org.junit.runner.RunWith; 31 32 @RunWith(AndroidJUnit4.class) 33 public class DataUsageInfoControllerTest { 34 35 private static final int NEGATIVE = -1; 36 private static final int ZERO = 0; 37 private static final int POSITIVE_SMALL = 1; 38 private static final int POSITIVE_LARGE = 5; 39 40 private DataUsageInfoController mInfoController; 41 private DataUsageInfo info; 42 43 @Before setUp()44 public void setUp() { 45 mInfoController = new DataUsageInfoController(); 46 info = new DataUsageInfo(); 47 } 48 49 @Test getSummaryLimit_LowUsageLowWarning_LimitUsed()50 public void getSummaryLimit_LowUsageLowWarning_LimitUsed() { 51 info.warningLevel = POSITIVE_SMALL; 52 info.limitLevel = POSITIVE_LARGE; 53 info.usageLevel = POSITIVE_SMALL; 54 55 assertThat(mInfoController.getSummaryLimit(info)).isEqualTo(info.limitLevel); 56 } 57 58 @Test getSummaryLimit_LowUsageEqualWarning_LimitUsed()59 public void getSummaryLimit_LowUsageEqualWarning_LimitUsed() { 60 info.warningLevel = POSITIVE_LARGE; 61 info.limitLevel = POSITIVE_LARGE; 62 info.usageLevel = POSITIVE_SMALL; 63 64 assertThat(mInfoController.getSummaryLimit(info)).isEqualTo(info.limitLevel); 65 } 66 67 @Test getSummaryLimit_NoLimitNoUsage_WarningUsed()68 public void getSummaryLimit_NoLimitNoUsage_WarningUsed() { 69 info.warningLevel = POSITIVE_LARGE; 70 info.limitLevel = ZERO; 71 info.usageLevel = ZERO; 72 73 assertThat(mInfoController.getSummaryLimit(info)).isEqualTo(info.warningLevel); 74 } 75 76 @Test getSummaryLimit_NoLimitLowUsage_WarningUsed()77 public void getSummaryLimit_NoLimitLowUsage_WarningUsed() { 78 info.warningLevel = POSITIVE_LARGE; 79 info.limitLevel = ZERO; 80 info.usageLevel = POSITIVE_SMALL; 81 82 assertThat(mInfoController.getSummaryLimit(info)).isEqualTo(info.warningLevel); 83 } 84 85 @Test getSummaryLimit_LowWarningNoLimit_UsageUsed()86 public void getSummaryLimit_LowWarningNoLimit_UsageUsed() { 87 info.warningLevel = POSITIVE_SMALL; 88 info.limitLevel = ZERO; 89 info.usageLevel = POSITIVE_LARGE; 90 91 assertThat(mInfoController.getSummaryLimit(info)).isEqualTo(info.usageLevel); 92 } 93 94 @Test getSummaryLimit_LowWarningLowLimit_UsageUsed()95 public void getSummaryLimit_LowWarningLowLimit_UsageUsed() { 96 info.warningLevel = POSITIVE_SMALL; 97 info.limitLevel = POSITIVE_SMALL; 98 info.usageLevel = POSITIVE_LARGE; 99 100 assertThat(mInfoController.getSummaryLimit(info)).isEqualTo(info.usageLevel); 101 } 102 getDefaultNetworkPolicy()103 private NetworkPolicy getDefaultNetworkPolicy() { 104 NetworkTemplate template = NetworkTemplate.buildTemplateWifi( 105 NetworkTemplate.WIFI_NETWORKID_ALL, null /* subscriberId */); 106 int cycleDay = -1; 107 String cycleTimezone = "UTC"; 108 long warningBytes = -1; 109 long limitBytes = -1; 110 return new NetworkPolicy(template, cycleDay, cycleTimezone, warningBytes, limitBytes, true); 111 } 112 113 @Test updateDataLimit_NullArguments_NoError()114 public void updateDataLimit_NullArguments_NoError() { 115 mInfoController.updateDataLimit(null, null); 116 mInfoController.updateDataLimit(info, null); 117 mInfoController.updateDataLimit(null, getDefaultNetworkPolicy()); 118 } 119 120 @Test updateDataLimit_NegativeWarning_UpdatedToZero()121 public void updateDataLimit_NegativeWarning_UpdatedToZero() { 122 NetworkPolicy policy = getDefaultNetworkPolicy(); 123 policy.warningBytes = NEGATIVE; 124 125 mInfoController.updateDataLimit(info, policy); 126 127 assertThat(info.warningLevel).isEqualTo(ZERO); 128 } 129 130 @Test updateDataLimit_WarningZero_UpdatedToZero()131 public void updateDataLimit_WarningZero_UpdatedToZero() { 132 NetworkPolicy policy = getDefaultNetworkPolicy(); 133 policy.warningBytes = ZERO; 134 135 mInfoController.updateDataLimit(info, policy); 136 137 assertThat(info.warningLevel).isEqualTo(ZERO); 138 } 139 140 @Test updateDataLimit_WarningPositive_UpdatedToWarning()141 public void updateDataLimit_WarningPositive_UpdatedToWarning() { 142 NetworkPolicy policy = getDefaultNetworkPolicy(); 143 policy.warningBytes = POSITIVE_SMALL; 144 145 mInfoController.updateDataLimit(info, policy); 146 147 assertThat(info.warningLevel).isEqualTo(policy.warningBytes); 148 } 149 150 @Test updateDataLimit_LimitNegative_UpdatedToZero()151 public void updateDataLimit_LimitNegative_UpdatedToZero() { 152 NetworkPolicy policy = getDefaultNetworkPolicy(); 153 policy.limitBytes = NEGATIVE; 154 155 mInfoController.updateDataLimit(info, policy); 156 157 assertThat(info.limitLevel).isEqualTo(ZERO); 158 } 159 160 @Test updateDataLimit_LimitZero_UpdatedToZero()161 public void updateDataLimit_LimitZero_UpdatedToZero() { 162 NetworkPolicy policy = getDefaultNetworkPolicy(); 163 policy.limitBytes = ZERO; 164 165 mInfoController.updateDataLimit(info, policy); 166 167 assertThat(info.limitLevel).isEqualTo(ZERO); 168 } 169 170 @Test updateDataLimit_LimitPositive_UpdatedToLimit()171 public void updateDataLimit_LimitPositive_UpdatedToLimit() { 172 NetworkPolicy policy = getDefaultNetworkPolicy(); 173 policy.limitBytes = POSITIVE_SMALL; 174 175 mInfoController.updateDataLimit(info, policy); 176 177 assertThat(info.limitLevel).isEqualTo(policy.limitBytes); 178 } 179 } 180