Best Rod code snippet using rod.Method
setup_test.go
Source:setup_test.go
1package rod_test2import (3 "context"4 "errors"5 "flag"6 "fmt"7 "log"8 "net/http"9 "os"10 "path/filepath"11 "runtime"12 "sync"13 "sync/atomic"14 "testing"15 "time"16 "github.com/go-rod/rod"17 "github.com/go-rod/rod/lib/cdp"18 "github.com/go-rod/rod/lib/defaults"19 "github.com/go-rod/rod/lib/launcher"20 "github.com/go-rod/rod/lib/proto"21 "github.com/go-rod/rod/lib/utils"22 "github.com/ysmood/got"23 "github.com/ysmood/gotrace"24 "github.com/ysmood/gson"25)26var TimeoutEach = flag.Duration("timeout-each", time.Minute, "timeout for each test")27var LogDir = slash(fmt.Sprintf("tmp/cdp-log/%s", time.Now().Format("2006-01-02_15-04-05")))28func init() {29 got.DefaultFlags("timeout=5m", "run=/")30 utils.E(os.MkdirAll(slash("tmp/cdp-log"), 0755))31 launcher.NewBrowser().MustGet() // preload browser to local32}33var testerPool TesterPool34func TestMain(m *testing.M) {35 testerPool = newTesterPool()36 code := m.Run()37 if code != 0 {38 os.Exit(code)39 }40 testerPool.cleanup()41 if err := gotrace.Check(0, gotrace.IgnoreFuncs("internal/poll.runtime_pollWait")); err != nil {42 log.Fatal(err)43 }44}45var setup = func(t *testing.T) G {46 return testerPool.get(t)47}48// G is a tester. Testers are thread-safe, they shouldn't race each other.49type G struct {50 got.G51 // mock client for proxy the cdp requests52 mc *MockClient53 // a random browser instance from the pool. If you have changed state of it, you must reset it54 // or it may affect other test cases.55 browser *rod.Browser56 // a random page instance from the pool. If you have changed state of it, you must reset it57 // or it may affect other test cases.58 page *rod.Page59 // use it to cancel the TimeoutEach for each test case60 cancelTimeout func()61}62// TesterPool if we don't use pool to cache, the total time will be much longer.63type TesterPool struct {64 pool chan *G65 parallel int66}67func newTesterPool() TesterPool {68 parallel := got.Parallel()69 if parallel == 0 {70 parallel = runtime.GOMAXPROCS(0)71 }72 fmt.Println("parallel test", parallel)73 cp := TesterPool{74 pool: make(chan *G, parallel),75 parallel: parallel,76 }77 for i := 0; i < parallel; i++ {78 cp.pool <- nil79 }80 return cp81}82// new tester83func (tp TesterPool) new() *G {84 u := launcher.New().MustLaunch()85 mc := newMockClient(u)86 browser := rod.New().Client(mc).MustConnect().MustIgnoreCertErrors(false)87 pages := browser.MustPages()88 var page *rod.Page89 if pages.Empty() {90 page = browser.MustPage()91 } else {92 page = pages.First()93 }94 return &G{95 mc: mc,96 browser: browser,97 page: page,98 }99}100// get a tester101func (tp TesterPool) get(t *testing.T) G {102 if got.Parallel() != 1 {103 t.Parallel()104 }105 tester := <-tp.pool106 if tester == nil {107 tester = tp.new()108 }109 t.Cleanup(func() { tp.pool <- tester })110 tester.G = got.New(t)111 tester.mc.t = t112 tester.mc.log.SetOutput(tester.Open(true, LogDir, tester.mc.id, t.Name()+".log"))113 tester.checkLeaking()114 return *tester115}116func (tp TesterPool) cleanup() {117 for i := 0; i < tp.parallel; i++ {118 if t := <-testerPool.pool; t != nil {119 t.browser.MustClose()120 }121 }122}123func (g G) enableCDPLog() {124 g.mc.principal.Logger(rod.DefaultLogger)125}126func (g G) dump(args ...interface{}) {127 g.Log(utils.Dump(args))128}129func (g G) blank() string {130 return g.srcFile("./fixtures/blank.html")131}132// Get abs file path from fixtures folder, such as "file:///a/b/click.html".133// Usually the path can be used for html src attribute like:134//135// <img src="file:///a/b">136func (g G) srcFile(path string) string {137 g.Helper()138 f, err := filepath.Abs(slash(path))139 g.E(err)140 return "file://" + f141}142func (g G) newPage(u ...string) *rod.Page {143 g.Helper()144 p := g.browser.MustPage(u...)145 g.Cleanup(func() {146 if !g.Failed() {147 p.MustClose()148 }149 })150 return p151}152func (g *G) checkLeaking() {153 ig := gotrace.CombineIgnores(gotrace.IgnoreCurrent(), gotrace.IgnoreNonChildren())154 gotrace.CheckLeak(g.Testable, 0, ig)155 self := gotrace.Get(false)[0]156 g.cancelTimeout = g.DoAfter(*TimeoutEach, func() {157 t := gotrace.Get(true).Filter(func(t *gotrace.Trace) bool {158 if t.GoroutineID == self.GoroutineID {159 return false160 }161 return ig(t)162 }).String()163 panic(fmt.Sprintf(`[rod_test.TimeoutEach] %s timeout after %v164running goroutines: %s`, g.Name(), *TimeoutEach, t))165 })166 g.Cleanup(func() {167 if g.Failed() {168 return169 }170 // close all other pages other than g.page171 res, err := proto.TargetGetTargets{}.Call(g.browser)172 g.E(err)173 for _, info := range res.TargetInfos {174 if info.TargetID != g.page.TargetID {175 g.E(proto.TargetCloseTarget{TargetID: info.TargetID}.Call(g.browser))176 }177 }178 if g.browser.LoadState(g.page.SessionID, &proto.FetchEnable{}) {179 g.Logf("leaking FetchEnable")180 g.FailNow()181 }182 g.mc.setCall(nil)183 })184}185type Call func(ctx context.Context, sessionID, method string, params interface{}) ([]byte, error)186var _ rod.CDPClient = &MockClient{}187type MockClient struct {188 sync.RWMutex189 id string190 t got.Testable191 log *log.Logger192 principal *cdp.Client193 call Call194 event <-chan *cdp.Event195}196var mockClientCount int32197func newMockClient(u string) *MockClient {198 id := fmt.Sprintf("%02d", atomic.AddInt32(&mockClientCount, 1))199 // create init log file200 utils.E(os.MkdirAll(filepath.Join(LogDir, id), 0755))201 f, err := os.Create(filepath.Join(LogDir, id, "_.log"))202 log := log.New(f, "", log.Ltime)203 utils.E(err)204 client := cdp.New().Logger(utils.MultiLogger(defaults.CDP, log)).Start(cdp.MustConnectWS(u))205 return &MockClient{id: id, principal: client, log: log}206}207func (mc *MockClient) Event() <-chan *cdp.Event {208 if mc.event != nil {209 return mc.event210 }211 return mc.principal.Event()212}213func (mc *MockClient) Call(ctx context.Context, sessionID, method string, params interface{}) ([]byte, error) {214 return mc.getCall()(ctx, sessionID, method, params)215}216func (mc *MockClient) getCall() Call {217 mc.RLock()218 defer mc.RUnlock()219 if mc.call == nil {220 return mc.principal.Call221 }222 return mc.call223}224func (mc *MockClient) setCall(fn Call) {225 mc.Lock()226 defer mc.Unlock()227 if mc.call != nil {228 mc.t.Logf("leaking MockClient.stub")229 mc.t.Fail()230 }231 mc.call = fn232}233func (mc *MockClient) resetCall() {234 mc.Lock()235 defer mc.Unlock()236 mc.call = nil237}238// Use it to find out which cdp call to intercept. Put a print like log.Println("*****") after the cdp call you want to intercept.239// The output of the test should has something like:240//241// [stubCounter] begin242// [stubCounter] 1, proto.DOMResolveNode{}243// [stubCounter] 1, proto.RuntimeCallFunctionOn{}244// [stubCounter] 2, proto.RuntimeCallFunctionOn{}245// 01:49:43 *****246//247// So the 3rd call is the one we want to intercept, then you can use the output with s.at or s.errorAt.248func (mc *MockClient) stubCounter() {249 l := sync.Mutex{}250 mCount := map[string]int{}251 fmt.Fprintln(os.Stdout, "[stubCounter] begin")252 mc.setCall(func(ctx context.Context, sessionID, method string, params interface{}) ([]byte, error) {253 l.Lock()254 mCount[method]++255 m := fmt.Sprintf("%d, proto.%s{}", mCount[method], proto.GetType(method).Name())256 _, _ = fmt.Fprintln(os.Stdout, "[stubCounter]", m)257 l.Unlock()258 return mc.principal.Call(ctx, sessionID, method, params)259 })260}261type StubSend func() (gson.JSON, error)262// When call the cdp.Client.Call the nth time use fn instead.263// Use p to filter method.264func (mc *MockClient) stub(nth int, p proto.Request, fn func(send StubSend) (gson.JSON, error)) {265 if p == nil {266 mc.t.Logf("p must be specified")267 mc.t.FailNow()268 }269 count := int64(0)270 mc.setCall(func(ctx context.Context, sessionID, method string, params interface{}) ([]byte, error) {271 if method == p.ProtoReq() {272 if int(atomic.AddInt64(&count, 1)) == nth {273 mc.resetCall()274 j, err := fn(func() (gson.JSON, error) {275 b, err := mc.principal.Call(ctx, sessionID, method, params)276 return gson.New(b), err277 })278 if err != nil {279 return nil, err280 }281 return j.MarshalJSON()282 }283 }284 return mc.principal.Call(ctx, sessionID, method, params)285 })286}287// When call the cdp.Client.Call the nth time return error.288// Use p to filter method.289func (mc *MockClient) stubErr(nth int, p proto.Request) {290 mc.stub(nth, p, func(send StubSend) (gson.JSON, error) {291 return gson.New(nil), errors.New("mock error")292 })293}294type MockRoundTripper struct {295 res *http.Response296 err error297}298func (mrt *MockRoundTripper) RoundTrip(*http.Request) (*http.Response, error) {299 return mrt.res, mrt.err300}301type MockReader struct {302 err error303}304func (mr *MockReader) Read(p []byte) (n int, err error) {305 return 0, mr.err306}307func TestLintIgnore(t *testing.T) {308 t.Skip()309 _ = rod.Try(func() {310 tt := G{}311 tt.dump()312 tt.enableCDPLog()313 mc := &MockClient{}314 mc.stubCounter()315 })316}317var slash = filepath.FromSlash...
main.go
Source:main.go
...51 "\n\n_<sub>generated by [check-issue](https://github.com/go-rod/rod/actions/runs/%s)</sub>_",52 os.Getenv("GITHUB_RUN_ID"),53 )54 q := req(fmt.Sprintf("/repos/go-rod/rod/issues/%d/comments", id))55 q.Method = http.MethodPost56 q.Body = ioutil.NopCloser(bytes.NewBuffer(utils.MustToJSONBytes(map[string]string{"body": msg})))57 res, err := http.DefaultClient.Do(q)58 utils.E(err)59 defer func() { _ = res.Body.Close() }()60 resE(res)61}62func deleteComments(id int) {63 q := req(fmt.Sprintf("/repos/go-rod/rod/issues/%d/comments", id))64 res, err := http.DefaultClient.Do(q)65 utils.E(err)66 resE(res)67 list := gson.New(res.Body)68 for _, c := range list.Arr() {69 if c.Get("user.login").Str() == "rod-robot" &&70 strings.Contains(c.Get("body").Str(), "[check-issue]") {71 iid := c.Get("id").Int()72 q := req(fmt.Sprintf("/repos/go-rod/rod/issues/comments/%d", iid))73 q.Method = http.MethodDelete74 res, err := http.DefaultClient.Do(q)75 utils.E(err)76 resE(res)77 }78 }79}80func req(u string) *http.Request {81 token := os.Getenv("ROD_GITHUB_ROBOT")82 if token == "" {83 panic("missing github token")84 }85 r, err := http.NewRequest(http.MethodGet, "https://api.github.com"+u, nil)86 utils.E(err)87 r.Header.Add("Authorization", "token "+token)88 return r89}90func resE(res *http.Response) {91 if res.StatusCode >= 400 {92 str, err := ioutil.ReadAll(res.Body)93 utils.E(err)94 panic(string(str))95 }96}...
Method
Using AI Code Generation
1import "fmt"2func main() {3 r.Method()4}5import "fmt"6func main() {7 r.Method()8}9import "fmt"10func main() {11 r.Method()12}13import "fmt"14func main() {15 r.Method()16}17import "fmt"18func main() {19 r.Method()20}21import "fmt"22func main() {23 r.Method()24}25import "fmt"26func main() {27 r.Method()28}29import "fmt"30func main() {31 r.Method()32}33import "fmt"34func main() {35 r.Method()36}37import "fmt"38func main() {39 r.Method()40}41import "fmt"42func main() {43 r.Method()44}45import "fmt"46func main() {47 r.Method()48}49import "fmt"50func main() {51 r.Method()52}53import
Method
Using AI Code Generation
1import "fmt"2func main() {3 r := rod{}4 r.Method()5}6import "fmt"7func main() {8 r := rod{}9 r.Method()10}11import "fmt"12func main() {13 r := rod{}14 r.Method()15}16import "fmt"17func main() {18 r := rod{}19 r.Method()20}21import "fmt"22func main() {23 r := rod{}24 r.Method()25}26import "fmt"27func main() {28 r := rod{}29 r.Method()30}31import "fmt"32func main() {33 r := rod{}34 r.Method()35}36import "fmt"37func main() {38 r := rod{}39 r.Method()40}41import "fmt"42func main() {43 r := rod{}44 r.Method()45}46import "fmt"47func main() {48 r := rod{}49 r.Method()50}51import "fmt"52func main() {53 r := rod{}54 r.Method()55}56import "fmt"57func main() {58 r := rod{}59 r.Method()60}61import "fmt"62func main() {63 r := rod{}64 r.Method()65}66import
Method
Using AI Code Generation
1import (2func main() {3 r.Method()4}5import (6func main() {7 r.Method()8}9import (10func main() {11 r.Method()12}13import (14func main() {15 r.Method()16}17import (18func main() {19 r.Method()20}21import (22func main() {23 r.Method()24}25import (26func main() {27 r.Method()28}29import (30func main() {31 r.Method()32}33import (34func main() {35 r.Method()36}37import (38func main() {39 r.Method()40}41import (42func main() {43 r.Method()44}45import (
Method
Using AI Code Generation
1import (2type rod struct {3}4func (r rod) method() {5 fmt.Println("rod method")6}7func main() {8 r.method()9}10import (11type rod struct {12}13func (r *rod) method() {14 fmt.Println("rod method")15}16func main() {17 r.method()18}19import (20type rod struct {21}22func (r *rod) method() {23 fmt.Println("rod method")24}25func main() {26 r.method()27}28import (29type rod struct {30}31func (r rod) method() {32 fmt.Println("rod method")33}34func main() {35 r.method()36}37import (38type rod struct {39}40func (r *rod) method() {41 fmt.Println("rod method")42}43func main() {44 r.method()45}46import (47type rod struct {48}49func (r rod) method() {50 fmt.Println("rod method")51}52func main() {53 r.method()54}55import (56type rod struct {57}58func (r *rod) method() {59 fmt.Println("rod method")60}61func main() {62 r.method()63}64import (65type rod struct {66}67func (r rod) method() {
Method
Using AI Code Generation
1import "fmt"2func main() {3 a.Method(1)4 a.Method(2)5}6import "fmt"7func main() {8 a.Method(1)9 a.Method(2)10}11import "fmt"12func main() {13 a.Method(1)14 a.Method(2)15}16import "fmt"17func main() {18 a.Method(1)19 a.Method(2)20}21import "fmt"22func main() {23 a.Method(1)24 a.Method(2)25}26import "fmt"27func main() {28 a.Method(1)29 a.Method(2)30}31import "fmt"32func main() {33 a.Method(1)34 a.Method(2)35}36import "fmt"37func main() {38 a.Method(1)39 a.Method(2)40}41import "fmt"42func main() {43 a.Method(1)44 a.Method(2)45}
Method
Using AI Code Generation
1import (2func main() {3 rod := rods.NewRod(1.5, 3.0)4 fmt.Println(rod.Length())5}6import (7func main() {8 rod := rods.NewRod(1.5, 3.0)9 fmt.Println(rod.Width())10}11import (12func main() {13 rod := rods.NewRod(1.5, 3.0)14 fmt.Println(rod.Area())15}16import (17func main() {18 rod := rods.NewRod(1.5, 3.0)19 fmt.Println(rod.Volume())20}21import (22func main() {23 rod := rods.NewRod(1.5, 3.0)24 fmt.Println(rod.Weight(5.5))25}26import (27func main() {28 rod := rods.NewRod(1.5, 3.0)29 fmt.Println(rod.Weight(5.5))30 fmt.Println(rod.Length())31 fmt.Println(rod.Width())32 fmt.Println(rod.Area())33 fmt.Println(rod.Volume())34}35import (36func main() {37 rod := rods.NewRod(1.5, 3.0)38 fmt.Println(rod.Weight(5.5))39 fmt.Println(rod.Length())40 fmt.Println(rod.Width())41 fmt.Println(rod.Area())42 fmt.Println(rod.Volume())43 fmt.Println(rod)44}45import (46func main() {
Method
Using AI Code Generation
1import (2func main() {3 r := rod.NewRod()4 r.Method()5}6import (7func main() {8 r := rod.NewRod()9 r.Method()10}11import (12func main() {13 r := rod.NewRod()14 r.Method()15}16import (17func main() {18 r := rod.NewRod()19 r.Method()20}21import (22func main() {23 r := rod.NewRod()24 r.Method()25}26import (27func main() {28 r := rod.NewRod()29 r.Method()30}31import (32func main() {33 r := rod.NewRod()34 r.Method()35}36import (37func main() {38 r := rod.NewRod()39 r.Method()40}41import (42func main() {
Method
Using AI Code Generation
1import (2func main() {3 rod1 := rod.New(1, 2)4 fmt.Println(rod1.Method())5}6import (7type Rod struct {8}9func New(l float64, d float64) *Rod {10 return &Rod{l, d}11}12func (r Rod) Method() float64 {13 return math.Sqrt(r.Length * r.Length + r.Dia * r.Dia)14}15import (16type Rod struct {17}18func New(l float64, d float64) *Rod {19 return &Rod{l, d}20}21func (r Rod) Method() float64 {22 return math.Sqrt(r.Length * r.Length + r.Dia * r.Dia)23}24import (25type Rod struct {26}27func New(l float64, d float64) *Rod {28 return &Rod{l, d}29}30func (r Rod) Method() float64 {31 return math.Sqrt(r.Length * r.Length + r.Dia * r.Dia)32}
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.
You could also refer to video tutorials over LambdaTest YouTube channel to get step by step demonstration from industry experts.
Get 100 minutes of automation test minutes FREE!!