com.yammer.tenacity.testing.TenacityTestRule Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of tenacity-testing Show documentation

There is a newer version: 1.2.0.dw9

package com.yammer.tenacity.testing;

import com.netflix.config.ConfigurationManager;
import com.netflix.hystrix.Hystrix;
import com.netflix.hystrix.metric.consumer.*;
import org.apache.commons.configuration.AbstractConfiguration;
import org.junit.rules.TestRule;
import org.junit.runner.Description;
import org.junit.runners.model.Statement;

import java.util.concurrent.TimeUnit;

public class TenacityTestRule implements TestRule {
    private void setup() {
        resetStreams();
        Hystrix.reset();
        final AbstractConfiguration configuration = ConfigurationManager.getConfigInstance();
        configuration.setProperty("hystrix.command.default.metrics.healthSnapshot.intervalInMilliseconds", "100");
    }

    public void teardown() {
        Hystrix.reset(1, TimeUnit.SECONDS);
        ConfigurationManager.getConfigInstance().clear();
    }

    private void resetStreams() {
        /* BucketedCounterStream */
        CumulativeCommandEventCounterStream.reset();
        RollingCommandEventCounterStream.reset();
        CumulativeCollapserEventCounterStream.reset();
        RollingCollapserEventCounterStream.reset();
        CumulativeThreadPoolEventCounterStream.reset();
        RollingThreadPoolEventCounterStream.reset();
        HealthCountsStream.reset();
        /* --------------------- */

        /* RollingConcurrencyStream */
        RollingThreadPoolMaxConcurrencyStream.reset();
        RollingCommandMaxConcurrencyStream.reset();
        /* ------------------------ */

        /* RollingDistributionStream */
        RollingCommandLatencyDistributionStream.reset();
        RollingCommandUserLatencyDistributionStream.reset();
        RollingCollapserBatchSizeDistributionStream.reset();
        /* ------------------------- */
    }

    @Override
    public Statement apply(final Statement base, Description description) {
        return new Statement() {
            @Override
            public void evaluate() throws Throwable {
                try {
                    setup();
                    base.evaluate();
                } finally {
                    teardown();
                }
            }
        };
    }
}