mirror of
https://github.com/signalapp/Signal-Server
synced 2026-04-22 23:28:04 +01:00
Extend Experiment to report more detail when results don't match.
This commit is contained in:
committed by
Jon Chambers
parent
75cbfa2898
commit
a0bebca1e6
@@ -1,47 +1,36 @@
|
||||
package org.whispersystems.textsecuregcm.experiment;
|
||||
|
||||
import io.micrometer.core.instrument.Meter;
|
||||
import io.micrometer.core.instrument.MeterRegistry;
|
||||
import io.micrometer.core.instrument.Tag;
|
||||
import io.micrometer.core.instrument.Timer;
|
||||
import io.micrometer.core.instrument.distribution.DistributionStatisticConfig;
|
||||
import org.LatencyUtils.PauseDetector;
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
import org.mockito.ArgumentCaptor;
|
||||
import org.mockito.ArgumentMatchers;
|
||||
|
||||
import java.util.HashSet;
|
||||
import java.util.Set;
|
||||
import java.util.concurrent.CompletableFuture;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
|
||||
import static org.junit.Assert.*;
|
||||
import static org.mockito.ArgumentMatchers.any;
|
||||
import static org.mockito.ArgumentMatchers.anyLong;
|
||||
import static org.mockito.ArgumentMatchers.anyString;
|
||||
import static org.mockito.ArgumentMatchers.eq;
|
||||
import static org.mockito.Mockito.mock;
|
||||
import static org.mockito.Mockito.never;
|
||||
import static org.mockito.Mockito.verify;
|
||||
import static org.mockito.Mockito.verifyNoMoreInteractions;
|
||||
import static org.mockito.Mockito.when;
|
||||
|
||||
public class ExperimentTest {
|
||||
|
||||
private Timer matchTimer;
|
||||
private Timer mismatchTimer;
|
||||
private Timer errorTimer;
|
||||
private Timer bothPresentMismatchTimer;
|
||||
private Timer controlNullMismatchTimer;
|
||||
private Timer experimentNullMismatchTimer;
|
||||
|
||||
private Experiment experiment;
|
||||
|
||||
@Before
|
||||
public void setUp() {
|
||||
matchTimer = mock(Timer.class);
|
||||
mismatchTimer = mock(Timer.class);
|
||||
errorTimer = mock(Timer.class);
|
||||
bothPresentMismatchTimer = mock(Timer.class);
|
||||
controlNullMismatchTimer = mock(Timer.class);
|
||||
experimentNullMismatchTimer = mock(Timer.class);
|
||||
|
||||
experiment = new Experiment(matchTimer, mismatchTimer, errorTimer);
|
||||
experiment = new Experiment(matchTimer, errorTimer, bothPresentMismatchTimer, controlNullMismatchTimer, experimentNullMismatchTimer);
|
||||
}
|
||||
|
||||
@Test
|
||||
@@ -51,9 +40,21 @@ public class ExperimentTest {
|
||||
}
|
||||
|
||||
@Test
|
||||
public void compareFutureResultMismatch() {
|
||||
public void compareFutureResultMismatchBothPresent() {
|
||||
experiment.compareFutureResult(12, CompletableFuture.completedFuture(77));
|
||||
verify(mismatchTimer).record(anyLong(), eq(TimeUnit.NANOSECONDS));
|
||||
verify(bothPresentMismatchTimer).record(anyLong(), eq(TimeUnit.NANOSECONDS));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void compareFutureResultMismatchControlNull() {
|
||||
experiment.compareFutureResult(null, CompletableFuture.completedFuture(77));
|
||||
verify(controlNullMismatchTimer).record(anyLong(), eq(TimeUnit.NANOSECONDS));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void compareFutureResultMismatchExperimentNull() {
|
||||
experiment.compareFutureResult(12, CompletableFuture.completedFuture(null));
|
||||
verify(experimentNullMismatchTimer).record(anyLong(), eq(TimeUnit.NANOSECONDS));
|
||||
}
|
||||
|
||||
@Test
|
||||
@@ -62,12 +63,6 @@ public class ExperimentTest {
|
||||
verify(errorTimer).record(anyLong(), eq(TimeUnit.NANOSECONDS));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void compareFutureResultNoExperimentData() {
|
||||
experiment.compareFutureResult(12, CompletableFuture.completedFuture(null));
|
||||
verifyNoMoreInteractions(matchTimer, mismatchTimer, errorTimer);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void compareSupplierResultMatch() {
|
||||
experiment.compareSupplierResult(12, () -> 12);
|
||||
@@ -75,9 +70,21 @@ public class ExperimentTest {
|
||||
}
|
||||
|
||||
@Test
|
||||
public void compareSupplierResultMismatch() {
|
||||
public void compareSupplierResultMismatchBothPresent() {
|
||||
experiment.compareSupplierResult(12, () -> 77);
|
||||
verify(mismatchTimer).record(anyLong(), eq(TimeUnit.NANOSECONDS));
|
||||
verify(bothPresentMismatchTimer).record(anyLong(), eq(TimeUnit.NANOSECONDS));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void compareSupplierResultMismatchControlNull() {
|
||||
experiment.compareSupplierResult(null, () -> 77);
|
||||
verify(controlNullMismatchTimer).record(anyLong(), eq(TimeUnit.NANOSECONDS));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void compareSupplierResultMismatchExperimentNull() {
|
||||
experiment.compareSupplierResult(12, () -> null);
|
||||
verify(experimentNullMismatchTimer).record(anyLong(), eq(TimeUnit.NANOSECONDS));
|
||||
}
|
||||
|
||||
@Test
|
||||
@@ -85,10 +92,4 @@ public class ExperimentTest {
|
||||
experiment.compareSupplierResult(12, () -> { throw new RuntimeException("OH NO"); });
|
||||
verify(errorTimer).record(anyLong(), eq(TimeUnit.NANOSECONDS));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void compareSupplierResultNoExperimentData() {
|
||||
experiment.compareSupplierResult(12, () -> null);
|
||||
verifyNoMoreInteractions(mismatchTimer, mismatchTimer, errorTimer);
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user