/*
 * Copyright (C) 2017 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.server.wifi.aware;

import static android.net.wifi.aware.WifiAwareNetworkSpecifier.NETWORK_SPECIFIER_TYPE_IB;
import static android.net.wifi.aware.WifiAwareNetworkSpecifier.NETWORK_SPECIFIER_TYPE_IB_ANY_PEER;

import static com.android.server.wifi.aware.WifiAwareMetrics.addNanHalStatusToHistogram;
import static com.android.server.wifi.aware.WifiAwareMetrics.histogramToProtoArray;

import static org.hamcrest.core.IsEqual.equalTo;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.when;

import android.app.AppOpsManager;
import android.content.Context;
import android.net.wifi.aware.WifiAwareManager;
import android.net.wifi.aware.WifiAwareNetworkSpecifier;
import android.util.LocalLog;
import android.util.Log;
import android.util.SparseArray;
import android.util.SparseIntArray;

import androidx.test.filters.SmallTest;

import com.android.dx.mockito.inline.extended.ExtendedMockito;
import com.android.server.wifi.Clock;
import com.android.server.wifi.WifiBaseTest;
import com.android.server.wifi.hal.WifiNanIface.NanStatusCode;
import com.android.server.wifi.proto.WifiStatsLog;
import com.android.server.wifi.proto.nano.WifiMetricsProto;
import com.android.server.wifi.util.MetricsUtils;
import com.android.server.wifi.util.WifiPermissionsUtil;

import org.junit.After;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.rules.ErrorCollector;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import org.mockito.MockitoSession;
import org.mockito.quality.Strictness;

import java.io.PrintWriter;
import java.io.StringWriter;
import java.util.HashMap;
import java.util.Map;

/**
 * Unit test harness for WifiAwareMetrics
 */
@SmallTest
public class WifiAwareMetricsTest extends WifiBaseTest {
    @Mock Clock mClock;
    @Mock private Context mMockContext;
    @Mock private AppOpsManager mMockAppOpsManager;
    @Mock private WifiPermissionsUtil mWifiPermissionsUtil;
    @Rule public ErrorCollector collector = new ErrorCollector();
    @Mock private PairingConfigManager mPairingConfigManager;

    private WifiAwareMetrics mDut;
    private LocalLog mLocalLog = new LocalLog(512);
    private MockitoSession mSession;

    // Histogram definition: start[i] = b + p * m^i with s sub-buckets, i=0,...,n-1

    /**
     * Histogram of following buckets, start[i] = 0 + 1 * 10^i with 9 sub-buckets, i=0,...,5
     * 1 - 10: 9 sub-buckets each of width 1
     * 10 - 100: 10
     * 100 - 10e3: 10^2
     * 10e3 - 10e4: 10^3
     * 10e4 - 10e5: 10^4
     * 10e5 - 10e6: 10^5
     */
    private static final MetricsUtils.LogHistParms HIST1 = new MetricsUtils.LogHistParms(0, 1,
            10, 9, 6);

    /**
     * Histogram of following buckets, start[i] = -20 + 2 * 5^i with 40 sub-buckets, i=0,...,2
     * -18 - -10: 40 sub-bucket each of width 0.2
     * -10 - 30: 1
     * 30 - 230: 5
     */
    private static final MetricsUtils.LogHistParms HIST2 = new MetricsUtils.LogHistParms(-20, 2,
            5, 40, 3);

    // Linear histogram of following buckets:
    //   <10
    //   [10, 30)
    //   [30, 60)
    //   [60, 100)
    //   >100
    private static final int[] HIST_LINEAR = { 10, 30, 60, 100 };

    /**
     * Pre-test configuration. Initialize and install mocks.
     */
    @Before
    public void setUp() throws Exception {
        MockitoAnnotations.initMocks(this);
        when(mMockContext.getSystemService(Context.APP_OPS_SERVICE)).thenReturn(mMockAppOpsManager);
        setTime(0);

        mDut = new WifiAwareMetrics(mClock);
        mSession = ExtendedMockito.mockitoSession()
                .strictness(Strictness.LENIENT)
                .mockStatic(WifiStatsLog.class)
                .startMocking();
    }

    @After
    public void tearDown() {
        mSession.finishMocking();
    }

    /**
     * Validates that recordEnableUsage() and recordDisableUsage() record valid metrics.
     */
    @Test
    public void testEnableDisableUsageMetrics() {
        WifiMetricsProto.WifiAwareLog log;

        // create 2 records
        setTime(5);
        mDut.recordEnableUsage();
        setTime(10);
        mDut.recordDisableUsage();
        setTime(11);
        mDut.recordEnableUsage();
        setTime(12);
        mDut.recordDisableUsage();

        setTime(14);
        log = mDut.consolidateProto();
        collector.checkThat(countAllHistogramSamples(log.histogramAwareAvailableDurationMs),
                equalTo(2));
        validateProtoHistBucket("Duration[0] #1", log.histogramAwareAvailableDurationMs[0], 1, 2,
                1);
        validateProtoHistBucket("Duration[1] #1", log.histogramAwareAvailableDurationMs[1], 5, 6,
                1);
        collector.checkThat(log.availableTimeMs, equalTo(6L));

        // create another partial record
        setTime(15);
        mDut.recordEnableUsage();

        setTime(17);
        log = mDut.consolidateProto();
        collector.checkThat(countAllHistogramSamples(log.histogramAwareAvailableDurationMs),
                equalTo(2));
        validateProtoHistBucket("Duration[0] #2", log.histogramAwareAvailableDurationMs[0], 1, 2,
                1);
        validateProtoHistBucket("Duration[1] #2", log.histogramAwareAvailableDurationMs[1], 5, 6,
                1);
        collector.checkThat(log.availableTimeMs, equalTo(8L)); // the partial record of 2ms


        // clear and continue that partial record (verify completed)
        mDut.clear();
        setTime(23);
        mDut.recordDisableUsage();

        log = mDut.consolidateProto();
        collector.checkThat(countAllHistogramSamples(log.histogramAwareAvailableDurationMs),
                equalTo(1));
        validateProtoHistBucket("Duration[0] #3", log.histogramAwareAvailableDurationMs[0], 8, 9,
                1);
        collector.checkThat(log.availableTimeMs, equalTo(6L)); // the remnant record of 6ms

        // clear and verify empty records
        mDut.clear();
        log = mDut.consolidateProto();
        collector.checkThat(countAllHistogramSamples(log.histogramAwareAvailableDurationMs),
                equalTo(0));
    }

    /**
     * Validates that recordEnableAware() and recordDisableAware() record valid metrics.
     */
    @Test
    public void testEnableDisableAwareMetrics() {
        WifiMetricsProto.WifiAwareLog log;

        // create 2 records
        setTime(5);
        mDut.recordEnableAware();
        setTime(10);
        mDut.recordDisableAware();
        setTime(11);
        mDut.recordEnableAware();
        setTime(12);
        mDut.recordDisableAware();

        setTime(14);
        log = mDut.consolidateProto();
        collector.checkThat(countAllHistogramSamples(log.histogramAwareEnabledDurationMs),
                equalTo(2));
        validateProtoHistBucket("Duration[0] #1", log.histogramAwareEnabledDurationMs[0], 1, 2,
                1);
        validateProtoHistBucket("Duration[1] #1", log.histogramAwareEnabledDurationMs[1], 5, 6,
                1);
        collector.checkThat(log.enabledTimeMs, equalTo(6L));

        // create another partial record
        setTime(15);
        mDut.recordEnableAware();

        setTime(17);
        log = mDut.consolidateProto();
        collector.checkThat(countAllHistogramSamples(log.histogramAwareEnabledDurationMs),
                equalTo(2));
        validateProtoHistBucket("Duration[0] #2", log.histogramAwareEnabledDurationMs[0], 1, 2,
                1);
        validateProtoHistBucket("Duration[1] #2", log.histogramAwareEnabledDurationMs[1], 5, 6,
                1);
        collector.checkThat(log.enabledTimeMs, equalTo(8L)); // the partial record of 2ms


        // clear and continue that partial record (verify completed)
        mDut.clear();
        setTime(23);
        mDut.recordDisableAware();

        log = mDut.consolidateProto();
        collector.checkThat(countAllHistogramSamples(log.histogramAwareEnabledDurationMs),
                equalTo(1));
        validateProtoHistBucket("Duration[0] #3", log.histogramAwareEnabledDurationMs[0], 8, 9,
                1);
        collector.checkThat(log.enabledTimeMs, equalTo(6L)); // the remnant record of 6ms

        // clear and verify empty records
        mDut.clear();
        log = mDut.consolidateProto();
        collector.checkThat(countAllHistogramSamples(log.histogramAwareEnabledDurationMs),
                equalTo(0));
    }

    @Test
    public void testAttachSessionMetrics() {
        final int uid1 = 1005;
        final int uid2 = 1006;
        final String tag1 = "tag1";
        final String tag2 = "tag2";
        final SparseArray<WifiAwareClientState> clients = new SparseArray<>();
        WifiMetricsProto.WifiAwareLog log;

        setTime(5);

        // uid1: session 1
        clients.put(10,
                new WifiAwareClientState(mMockContext, 10, uid1, 0, null, null, null, null, false,
                        mClock.getElapsedSinceBootMillis(), mWifiPermissionsUtil, null,
                        false, 6));
        mDut.recordAttachSession(uid1, false, clients, 6, tag1);

        // uid1: session 2
        clients.put(11,
                new WifiAwareClientState(mMockContext, 11, uid1, 0, null, null, null, null, false,
                        mClock.getElapsedSinceBootMillis(), mWifiPermissionsUtil, null,
                        false, 6));
        mDut.recordAttachSession(uid1, false, clients, 6, tag1);

        // uid2: session 1
        clients.put(12,
                new WifiAwareClientState(mMockContext, 12, uid2, 0, null, null, null, null, false,
                        mClock.getElapsedSinceBootMillis(), mWifiPermissionsUtil, null,
                        false, 6));
        mDut.recordAttachSession(uid2, false, clients, 6, tag2);

        // uid2: session 2
        clients.put(13,
                new WifiAwareClientState(mMockContext, 13, uid2, 0, null, null, null, null, true,
                        mClock.getElapsedSinceBootMillis(), mWifiPermissionsUtil, null,
                        false, 6));
        mDut.recordAttachSession(uid2, true, clients, 6, tag2);

        // uid2: delete session 1
        setTime(10);
        mDut.recordAttachSessionDuration(clients.get(12).getCreationTime());
        clients.delete(12);

        // uid2: delete session 2
        setTime(15);
        mDut.recordAttachSessionDuration(clients.get(13).getCreationTime());
        clients.delete(13);

        // uid2: session 3
        clients.put(14,
                new WifiAwareClientState(mMockContext, 14, uid2, 0, null, null, null, null, false,
                        mClock.getElapsedSinceBootMillis(), mWifiPermissionsUtil, null,
                        false, 6));
        mDut.recordAttachSession(uid2, false, clients, 6, tag2);

        // a few failures
        mDut.recordAttachStatus(NanStatusCode.INTERNAL_FAILURE, 6, tag1, uid1);
        mDut.recordAttachStatus(NanStatusCode.INTERNAL_FAILURE, 6, tag2, uid2);
        mDut.recordAttachStatus(-5, 6, tag2, uid2); // invalid

        // verify
        log = mDut.consolidateProto();

        collector.checkThat("numApps", log.numApps, equalTo(2));
        collector.checkThat("numAppsUsingIdentityCallback", log.numAppsUsingIdentityCallback,
                equalTo(1));
        collector.checkThat("maxConcurrentAttachSessionsInApp",
                log.maxConcurrentAttachSessionsInApp, equalTo(2));
        collector.checkThat("histogramAttachSessionStatus.length",
                log.histogramAttachSessionStatus.length, equalTo(3)); // 3 buckets
        validateNanStatusProtoHistBucket("Bucket[SUCCESS]",
                log.histogramAttachSessionStatus[0],
                WifiMetricsProto.WifiAwareLog.SUCCESS, 5);
        validateNanStatusProtoHistBucket("Bucket[INTERNAL_FAILURE]",
                log.histogramAttachSessionStatus[1],
                WifiMetricsProto.WifiAwareLog.INTERNAL_FAILURE, 2);
        validateNanStatusProtoHistBucket("Bucket[UNKNOWN_HAL_STATUS]",
                log.histogramAttachSessionStatus[2],
                WifiMetricsProto.WifiAwareLog.UNKNOWN_HAL_STATUS, 1);
        collector.checkThat("histogramAttachDurationMs.length",
                log.histogramAttachDurationMs.length, equalTo(2));
        validateProtoHistBucket("Duration[0]", log.histogramAttachDurationMs[0], 5, 6, 1);
        validateProtoHistBucket("Duration[1]", log.histogramAttachDurationMs[1], 10, 20, 1);
        ExtendedMockito.verify(() -> WifiStatsLog.write(
                WifiStatsLog.WIFI_AWARE_ATTACH_REPORTED,
                WifiStatsLog.WIFI_AWARE_ATTACH_REPORTED__STATUS__ST_SUCCESS,
                WifiStatsLog.WIFI_AWARE_ATTACH_REPORTED__CALLER_TYPE__OTHERS, tag1, uid1),
                times(2));
        ExtendedMockito.verify(() -> WifiStatsLog.write(
                        WifiStatsLog.WIFI_AWARE_ATTACH_REPORTED,
                        WifiStatsLog.WIFI_AWARE_ATTACH_REPORTED__STATUS__ST_SUCCESS,
                        WifiStatsLog.WIFI_AWARE_ATTACH_REPORTED__CALLER_TYPE__OTHERS, tag2, uid2),
                times(3));
        ExtendedMockito.verify(() -> WifiStatsLog.write(
                WifiStatsLog.WIFI_AWARE_ATTACH_REPORTED,
                WifiStatsLog.WIFI_AWARE_ATTACH_REPORTED__STATUS__ST_INTERNAL_FAILURE,
                WifiStatsLog.WIFI_AWARE_ATTACH_REPORTED__CALLER_TYPE__OTHERS, tag1, uid1),
                times(1));
        ExtendedMockito.verify(() -> WifiStatsLog.write(
                WifiStatsLog.WIFI_AWARE_ATTACH_REPORTED,
                WifiStatsLog.WIFI_AWARE_ATTACH_REPORTED__STATUS__ST_INTERNAL_FAILURE,
                WifiStatsLog.WIFI_AWARE_ATTACH_REPORTED__CALLER_TYPE__OTHERS, tag2, uid2),
                times(1));
        ExtendedMockito.verify(() -> WifiStatsLog.write(
                WifiStatsLog.WIFI_AWARE_ATTACH_REPORTED,
                WifiStatsLog.WIFI_AWARE_ATTACH_REPORTED__STATUS__ST_GENERIC_FAILURE,
                WifiStatsLog.WIFI_AWARE_ATTACH_REPORTED__CALLER_TYPE__OTHERS, tag2, uid2),
                times(1));
    }

    @Test
    public void testDiscoverySessionMetrics() {
        final int uid1 = 1005;
        final int uid2 = 1006;
        final int uid3 = 1007;
        final int sessionId = 1;
        final String tag1 = "tag1";
        final String tag2 = "tag2";
        final String tag3 = "tag3";
        final SparseArray<WifiAwareClientState> clients = new SparseArray<>();
        WifiMetricsProto.WifiAwareLog log;

        setTime(5);
        WifiAwareClientState client1 = new WifiAwareClientState(mMockContext, 10, uid1, 0, null,
                null, null, null, false, 0, mWifiPermissionsUtil, null, false, 6);
        WifiAwareClientState client2 = new WifiAwareClientState(mMockContext, 11, uid2, 0, null,
                null, null, null, false, 0, mWifiPermissionsUtil, null, false, 6);
        WifiAwareClientState client3 = new WifiAwareClientState(mMockContext, 12, uid3, 0, null,
                null, null, null, false, 0, mWifiPermissionsUtil, null, false, 6);
        clients.put(10, client1);
        clients.put(11, client2);
        clients.put(12, client3);

        // uid1: publish session 1
        client1.addSession(
                new WifiAwareDiscoverySessionState(
                        /* wifiAwareNativeApi= */ null,
                        /* sessionId= */ 100,
                        /* pubSubId= */ (byte) 0,
                        /* callback= */ null,
                        /* isPublishSession= */ true,
                        /* isRangingEnabled= */ false,
                        /* creationTime= */ mClock.getElapsedSinceBootMillis(),
                        /* instantModeEnabled= */ false,
                        /* instantModeBand= */ 0,
                        /* isSuspendable= */ false,
                        /* pairingConfig= */ null));
        mDut.recordDiscoverySession(uid1, clients);
        mDut.recordDiscoveryStatus(uid1, NanStatusCode.SUCCESS, true, 100, 6, tag1);

        // uid1: publish session 2
        client1.addSession(
                new WifiAwareDiscoverySessionState(
                        /* wifiAwareNativeApi= */ null,
                        /* sessionId= */ 101,
                        /* pubSubId= */ (byte) 0,
                        /* callback= */ null,
                        /* isPublishSession= */ true,
                        /* isRangingEnabled= */ false,
                        /* creationTime= */ mClock.getElapsedSinceBootMillis(),
                        /* instantModeEnabled= */ false,
                        /* instantModeBand= */ 0,
                        /* isSuspendable= */ false,
                        /* pairingConfig= */ null));
        mDut.recordDiscoverySession(uid1, clients);
        mDut.recordDiscoveryStatus(uid1, NanStatusCode.SUCCESS, true, 101, 6, tag1);

        // uid3: publish session 3 with ranging
        client3.addSession(
                new WifiAwareDiscoverySessionState(
                        /* wifiAwareNativeApi= */ null,
                        /* sessionId= */ 111,
                        /* pubSubId= */ (byte) 0,
                        /* callback= */ null,
                        /* isPublishSession= */ true,
                        /* isRangingEnabled= */ true,
                        /* creationTime= */ mClock.getElapsedSinceBootMillis(),
                        /* instantModeEnabled= */ false,
                        /* instantModeBand= */ 0,
                        /* isSuspendable= */ false,
                        /* pairingConfig= */ null));
        mDut.recordDiscoverySessionWithRanging(uid3, false, -1, -1, clients);
        mDut.recordDiscoveryStatus(uid3, NanStatusCode.SUCCESS, true, 111, 6, tag3);

        // uid2: subscribe session 1
        client2.addSession(
                new WifiAwareDiscoverySessionState(
                        /* wifiAwareNativeApi= */ null,
                        /* sessionId= */ 102,
                        /* pubSubId= */ (byte) 0,
                        /* callback= */ null,
                        /* isPublishSession= */ false,
                        /* isRangingEnabled= */ false,
                        /* creationTime= */ mClock.getElapsedSinceBootMillis(),
                        /* instantModeEnabled= */ false,
                        /* instantModeBand= */ 0,
                        /* isSuspendable= */ false,
                        /* pairingConfig= */ null));
        mDut.recordDiscoverySession(uid2, clients);
        mDut.recordDiscoveryStatus(uid2, NanStatusCode.SUCCESS, false, 102, 6, tag2);

        // uid2: publish session 2
        client2.addSession(
                new WifiAwareDiscoverySessionState(
                        /* wifiAwareNativeApi= */ null,
                        /* sessionId= */ 103,
                        /* pubSubId= */ (byte) 0,
                        /* callback= */ null,
                        /* isPublishSession= */ true,
                        /* isRangingEnabled= */ false,
                        /* creationTime= */ mClock.getElapsedSinceBootMillis(),
                        /* instantModeEnabled= */ false,
                        /* instantModeBand= */ 0,
                        /* isSuspendable= */ false,
                        /* pairingConfig= */ null));
        mDut.recordDiscoverySession(uid2, clients);
        mDut.recordDiscoveryStatus(uid2, NanStatusCode.SUCCESS, false, 103, 6, tag2);

        // uid3: subscribe session 3 with ranging: min
        client3.addSession(
                new WifiAwareDiscoverySessionState(
                        /* wifiAwareNativeApi= */ null,
                        /* sessionId= */ 112,
                        /* pubSubId= */ (byte) 0,
                        /* callback= */ null,
                        /* isPublishSession= */ false,
                        /* isRangingEnabled= */ true,
                        /* creationTime= */ mClock.getElapsedSinceBootMillis(),
                        /* instantModeEnabled= */ false,
                        /* instantModeBand= */ 0,
                        /* isSuspendable= */ false,
                        /* pairingConfig= */ null));
        mDut.recordDiscoverySessionWithRanging(uid3, true, 10, -1, clients);
        mDut.recordDiscoveryStatus(uid3, NanStatusCode.SUCCESS, false, 112, 6, tag3);

        // uid3: subscribe session 3 with ranging: max
        client3.addSession(
                new WifiAwareDiscoverySessionState(
                        /* wifiAwareNativeApi= */ null,
                        /* sessionId= */ 113,
                        /* pubSubId= */ (byte) 0,
                        /* callback= */ null,
                        /* isPublishSession= */ false,
                        /* isRangingEnabled= */ true,
                        /* creationTime= */ mClock.getElapsedSinceBootMillis(),
                        /* instantModeEnabled= */ false,
                        /* instantModeBand= */ 0,
                        /* isSuspendable= */ false,
                        /* pairingConfig= */ null));
        mDut.recordDiscoverySessionWithRanging(uid3, true, -1, 50, clients);
        mDut.recordDiscoveryStatus(uid3, NanStatusCode.SUCCESS, false, 113, 6, tag3);

        // uid3: subscribe session 3 with ranging: minmax
        client3.addSession(
                new WifiAwareDiscoverySessionState(
                        /* wifiAwareNativeApi= */ null,
                        /* sessionId= */ 114,
                        /* pubSubId= */ (byte) 0,
                        /* callback= */ null,
                        /* isPublishSession= */ false,
                        /* isRangingEnabled= */ true,
                        /* creationTime= */ mClock.getElapsedSinceBootMillis(),
                        /* instantModeEnabled= */ false,
                        /* instantModeBand= */ 0,
                        /* isSuspendable= */ false,
                        /* pairingConfig= */ null));
        mDut.recordDiscoverySessionWithRanging(uid3, true, 0, 110, clients);
        mDut.recordDiscoveryStatus(uid3, NanStatusCode.SUCCESS, false, 114, 6, tag3);

        // uid1: delete session 1
        setTime(10);
        mDut.recordDiscoverySessionDuration(client1.getSession(100).getCreationTime(),
                client1.getSession(100).isPublishSession(), 0);
        client1.removeSession(100);

        // uid2: delete session 1
        setTime(15);
        mDut.recordDiscoverySessionDuration(client2.getSession(102).getCreationTime(),
                client2.getSession(102).isPublishSession(), 0);
        client2.removeSession(102);

        // uid2: subscribe session 3
        mDut.recordDiscoverySession(uid2, clients);
        client2.addSession(
                new WifiAwareDiscoverySessionState(
                        /* wifiAwareNativeApi= */ null,
                        /* sessionId= */ 104,
                        /* pubSubId= */ (byte) 0,
                        /* callback= */ null,
                        /* isPublishSession= */ false,
                        /* isRangingEnabled= */ false,
                        /* creationTime= */ mClock.getElapsedSinceBootMillis(),
                        /* instantModeEnabled= */ false,
                        /* instantModeBand= */ 0,
                        /* isSuspendable= */ false,
                        /* pairingConfig= */ null));

        // a few failures
        mDut.recordDiscoveryStatus(uid1, NanStatusCode.INTERNAL_FAILURE, true, 6, tag1);
        mDut.recordDiscoveryStatus(uid2, NanStatusCode.INTERNAL_FAILURE, false, 6, tag2);
        mDut.recordDiscoveryStatus(uid2, NanStatusCode.NO_RESOURCES_AVAILABLE, false, 6, tag2);
        mDut.recordAttachStatus(-5, 6, tag1, uid1); // invalid

        // verify
        log = mDut.consolidateProto();

        collector.checkThat("maxConcurrentPublishInApp", log.maxConcurrentPublishInApp, equalTo(2));
        collector.checkThat("maxConcurrentSubscribeInApp", log.maxConcurrentSubscribeInApp,
                equalTo(3));
        collector.checkThat("maxConcurrentDiscoverySessionsInApp",
                log.maxConcurrentDiscoverySessionsInApp, equalTo(4));
        collector.checkThat("maxConcurrentPublishInSystem", log.maxConcurrentPublishInSystem,
                equalTo(4));
        collector.checkThat("maxConcurrentSubscribeInSystem", log.maxConcurrentSubscribeInSystem,
                equalTo(4));
        collector.checkThat("maxConcurrentDiscoverySessionsInSystem",
                log.maxConcurrentDiscoverySessionsInSystem, equalTo(8));
        collector.checkThat("histogramPublishStatus.length",
                log.histogramPublishStatus.length, equalTo(2)); // 2 buckets
        validateNanStatusProtoHistBucket("Bucket[SUCCESS]",
                log.histogramPublishStatus[0],
                WifiMetricsProto.WifiAwareLog.SUCCESS, 3);
        validateNanStatusProtoHistBucket("Bucket[INTERNAL_FAILURE]",
                log.histogramPublishStatus[1],
                WifiMetricsProto.WifiAwareLog.INTERNAL_FAILURE, 1);
        collector.checkThat("histogramSubscribeStatus.length",
                log.histogramSubscribeStatus.length, equalTo(3)); // 3 buckets
        validateNanStatusProtoHistBucket("Bucket[SUCCESS]",
                log.histogramSubscribeStatus[0],
                WifiMetricsProto.WifiAwareLog.SUCCESS, 5);
        validateNanStatusProtoHistBucket("Bucket[INTERNAL_FAILURE]",
                log.histogramSubscribeStatus[1],
                WifiMetricsProto.WifiAwareLog.INTERNAL_FAILURE, 1);
        validateNanStatusProtoHistBucket("Bucket[NO_RESOURCES_AVAILABLE]",
                log.histogramSubscribeStatus[2],
                WifiMetricsProto.WifiAwareLog.NO_RESOURCES_AVAILABLE, 1);
        collector.checkThat("numAppsWithDiscoverySessionFailureOutOfResources",
                log.numAppsWithDiscoverySessionFailureOutOfResources, equalTo(1));
        validateProtoHistBucket("Publish Duration[0]", log.histogramPublishSessionDurationMs[0], 5,
                6, 1);
        validateProtoHistBucket("Subscribe Duration[0]", log.histogramSubscribeSessionDurationMs[0],
                10, 20, 1);

        collector.checkThat("maxConcurrentPublishWithRangingInApp",
                log.maxConcurrentPublishWithRangingInApp, equalTo(1));
        collector.checkThat("maxConcurrentSubscribeWithRangingInApp",
                log.maxConcurrentSubscribeWithRangingInApp, equalTo(3));
        collector.checkThat("maxConcurrentPublishWithRangingInSystem",
                log.maxConcurrentPublishWithRangingInSystem, equalTo(1));
        collector.checkThat("maxConcurrentSubscribeWithRangingInSystem",
                log.maxConcurrentSubscribeWithRangingInSystem, equalTo(3));
        collector.checkThat("numSubscribesWithRanging", log.numSubscribesWithRanging, equalTo(3));
        collector.checkThat("histogramSubscribeGeofenceMin.length",
                log.histogramSubscribeGeofenceMin.length, equalTo(2));
        collector.checkThat("histogramSubscribeGeofenceMax.length",
                log.histogramSubscribeGeofenceMax.length, equalTo(2));
        validateProtoHistBucket("histogramSubscribeGeofenceMin[0]",
                log.histogramSubscribeGeofenceMin[0], Integer.MIN_VALUE, 10, 1);
        validateProtoHistBucket("histogramSubscribeGeofenceMin[1]",
                log.histogramSubscribeGeofenceMin[1], 10, 30, 1);
        validateProtoHistBucket("histogramSubscribeGeofenceMax[0]",
                log.histogramSubscribeGeofenceMax[0], 30, 60, 1);
        validateProtoHistBucket("histogramSubscribeGeofenceMax[1]",
                log.histogramSubscribeGeofenceMax[1], 100, Integer.MAX_VALUE, 1);
    }

    /**
     * Validate the data-path (NDP & NDI) metrics.
     */
    @Test
    public void testDataPathMetrics() {
        final int uid1 = 1005;
        final String package1 = "com.test1";
        final int uid2 = 1006;
        final String package2 = "com.test2";
        final String ndi0 = "aware_data0";
        final String ndi1 = "aware_data1";
        final String tag1 = "tag1";
        final String tag2 = "tag2";
        final String tag3 = "tag3";
        final int[] sessionIds = {1, 2, 3, 4, 5, 6, 7, 8, 9};
        final int role_init = WifiAwareManager.WIFI_AWARE_DATA_PATH_ROLE_INITIATOR;
        final int role_resp = WifiAwareManager.WIFI_AWARE_DATA_PATH_ROLE_RESPONDER;
        Map<WifiAwareNetworkSpecifier, WifiAwareDataPathStateManager.AwareNetworkRequestInformation>
                networkRequestCache = new HashMap<>();
        WifiMetricsProto.WifiAwareLog log;

        setTime(0);
        mDut.recordDiscoveryStatus(uid1, NanStatusCode.SUCCESS, true, sessionIds[0], 6, tag1);

        setTime(5);
        // uid1: ndp (non-secure) on ndi0
        addNetworkInfoToCache(networkRequestCache, 10, uid1, package1, ndi0, null);

        mDut.recordNdpCreation(uid1, package1, networkRequestCache);
        setTime(7); // 2ms creation time, 7ms discovery + NDP latency
        mDut.recordNdpStatus(NanStatusCode.SUCCESS, false, role_init, 5, sessionIds[0], 5120);

        // uid2: ndp (non-secure) on ndi0
        setTime(5);
        mDut.recordDiscoveryStatus(uid1, NanStatusCode.SUCCESS, true, sessionIds[1], 6, tag1);
        setTime(7);
        WifiAwareNetworkSpecifier ns = addNetworkInfoToCache(networkRequestCache, 11, uid2,
                package2, ndi0, null);
        mDut.recordNdpCreation(uid2, package2, networkRequestCache);
        setTime(10); // 3 ms creation time, 5ms discovery + NDP latency
        mDut.recordNdpStatus(NanStatusCode.SUCCESS, false, role_resp, 7, sessionIds[1], 2412);

        // uid2: ndp (secure) on ndi1 (OOB)
        setTime(8);
        mDut.recordDiscoveryStatus(uid2, NanStatusCode.SUCCESS, true, sessionIds[2], 6, tag2);
        setTime(10);
        addNetworkInfoToCache(networkRequestCache, 12, uid2, package2, ndi1,
                "passphrase of some kind");
        mDut.recordNdpCreation(uid2, package2, networkRequestCache);
        setTime(25); // 15 ms creation time, 17ms discovery + NDP latency
        mDut.recordNdpStatus(NanStatusCode.SUCCESS, true, role_init, 10, sessionIds[2], 5180);

        // uid2: ndp (secure) on ndi0 (OOB)
        setTime(20);
        mDut.recordDiscoveryStatus(uid2, NanStatusCode.SUCCESS, true, sessionIds[3], 6, tag2);
        setTime(25);
        addNetworkInfoToCache(networkRequestCache, 13, uid2, package2, ndi0,
                "super secret password");
        mDut.recordNdpCreation(uid2, package2, networkRequestCache);
        setTime(36); // 11 ms creation time
        mDut.recordNdpStatus(NanStatusCode.SUCCESS, true, role_resp, 25, sessionIds[3], 2437);

        // uid2: delete the first NDP
        networkRequestCache.remove(ns);

        // uid2: ndp (non-secure) on ndi0
        setTime(32);
        mDut.recordDiscoveryStatus(uid2, NanStatusCode.SUCCESS, true, sessionIds[4], 6, tag2);
        setTime(36);
        addNetworkInfoToCache(networkRequestCache, 14, uid2, package2, ndi0, null);
        mDut.recordNdpCreation(uid2, package2, networkRequestCache);
        setTime(37); // 1 ms creation time!
        mDut.recordNdpStatus(NanStatusCode.SUCCESS, false, role_resp, 36, sessionIds[4], 5180);

        // a few error codes
        mDut.recordNdpStatus(NanStatusCode.INTERNAL_FAILURE, false, role_resp, 0, sessionIds[5]);
        mDut.recordNdpStatus(NanStatusCode.INTERNAL_FAILURE, false, role_init, 0, sessionIds[6]);
        mDut.recordNdpStatus(NanStatusCode.NO_RESOURCES_AVAILABLE, false, role_resp, 0,
                sessionIds[7]);

        // and some durations
        setTime(150);
        mDut.recordNdpSessionDuration(7);   // 143ms
        mDut.recordNdpSessionDuration(10);  // 140ms
        mDut.recordNdpSessionDuration(25);  // 125ms
        mDut.recordNdpSessionDuration(140); // 10ms

        //verify
        log = mDut.consolidateProto();

        collector.checkThat("maxConcurrentNdiInApp", log.maxConcurrentNdiInApp, equalTo(2));
        collector.checkThat("maxConcurrentNdiInSystem", log.maxConcurrentNdiInSystem, equalTo(2));
        collector.checkThat("maxConcurrentNdpInApp", log.maxConcurrentNdpInApp, equalTo(3));
        collector.checkThat("maxConcurrentNdpInSystem", log.maxConcurrentNdpInSystem, equalTo(4));
        collector.checkThat("maxConcurrentSecureNdpInApp", log.maxConcurrentSecureNdpInApp,
                equalTo(2));
        collector.checkThat("maxConcurrentSecureNdpInSystem", log.maxConcurrentSecureNdpInSystem,
                equalTo(2));
        collector.checkThat("maxConcurrentNdpPerNdi", log.maxConcurrentNdpPerNdi, equalTo(3));
        collector.checkThat("histogramRequestNdpStatus.length",
                log.histogramRequestNdpStatus.length, equalTo(3));
        validateNanStatusProtoHistBucket("Bucket[SUCCESS]",
                log.histogramRequestNdpStatus[0],
                WifiMetricsProto.WifiAwareLog.SUCCESS, 3);
        validateNanStatusProtoHistBucket("Bucket[INTERNAL_FAILURE]",
                log.histogramRequestNdpStatus[1],
                WifiMetricsProto.WifiAwareLog.INTERNAL_FAILURE, 2);
        validateNanStatusProtoHistBucket("Bucket[UNKNOWN_HAL_STATUS]",
                log.histogramRequestNdpStatus[2],
                WifiMetricsProto.WifiAwareLog.NO_RESOURCES_AVAILABLE, 1);
        collector.checkThat("histogramRequestNdpOobStatus.length",
                log.histogramRequestNdpOobStatus.length, equalTo(1));
        validateNanStatusProtoHistBucket("Bucket[SUCCESS]",
                log.histogramRequestNdpOobStatus[0],
                WifiMetricsProto.WifiAwareLog.SUCCESS, 2);

        collector.checkThat("ndpCreationTimeMsMin", log.ndpCreationTimeMsMin, equalTo(1L));
        collector.checkThat("ndpCreationTimeMsMax", log.ndpCreationTimeMsMax, equalTo(15L));
        collector.checkThat("ndpCreationTimeMsSum", log.ndpCreationTimeMsSum, equalTo(32L));
        collector.checkThat("ndpCreationTimeMsSumOfSq", log.ndpCreationTimeMsSumOfSq,
                equalTo(360L));
        collector.checkThat("ndpCreationTimeMsNumSamples", log.ndpCreationTimeMsNumSamples,
                equalTo(5L));
        validateProtoHistBucket("Creation[0]", log.histogramNdpCreationTimeMs[0], 1, 2, 1);
        validateProtoHistBucket("Creation[1]", log.histogramNdpCreationTimeMs[1], 2, 3, 1);
        validateProtoHistBucket("Creation[2]", log.histogramNdpCreationTimeMs[2], 3, 4, 1);
        validateProtoHistBucket("Creation[3]", log.histogramNdpCreationTimeMs[3], 10, 20, 2);

        validateProtoHistBucket("Duration[0]", log.histogramNdpSessionDurationMs[0], 10, 20, 1);
        validateProtoHistBucket("Duration[1]", log.histogramNdpSessionDurationMs[1], 100, 200, 3);
        ExtendedMockito.verify(() -> WifiStatsLog.write(
                WifiStatsLog.WIFI_AWARE_NDP_REPORTED,
                WifiStatsLog.WIFI_AWARE_NDP_REPORTED__ROLE__ROLE_INITIATOR,
                false, WifiStatsLog.WIFI_AWARE_NDP_REPORTED__STATUS__ST_SUCCESS,
                2, 7, 5120, false,
                WifiStatsLog.WIFI_AWARE_NDP_REPORTED__CALLER_TYPE__OTHERS, tag1, uid1));
        ExtendedMockito.verify(() -> WifiStatsLog.write(
                WifiStatsLog.WIFI_AWARE_NDP_REPORTED,
                WifiStatsLog.WIFI_AWARE_NDP_REPORTED__ROLE__ROLE_RESPONDER,
                false, WifiStatsLog.WIFI_AWARE_NDP_REPORTED__STATUS__ST_SUCCESS,
                3, 5, 2412, false,
                WifiStatsLog.WIFI_AWARE_NDP_REPORTED__CALLER_TYPE__OTHERS, tag1, uid1));
        ExtendedMockito.verify(() -> WifiStatsLog.write(
                WifiStatsLog.WIFI_AWARE_NDP_REPORTED,
                WifiStatsLog.WIFI_AWARE_NDP_REPORTED__ROLE__ROLE_INITIATOR,
                true, WifiStatsLog.WIFI_AWARE_NDP_REPORTED__STATUS__ST_SUCCESS,
                15, 17, 5180, false,
                WifiStatsLog.WIFI_AWARE_NDP_REPORTED__CALLER_TYPE__OTHERS, tag2, uid2));
        ExtendedMockito.verify(() -> WifiStatsLog.write(
                WifiStatsLog.WIFI_AWARE_NDP_REPORTED,
                WifiStatsLog.WIFI_AWARE_NDP_REPORTED__ROLE__ROLE_RESPONDER,
                true, WifiStatsLog.WIFI_AWARE_NDP_REPORTED__STATUS__ST_SUCCESS,
                11, 16, 2437, false,
                WifiStatsLog.WIFI_AWARE_NDP_REPORTED__CALLER_TYPE__OTHERS, tag2, uid2));
        ExtendedMockito.verify(() -> WifiStatsLog.write(
                WifiStatsLog.WIFI_AWARE_NDP_REPORTED,
                WifiStatsLog.WIFI_AWARE_NDP_REPORTED__ROLE__ROLE_RESPONDER,
                false, WifiStatsLog.WIFI_AWARE_NDP_REPORTED__STATUS__ST_SUCCESS,
                1, 5, 5180, false,
                WifiStatsLog.WIFI_AWARE_NDP_REPORTED__CALLER_TYPE__OTHERS, tag2, uid2));
        ExtendedMockito.verify(() -> WifiStatsLog.write(
                WifiStatsLog.WIFI_AWARE_NDP_REPORTED,
                WifiStatsLog.WIFI_AWARE_NDP_REPORTED__ROLE__ROLE_RESPONDER,
                false, WifiStatsLog.WIFI_AWARE_NDP_REPORTED__STATUS__ST_INTERNAL_FAILURE,
                37, 37, 0, false,
                WifiStatsLog.WIFI_AWARE_NDP_REPORTED__CALLER_TYPE__UNKNOWN, null, 0));
        ExtendedMockito.verify(() -> WifiStatsLog.write(
                WifiStatsLog.WIFI_AWARE_NDP_REPORTED,
                WifiStatsLog.WIFI_AWARE_NDP_REPORTED__ROLE__ROLE_INITIATOR,
                false, WifiStatsLog.WIFI_AWARE_NDP_REPORTED__STATUS__ST_INTERNAL_FAILURE,
                37, 37, 0, false,
                WifiStatsLog.WIFI_AWARE_NDP_REPORTED__CALLER_TYPE__UNKNOWN, null, 0));
        ExtendedMockito.verify(() -> WifiStatsLog.write(
                WifiStatsLog.WIFI_AWARE_NDP_REPORTED,
                WifiStatsLog.WIFI_AWARE_NDP_REPORTED__ROLE__ROLE_RESPONDER,
                false, WifiStatsLog.WIFI_AWARE_NDP_REPORTED__STATUS__ST_NO_RESOURCES_AVAILABLE,
                37, 37, 0, false,
                WifiStatsLog.WIFI_AWARE_NDP_REPORTED__CALLER_TYPE__UNKNOWN, null, 0));
    }

    /**
     * Validate that the histogram configuration is initialized correctly: bucket starting points
     * and sub-bucket widths.
     */
    @Test
    public void testHistParamInit() {
        collector.checkThat("HIST1.mLog", HIST1.mLog, equalTo(Math.log(10)));
        collector.checkThat("HIST1.bb[0]", HIST1.bb[0], equalTo(1.0));
        collector.checkThat("HIST1.bb[1]", HIST1.bb[1], equalTo(10.0));
        collector.checkThat("HIST1.bb[2]", HIST1.bb[2], equalTo(100.0));
        collector.checkThat("HIST1.bb[3]", HIST1.bb[3], equalTo(1000.0));
        collector.checkThat("HIST1.bb[4]", HIST1.bb[4], equalTo(10000.0));
        collector.checkThat("HIST1.bb[5]", HIST1.bb[5], equalTo(100000.0));
        collector.checkThat("HIST1.sbw[0]", HIST1.sbw[0], equalTo(1.0));
        collector.checkThat("HIST1.sbw[1]", HIST1.sbw[1], equalTo(10.0));
        collector.checkThat("HIST1.sbw[2]", HIST1.sbw[2], equalTo(100.0));
        collector.checkThat("HIST1.sbw[3]", HIST1.sbw[3], equalTo(1000.0));
        collector.checkThat("HIST1.sbw[4]", HIST1.sbw[4], equalTo(10000.0));
        collector.checkThat("HIST1.sbw[5]", HIST1.sbw[5], equalTo(100000.0));

        collector.checkThat("HIST2.mLog", HIST1.mLog, equalTo(Math.log(10)));
        collector.checkThat("HIST2.bb[0]", HIST2.bb[0], equalTo(-18.0));
        collector.checkThat("HIST2.bb[1]", HIST2.bb[1], equalTo(-10.0));
        collector.checkThat("HIST2.bb[2]", HIST2.bb[2], equalTo(30.0));
        collector.checkThat("HIST2.sbw[0]", HIST2.sbw[0], equalTo(0.2));
        collector.checkThat("HIST2.sbw[1]", HIST2.sbw[1], equalTo(1.0));
        collector.checkThat("HIST2.sbw[2]", HIST2.sbw[2], equalTo(5.0));
    }

    /**
     * Validate the conversion to a NanStatusType proto raw histogram.
     */
    @Test
    public void testNanStatusTypeHistogram() {
        SparseIntArray statusHistogram = new SparseIntArray();

        addNanHalStatusToHistogram(NanStatusCode.SUCCESS, statusHistogram);
        addNanHalStatusToHistogram(-1, statusHistogram);
        addNanHalStatusToHistogram(NanStatusCode.ALREADY_ENABLED, statusHistogram);
        addNanHalStatusToHistogram(NanStatusCode.SUCCESS, statusHistogram);
        addNanHalStatusToHistogram(NanStatusCode.INTERNAL_FAILURE, statusHistogram);
        addNanHalStatusToHistogram(NanStatusCode.SUCCESS, statusHistogram);
        addNanHalStatusToHistogram(NanStatusCode.INTERNAL_FAILURE, statusHistogram);
        addNanHalStatusToHistogram(55, statusHistogram);
        addNanHalStatusToHistogram(65, statusHistogram);

        WifiMetricsProto.WifiAwareLog.NanStatusHistogramBucket[] sh = histogramToProtoArray(
                statusHistogram);
        collector.checkThat("Number of buckets", sh.length, equalTo(4));
        validateNanStatusProtoHistBucket("Bucket[SUCCESS]", sh[0],
                WifiMetricsProto.WifiAwareLog.SUCCESS, 3);
        validateNanStatusProtoHistBucket("Bucket[INTERNAL_FAILURE]", sh[1],
                WifiMetricsProto.WifiAwareLog.INTERNAL_FAILURE, 2);
        validateNanStatusProtoHistBucket("Bucket[ALREADY_ENABLED]", sh[2],
                WifiMetricsProto.WifiAwareLog.ALREADY_ENABLED, 1);
        validateNanStatusProtoHistBucket("Bucket[UNKNOWN_HAL_STATUS]", sh[3],
                WifiMetricsProto.WifiAwareLog.UNKNOWN_HAL_STATUS, 3);
    }

    @Test
    public void testNdpRequestTypeHistogram() {
        mDut.recordNdpRequestType(NETWORK_SPECIFIER_TYPE_IB);
        mDut.recordNdpRequestType(NETWORK_SPECIFIER_TYPE_IB);
        mDut.recordNdpRequestType(NETWORK_SPECIFIER_TYPE_IB_ANY_PEER);

        WifiMetricsProto.WifiAwareLog log;
        log = mDut.consolidateProto();

        validateNdpRequestProtoHistBucket("", log.histogramNdpRequestType[0],
                WifiMetricsProto.WifiAwareLog.NETWORK_SPECIFIER_TYPE_IB, 2);
        validateNdpRequestProtoHistBucket("", log.histogramNdpRequestType[1],
                WifiMetricsProto.WifiAwareLog.NETWORK_SPECIFIER_TYPE_IB_ANY_PEER, 1);
    }

    // utilities

    /**
     * Mock the elapsed time since boot to the input argument.
     */
    private void setTime(long timeMs) {
        when(mClock.getElapsedSinceBootMillis()).thenReturn(timeMs);
    }

    /**
     * Sum all the 'count' entries in the histogram array.
     */
    private int countAllHistogramSamples(WifiMetricsProto.WifiAwareLog.HistogramBucket[] hba) {
        int sum = 0;
        for (WifiMetricsProto.WifiAwareLog.HistogramBucket hb: hba) {
            sum += hb.count;
        }
        return sum;
    }

    private int countAllHistogramSamples(
            WifiMetricsProto.WifiAwareLog.NanStatusHistogramBucket[] nshba) {
        int sum = 0;
        for (WifiMetricsProto.WifiAwareLog.NanStatusHistogramBucket nshb: nshba) {
            sum += nshb.count;
        }
        return sum;
    }

    private void validateProtoHistBucket(String logPrefix,
            WifiMetricsProto.WifiAwareLog.HistogramBucket bucket, long start, long end, int count) {
        collector.checkThat(logPrefix + ": start", bucket.start, equalTo(start));
        collector.checkThat(logPrefix + ": end", bucket.end, equalTo(end));
        collector.checkThat(logPrefix + ": count", bucket.count, equalTo(count));
    }

    private void validateNanStatusProtoHistBucket(String logPrefix,
            WifiMetricsProto.WifiAwareLog.NanStatusHistogramBucket bucket, int type, int count) {
        collector.checkThat(logPrefix + ": type", bucket.nanStatusType, equalTo(type));
        collector.checkThat(logPrefix + ": count", bucket.count, equalTo(count));
    }

    private void validateNdpRequestProtoHistBucket(String logPrefix,
            WifiMetricsProto.WifiAwareLog.NdpRequestTypeHistogramBucket bucket, int type,
            int count) {
        collector.checkThat(logPrefix + ": type", bucket.ndpRequestType, equalTo(type));
        collector.checkThat(logPrefix + ": count", bucket.count, equalTo(count));
    }

    private WifiAwareNetworkSpecifier addNetworkInfoToCache(
            Map<WifiAwareNetworkSpecifier, WifiAwareDataPathStateManager
                    .AwareNetworkRequestInformation> networkRequestCache,
            int index, int uid, String packageName, String interfaceName, String passphrase) {
        WifiAwareNetworkSpecifier ns = new WifiAwareNetworkSpecifier(0, 0, 0, index, 0, null, null,
                passphrase, 0, 0);
        WifiAwareDataPathStateManager.AwareNetworkRequestInformation anri =
                new WifiAwareDataPathStateManager.AwareNetworkRequestInformation();
        anri.networkSpecifier = ns;
        anri.state = WifiAwareDataPathStateManager.AwareNetworkRequestInformation.STATE_CONFIRMED;
        anri.uid = uid;
        anri.packageName = packageName;
        anri.interfaceName = interfaceName;

        networkRequestCache.put(ns, anri);
        return ns;
    }

    private void dumpDut(String prefix) {
        StringWriter sw = new StringWriter();
        mDut.dump(null, new PrintWriter(sw), null);
        Log.e("WifiAwareMetrics", prefix + sw.toString());
    }
}