Join the 1-day Testing & QA Summit featuring 15+ Expert Speakers.Register for FREE! Join TestMu Conference

How to use Context method of rod Package

Best Rod code snippet using rod.Context

rod.go

Source:rod.go

...29	config   bbb.Config30	rod      Config31	launcher *launcher.Launcher32	browser  *rod.Browser33	ctx      context.Context34}35func (b *bb) Init(ctx context.Context) error {36	b.ctx = ctx37	b.launcher = launcher.New().Context(ctx)38	if b.rod.ShowUi {39		b.launcher.Headless(false)40	}41	url, err := b.launcher.Launch()42	if err != nil {43		b.Done()44		return err45	}46	b.browser = rod.New().Context(ctx).ControlURL(url)47	err = b.browser.Connect()48	if err != nil {49		b.browser = nil50		b.Done()51		return err52	}53	//TODO detect browser close in some way54	return nil55}56func (b *bb) Generate() <-chan bbb.Task {57	return bbb.Generator(b.ctx, b.config.Urls, b.config.Log)58}59func (b *bb) Worker(tasks <-chan bbb.Task) {60	page := stealth.MustPage(b.browser).Context(b.ctx)61	errP := &errPusher{}62	ddos := &ddosDetector{}63	go page.EachEvent(func(e *proto.NetworkResponseReceived) {64		b.config.Log.LogFn(b.ctx, bbb.LogLevelVerbose, fmt.Sprintf("Fetched %q, status %d", e.Response.URL, e.Response.Status))65		if strings.Contains(e.Response.URL, "ddos") {66			ddos.setDdos()67		}68		if e.Type == "Document" && e.Response.Status >= http.StatusBadRequest {69			if e.Response.StatusText != "" {70				errP.Push(errors.New(e.Response.StatusText))71			} else if e.Response.Status >= http.StatusInternalServerError {72				errP.Push(fmt.Errorf("status %d", e.Response.Status))73			}74		}75	})()76	for t := range tasks {77		doTask(page, t, b.config, errP, ddos)78	}79}80func doTask(page *rod.Page, t bbb.Task, config bbb.Config, errP *errPusher, ddos *ddosDetector) {81	defer rec(page.GetContext(), t, config.Log)82	errc := make(chan error, 1)83	errP.Install(errc)84	defer errP.Cleanup()85	timeout, hasTimeout := config.Workers.GetTimeout()86	if hasTimeout {87		p, cancel := ddos.withTimeout(page, timeout, config.Log)88		defer cancel()89		page = p90	}91	config.Log.LogFn(page.GetContext(), bbb.LogLevelVerbose, "Navigating to "+t.Url)92	page.MustNavigate(t.Url)93	page.MustWaitLoad()94	select {95	case err := <-errc:96		config.Log.UpdateFn(page.GetContext(), bbb.UrlUpdate{97			Url:   t.UrlConfig,98			Index: t.Index,99			Err:   err,100		})101	default:102		config.Log.UpdateFn(page.GetContext(), bbb.UrlUpdate{103			Url:   t.UrlConfig,104			Index: t.Index,105			Ok:    true,106		})107	}108}109type errPusher struct {110	ch atomic.Value111}112func (e *errPusher) Push(err error) {113	i := e.ch.Load()114	if i == nil {115		return116	}117	ch := i.(chan error)118	if ch == nil {119		return120	}121	select {122	case ch <- err:123	default:124	}125}126func (e *errPusher) Install(ch chan error) {127	e.ch.Store(ch)128}129func (e *errPusher) Cleanup() {130	var ch chan error131	e.ch.Store(ch)132}133type ddosDetector struct {134	ch atomic.Value135}136func (d *ddosDetector) withTimeout(parent *rod.Page, dur time.Duration, log bbb.LogConfig) (*rod.Page, context.CancelFunc) {137	//ctx, cancel := context.WithCancel(parent)138	page, cancel := parent.WithCancel()139	done := make(chan struct{})140	go func() {141		timer := time.NewTimer(dur)142		defer timer.Stop()143		select {144		case <-timer.C:145			cancel()146		case <-done:147		case <-d.ddos():148			log.LogFn(parent.GetContext(), bbb.LogLevelVerbose, "timeout cancelled 'cause of ddos protection")149		}150	}()151	return page, func() {152		close(done)153		cancel()154	}155}156func (d *ddosDetector) ddos() <-chan struct{} {157	ch := make(chan struct{}, 1)158	d.ch.Store(ch)159	return ch160}161func (d *ddosDetector) setDdos() {162	i := d.ch.Load()163	if i == nil {164		return165	}166	ch := i.(chan struct{})167	if ch == nil {168		return169	}170	select {171	case ch <- struct{}{}:172	default:173	}174}175func rec(ctx context.Context, t bbb.Task, log bbb.LogConfig) {176	crash := recover()177	if crash == nil {178		return179	}180	var err error181	if e, ok := crash.(error); ok {182		err = e183	} else {184		err = fmt.Errorf("%v", crash)185	}186	log.UpdateFn(ctx, bbb.UrlUpdate{187		Url:   t.UrlConfig,188		Index: t.Index,189		Err:   err,...

use_gorod.go

Source:use_gorod.go

...63	// defer m.Browser.MustClose()64	return m65}66func goRodRender(w http.ResponseWriter, r *http.Request) {67	cfg := r.Context().Value("cfg").(config.IManager)68	cdp := r.Context().Value("cdp").(IManager)69	mapping := cfg.GetMapByHost(r.Host)70	if mapping == nil {71		_, _ = fmt.Fprint(w, "")72		return73	}74	// browser.Logger(rod.DefaultLogger).Trace(true)75	page := cdp.GetBrowser().MustPage("")76	wait := page.MustWaitRequestIdle()77	err := rod.Try(func() {78		page.79			Context(r.Context()).80			Timeout(10 * time.Second).81			MustNavigate(fmt.Sprintf("%s%s", mapping.ToBaseUrl, r.URL.Path))82	})83	if errors.Is(err, context.DeadlineExceeded) {84		// in case want to handle on timeout differently85		_, _ = fmt.Fprint(w, "")86		return87	} else if err != nil {88		_, _ = fmt.Fprint(w, "")89		return90	}91	// Custom function to add script tag with its content to body92	addScriptTagToBody := func(p *rod.Page, value string) error {93		var addScriptHelper = &js.Function{...

main.go

Source:main.go

...11// 	defer ccancel()12// 	var wg sync.WaitGroup13// 	for i := 0; i < 10; i++ {14// 		wg.Add(1)15// ctx, _ = chromedp.NewContext(ctx)16// fmt.Println("starting: routine ", i)17// go chromedp.Run(context.Background(), chromedp.Navigate("https://twitch.tv/superserverbrasil"), chromedp.ActionFunc(func(ctx context.Context) error {18// 	fmt.Println("go routine ", i)19// 	defer wg.Done()20// 	time.Sleep(time.Minute * 10)21// 	return nil22// }))23// }24// defer cancel()25// time.After(time.Minute)26// 	wg.Wait()27// }28// import (29// 	"fmt"30// 	"log"31// 	"gopkg.in/headzoo/surf.v1"...

Context

Using AI Code Generation

1import (2func main() {3	browser := rod.New().Connect()4	page := browser.Page("")5	page.WaitLoad()6	input := page.Element(`input[name="q"]`)7	input.Input("rod")8	input.MustPress(input.Enter)9	page.WaitLoad()10	text := page.MustElement("#search .g").MustText()11	fmt.Println(text)12	browser.Close()13}14import (15func main() {16	browser := rod.New().Connect()17	page := browser.Page("")18	page.WaitLoad()19	input := page.Element(`input[name="q"]`)20	input.Input("rod")21	input.MustPress(input.Enter)22	page.WaitLoad()23	text := page.MustElement("#search .g").MustText()24	fmt.Println(text)25	browser.Close()26}27import (

Context

Using AI Code Generation

1import (2func main() {3	browser := rod.New().ControlURL(launcher.New().MustLaunch()).MustConnect()4	contextID := page.MustContextID()5	contextID2 := page2.MustContextID()6	contextID3 := page3.MustContextID()7	contextID4 := page4.MustContextID()8	contextID5 := page5.MustContextID()9	contextID6 := page6.MustContextID()10	contextID7 := page7.MustContextID()11	contextID8 := page8.MustContextID()12	contextID9 := page9.MustContextID()13	contextID10 := page10.MustContextID()

Context

Using AI Code Generation

1import (2func main() {3	l := launcher.New().Bin("chromium").Headless(false).MustLaunch()4	browser := rod.New().ControlURL(l).MustConnect()5	title := page.MustTitle()6	fmt.Println(title)7	cookies := page.MustCookies()8	fmt.Println(cookies)

Context

Using AI Code Generation

1import (2func main() {3	browser := rod.New().ControlURL(launcher.New().MustLaunch()).MustConnect()4	defer browser.Close()5	page.MustElement("#tsf > div:nth-child(2) > div > div.RNNXgb > div > div.a4bIc > input").MustInput("Rod Page Object")6	page.MustElement("#tsf > div:nth-child(2) > div > div.FPdoLc.VlcLAe > center > input.gNO89b").MustClick()7	time.Sleep(5 * time.Second)8	title := page.MustTitle()9	fmt.Println(title)10}11import (12func main() {13	browser := rod.New().ControlURL(launcher.New().MustLaunch()).MustConnect()14	defer browser.Close()15	page.MustElement("#tsf > div:nth-child(2) > div > div.RNNXgb > div > div.a4bIc > input").MustInput("Rod Page Object")16	page.MustElement("#tsf > div:nth-child(2) > div > div.FPdoLc.VlcLAe > center > input.gNO89b").MustClick()17	page.MustWaitLoad()18	title := page.MustTitle()19	fmt.Println(title)20}21import (

Context

Using AI Code Generation

1import (2func main() {3	browser := rod.New().MustConnect()4	pngFile, err := os.Create("google.png")5	if err != nil {6		fmt.Println(err)7	}8	defer pngFile.Close()9	img, err := page.Element("body").Screenshot()10	if err != nil {11		fmt.Println(err)12	}13	err = png.Encode(pngFile, img)14	if err != nil {15		fmt.Println(err)16	}17}18import (19func main() {20	browser := rod.New().MustConnect()21	pngFile, err := os.Create("google.png")22	if err != nil {23		fmt.Println(err)24	}25	defer pngFile.Close()26	img, err := page.Element("body").Screenshot()27	if err != nil {28		fmt.Println(err)29	}30	err = png.Encode(pngFile, img)31	if err != nil {32		fmt.Println(err)33	}34}35import (36func main() {37	browser := rod.New().MustConnect()38	pngFile, err := os.Create("google.png")39	if err != nil {40		fmt.Println(err)41	}42	defer pngFile.Close()43	img, err := page.Element("body").Screenshot()44	if err != nil {45		fmt.Println(err)46	}47	err = png.Encode(pngFile, img)48	if err != nil {49		fmt.Println(err)

Context

Using AI Code Generation

1import (2func main() {3	browser := rod.New().Connect()4	page := browser.Page("")5	title := page.Title()6	fmt.Println(title)7	url := page.URL()8	fmt.Println(url)9	page2 := browser.Page("")10	title2 := page2.Title()11	fmt.Println(title2)12	url2 := page2.URL()13	fmt.Println(url2)14	browser.Close()15}16import (17func main() {18	browser := rod.New().Connect()19	page := browser.Page("")20	title := page.Title()21	fmt.Println(title)22	url := page.URL()23	fmt.Println(url)24	page2 := browser.Page("")25	title2 := page2.Title()26	fmt.Println(title2)27	url2 := page2.URL()28	fmt.Println(url2)29	browser.Close()30}31import (32func main() {33	browser := rod.New().Connect()34	page := browser.Page("")35	title := page.Title()36	fmt.Println(title)

Context

Using AI Code Generation

1import (2func main() {3	browser := rod.New().Connect()4	page.Keyboard.Press('a', 'b', 'c', rod.Enter)5	fmt.Println(page.Element("body").Text())6}7import (8func main() {9	browser := rod.New().Connect()10	page.Keyboard.Press('a', 'b', 'c', rod.Enter)11	fmt.Println(page.Element("body").Text())12}13import (14func main() {15	browser := rod.New().Connect()16	page.Keyboard.Press('a', 'b', 'c', rod.Enter)17	fmt.Println(page.Element("body").Text())18}19import (20func main() {21	browser := rod.New().Connect()22	page.Keyboard.Press('a', 'b', 'c', rod.Enter)23	fmt.Println(page.Element("body").Text())24}25import (26func main() {27	browser := rod.New().Connect()28	page.Keyboard.Press('a', 'b', 'c', rod.Enter)29	fmt.Println(page.Element("body").Text())30}31import (32func main() {33	browser := rod.New().Connect()34	page.Keyboard.Press('a', 'b', 'c', rod.Enter)35	fmt.Println(page.Element("body").Text())36}

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.