Call UI.detach in UI cleanup (#9755)

Change-Id: I0050994247018270f8b39b103bcfdce2b5053c6f

1 files changed, 48 insertions, 0 deletions

diff --git a/server/tests/src/com/vaadin/tests/server/TestUICleanup.java b/server/tests/src/com/vaadin/tests/server/TestUICleanup.java
new file mode 100644
index 0000000000..83d6838bfe
--- /dev/null
+++ b/server/tests/src/com/vaadin/tests/server/TestUICleanup.java
@@ -0,0 +1,48 @@
+package com.vaadin.tests.server;
+
+import org.easymock.EasyMock;
+import org.junit.Test;
+
+import com.vaadin.server.VaadinRequest;
+import com.vaadin.server.VaadinService;
+import com.vaadin.server.VaadinServiceSession;
+import com.vaadin.ui.UI;
+import com.vaadin.ui.UI.CleanupEvent;
+import com.vaadin.ui.UI.CleanupListener;
+
+public class TestUICleanup {
+
+    @Test
+    public void uiDetach() {
+        VaadinService service = EasyMock.createMock(VaadinService.class);
+
+        UI ui = EasyMock.createMock(UI.class);
+        ui.setSession(null);
+        ui.detach();
+        ui.fireCleanupEvent();
+        EasyMock.expect(ui.getUIId()).andReturn(1);
+
+        EasyMock.replay(service, ui);
+
+        VaadinServiceSession session = new VaadinServiceSession(service);
+
+        session.cleanupUI(ui);
+    }
+
+    @Test
+    public void uiCleanupListeners() {
+        CleanupListener listener = EasyMock.createMock(CleanupListener.class);
+        listener.cleanup(EasyMock.anyObject(CleanupEvent.class));
+
+        EasyMock.replay(listener);
+
+        UI ui = new UI() {
+            @Override
+            protected void init(VaadinRequest request) {
+            }
+        };
+
+        ui.addCleanupListener(listener);
+        ui.fireCleanupEvent();
+    }
+}