fix: one more tests fix

vaadin · Nov 26, 2020 · 51291db · 51291db
1 parent 3dd49c9
commit 51291db
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/flow-lit-template/src/test/java/com/vaadin/flow/server/MockVaadinServletService.java b/flow-lit-template/src/test/java/com/vaadin/flow/server/MockVaadinServletService.java
@@ -95,6 +95,8 @@ protected List<RequestHandler> createRequestHandlers()
 
     public void init(Instantiator instantiator) {
         this.instantiator = instantiator;
+
+        init();
     }
 
     @Override
@@ -118,7 +120,9 @@ public void init() {
                     .thenReturn(resourceProvider);
             Mockito.when(context.getAttribute(Lookup.class.getName()))
                     .thenReturn(lookup);
-            getServlet().init(config);
+            if (getServlet().getServletConfig() == null) {
+                getServlet().init(config);
+            }
             super.init();
         } catch (ServiceException | ServletException e) {
             throw new RuntimeException(e);

diff --git a/flow-polymer-template/src/test/java/com/vaadin/flow/server/MockVaadinServletService.java b/flow-polymer-template/src/test/java/com/vaadin/flow/server/MockVaadinServletService.java
@@ -95,6 +95,8 @@ protected List<RequestHandler> createRequestHandlers()
 
     public void init(Instantiator instantiator) {
         this.instantiator = instantiator;
+
+        init();
     }
 
     @Override
@@ -118,7 +120,9 @@ public void init() {
                     .thenReturn(resourceProvider);
             Mockito.when(context.getAttribute(Lookup.class.getName()))
                     .thenReturn(lookup);
-            getServlet().init(config);
+            if (getServlet().getServletConfig() == null) {
+                getServlet().init(config);
+            }
             super.init();
         } catch (ServiceException | ServletException e) {
             throw new RuntimeException(e);