summaryrefslogtreecommitdiff
path: root/tests/wifitests/src/com/android/server/wifi/util/IntCounterTest.java
blob: e3029ea578c13c343b0169b29d4dc409828c2830 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
/*
 * Copyright (C) 2019 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.util;

import static com.android.server.wifi.WifiMetricsTestUtil.assertKeyCountsEqual;
import static com.android.server.wifi.WifiMetricsTestUtil.buildInt32Count;

import androidx.test.filters.SmallTest;

import com.android.server.wifi.WifiBaseTest;
import com.android.server.wifi.proto.nano.WifiMetricsProto.Int32Count;

import org.junit.Test;


/**
 * Unit tests for IntCounter.
 */
@SmallTest
public class IntCounterTest extends WifiBaseTest {

    private static final int[] TEST_KEYS = {
            100, 20, 34, 5656, 3535, 6456, -1231, -4235, 20, 3535, -5, 100, 6456, 34, -4235, -4235
    };

    /**
     * Tests when the counter is empty.
     */
    @Test
    public void testEmpty() {
        IntCounter counter = new IntCounter();
        assertKeyCountsEqual(new Int32Count[0], counter.toProto());
    }

    /**
     * Tests adding to the counter.
     */
    @Test
    public void testAddToCounter() {
        IntCounter counter = new IntCounter();

        for (int k : TEST_KEYS) {
            counter.increment(k);
        }

        Int32Count[] expected = {
                buildInt32Count(-4235, 3),
                buildInt32Count(-1231, 1),
                buildInt32Count(-5, 1),
                buildInt32Count(20, 2),
                buildInt32Count(34, 2),
                buildInt32Count(100, 2),
                buildInt32Count(3535, 2),
                buildInt32Count(5656, 1),
                buildInt32Count(6456, 2),
        };
        assertKeyCountsEqual(expected, counter.toProto());
    }

    /**
     * Tests adding to clamped counter.
     */
    @Test
    public void testAddToClampedCounter() {
        IntCounter counter = new IntCounter(-5, 100);

        for (int k : TEST_KEYS) {
            counter.increment(k);
        }

        Int32Count[] expected = {
                buildInt32Count(-5, 5),
                buildInt32Count(20, 2),
                buildInt32Count(34, 2),
                buildInt32Count(100, 7),
        };
        assertKeyCountsEqual(expected, counter.toProto());
    }
}