impl/operations/OperationMonitorTest.java

	OperationMonitorTest.java revision 7b8fd79c6177846da551bb2cd5e1579b7c650a3c
/*
* The contents of this file are subject to the terms of the Common Development and
* Distribution License (the License). You may not use this file except in compliance with the
* License.
*
* You can obtain a copy of the License at legal/CDDLv1.0.txt. See the License for the
* specific language governing permission and limitations under the License.
*
* When distributing Covered Software, include this CDDL Header Notice in each file and include
* the License file at legal/CDDLv1.0.txt. If applicable, add the following below the CDDL
* Header, with the fields enclosed by brackets [] replaced by your own identifying
* information: "Portions copyright [year] [name of copyright owner]".
*
* Copyright 2013-2014 ForgeRock AS.
*/

package org.forgerock.openam.cts.monitoring.impl.operations;

import org.forgerock.openam.cts.monitoring.impl.RateTimer;
import org.forgerock.openam.cts.monitoring.impl.RateWindow;
import org.forgerock.openam.cts.monitoring.impl.TestCurrentMillis;
import static org.mockito.BDDMockito.given;
import static org.mockito.Mockito.mock;
import static org.testng.Assert.assertEquals;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;

public class OperationMonitorTest {

    private OperationMonitor operationMonitor;

    private RateWindow rateWindow;

    @BeforeMethod
    public void setUp() {

        RateTimer timer = mock(RateTimer.class);
        rateWindow = mock(RateWindow.class);

        operationMonitor = new OperationMonitor(timer, rateWindow);
    }

    @Test
    public void shouldGetCountReturningZeroWhenNoIncrement() {

        //Given

        //When
        long count = operationMonitor.getCount();

        //Then
        assertEquals(count, 0);
    }

    @Test
    public void shouldGetAverageRate() {

        final double SOLUTION = 5D;

        //Given
        given(rateWindow.getAverageRate()).willReturn(SOLUTION);

        //When
        double rate = operationMonitor.getAverageRate();

        //Then
        assertEquals(rate, SOLUTION);
    }

    @Test
    public void shouldGetMinRate() {

        final long SOLUTION = 3L;

        //Given
        given(rateWindow.getMinRate()).willReturn(SOLUTION);

        //When
        long rate = operationMonitor.getMinRate();

        //Then
        assertEquals(rate, SOLUTION);
    }

    @Test
    public void shouldGetMaxRate() {

        final long SOLUTION = 4L;

        //Given
        given(rateWindow.getMaxRate()).willReturn(SOLUTION);

        //When
        long rate = operationMonitor.getMaxRate();

        //Then
        assertEquals(rate, SOLUTION);
    }

    @Test
    public void shouldGetCountReturningOneAfterIncrement() {

        final long SOLUTION = 1L;

        //Given
        operationMonitor.increment();

        //When
        long count = operationMonitor.getCount();

        //Then
        assertEquals(count, SOLUTION);
    }

    @Test
    public void shouldGetCountReturningSixAfterMultipleIncrement() {

        //Given
        final int NUM = 6;

        for(int i = 0; i < NUM; i++) {
            operationMonitor.increment();
        }

        //When
        long count = operationMonitor.getCount();

        //Then
        assertEquals(count, NUM);
    }

    @Test
    public void recalculatingRateShouldNotBlockGetRateIfNotModifying() {

        //Given
        RateTimer timer = new TestCurrentMillis();
        RateWindow rateWindow = new RateWindow(timer, 10, 1000L);

        final OperationMonitor monitor = new OperationMonitor(timer, rateWindow);

        //When
        new Thread(new Runnable() {
            public void run() {
                monitor.increment();
            }
        }).start();

        double rate = monitor.getAverageRate();

        //Then
        assertEquals(rate, 0D);
    }

}