Extract MockUI class

Change-Id: I2a85520c89b952bed3308156afa8aff3f80082bd

1 files changed, 1 insertions, 13 deletions

diff --git a/server/tests/src/com/vaadin/ui/LabelDataSourceTest.java b/server/tests/src/com/vaadin/ui/LabelDataSourceTest.java
index 8ec3ca1245..030504cf2b 100644
--- a/server/tests/src/com/vaadin/ui/LabelDataSourceTest.java
+++ b/server/tests/src/com/vaadin/ui/LabelDataSourceTest.java
@@ -22,9 +22,9 @@ import org.junit.Before;
 import org.junit.Test;
 
 import com.vaadin.data.util.ObjectProperty;
-import com.vaadin.server.VaadinRequest;
 import com.vaadin.server.VaadinSession;
 import com.vaadin.tests.util.AlwaysLockedVaadinSession;
+import com.vaadin.tests.util.MockUI;
 
 public class LabelDataSourceTest {
 
@@ -104,18 +104,6 @@ public class LabelDataSourceTest {
         Assert.assertEquals("after", label.getValue());
     }
 
-    public class MockUI extends UI {
-
-        public MockUI() {
-            setSession(vaadinSession);
-        }
-
-        @Override
-        protected void init(VaadinRequest request) {
-        }
-
-    }
-
     @Test
     public void attachToSessionWithDifferentLocale() {
         label.setValue("before");