diff --git a/gerrit-acceptance-framework/src/test/java/com/google/gerrit/acceptance/EventRecorder.java b/gerrit-acceptance-framework/src/test/java/com/google/gerrit/acceptance/EventRecorder.java index 6cc8d3c628..fbdfee6768 100644 --- a/gerrit-acceptance-framework/src/test/java/com/google/gerrit/acceptance/EventRecorder.java +++ b/gerrit-acceptance-framework/src/test/java/com/google/gerrit/acceptance/EventRecorder.java @@ -16,7 +16,6 @@ package com.google.gerrit.acceptance; import static com.google.common.truth.Truth.assertThat; -import com.google.common.base.Function; import com.google.common.collect.FluentIterable; import com.google.common.collect.ImmutableList; import com.google.common.collect.LinkedListMultimap; @@ -90,16 +89,6 @@ public class EventRecorder { return String.format("%s-%s-%s", type, project, ref); } - private static class RefEventTransformer - implements Function { - - @SuppressWarnings("unchecked") - @Override - public T apply(RefEvent e) { - return (T) e; - } - } - private ImmutableList getRefUpdatedEvents(String project, String refName, int expectedSize) { String key = refEventKey(RefUpdatedEvent.TYPE, project, refName); @@ -111,7 +100,7 @@ public class EventRecorder { assertThat(recordedEvents).containsKey(key); ImmutableList events = FluentIterable .from(recordedEvents.get(key)) - .transform(new RefEventTransformer()) + .transform(RefUpdatedEvent.class::cast) .toList(); assertThat(events).hasSize(expectedSize); return events; @@ -128,7 +117,7 @@ public class EventRecorder { assertThat(recordedEvents).containsKey(key); ImmutableList events = FluentIterable .from(recordedEvents.get(key)) - .transform(new RefEventTransformer()) + .transform(ChangeMergedEvent.class::cast) .toList(); assertThat(events).hasSize(expectedSize); return events; @@ -144,7 +133,7 @@ public class EventRecorder { assertThat(recordedEvents).containsKey(key); ImmutableList events = FluentIterable .from(recordedEvents.get(key)) - .transform(new RefEventTransformer()) + .transform(ReviewerDeletedEvent.class::cast) .toList(); assertThat(events).hasSize(expectedSize); return events; @@ -217,4 +206,4 @@ public class EventRecorder { public void close() { eventListenerRegistration.remove(); } -} \ No newline at end of file +}