diff --git a/vaadin-spring/src/main/java/com/vaadin/flow/spring/VaadinServletContextInitializer.java b/vaadin-spring/src/main/java/com/vaadin/flow/spring/VaadinServletContextInitializer.java index 9c0f45b3d..7803d7a1c 100644 --- a/vaadin-spring/src/main/java/com/vaadin/flow/spring/VaadinServletContextInitializer.java +++ b/vaadin-spring/src/main/java/com/vaadin/flow/spring/VaadinServletContextInitializer.java @@ -258,8 +258,12 @@ protected Collection> getServiceTypes() { @Override protected Lookup createLookup( Map, Collection> services) { - services.put(Executor.class, Collections - .singleton(appContext.getBean(TaskExecutor.class))); + Map executors = appContext + .getBeansOfType(TaskExecutor.class); + if (!executors.isEmpty()) { + services.put(Executor.class, Collections + .singleton(executors.values().iterator().next())); + } return super.createLookup(services); } @@ -739,7 +743,8 @@ private List getDefaultPackages() { private List getLookupPackages() { return Stream.concat(getDefaultPackages().stream(), - Stream.of("com.vaadin.flow.server.frontend.fusion", "com.vaadin.flow.component.polymertemplate.rpc")) + Stream.of("com.vaadin.flow.server.frontend.fusion", + "com.vaadin.flow.component.polymertemplate.rpc")) .collect(Collectors.toList()); } diff --git a/vaadin-spring/src/test/java/com/vaadin/flow/spring/VaadinServletContextInitializerTest.java b/vaadin-spring/src/test/java/com/vaadin/flow/spring/VaadinServletContextInitializerTest.java index acc813098..1d1a2e10b 100644 --- a/vaadin-spring/src/test/java/com/vaadin/flow/spring/VaadinServletContextInitializerTest.java +++ b/vaadin-spring/src/test/java/com/vaadin/flow/spring/VaadinServletContextInitializerTest.java @@ -70,8 +70,8 @@ public class VaadinServletContextInitializerTest { public void init() { MockitoAnnotations.openMocks(this); - Mockito.when(applicationContext.getBean(TaskExecutor.class)) - .thenReturn(executor); + Mockito.when(applicationContext.getBeansOfType(TaskExecutor.class)) + .thenReturn(Collections.singletonMap("foo", executor)); PowerMockito.mockStatic( VaadinServletContextInitializer.SpringStubServletConfig.class); @@ -107,7 +107,7 @@ public void onStartup_devModeNotInitialized_devModeInitialized() theMock.verifyNoMoreInteractions(); } - Mockito.verify(applicationContext).getBean(TaskExecutor.class); + Mockito.verify(applicationContext).getBeansOfType(TaskExecutor.class); Mockito.verify(servletContext).setAttribute( Mockito.eq(Lookup.class.getName()), capture.capture()); Lookup lookup = capture.getValue();