Test your AI Agents with the all-new Agent to Agent Testing Platform.Learn More

Automation Advisor
Rod
rod
HijackRequests

How to use HijackRequests method of rod Package

Best Rod code snippet using rod.HijackRequests

hijack_test.go

Source:hijack_test.go

...28		g.Eq("a", string(b))29		g.HandleHTTP(".html", "test")(w, r)30	})31	s.Route("/b", "", "b")32	router := g.page.HijackRequests()33	defer router.MustStop()34	router.MustAdd(s.URL("/a"), func(ctx *rod.Hijack) {35		r := ctx.Request.SetContext(g.Context())36		r.Req().Header.Set("Test", "header") // override request header37		r.SetBody([]byte("test"))            // override request body38		r.SetBody(123)                       // override request body39		r.SetBody(r.Body())                  // override request body40		type MyState struct {41			Val int42		}43		ctx.CustomState = &MyState{10}44		g.Eq(http.MethodPost, r.Method())45		g.Eq(s.URL("/a"), r.URL().String())46		g.Eq(proto.NetworkResourceTypeXHR, ctx.Request.Type())47		g.Is(ctx.Request.IsNavigation(), false)48		g.Has(ctx.Request.Header("Origin"), s.URL())49		g.Len(ctx.Request.Headers(), 6)50		g.True(ctx.Request.JSONBody().Nil())51		// send request load response from real destination as the default value to hijack52		ctx.MustLoadResponse()53		g.Eq(200, ctx.Response.Payload().ResponseCode)54		// override status code55		ctx.Response.Payload().ResponseCode = http.StatusCreated56		g.Eq("4", ctx.Response.Headers().Get("Content-Length"))57		g.Has(ctx.Response.Headers().Get("Content-Type"), "text/html; charset=utf-8")58		// override response header59		ctx.Response.SetHeader("Set-Cookie", "key=val")60		// override response body61		ctx.Response.SetBody([]byte("test"))62		ctx.Response.SetBody("test")63		ctx.Response.SetBody(map[string]string{64			"text": "test",65		})66		g.Eq("{\"text\":\"test\"}", ctx.Response.Body())67	})68	router.MustAdd(s.URL("/b"), func(ctx *rod.Hijack) {69		panic("should not come to here")70	})71	router.MustRemove(s.URL("/b"))72	router.MustAdd(s.URL("/b"), func(ctx *rod.Hijack) {73		// transparent proxy74		ctx.MustLoadResponse()75	})76	go router.Run()77	g.page.MustNavigate(s.URL())78	g.Eq("201 test key=val", g.page.MustElement("#a").MustText())79	g.Eq("b", g.page.MustElement("#b").MustText())80}81func TestHijackContinue(t *testing.T) {82	g := setup(t)83	s := g.Serve().Route("/", ".html", `<body>ok</body>`)84	router := g.page.HijackRequests()85	defer router.MustStop()86	wg := &sync.WaitGroup{}87	wg.Add(1)88	router.MustAdd(s.URL("/a"), func(ctx *rod.Hijack) {89		ctx.ContinueRequest(&proto.FetchContinueRequest{})90		wg.Done()91	})92	go router.Run()93	g.page.MustNavigate(s.URL("/a"))94	g.Eq("ok", g.page.MustElement("body").MustText())95	wg.Wait()96}97func TestHijackMockWholeResponseEmptyBody(t *testing.T) {98	g := setup(t)99	router := g.page.HijackRequests()100	defer router.MustStop()101	router.MustAdd("*", func(ctx *rod.Hijack) {102		ctx.Response.SetBody("")103	})104	go router.Run()105	// needs to timeout or will hang when "omitempty" does not get removed from body in fulfillRequest106	timed := g.page.Timeout(time.Second)107	timed.MustNavigate(g.Serve().Route("/", ".txt", "OK").URL())108	g.Eq("", g.page.MustElement("body").MustText())109}110func TestHijackMockWholeResponseNoBody(t *testing.T) {111	// TODO: remove the skip112	t.Skip("Because of flaky test result")113	g := setup(t)114	router := g.page.HijackRequests()115	defer router.MustStop()116	// intercept and reply without setting a body117	router.MustAdd("*", func(ctx *rod.Hijack) {118		// we don't set any body here119	})120	go router.Run()121	// has to timeout as it will lock up the browser reading the reply.122	err := g.page.Timeout(time.Second).Navigate(g.Serve().Route("/", "").URL())123	g.Is(err, context.DeadlineExceeded)124}125func TestHijackMockWholeResponse(t *testing.T) {126	g := setup(t)127	router := g.page.HijackRequests()128	defer router.MustStop()129	router.MustAdd("*", func(ctx *rod.Hijack) {130		ctx.Response.SetHeader("Content-Type", mime.TypeByExtension(".html"))131		ctx.Response.SetBody("<body>ok</body>")132	})133	go router.Run()134	g.page.MustNavigate("http://test.com")135	g.Eq("ok", g.page.MustElement("body").MustText())136}137func TestHijackSkip(t *testing.T) {138	g := setup(t)139	s := g.Serve()140	router := g.page.HijackRequests()141	defer router.MustStop()142	wg := &sync.WaitGroup{}143	wg.Add(2)144	router.MustAdd(s.URL("/a"), func(ctx *rod.Hijack) {145		ctx.Skip = true146		wg.Done()147	})148	router.MustAdd(s.URL("/a"), func(ctx *rod.Hijack) {149		ctx.ContinueRequest(&proto.FetchContinueRequest{})150		wg.Done()151	})152	go router.Run()153	g.page.MustNavigate(s.URL("/a"))154	wg.Wait()155}156func TestHijackOnErrorLog(t *testing.T) {157	g := setup(t)158	s := g.Serve().Route("/", ".html", `<body>ok</body>`)159	router := g.page.HijackRequests()160	defer router.MustStop()161	wg := &sync.WaitGroup{}162	wg.Add(1)163	var err error164	router.MustAdd(s.URL("/a"), func(ctx *rod.Hijack) {165		ctx.OnError = func(e error) {166			err = e167			wg.Done()168		}169		ctx.ContinueRequest(&proto.FetchContinueRequest{})170	})171	go router.Run()172	g.mc.stub(1, proto.FetchContinueRequest{}, func(send StubSend) (gson.JSON, error) {173		return gson.New(nil), errors.New("err")174	})175	go func() {176		_ = g.page.Context(g.Context()).Navigate(s.URL("/a"))177	}()178	wg.Wait()179	g.Eq(err.Error(), "err")180}181func TestHijackFailRequest(t *testing.T) {182	g := setup(t)183	s := g.Serve().Route("/page", ".html", `<html>184	<body></body>185	<script>186		fetch('/a').catch(async (err) => {187			document.title = err.message188		})189	</script></html>`)190	router := g.browser.HijackRequests()191	defer router.MustStop()192	router.MustAdd(s.URL("/a"), func(ctx *rod.Hijack) {193		ctx.Response.Fail(proto.NetworkErrorReasonAborted)194	})195	go router.Run()196	g.page.MustNavigate(s.URL("/page")).MustWaitLoad()197	g.page.MustWait(`() => document.title === 'Failed to fetch'`)198	{ // test error log199		g.mc.stub(1, proto.FetchFailRequest{}, func(send StubSend) (gson.JSON, error) {200			_, _ = send()201			return gson.JSON{}, errors.New("err")202		})203		_ = g.page.Navigate(s.URL("/a"))204	}205}206func TestHijackLoadResponseErr(t *testing.T) {207	g := setup(t)208	p := g.newPage().Context(g.Context())209	router := p.HijackRequests()210	defer router.MustStop()211	wg := &sync.WaitGroup{}212	wg.Add(1)213	router.MustAdd("http://test.com/a", func(ctx *rod.Hijack) {214		g.Err(ctx.LoadResponse(&http.Client{215			Transport: &MockRoundTripper{err: errors.New("err")},216		}, true))217		g.Err(ctx.LoadResponse(&http.Client{218			Transport: &MockRoundTripper{res: &http.Response{219				StatusCode: 200,220				Body:       ioutil.NopCloser(&MockReader{err: errors.New("err")}),221			}},222		}, true))223		wg.Done()224		ctx.Response.Fail(proto.NetworkErrorReasonAborted)225	})226	go router.Run()227	_ = p.Navigate("http://test.com/a")228	wg.Wait()229}230func TestHijackResponseErr(t *testing.T) {231	g := setup(t)232	s := g.Serve().Route("/", ".html", `ok`)233	p := g.newPage().Context(g.Context())234	router := p.HijackRequests()235	defer router.MustStop()236	wg := &sync.WaitGroup{}237	wg.Add(1)238	router.MustAdd(s.URL("/a"), func(ctx *rod.Hijack) { // to ignore favicon239		ctx.OnError = func(err error) {240			g.Err(err)241			wg.Done()242		}243		ctx.MustLoadResponse()244		g.mc.stub(1, proto.FetchFulfillRequest{}, func(send StubSend) (gson.JSON, error) {245			res, _ := send()246			return res, errors.New("err")247		})248	})...

HijackRequests

Using AI Code Generation

1import (2func main() {3l := launcher.New()4l.Headless(false)5l.Debug(true)6l.Use("chromium")7l.Flag("disable-web-security", true)8l.Flag("allow-running-insecure-content", true)9l.Flag("disable-background-networking", true)10l.Flag("disable-background-timer-throttling", true)11l.Flag("disable-client-side-phishing-detection", true)12l.Flag("disable-default-apps", true)13l.Flag("disable-dev-shm-usage", true)14l.Flag("disable-extensions", true)15l.Flag("disable-features", "site-per-process")16l.Flag("disable-hang-monitor", true)17l.Flag("disable-popup-blocking", true)18l.Flag("disable-prompt-on-repost", true)19l.Flag("disable-sync", true)20l.Flag("disable-translate", true)21l.Flag("metrics-recording-only", true)22l.Flag("no-first-run", true)23l.Flag("safebrowsing-disable-auto-update", true)24l.Flag("enable-automation", true)25l.Flag("password-store", "basic")26l.Flag("use-mock-keychain", true)27l.Flag("mute-audio", true)28l.Flag("hide-scrollbars", true)29l.Flag("disable-gpu", true)30l.Flag("window-size", "1920,1080")31l.Flag("ignore-certificate-errors", true)32l.Flag("ignore-certificate-errors-spki-list", true)33l.Flag("ignore-ssl-errors", true)34l.Flag("ignore-certificate

HijackRequests

Using AI Code Generation

1import (2func main() {3	browser := rod.New().Connect()4	page.HijackRequests(func(request *proto.NetworkRequestWillBeSent) {5		fmt.Println(request.Request.URL)6	})7	page.WaitRequestIdle()8}

HijackRequests

Using AI Code Generation

1import (2func main() {3	browser := rod.New().Client(launcher.New().HijackRequests(func(req *http.Request) {4		req.Header.Set("User-Agent", "Rod")5	}).MustLaunch()).MustConnect()6	title := page.MustElement("h1").MustText()7	fmt.Println(title)8	browser.MustClose()9}10import (11func main() {12	browser := rod.New().Client(launcher.New().HijackResponses(func(res *http.Response) {13		res.Header.Set("X-My-Header", "Rod")14	}).MustLaunch()).MustConnect()15	title := page.MustElement("h1").MustText()16	fmt.Println(title)17	browser.MustClose()18}

HijackRequests

Using AI Code Generation

1import (2func main() {3	browser := rod.New().ControlURL(launcher.New().Headless(false).MustLaunch()).MustConnect()4	page.MustEvalOnNewDocument(`Object.defineProperty(navigator, 'webdriver', { get: () => false })`)5	page.MustHijackRequests(func(req *rod.Hijack) {6		req.MustContinue()7	})8	page.MustWaitLoad()9}10import (11func main() {12	browser := rod.New().ControlURL(launcher.New().Headless(false).MustLaunch()).MustConnect()13	page.MustEvalOnNewDocument(`Object.defineProperty(navigator, 'webdriver', { get: () => false })`)14	page.MustInterceptRequests(func(req *rod.Intercept) {15		req.MustContinue()16	})17	page.MustWaitLoad()18}19import (20func main() {21	browser := rod.New().ControlURL(launcher.New().Headless(false).MustLaunch()).MustConnect()

HijackRequests

Using AI Code Generation

1import (2func main() {3    fmt.Println(page.MustElement("input").MustAttribute("name"))4}5import (6func main() {7    browser := rod.New().MustConnect()8}9import (10func main() {11    fmt.Println(page.MustElement("input").MustAttribute("name"))12}13import (14func main() {15    fmt.Println(page.MustElement("input").MustAttribute("name"))16}17import (18func main() {19    fmt.Println(page.MustElement("input").MustAttribute("name"))20}21import (22func main() {23    fmt.Println(page.MustElement("input").MustAttribute("name"))24}25import (26func main() {

HijackRequests

Using AI Code Generation

1import (2func main() {3    page.HijackRequests(func(req *rod.Hijack) {4        fmt.Println(req.URL())5        req.Continue()6    })7    page.Element("input").Input("rod")8    page.Keyboard.Press('Enter')9    page.WaitLoad()10    fmt.Println(page.MustElement("h3").MustText())11}

HijackRequests

Using AI Code Generation

1import (2func main() {3	browser := rod.New().ControlURL(launcher.New().MustLaunch()).MustConnect()4	page.MustHijackRequests(func(req *rod.Hijack) {5			fmt.Println("Favicon requested")6		}7		req.MustLoadResponse()8	})9	page.MustWaitLoad()10}

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.