fix: add missing annotations

FlowingCode · Mar 7, 2023 · e3427b2 · e3427b2
1 parent c09855a
commit e3427b2
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/src/test/java/com/flowingcode/vaadin/addons/twincolgrid/FilterableDemo.java b/src/test/java/com/flowingcode/vaadin/addons/twincolgrid/FilterableDemo.java
@@ -20,6 +20,7 @@
 
 package com.flowingcode.vaadin.addons.twincolgrid;
 
+import com.flowingcode.vaadin.addons.demo.DemoSource;
 import com.vaadin.flow.component.orderedlayout.VerticalLayout;
 import com.vaadin.flow.router.PageTitle;
 import com.vaadin.flow.router.Route;
@@ -31,6 +32,7 @@
 
 @SuppressWarnings("serial")
 @PageTitle("Filterable")
+@DemoSource
 @Route(value = "twincolgrid/filterable", layout = TwincolDemoView.class)
 public class FilterableDemo extends VerticalLayout {
 

diff --git a/src/test/java/com/flowingcode/vaadin/addons/twincolgrid/LazyFilterableDemo.java b/src/test/java/com/flowingcode/vaadin/addons/twincolgrid/LazyFilterableDemo.java
@@ -20,6 +20,7 @@
 
 package com.flowingcode.vaadin.addons.twincolgrid;
 
+import com.flowingcode.vaadin.addons.demo.DemoSource;
 import com.vaadin.flow.component.button.Button;
 import com.vaadin.flow.component.grid.Grid;
 import com.vaadin.flow.component.notification.Notification;
@@ -38,28 +39,28 @@
 
 @SuppressWarnings("serial")
 @PageTitle("Lazy Filterable")
+@DemoSource
 @Route(value = "twincolgrid/lazyfilterable", layout = TwincolDemoView.class)
 public class LazyFilterableDemo extends VerticalLayout {
 
   public LazyFilterableDemo() {
     BookService bookService = new BookService();
+    BookFilter bookFilter = new BookFilter();
 
-    DataProvider<Book, BookFilter> availableDataProvider =
+    DataProvider<Book, Void> availableDataProvider =
         DataProvider.fromFilteringCallbacks(
             query -> {
-              BookFilter filter = query.getFilter().orElseGet(BookFilter::new);
-              filter.setSorting(query.getSortOrders().stream()
+              bookFilter.setSorting(query.getSortOrders().stream()
                   .map(q -> new SortOrder<>(q.getSorted(), q.getDirection()))
                   .collect(Collectors.toList()));
-              return bookService.fetch(query.getOffset(), query.getLimit(), filter);
+              return bookService.fetch(query.getOffset(), query.getLimit(), bookFilter);
             },
-            query -> bookService.count(query.getFilter().orElseGet(BookFilter::new)));
+            query -> bookService.count(bookFilter));
 
     Grid<Book> availableGrid = new Grid<>();
     availableGrid.setDataProvider(availableDataProvider);
 
     Grid<Book> selectionGrid = new Grid<>();
-    BookFilter bookFilter = new BookFilter();
 
     final TwinColGrid<Book> twinColGrid =
         new TwinColGrid<>(availableGrid, selectionGrid)