diff --git a/browser/browser_context_mapping.go b/browser/browser_context_mapping.go index b7bb9e190..d9ed4cae6 100644 --- a/browser/browser_context_mapping.go +++ b/browser/browser_context_mapping.go @@ -43,7 +43,10 @@ func mapBrowserContext(vu moduleVU, bc *common.BrowserContext) mapping { //nolin return bc.AddInitScript(source) //nolint:wrapcheck }, - "browser": bc.Browser, + "browser": func() mapping { + // the browser is grabbed from VU. + return mapBrowser(vu) + }, "clearCookies": bc.ClearCookies, "clearPermissions": bc.ClearPermissions, "close": bc.Close, diff --git a/browser/page_mapping.go b/browser/page_mapping.go index 45704aa37..cafc79f42 100644 --- a/browser/page_mapping.go +++ b/browser/page_mapping.go @@ -35,8 +35,10 @@ func mapPage(vu moduleVU, p *common.Page) mapping { //nolint:gocognit,cyclop return p.Close(opts) //nolint:wrapcheck }, - "content": p.Content, - "context": p.Context, + "content": p.Content, + "context": func() mapping { + return mapBrowserContext(vu, p.Context()) + }, "dblclick": p.Dblclick, "dispatchEvent": func(selector, typ string, eventInit, opts goja.Value) error { popts := common.NewFrameDispatchEventOptions(p.Timeout())