Use event objects for future-proofing UIProvider API (#9721)

1 files changed, 3 insertions, 2 deletions

diff --git a/server/tests/src/com/vaadin/tests/server/component/root/CustomUIClassLoader.java b/server/tests/src/com/vaadin/tests/server/component/root/CustomUIClassLoader.java
index 858a18b223..e84d1c9939 100644
--- a/server/tests/src/com/vaadin/tests/server/component/root/CustomUIClassLoader.java
+++ b/server/tests/src/com/vaadin/tests/server/component/root/CustomUIClassLoader.java
@@ -11,6 +11,7 @@ import org.easymock.EasyMock;
 import com.vaadin.DefaultDeploymentConfiguration;
 import com.vaadin.server.DefaultUIProvider;
 import com.vaadin.server.DeploymentConfiguration;
+import com.vaadin.server.UIClassSelectionEvent;
 import com.vaadin.server.VaadinRequest;
 import com.vaadin.server.VaadinService;
 import com.vaadin.server.VaadinSession;
@@ -59,7 +60,7 @@ public class CustomUIClassLoader extends TestCase {
 
         DefaultUIProvider uiProvider = new DefaultUIProvider();
         Class<? extends UI> uiClass = uiProvider
-                .getUIClass(createRequestMock(null));
+                .getUIClass(new UIClassSelectionEvent(createRequestMock(null)));
 
         assertEquals(MyUI.class, uiClass);
     }
@@ -103,7 +104,7 @@ public class CustomUIClassLoader extends TestCase {
 
         DefaultUIProvider uiProvider = new DefaultUIProvider();
         Class<? extends UI> uiClass = uiProvider
-                .getUIClass(createRequestMock(loggingClassLoader));
+                .getUIClass(new UIClassSelectionEvent(createRequestMock(null)));
 
         assertEquals(MyUI.class, uiClass);
         assertEquals(1, loggingClassLoader.requestedClasses.size());