container-disc/src/test/java/com/yahoo/container/jdisc/metric/MetricConsumerProviderTest.java


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

// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.container.jdisc.metric;

import com.yahoo.jdisc.application.MetricConsumer;
import org.junit.jupiter.api.Test;
import org.mockito.Mockito;

import static org.junit.jupiter.api.Assertions.assertNotSame;
import static org.junit.jupiter.api.Assertions.assertSame;

/**
 * @author Simon Thoresen Hult
 */
public class MetricConsumerProviderTest {

    @Test
    void requireThatSingleConsumerIsNotDelegated() {
        MetricConsumer consumer = Mockito.mock(MetricConsumer.class);
        MetricConsumerProvider provider = MetricConsumerProviders.newSingletonFactories(consumer);
        assertSame(consumer, provider.newInstance());
    }

    @Test
    void requireThatMultipleConsumersAreDelegated() {
        MetricConsumer foo = Mockito.mock(MetricConsumer.class);
        MetricConsumer bar = Mockito.mock(MetricConsumer.class);
        MetricConsumerProvider provider = MetricConsumerProviders.newSingletonFactories(foo, bar);
        MetricConsumer consumer = provider.newInstance();
        assertNotSame(foo, consumer);
        assertNotSame(bar, consumer);
        consumer.add("foo", 6, null);
        Mockito.verify(foo, Mockito.times(1)).add("foo", 6, null);
        Mockito.verify(bar, Mockito.times(1)).add("foo", 6, null);
    }

}