diff --git a/sdks/java/io/rrio/src/test/java/org/apache/beam/io/requestresponse/CallTest.java b/sdks/java/io/rrio/src/test/java/org/apache/beam/io/requestresponse/CallTest.java index 23ced30ada55..0764ab8db405 100644 --- a/sdks/java/io/rrio/src/test/java/org/apache/beam/io/requestresponse/CallTest.java +++ b/sdks/java/io/rrio/src/test/java/org/apache/beam/io/requestresponse/CallTest.java @@ -123,7 +123,7 @@ public void givenCallerThrowsQuotaException_emitsIntoFailurePCollection() { @Test public void givenCallerTimeout_emitsFailurePCollection() { - Duration timeout = Duration.standardMinutes(1L); + Duration timeout = Duration.standardSeconds(1L); Result result = pipeline .apply(Create.of(new Request("a"))) @@ -182,8 +182,7 @@ public void givenSetupThrowsQuotaException_throwsError() { @Test public void givenSetupTimeout_throwsError() { - Duration timeout = Duration.standardMinutes(1L); - + Duration timeout = Duration.standardSeconds(1L); pipeline .apply(Create.of(new Request(""))) .apply( @@ -231,7 +230,7 @@ public void givenTeardownThrowsQuotaException_throwsError() { @Test public void givenTeardownTimeout_throwsError() { - Duration timeout = Duration.standardMinutes(1L); + Duration timeout = Duration.standardSeconds(1L); pipeline .apply(Create.of(new Request(""))) .apply( @@ -271,7 +270,7 @@ public void givenValidCaller_emitValidResponse() { private static class ValidCaller implements Caller { @Override - public Response call(Request request) throws UserCodeExecutionException { + public Response call(Request request) { return new Response(request.id); } } @@ -282,7 +281,7 @@ private static class UnSerializableCaller implements Caller { private final UnSerializable nestedThing = new UnSerializable(); @Override - public Response call(Request request) throws UserCodeExecutionException { + public Response call(Request request) { return new Response(request.id); } } @@ -291,10 +290,10 @@ private static class UnSerializableCallerWithSetupTeardown extends UnSerializabl implements SetupTeardown { @Override - public void setup() throws UserCodeExecutionException {} + public void setup() {} @Override - public void teardown() throws UserCodeExecutionException {} + public void teardown() {} } private static class UnSerializable {} @@ -358,11 +357,11 @@ private static class CallerExceedsTimeout implements Caller { private final Duration timeout; CallerExceedsTimeout(Duration timeout) { - this.timeout = timeout.plus(Duration.standardSeconds(1L)); + this.timeout = timeout.plus(Duration.standardSeconds(10L)); } @Override - public Response call(Request request) throws UserCodeExecutionException { + public Response call(Request request) { sleep(timeout); return new Response(request.id); } @@ -397,16 +396,16 @@ private static class SetupExceedsTimeout implements SetupTeardown { private final Duration timeout; private SetupExceedsTimeout(Duration timeout) { - this.timeout = timeout.plus(Duration.standardSeconds(1L)); + this.timeout = timeout.plus(Duration.standardSeconds(10L)); } @Override - public void setup() throws UserCodeExecutionException { + public void setup() { sleep(timeout); } @Override - public void teardown() throws UserCodeExecutionException {} + public void teardown() {} } private static class SetupThrowsUserCodeExecutionException implements SetupTeardown { @@ -416,7 +415,7 @@ public void setup() throws UserCodeExecutionException { } @Override - public void teardown() throws UserCodeExecutionException {} + public void teardown() {} } private static class SetupThrowsUserCodeQuotaException implements SetupTeardown { @@ -426,7 +425,7 @@ public void setup() throws UserCodeExecutionException { } @Override - public void teardown() throws UserCodeExecutionException {} + public void teardown() {} } private static class SetupThrowsUserCodeTimeoutException implements SetupTeardown { @@ -436,28 +435,28 @@ public void setup() throws UserCodeExecutionException { } @Override - public void teardown() throws UserCodeExecutionException {} + public void teardown() {} } private static class TeardownExceedsTimeout implements SetupTeardown { private final Duration timeout; private TeardownExceedsTimeout(Duration timeout) { - this.timeout = timeout.plus(Duration.standardSeconds(1L)); + this.timeout = timeout.plus(Duration.standardSeconds(10L)); } @Override - public void setup() throws UserCodeExecutionException {} + public void setup() {} @Override - public void teardown() throws UserCodeExecutionException { + public void teardown() { sleep(timeout); } } private static class TeardownThrowsUserCodeExecutionException implements SetupTeardown { @Override - public void setup() throws UserCodeExecutionException {} + public void setup() {} @Override public void teardown() throws UserCodeExecutionException { @@ -467,7 +466,7 @@ public void teardown() throws UserCodeExecutionException { private static class TeardownThrowsUserCodeQuotaException implements SetupTeardown { @Override - public void setup() throws UserCodeExecutionException {} + public void setup() {} @Override public void teardown() throws UserCodeExecutionException { @@ -477,7 +476,7 @@ public void teardown() throws UserCodeExecutionException { private static class TeardownThrowsUserCodeTimeoutException implements SetupTeardown { @Override - public void setup() throws UserCodeExecutionException {} + public void setup() {} @Override public void teardown() throws UserCodeExecutionException { @@ -519,14 +518,12 @@ private static class DeterministicRequestCoder extends CustomCoder<@NonNull Requ private static final Coder ID_CODER = StringUtf8Coder.of(); @Override - public void encode(Request value, @NotNull OutputStream outStream) - throws CoderException, IOException { + public void encode(Request value, @NotNull OutputStream outStream) throws IOException { ID_CODER.encode(checkStateNotNull(value).id, outStream); } @Override - public @NonNull Request decode(@NotNull InputStream inStream) - throws CoderException, IOException { + public @NonNull Request decode(@NotNull InputStream inStream) throws IOException { String id = ID_CODER.decode(inStream); return new Request(id); } @@ -542,7 +539,7 @@ private static class DeterministicResponseCoder extends CustomCoder { @Override public void encode(@Nullable Response value, @NotNull OutputStream outStream) - throws CoderException, IOException { + throws IOException { if (value == null) { ID_CODER.encode(null, outStream); return; @@ -551,7 +548,7 @@ public void encode(@Nullable Response value, @NotNull OutputStream outStream) } @Override - public Response decode(@NotNull InputStream inStream) throws CoderException, IOException { + public Response decode(@NotNull InputStream inStream) throws IOException { try { String id = ID_CODER.decode(inStream); return new Response(id);