core/tests/coretests/src/com/android/internal/util/ProgressReporterTest.java

/*
 * Copyright (C) 2016 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.internal.util;

import junit.framework.TestCase;

public class ProgressReporterTest extends TestCase {
    private ProgressReporter r;

    @Override
    protected void setUp() throws Exception {
        super.setUp();
        r = new ProgressReporter(0, null);
    }

    private void assertProgress(int expected) {
        assertEquals(expected, r.getProgress());
    }

    private void assertRange(int start, int len) {
        final int[] range = r.getSegmentRange();
        assertEquals("start", start, range[0]);
        assertEquals("len", len, range[1]);
    }

    public void testBasic() throws Exception {
        assertProgress(0);

        r.setProgress(20);
        assertProgress(20);

        r.setProgress(-20);
        assertProgress(0);

        r.setProgress(1024);
        assertProgress(100);
    }

    public void testSegment() throws Exception {
        r.setProgress(20);
        assertProgress(20);

        final int[] lastRange = r.startSegment(40);
        {
            assertProgress(20);

            r.setProgress(50);
            assertProgress(40);
        }
        r.endSegment(lastRange);
        assertProgress(60);

        r.setProgress(80);
        assertProgress(80);
    }

    public void testSegmentOvershoot() throws Exception {
        r.setProgress(20);
        assertProgress(20);

        final int[] lastRange = r.startSegment(40);
        {
            r.setProgress(-100, 2);
            assertProgress(20);

            r.setProgress(1, 2);
            assertProgress(40);

            r.setProgress(100, 2);
            assertProgress(60);
        }
        r.endSegment(lastRange);
        assertProgress(60);
    }

    public void testSegmentNested() throws Exception {
        r.setProgress(20);
        assertProgress(20);
        assertRange(0, 100);

        final int[] lastRange = r.startSegment(40);
        assertRange(20, 40);
        {
            r.setProgress(50);
            assertProgress(40);

            final int[] lastRange2 = r.startSegment(25);
            assertRange(40, 10);
            {
                r.setProgress(0);
                assertProgress(40);

                r.setProgress(50);
                assertProgress(45);

                r.setProgress(100);
                assertProgress(50);
            }
            r.endSegment(lastRange2);
            assertProgress(50);
        }
        r.endSegment(lastRange);
        assertProgress(60);
    }
}