diff --git a/video/src/test/java/com/example/video/DetectLogoGcsTest.java b/video/src/test/java/com/example/video/DetectLogoGcsTest.java index 7695de3108d..11715715503 100644 --- a/video/src/test/java/com/example/video/DetectLogoGcsTest.java +++ b/video/src/test/java/com/example/video/DetectLogoGcsTest.java @@ -1,5 +1,5 @@ /* - * Copyright 2020 Google Inc. + * Copyright 2021 Google Inc. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. @@ -18,6 +18,7 @@ import static com.google.common.truth.Truth.assertThat; +import com.google.cloud.testing.junit4.MultipleAttemptsRule; import java.io.ByteArrayOutputStream; import java.io.IOException; import java.io.PrintStream; @@ -25,6 +26,7 @@ import java.util.concurrent.TimeoutException; import org.junit.After; import org.junit.Before; +import org.junit.Rule; import org.junit.Test; public class DetectLogoGcsTest { @@ -43,6 +45,8 @@ public void tearDown() { System.setOut(null); } + @Rule public MultipleAttemptsRule multipleAttemptsRule = new MultipleAttemptsRule(3); + @Test public void testLogoDetectGcs() throws IOException, ExecutionException, InterruptedException, TimeoutException {