From 3ae92ebb9179bd95394d3eb5a428068241f098b0 Mon Sep 17 00:00:00 2001 From: Mark Paluch Date: Sun, 30 Jun 2019 23:42:16 +0200 Subject: [PATCH] Polishing. Fix type parameter in stubbing captor. --- .../lease/SecretLeaseContainerUnitTests.java | 20 +++++++++++-------- 1 file changed, 12 insertions(+), 8 deletions(-) diff --git a/spring-vault-core/src/test/java/org/springframework/vault/core/lease/SecretLeaseContainerUnitTests.java b/spring-vault-core/src/test/java/org/springframework/vault/core/lease/SecretLeaseContainerUnitTests.java index f5f901fb..26da1414 100644 --- a/spring-vault-core/src/test/java/org/springframework/vault/core/lease/SecretLeaseContainerUnitTests.java +++ b/spring-vault-core/src/test/java/org/springframework/vault/core/lease/SecretLeaseContainerUnitTests.java @@ -285,20 +285,22 @@ class SecretLeaseContainerUnitTests { verifyZeroInteractions(scheduledFuture); verify(taskScheduler, times(2)).schedule(captor.capture(), any(Trigger.class)); - ArgumentCaptor createdEvents = ArgumentCaptor + ArgumentCaptor createdEvents = ArgumentCaptor .forClass(SecretLeaseCreatedEvent.class); verify(leaseListenerAdapter, times(3)).onLeaseEvent(createdEvents.capture()); - List events = createdEvents.getAllValues(); + List events = createdEvents.getAllValues(); assertThat(events).hasSize(3); assertThat(events.get(0)).isInstanceOf(SecretLeaseCreatedEvent.class); - assertThat(events.get(0).getSecrets()).containsOnlyKeys("key"); + assertThat(((SecretLeaseCreatedEvent) events.get(0)).getSecrets()) + .containsOnlyKeys("key"); assertThat(events.get(1)).isInstanceOf(SecretLeaseExpiredEvent.class); assertThat(events.get(2)).isInstanceOf(SecretLeaseCreatedEvent.class); - assertThat(events.get(2).getSecrets()).containsOnlyKeys("foo"); + assertThat(((SecretLeaseCreatedEvent) events.get(2)).getSecrets()) + .containsOnlyKeys("foo"); } @Test @@ -344,20 +346,22 @@ class SecretLeaseContainerUnitTests { captor.getValue().run(); verify(taskScheduler, times(2)).schedule(captor.capture(), any(Trigger.class)); - ArgumentCaptor createdEvents = ArgumentCaptor + ArgumentCaptor createdEvents = ArgumentCaptor .forClass(SecretLeaseCreatedEvent.class); verify(leaseListenerAdapter, times(3)).onLeaseEvent(createdEvents.capture()); - List events = createdEvents.getAllValues(); + List events = createdEvents.getAllValues(); assertThat(events).hasSize(3); assertThat(events.get(0)).isInstanceOf(SecretLeaseCreatedEvent.class); - assertThat(events.get(0).getSecrets()).containsOnlyKeys("key"); + assertThat(((SecretLeaseCreatedEvent) events.get(0)).getSecrets()) + .containsOnlyKeys("key"); assertThat(events.get(1)).isInstanceOf(SecretLeaseExpiredEvent.class); assertThat(events.get(2)).isInstanceOf(SecretLeaseCreatedEvent.class); - assertThat(events.get(2).getSecrets()).containsOnlyKeys("foo"); + assertThat(((SecretLeaseCreatedEvent) events.get(2)).getSecrets()) + .containsOnlyKeys("foo"); } @Test