Fix error on page.query when no matches with given selector

browser/mapping.go

@@ -253,6 +253,12 @@ func mapElementHandle(vu moduleVU, eh *common.ElementHandle) mapping {
 		if err != nil {
 			return nil, err //nolint:wrapcheck
 		}
+		// ElementHandle can be null when the selector does not match any elements.
+		// We do not want to map nil elementHandles since the expectation is a
+		// null result in the test script for this case.
+		if eh == nil {
+			return nil, nil //nolint:nilnil
+		}
 		ehm := mapElementHandle(vu, eh)
 		return ehm, nil
 	}
@@ -396,6 +402,12 @@ func mapFrame(vu moduleVU, f *common.Frame) mapping {
 		if err != nil {
 			return nil, err //nolint:wrapcheck
 		}
+		// ElementHandle can be null when the selector does not match any elements.
+		// We do not want to map nil elementHandles since the expectation is a
+		// null result in the test script for this case.
+		if eh == nil {
+			return nil, nil //nolint:nilnil
+		}
 		ehm := mapElementHandle(vu, eh)
 		return ehm, nil
 	}
@@ -594,6 +606,12 @@ func mapPage(vu moduleVU, p *common.Page) mapping {
 		if err != nil {
 			return nil, err //nolint:wrapcheck
 		}
+		// ElementHandle can be null when the selector does not match any elements.
+		// We do not want to map nil elementHandles since the expectation is a
+		// null result in the test script for this case.
+		if eh == nil {
+			return nil, nil //nolint:nilnil
+		}
 		ehm := mapElementHandle(vu, eh)
 		return ehm, nil
 	}

common/element_handle.go

@@ -1042,7 +1042,7 @@ func (h *ElementHandle) Query(selector string) (*ElementHandle, error) {
 		return nil, fmt.Errorf("querying selector %q: %w", selector, err)
 	}
 	if result == nil {
-		return nil, fmt.Errorf("querying selector %q", selector)
+		return nil, nil //nolint:nilnil
 	}
 	handle, ok := result.(JSHandleAPI)
 	if !ok {

tests/element_handle_test.go

@@ -439,3 +439,15 @@ func TestElementHandlePress(t *testing.T) {
 
 	require.Equal(t, "AbC", el.InputValue(nil))
 }
+
+func TestElementHandleQuery(t *testing.T) {
+	t.Parallel()
+
+	p := newTestBrowser(t).NewPage(nil)
+	p.SetContent(`<div id="foo">hello</div>`, nil)
+
+	element, err := p.Query("bar")
+
+	require.NoError(t, err)
+	require.Nil(t, element)
+}