Ensure PushConnection is properly cleaned up on disconnect (#12226, #12522)

Change-Id: I0bab199632554655ef92a624f5654852b4b157d1

1 files changed, 119 insertions, 0 deletions

diff --git a/uitest/src/com/vaadin/tests/push/EnableDisablePush.java b/uitest/src/com/vaadin/tests/push/EnableDisablePush.java
new file mode 100644
index 0000000000..1911c66c2d
--- /dev/null
+++ b/uitest/src/com/vaadin/tests/push/EnableDisablePush.java
@@ -0,0 +1,119 @@
+package com.vaadin.tests.push;
+
+import java.util.Date;
+import java.util.Timer;
+import java.util.TimerTask;
+import java.util.concurrent.TimeUnit;
+
+import com.vaadin.server.VaadinRequest;
+import com.vaadin.shared.communication.PushMode;
+import com.vaadin.tests.components.AbstractTestUI;
+import com.vaadin.tests.util.Log;
+import com.vaadin.ui.Button;
+import com.vaadin.ui.UIDetachedException;
+
+public class EnableDisablePush extends AbstractTestUI {
+
+    private int c = 0;
+
+    private Log log = new Log(15);
+
+    private final Timer timer = new Timer(true);
+
+    private final class CounterTask extends TimerTask {
+
+        @Override
+        public void run() {
+
+            try {
+                while (true) {
+                    TimeUnit.MILLISECONDS.sleep(1000);
+
+                    access(new Runnable() {
+                        @Override
+                        public void run() {
+                            log.log("Counter = " + c++);
+                            if (c == 3) {
+                                log.log("Disabling polling, enabling push");
+                                getPushConfiguration().setPushMode(
+                                        PushMode.AUTOMATIC);
+                                setPollInterval(-1);
+                                log.log("Polling disabled, push enabled");
+                            }
+                        }
+                    });
+                }
+            } catch (InterruptedException e) {
+            } catch (UIDetachedException e) {
+            }
+        }
+    };
+
+    @Override
+    protected void setup(VaadinRequest request) {
+
+        getPushConfiguration().setPushMode(PushMode.AUTOMATIC);
+        log.log("Push enabled");
+
+        addComponent(new Button("Disable push", new Button.ClickListener() {
+            @Override
+            public void buttonClick(Button.ClickEvent event) {
+                log.log("Disabling push");
+                getPushConfiguration().setPushMode(PushMode.DISABLED);
+                log.log("Push disabled");
+            }
+        }));
+
+        addComponent(new Button("Enable push", new Button.ClickListener() {
+            @Override
+            public void buttonClick(Button.ClickEvent event) {
+                log.log("Enabling push");
+                getPushConfiguration().setPushMode(PushMode.AUTOMATIC);
+                log.log("Push enabled");
+            }
+        }));
+
+        addComponent(new Button("Disable polling", new Button.ClickListener() {
+            @Override
+            public void buttonClick(Button.ClickEvent event) {
+                log.log("Disabling poll");
+                setPollInterval(-1);
+                log.log("Poll disabled");
+            }
+        }));
+
+        addComponent(new Button("Enable polling", new Button.ClickListener() {
+            @Override
+            public void buttonClick(Button.ClickEvent event) {
+                log.log("Enabling poll");
+                setPollInterval(1000);
+                log.log("Poll enabled");
+            }
+        }));
+
+        addComponent(new Button(
+                "Disable push, re-enable from background thread",
+                new Button.ClickListener() {
+                    @Override
+                    public void buttonClick(Button.ClickEvent event) {
+                        log.log("Disabling push, enabling polling");
+                        getPushConfiguration().setPushMode(PushMode.DISABLED);
+                        setPollInterval(1000);
+                        timer.schedule(new CounterTask(), new Date());
+                        log.log("Push disabled, polling enabled");
+                    }
+                }));
+
+        addComponent(log);
+    }
+
+    @Override
+    protected String getTestDescription() {
+        return "Test dynamically enablind and disabling push";
+    }
+
+    @Override
+    protected Integer getTicketNumber() {
+        return 12226;
+    }
+}