Improve Page.Evaluate test

Resolves https://github.com/grafana/xk6-browser/pull/293/files/64af470d889b9e0bda962c83153033fd06ba412d..545a37210e884ccef65b3705aa80a121ffa8056d#r866708436
grafana · May 6, 2022 · 72a7ff4 · 72a7ff4
1 parent 7266f3a
commit 72a7ff4
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/tests/page_test.go b/tests/page_test.go
@@ -77,17 +77,21 @@ func TestPageEmulateMedia(t *testing.T) {
 func TestPageEvaluate(t *testing.T) {
 	t.Parallel()
 
-	t.Run("ok/func", func(t *testing.T) {
+	t.Run("ok/func_arg", func(t *testing.T) {
 		t.Parallel()
 
 		tb := newTestBrowser(t)
 		p := tb.NewPage(nil)
 
-		got := p.Evaluate(tb.rt.ToValue("() => document.location.toString()"))
+		got := p.Evaluate(
+			tb.rt.ToValue("(v) => { window.v = v; return window.v }"),
+			tb.rt.ToValue("test"),
+		)
 
 		require.IsType(t, tb.rt.ToValue(""), got)
-		gotVal, _ := got.(goja.Value)
-		assert.Equal(t, "about:blank", gotVal.Export())
+		gotVal, ok := got.(goja.Value)
+		require.True(t, ok)
+		assert.Equal(t, "test", gotVal.Export())
 	})
 
 	t.Run("err", func(t *testing.T) {