2 files changed, 128 insertions, 0 deletions
diff --git a/uitest/src/main/java/com/vaadin/tests/push/PushRemoveConnectors.java b/uitest/src/main/java/com/vaadin/tests/push/PushRemoveConnectors.java
new file mode 100644
index 0000000000..528492c33a
--- /dev/null
+++ b/uitest/src/main/java/com/vaadin/tests/push/PushRemoveConnectors.java
@@ -0,0 +1,92 @@
+package com.vaadin.tests.push;
+
+import java.util.concurrent.Executors;
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.ScheduledFuture;
+import java.util.concurrent.TimeUnit;
+
+import org.apache.commons.lang.SerializationUtils;
+
+import com.vaadin.annotations.Push;
+import com.vaadin.data.Property.ValueChangeEvent;
+import com.vaadin.data.Property.ValueChangeListener;
+import com.vaadin.server.VaadinRequest;
+import com.vaadin.tests.components.AbstractTestUIWithLog;
+import com.vaadin.ui.AbstractOrderedLayout;
+import com.vaadin.ui.Button;
+import com.vaadin.ui.Button.ClickEvent;
+import com.vaadin.ui.Button.ClickListener;
+import com.vaadin.ui.CheckBox;
+import com.vaadin.ui.HorizontalLayout;
+import com.vaadin.ui.Label;
+
+@Push
+public class PushRemoveConnectors extends AbstractTestUIWithLog {
+
+    private transient final ScheduledExecutorService threadPool = Executors
+            .newScheduledThreadPool(5);
+    static final String START = "start";
+    static final String STOP = "stop";
+    private AbstractOrderedLayout verticalLayout;
+    private transient ScheduledFuture<?> task = null;
+
+    @Override
+    protected void setup(VaadinRequest request) {
+        final CheckBox pollingEnabled = new CheckBox("Polling enabled");
+        pollingEnabled.addValueChangeListener(new ValueChangeListener() {
+            @Override
+            public void valueChange(ValueChangeEvent event) {
+                setPollInterval(pollingEnabled.getValue() ? 1000 : -1);
+            }
+        });
+
+        Button start = new Button("start");
+        start.setId(START);
+        start.addClickListener(new ClickListener() {
+
+            @Override
+            public void buttonClick(ClickEvent event) {
+                task = threadPool.scheduleAtFixedRate(new Runnable() {
+                    @Override
+                    public void run() {
+                        access(new Runnable() {
+                            public void run() {
+                                populate();
+                                log("Serialized session size: "
+                                        + getSessionSize());
+                            }
+                        });
+                    }
+                }, 1, 1, TimeUnit.SECONDS);
+            }
+        });
+        Button stop = new Button("stop");
+        stop.setId(STOP);
+        stop.addClickListener(new ClickListener() {
+            @Override
+            public void buttonClick(ClickEvent event) {
+                if (task != null) {
+                    task.cancel(true);
+                    task = null;
+                }
+
+            }
+        });
+        verticalLayout = new HorizontalLayout();
+        populate();
+        addComponents(pollingEnabled, start, stop, verticalLayout);
+    }
+
+    private void populate() {
+        verticalLayout.removeAllComponents();
+        for (int i = 0; i < 500; i++) {
+            Label l = new Label(".");
+            l.setSizeUndefined();
+            verticalLayout.addComponent(l);
+        }
+    }
+
+    private int getSessionSize() {
+        return SerializationUtils.serialize(getSession()).length;
+    }
+}
diff --git a/uitest/src/test/java/com/vaadin/tests/push/PushRemoveConnectorsTest.java b/uitest/src/test/java/com/vaadin/tests/push/PushRemoveConnectorsTest.java
new file mode 100644
index 0000000000..bcf93b0aa9
--- /dev/null
+++ b/uitest/src/test/java/com/vaadin/tests/push/PushRemoveConnectorsTest.java
@@ -0,0 +1,36 @@
+package com.vaadin.tests.push;
+
+import org.junit.Assert;
+import org.junit.Test;
+
+import com.vaadin.testbench.elements.ButtonElement;
+import com.vaadin.tests.tb3.SingleBrowserTest;
+
+public class PushRemoveConnectorsTest extends SingleBrowserTest {
+
+    @Test
+    public void testNoMemoryLeak() throws InterruptedException {
+        openTestURL();
+        $(ButtonElement.class).id(PushRemoveConnectors.START).click();
+        Thread.sleep(5000);
+        int last = getMemoryUsage();
+        int i = 0;
+        while (i++ < 10) {
+            Thread.sleep(5000);
+            int now = getMemoryUsage();
+            System.out.println("Memory usage: "+now);
+            if (last == now)
+                break;
+
+            last = now;
+        }
+        $(ButtonElement.class).id(PushRemoveConnectors.STOP).click();
+
+        Assert.assertNotEquals(10, i);
+    }
+
+    private int getMemoryUsage() {
+        return Integer.parseInt(getLogRow(0).replaceFirst(
+                ".*Serialized session size: ", ""));
+    }
+}