Made it easier to chain findAll() methods with the filter() method

serenity-bdd · Mar 16, 2024 · cbf3da5 · cbf3da5
1 parent 507df1f
commit cbf3da5
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/serenity-core/src/main/java/net/serenitybdd/core/pages/ListOfWebElementFacades.java b/serenity-core/src/main/java/net/serenitybdd/core/pages/ListOfWebElementFacades.java
@@ -37,8 +37,9 @@ public <T> List<T> map(Function<? super WebElementFacade, T> elementConverter) {
         return convert((list) -> list.stream().map(elementConverter).collect(Collectors.toList()));
     }
 
-    public List<? super WebElementFacade> filter(Predicate<? super WebElementFacade> condition) {
-        return convert((list) -> list.stream().filter(condition).collect(Collectors.toList()));
+    public ListOfWebElementFacades filter(Predicate<? super WebElementFacade> condition) {
+
+        return new ListOfWebElementFacades(convert((list) -> list.stream().filter(condition).collect(Collectors.toList())));
     }
 
     public void setFallback(Function<PageObject, ListOfWebElementFacades> fallbackMethod, PageObject page) {