Add waitForURL in frame mapping

Author: ankur22

internal/js/modules/k6/browser/browser/frame_mapping.go

@@ -2,6 +2,7 @@ package browser
 
 import (
 	"fmt"
+	"reflect"
 
 	"github.com/grafana/sobek"
 
@@ -422,6 +423,31 @@ func mapFrame(vu moduleVU, f *common.Frame) mapping {
 				return nil, nil
 			})
 		},
+		"waitForURL": func(url sobek.Value, opts sobek.Value) (*sobek.Promise, error) {
+			popts := common.NewFrameWaitForURLOptions(f.Timeout())
+			if err := popts.Parse(vu.Context(), opts); err != nil {
+				return nil, fmt.Errorf("parsing waitForURL options: %w", err)
+			}
+
+			var val string
+			switch url.ExportType() {
+			case reflect.TypeOf(string("")):
+				val = fmt.Sprintf("'%s'", url.String()) // Strings require quotes
+			default: // JS Regex, CSS, numbers or booleans
+				val = url.String() // No quotes
+			}
+
+			// Inject JS regex checker for URL pattern matching
+			ctx := vu.Context()
+			jsRegexChecker, err := injectRegexMatcherScript(ctx, vu, f.Page().TargetID())
+			if err != nil {
+				return nil, err
+			}
+
+			return k6ext.Promise(ctx, func() (result any, reason error) {
+				return nil, f.WaitForURL(val, popts, jsRegexChecker)
+			}), nil
+		},
 	}
 	maps["$"] = func(selector string) *sobek.Promise {
 		return k6ext.Promise(vu.Context(), func() (any, error) {

internal/js/modules/k6/browser/browser/mapping_test.go

@@ -433,6 +433,7 @@ type frameAPI interface { //nolint:interfacebloat
 	WaitForNavigation(opts sobek.Value) (*common.Response, error)
 	WaitForSelector(selector string, opts sobek.Value) (*common.ElementHandle, error)
 	WaitForTimeout(timeout int64)
+	WaitForURL(url string, opts sobek.Value) (*sobek.Promise, error)
 }
 
 // elementHandleAPI is the interface of an in-page DOM element.