zipkin.server.ZipkinServerConfigurationTest.java Source code

Introduction

Here is the source code for zipkin.server.ZipkinServerConfigurationTest.java

Source

/**
 * Copyright 2015-2017 The OpenZipkin Authors
 *
 * 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 zipkin.server;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.springframework.beans.factory.NoSuchBeanDefinitionException;
import org.springframework.boot.actuate.health.HealthAggregator;
import org.springframework.boot.actuate.health.OrderedHealthAggregator;
import org.springframework.boot.actuate.metrics.Metric;
import org.springframework.boot.actuate.metrics.buffer.CounterBuffers;
import org.springframework.boot.actuate.metrics.buffer.GaugeBuffers;
import org.springframework.boot.autoconfigure.PropertyPlaceholderAutoConfiguration;
import org.springframework.context.annotation.AnnotationConfigApplicationContext;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Import;

import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.Matchers.equalTo;
import static org.hamcrest.Matchers.notNullValue;
import static org.junit.Assert.fail;

public class ZipkinServerConfigurationTest {
    AnnotationConfigApplicationContext context;

    @Before
    public void init() {
        context = new AnnotationConfigApplicationContext();
    }

    @After
    public void close() {
        if (context != null) {
            context.close();
        }
    }

    @Test
    public void ActuateCollectorMetrics_buffersAreNotPresent() {
        context.register(PropertyPlaceholderAutoConfiguration.class, ZipkinServerConfigurationTest.Config.class,
                ZipkinServerConfiguration.class);
        context.refresh();

        assertBeanNotPresent(CounterBuffers.class);
        assertBeanNotPresent(GaugeBuffers.class);

        assertMetrics();
    }

    @Test
    public void ActuateCollectorMetrics_buffersArePresent() {
        context.register(PropertyPlaceholderAutoConfiguration.class,
                ZipkinServerConfigurationTest.ConfigWithBuffers.class, ZipkinServerConfiguration.class);
        context.refresh();

        assertThat(context.getBean(CounterBuffers.class), notNullValue());
        assertThat(context.getBean(GaugeBuffers.class), notNullValue());

        assertMetrics();
    }

    private void assertMetrics() {
        ActuateCollectorMetrics metrics = context.getBean(ActuateCollectorMetrics.class);
        metrics.incrementBytes(20);
        assertThat(findMetric(metrics, "gauge.zipkin_collector.message_bytes").getValue(), equalTo(20.0d));
    }

    private Metric<?> findMetric(ActuateCollectorMetrics metrics, String metricName) {
        return metrics.metrics().stream().filter(m -> m.getName().equals(metricName)).findAny().get();
    }

    private void assertBeanNotPresent(Class<?> beanClass) {
        try {
            context.getBean(beanClass);
            fail("Unexpected bean for :" + beanClass);
        } catch (NoSuchBeanDefinitionException ex) {
            // expected
        }
    }

    private void assertBeanPresent(Class<?> beanClass) {
        assertThat(context.getBean(beanClass), notNullValue());
    }

    @Configuration
    public static class Config {
        @Bean
        public HealthAggregator healthAggregator() {
            return new OrderedHealthAggregator();
        }
    }

    @Configuration
    @Import(Config.class)
    public static class ConfigWithBuffers {
        @Bean
        public CounterBuffers counterBuffers() {
            return new CounterBuffers();
        }

        @Bean
        public GaugeBuffers gaugeBuffers() {
            return new GaugeBuffers();
        }
    }
}