Add a beforeunload dialog test

ankur22 · ankur22 · commit 5bbfb66d38e9 · 2022-11-30T09:57:37.000Z
This test will ensure that the beforeunload dialog box is accepted
when we try to reload the page. Unlike other alert boxes which appear
when the page loads, beforeunload boxes appear when the page unloads.
diff --git a/tests/frame_test.go b/tests/frame_test.go
@@ -3,6 +3,7 @@ package tests
 import (
 	"testing"
 
+	"github.com/dop251/goja"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 )
@@ -24,7 +25,6 @@ func TestFramePress(t *testing.T) {
 }
 
 func TestFrameDismissDialogBox(t *testing.T) {
-
 	t.Parallel()
 
 	tests := []struct {
@@ -39,6 +39,9 @@ func TestFrameDismissDialogBox(t *testing.T) {
 		{
 			name: "prompt",
 		},
+		{
+			name: "beforeunload",
+		},
 	}
 
 	for _, tt := range tests {
@@ -55,7 +58,16 @@ func TestFrameDismissDialogBox(t *testing.T) {
 				}{
 					WaitUntil: "networkidle",
 				})
-				b.promise(p.Goto(b.staticURL("dialog.html?dialogType="+tt.name), opts)).then(func() {
+				b.promise(p.Goto(b.staticURL("dialog.html?dialogType="+tt.name), opts)).then(func() *goja.Promise {
+					if tt.name == "beforeunload" {
+						return p.Click("#clickHere", nil)
+					}
+
+					result := p.TextContent("#textField", nil)
+					assert.EqualValues(t, tt.name+" dismissed", result)
+
+					return nil
+				}).then(func() {
 					result := p.TextContent("#textField", nil)
 					assert.EqualValues(t, tt.name+" dismissed", result)
 				})
diff --git a/tests/static/dialog.html b/tests/static/dialog.html
@@ -3,7 +3,7 @@
     <body>
         <div id='textField'>Hello World</div>
         <br />
-        <a href="/">click here</a>
+        <button id="clickHere" onclick="window.location.reload();">click here</button>
 
         <script>
             const queryString = window.location.search;
@@ -24,6 +24,12 @@
                     prompt("Add text and then click accept");
                     div.textContent = 'prompt dismissed';
                     break;
+                case "beforeunload":
+                    window.addEventListener('beforeunload', (event) => {
+                        event.returnValue = "Are you sure you want to leave?";
+                    });
+                    div.textContent = 'beforeunload dismissed';
+                    break;
             }
         </script>
     </body>
