Fix DefaultCounterService.reset

Fix DefaultCounterService.reset to call the reset method of the
underlying MetricWriter.

Fixes gh-1391
pull/1487/merge
Phillip Webb 10 years ago
parent 2188e3553c
commit 5c34b07742

@ -47,7 +47,7 @@ public class DefaultCounterService implements CounterService {
@Override
public void reset(String metricName) {
this.writer.increment(new Delta<Long>(wrap(metricName), 0L));
this.writer.reset(wrap(metricName));
}
private String wrap(String metricName) {

@ -17,7 +17,10 @@
package org.springframework.boot.actuate.metrics.writer;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.ArgumentCaptor;
import org.mockito.Captor;
import org.mockito.runners.MockitoJUnitRunner;
import static org.junit.Assert.assertEquals;
import static org.mockito.Mockito.mock;
@ -26,6 +29,7 @@ import static org.mockito.Mockito.verify;
/**
* Tests for {@link DefaultCounterService}.
*/
@RunWith(MockitoJUnitRunner.class)
public class DefaultCounterServiceTests {
private final MetricWriter repository = mock(MetricWriter.class);
@ -33,22 +37,28 @@ public class DefaultCounterServiceTests {
private final DefaultCounterService service = new DefaultCounterService(
this.repository);
@Captor
private ArgumentCaptor<Delta<Number>> captor;
@Test
public void incrementPrependsCounter() {
this.service.increment("foo");
@SuppressWarnings("rawtypes")
ArgumentCaptor<Delta> captor = ArgumentCaptor.forClass(Delta.class);
verify(this.repository).increment(captor.capture());
assertEquals("counter.foo", captor.getValue().getName());
verify(this.repository).increment(this.captor.capture());
assertEquals("counter.foo", this.captor.getValue().getName());
assertEquals(1L, this.captor.getValue().getValue());
}
@Test
public void decrementPrependsCounter() {
this.service.decrement("foo");
@SuppressWarnings("rawtypes")
ArgumentCaptor<Delta> captor = ArgumentCaptor.forClass(Delta.class);
verify(this.repository).increment(captor.capture());
assertEquals("counter.foo", captor.getValue().getName());
assertEquals(-1L, captor.getValue().getValue());
verify(this.repository).increment(this.captor.capture());
assertEquals("counter.foo", this.captor.getValue().getName());
assertEquals(-1L, this.captor.getValue().getValue());
}
@Test
public void resetResetsCounter() throws Exception {
this.service.reset("foo");
verify(this.repository).reset("counter.foo");
}
}

Loading…
Cancel
Save