Merge "Add DataStallStatsUtilsTest"

am: 6002e86ede

Change-Id: I257d3da9b47571bdbc4efce54ef5500b7c13a2be

1 files changed, 7 insertions, 1 deletions

diff --git a/src/com/android/networkstack/metrics/DataStallStatsUtils.java b/src/com/android/networkstack/metrics/DataStallStatsUtils.java
index 9308901..59e8fd3 100644
--- a/src/com/android/networkstack/metrics/DataStallStatsUtils.java
+++ b/src/com/android/networkstack/metrics/DataStallStatsUtils.java
@@ -21,6 +21,8 @@ import android.annotation.Nullable;
 import android.net.captiveportal.CaptivePortalProbeResult;
 import android.util.Log;
 
+import androidx.annotation.VisibleForTesting;
+
 import com.android.internal.util.HexDump;
 import com.android.server.connectivity.nano.DataStallEventProto;
 
@@ -38,7 +40,11 @@ public class DataStallStatsUtils {
     private static final String TAG = DataStallStatsUtils.class.getSimpleName();
     private static final boolean DBG = false;
 
-    private static int probeResultToEnum(@Nullable final CaptivePortalProbeResult result) {
+    /**
+     * Map {@link CaptivePortalProbeResult} to {@link DataStallEventProto}.
+     */
+    @VisibleForTesting
+    public static int probeResultToEnum(@Nullable final CaptivePortalProbeResult result) {
         if (result == null) return DataStallEventProto.INVALID;
 
         if (result.isSuccessful()) {