fix unit tests

StefanBratanov · Oct 17, 2023 · 735906c · 735906c
1 parent 1b14966
commit 735906c
Showing 1 changed file with 7 additions and 10 deletions.
diff --git a/...tor/client/src/test/java/tech/pegasys/teku/validator/client/ValidatorRegistratorTest.java b/...tor/client/src/test/java/tech/pegasys/teku/validator/client/ValidatorRegistratorTest.java
@@ -115,9 +115,8 @@ void setUp(final SpecContext specContext) {
         .thenAnswer(
             args -> {
               final Validator validator = (Validator) args.getArguments()[0];
-              return Optional.of(
-                  SafeFuture.completedFuture(
-                      createSignedValidatorRegistration(validator.getPublicKey())));
+              return SafeFuture.completedFuture(
+                  createSignedValidatorRegistration(validator.getPublicKey()));
             });
 
     when(proposerConfigPropertiesProvider.isReadyToProvideProperties()).thenReturn(true);
@@ -251,11 +250,10 @@ void registerValidatorsEvenIfOneRegistrationSigningFails() {
             args -> {
               final Validator validator = (Validator) args.getArguments()[0];
               if (validator.equals(validator2)) {
-                return Optional.of(SafeFuture.failedFuture(new IllegalStateException("oopsy")));
+                return SafeFuture.failedFuture(new IllegalStateException("oopsy"));
               }
-              return Optional.of(
-                  SafeFuture.completedFuture(
-                      createSignedValidatorRegistration(validator.getPublicKey())));
+              return SafeFuture.completedFuture(
+                  createSignedValidatorRegistration(validator.getPublicKey()));
             });
     runRegistrationFlowWithSubscription(0);
 
@@ -264,9 +262,8 @@ void registerValidatorsEvenIfOneRegistrationSigningFails() {
         .thenAnswer(
             args -> {
               final Validator validator = (Validator) args.getArguments()[0];
-              return Optional.of(
-                  SafeFuture.completedFuture(
-                      createSignedValidatorRegistration(validator.getPublicKey())));
+              return SafeFuture.completedFuture(
+                  createSignedValidatorRegistration(validator.getPublicKey()));
             });
     runRegistrationFlowWithSubscription(1);