Add missing ScreenshotRule to Spring integration test (#8455)

1 files changed, 6 insertions, 0 deletions

diff --git a/test/spring-boot/src/test/java/com/example/VaadinSpringBootSmokeIT.java b/test/spring-boot/src/test/java/com/example/VaadinSpringBootSmokeIT.java
index 2f1cb545d8..76fc8790e2 100644
--- a/test/spring-boot/src/test/java/com/example/VaadinSpringBootSmokeIT.java
+++ b/test/spring-boot/src/test/java/com/example/VaadinSpringBootSmokeIT.java
@@ -2,6 +2,7 @@ package com.example;
 
 import org.junit.Assert;
 import org.junit.Before;
+import org.junit.Rule;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.openqa.selenium.phantomjs.PhantomJSDriver;
@@ -10,6 +11,7 @@ import org.springframework.boot.test.context.SpringBootTest;
 import org.springframework.boot.test.context.SpringBootTest.WebEnvironment;
 import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
 
+import com.vaadin.testbench.ScreenshotOnFailureRule;
 import com.vaadin.testbench.TestBench;
 import com.vaadin.testbench.TestBenchTestCase;
 import com.vaadin.testbench.elements.ButtonElement;
@@ -24,6 +26,10 @@ import com.vaadin.testbench.parallel.Browser;
 @SpringBootTest(webEnvironment = WebEnvironment.RANDOM_PORT)
 public class VaadinSpringBootSmokeIT extends TestBenchTestCase {
 
+    @Rule
+    public ScreenshotOnFailureRule screenshotRule = new ScreenshotOnFailureRule(
+            this, true);
+
     @LocalServerPort
     Integer port;