Add tests for jshandle.evaluateHandle

ankur22 · ankur22 · commit e6451851d409 · 2024-06-11T15:46:35.000+01:00
diff --git a/tests/js_handle_test.go b/tests/js_handle_test.go
@@ -54,3 +54,56 @@ func TestJSHandleEvaluate(t *testing.T) {
 		})
 	}
 }
+
+func TestJSHandleEvaluateHandle(t *testing.T) {
+	t.Parallel()
+
+	tests := []struct {
+		name     string
+		pageFunc string
+		args     []any
+		expected string
+	}{
+		{
+			name: "no_args",
+			pageFunc: `handle => {
+				return {"innerText": handle.innerText};
+			}`,
+			args:     nil,
+			expected: `{"innerText":"Some title"}`,
+		},
+		{
+			name: "with_args",
+			pageFunc: `(handle, a, b) => {
+				return {"innerText": handle.innerText, "sum": a + b};
+			}`,
+			args:     []any{1, 2},
+			expected: `{"innerText":"Some title","sum":3}`,
+		},
+	}
+	for _, tt := range tests {
+		tt := tt
+
+		t.Run(tt.name, func(t *testing.T) {
+			t.Parallel()
+
+			tb := newTestBrowser(t)
+			p := tb.NewPage(nil)
+
+			err := p.SetContent(`<html><head><title>Some title</title></head></html>`, nil)
+			require.NoError(t, err)
+
+			result, err := p.EvaluateHandle(`() => document.head`)
+			require.NoError(t, err)
+			require.NotNil(t, result)
+
+			got, err := result.EvaluateHandle(tt.pageFunc, tt.args...)
+			assert.NoError(t, err)
+			assert.NotNil(t, got)
+
+			j, err := got.JSONValue()
+			assert.NoError(t, err)
+			assert.Equal(t, tt.expected, j)
+		})
+	}
+}
