test: do not use getInShadowRoot (#976)

vaadin · Dec 16, 2021 · 811b0a0 · 811b0a0
1 parent 42405ad
commit 811b0a0
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/...-tests/test-spring-common/src/test/java/com/vaadin/flow/spring/test/ParentTemplateIT.java b/...-tests/test-spring-common/src/test/java/com/vaadin/flow/spring/test/ParentTemplateIT.java
@@ -21,6 +21,8 @@
 import org.openqa.selenium.By;
 import org.openqa.selenium.WebElement;
 
+import com.vaadin.testbench.TestBenchElement;
+
 public class ParentTemplateIT extends AbstractSpringTest {
 
     @Override
@@ -32,26 +34,24 @@ protected String getTestPath() {
     public void customElementIsRegistered() throws Exception {
         open();
 
-        WebElement template = findElement(By.tagName("parent-template"));
-        WebElement div = getInShadowRoot(template, By.id("div"));
+        TestBenchElement template = $("parent-template").first();
+        TestBenchElement div = template.$("*").id("div");
 
         Assert.assertEquals("baz", div.getText());
 
-        WebElement child = getInShadowRoot(template, By.id("child"));
+        TestBenchElement child = template.$("*").id("child");
 
-        Assert.assertEquals("bar",
-                getInShadowRoot(child, By.id("info")).getText());
+        Assert.assertEquals("bar", child.$("*").id("info").getText());
     }
 
     @Test
     public void injectedComponentIsSpringManaged() throws Exception {
         open();
 
-        WebElement template = findElement(By.tagName("parent-template"));
+        TestBenchElement template = $("parent-template").first();
 
-        WebElement child = getInShadowRoot(template, By.id("child"));
+        TestBenchElement child = template.$("*").id("child");
 
-        Assert.assertEquals("foo",
-                getInShadowRoot(child, By.id("message")).getText());
+        Assert.assertEquals("foo", child.$("*").id("message").getText());
     }
 }